MPEG_TS.h

#pragma once
 
#include "Base.h"
 
#include <functional>
#include <unordered_map>
#include <vector>
 
namespace Cogs::Core{
 
struct MPEGProgramAssociationSection{
  uint32_t table_id;
  uint32_t section_syntax_indicator;
  uint32_t section_length;
  uint32_t transport_stream_id;
  uint32_t version_number;
  uint32_t current_next_indicator;
  uint32_t section_number;
  uint32_t last_section_number;
 
  uint32_t program_number[44];
  uint32_t PID[44]; // network_PID / program_map_PID
 
  uint32_t CRC32;
 
  bool HasPID(uint32_t pid, uint32_t &idx)
  {
    if(section_length <= 9) return false;
    uint32_t N = (section_length-9)/4;
    for(uint32_t i=0; i<N; i++){
      if(PID[i] == pid){
        idx = i;
        return true;
      }
    }
    return  false;
  }
 
  void print();
};
 
struct MPEGProgramMapEntity{
  uint32_t stream_type;
  uint32_t elementary_PID;
  uint32_t ES_info_length;
  // TODO Descriptors
};
 
struct MPEGProgramMapTable{
  uint32_t table_id;
  uint32_t section_syntax_indicator;
  uint32_t section_length;
  uint32_t program_number;
  uint32_t version_number;
  uint32_t current_next_indicator;
  uint32_t section_number;
  uint32_t last_section_number;
  uint32_t PCR_PID;
  uint32_t program_info_length;
 
  // TODO Descriptors
  std::vector<MPEGProgramMapEntity> entities;
 
  uint32_t CRC32 = 0;
 
  bool HasPID(uint32_t pid)
  {
    for(auto &tmp : entities){
      if(tmp.elementary_PID == pid) return true;
    }
    return  false;
  }
 
  void print();
};
 
struct MPEGNetworkInformationTable{
  bool HasPID(uint32_t /*pid*/)
  {
    return false;
  }
};
 
struct MPEGAdaptationField{
  uint32_t discontinuity_indicator;
  uint32_t random_access_indicator;
  uint32_t elementary_stream_priority_indicator;
  uint32_t PCR_flag;
  uint32_t OPCR_flag;
  uint32_t splicing_point_flag;
  uint32_t transport_private_data_flag;
  uint32_t adaptation_field_extension_flag;
  uint64_t program_clock_reference_base;
  uint32_t program_clock_reference_extension;
  uint64_t original_program_clock_reference_base;
  uint32_t original_program_clock_reference_extension;
  uint32_t splice_countdown;
  uint32_t transport_private_data_length;
  uint32_t private_data_byte[184];
  uint32_t adaptation_field_extension_length;
  uint32_t ltw_flag;
  uint32_t piecewise_rate_flag;
  uint32_t seamless_splice_flag;
  uint32_t ltw_valid_flag;
  uint32_t ltw_offset;
  uint32_t piecewise_rate;
  uint32_t Splice_type;
  uint64_t DTS_next_AU;
 
  void print();
};
 
struct MPEGTransportStreamHeader{
  uint32_t transport_error_indicator = 0;
  uint32_t payload_unit_start_indicator = 0;
  uint32_t transport_priority = 0;
  uint32_t PID = 0;
  uint32_t transport_scrambling_control = 0;
  uint32_t adaptation_field_control = 0;
  uint32_t continuity_counter = 0;
 
  void print();
};
 
struct PES_Sinc{
  std::vector<uint8_t> data;
};
 
struct TSCallbackData
{
  uint8_t *data;
  size_t size;
  uint64_t timestamp;
};
typedef std::function<void(TSCallbackData&)> TSCallback ;
 
struct MPEGTransportStreamData{
  MPEGTransportStreamHeader header;
  MPEGAdaptationField af;
 
  MPEGProgramAssociationSection pas;
  MPEGProgramMapTable pmt;
  MPEGNetworkInformationTable nit;
 
  std::unordered_map<uint32_t, PES_Sinc> pes_sinc;
 
  TSCallback callback;
};
 
void COGSCORE_DLL_API parse_mpeg_ts(MPEGTransportStreamData &parse, uint8_t *data, size_t size);
 
}// namespace ...