import 'dart:typed_data';

/// H.264帧生成器
/// 生成简单的H.264帧数据，用于测试
class H264FrameGenerator {
  // 视频宽度
  final int width;

  // 视频高度
  final int height;

  // 序列参数集 (SPS) - 一个简单的示例
  // 注意：这不是一个完全有效的SPS，仅用于测试
  final List<int> _spsData = [
    0x00,
    0x00,
    0x00,
    0x01,
    0x67,
    0x42,
    0x00,
    0x0A,
    0xF8,
    0x41,
    0xA2,
    0x00,
    0x00,
    0x03,
    0x00,
    0x01,
    0x00,
    0x00,
    0x03,
    0x00,
    0x32,
    0x0F,
    0x18,
    0x31,
    0x8C
  ];

  // 图像参数集 (PPS) - 一个简单的示例
  // 注意：这不是一个完全有效的PPS，仅用于测试
  final List<int> _ppsData = [0x00, 0x00, 0x00, 0x01, 0x68, 0xCE, 0x38, 0x80];

  // I帧的起始数据示例 - 这只是一个示例头部
  final List<int> _iFrameHeader = [
    0x00,
    0x00,
    0x00,
    0x01,
    0x65,
    0x88,
    0x80,
    0x00,
    0x00,
    0x03,
    0x00,
    0x02,
    0x00
  ];

  // P帧的起始数据示例 - 这只是一个示例头部
  final List<int> _pFrameHeader = [
    0x00,
    0x00,
    0x00,
    0x01,
    0x41,
    0x9A,
    0x1C,
    0x0D,
    0x3E,
    0x04
  ];

  // 当前帧计数
  int _frameCount = 0;

  // 每个I帧之间的P帧数量
  final int _pFramesPerIFrame = 9;

  /// 构造函数
  H264FrameGenerator({
    required this.width,
    required this.height,
  });

  /// 获取SPS和PPS数据
  Uint8List getConfigurationData() {
    // 组合SPS和PPS
    List<int> data = [];
    data.addAll(_spsData);
    data.addAll(_ppsData);
    return Uint8List.fromList(data);
  }

  /// 生成下一帧数据
  /// 返回 (帧数据, 是否为I帧)
  (Uint8List, bool) generateNextFrame() {
    _frameCount++;

    // 每10帧插入一个I帧
    bool isIFrame = _frameCount % (_pFramesPerIFrame + 1) == 1;

    List<int> frameData = [];

    if (isIFrame) {
      // I帧: 添加SPS和PPS，并添加I帧数据
      frameData.addAll(_spsData);
      frameData.addAll(_ppsData);
      frameData.addAll(_iFrameHeader);

      // 添加一些模拟的I帧数据
      // 实际中，这里应该是真实的编码数据
      for (int i = 0; i < 1000; i++) {
        frameData.add((i * 13) % 256);
      }
    } else {
      // P帧: 只添加P帧数据
      frameData.addAll(_pFrameHeader);

      // 添加一些模拟的P帧数据
      // 实际中，这里应该是真实的编码数据
      for (int i = 0; i < 300; i++) {
        frameData.add((i * 7 + _frameCount) % 256);
      }
    }

    return (Uint8List.fromList(frameData), isIFrame);
  }

  /// 重置帧计数
  void reset() {
    _frameCount = 0;
  }
}