the story so far
[asdcplib.git] / src / AS_DCP_PCM.cpp
index 6cf907e1c01be403d08c4ce005efce932e3329f1..f93815bad1a6a89013173e6f566f0d9cb4c2e0ed 100755 (executable)
@@ -1,5 +1,5 @@
 /*
-Copyright (c) 2004-2009, John Hurst
+Copyright (c) 2004-2010, John Hurst
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without
@@ -53,7 +53,7 @@ Result_t
 PCM_ADesc_to_MD(PCM::AudioDescriptor& ADesc, MXF::WaveAudioDescriptor* ADescObj)
 {
   ASDCP_TEST_NULL(ADescObj);
-  ADescObj->SampleRate = ADesc.SampleRate;
+  ADescObj->SampleRate = ADesc.EditRate;
   ADescObj->AudioSamplingRate = ADesc.AudioSamplingRate;
   ADescObj->Locked = ADesc.Locked;
   ADescObj->ChannelCount = ADesc.ChannelCount;
@@ -88,7 +88,7 @@ ASDCP::Result_t
 MD_to_PCM_ADesc(MXF::WaveAudioDescriptor* ADescObj, PCM::AudioDescriptor& ADesc)
 {
   ASDCP_TEST_NULL(ADescObj);
-  ADesc.SampleRate = ADescObj->SampleRate;
+  ADesc.EditRate = ADescObj->SampleRate;
   ADesc.AudioSamplingRate = ADescObj->AudioSamplingRate;
   ADesc.Locked = ADescObj->Locked;
   ADesc.ChannelCount = ADescObj->ChannelCount;
@@ -120,7 +120,7 @@ MD_to_PCM_ADesc(MXF::WaveAudioDescriptor* ADescObj, PCM::AudioDescriptor& ADesc)
 std::ostream&
 ASDCP::PCM::operator << (std::ostream& strm, const AudioDescriptor& ADesc)
 {
-  strm << "        SampleRate: " << ADesc.SampleRate.Numerator << "/" << ADesc.SampleRate.Denominator << std::endl;
+  strm << "        SampleRate: " << ADesc.EditRate.Numerator << "/" << ADesc.EditRate.Denominator << std::endl;
   strm << " AudioSamplingRate: " << ADesc.AudioSamplingRate.Numerator << "/" << ADesc.AudioSamplingRate.Denominator << std::endl;
   strm << "            Locked: " << (unsigned) ADesc.Locked << std::endl;
   strm << "      ChannelCount: " << (unsigned) ADesc.ChannelCount << std::endl;
@@ -141,7 +141,7 @@ ASDCP::PCM::AudioDescriptorDump(const AudioDescriptor& ADesc, FILE* stream)
     stream = stderr;
 
   fprintf(stream, "\
-        SampleRate: %d/%d\n\
+        EditRate: %d/%d\n\
  AudioSamplingRate: %d/%d\n\
             Locked: %u\n\
       ChannelCount: %u\n\
@@ -150,8 +150,8 @@ ASDCP::PCM::AudioDescriptorDump(const AudioDescriptor& ADesc, FILE* stream)
             AvgBps: %u\n\
      LinkedTrackID: %u\n\
  ContainerDuration: %u\n",
-         ADesc.SampleRate.Numerator ,ADesc.SampleRate.Denominator,
-         ADesc.AudioSamplingRate.Numerator ,ADesc.AudioSamplingRate.Denominator,
+         ADesc.EditRate.Numerator, ADesc.EditRate.Denominator,
+         ADesc.AudioSamplingRate.NumeratorADesc.AudioSamplingRate.Denominator,
          ADesc.Locked,
          ADesc.ChannelCount,
          ADesc.QuantizationBits,
@@ -194,11 +194,12 @@ calc_CBR_frame_size(ASDCP::WriterInfo& Info, const ASDCP::PCM::AudioDescriptor&
 class ASDCP::PCM::MXFReader::h__Reader : public ASDCP::h__Reader
 {
   ASDCP_NO_COPY_CONSTRUCT(h__Reader);
+  h__Reader();
 
 public:
   AudioDescriptor m_ADesc;
 
-  h__Reader() {}
+  h__Reader(const Dictionary& d) : ASDCP::h__Reader(d) {}
   ~h__Reader() {}
   Result_t    OpenRead(const char*);
   Result_t    ReadFrame(ui32_t, FrameBuffer&, AESDecContext*, HMACContext*);
@@ -224,19 +225,22 @@ ASDCP::PCM::MXFReader::h__Reader::OpenRead(const char* filename)
 
   // check for sample/frame rate sanity
   if ( ASDCP_SUCCESS(result)
-       && m_ADesc.SampleRate != EditRate_24
-       && m_ADesc.SampleRate != EditRate_48
-       && m_ADesc.SampleRate != EditRate_23_98 )
+       && m_ADesc.EditRate != EditRate_24
+       && m_ADesc.EditRate != EditRate_25
+       && m_ADesc.EditRate != EditRate_30
+       && m_ADesc.EditRate != EditRate_48
+       && m_ADesc.EditRate != EditRate_50
+       && m_ADesc.EditRate != EditRate_60
+       && m_ADesc.EditRate != EditRate_23_98 )
     {
-      DefaultLogSink().Error("PCM file SampleRate is not 24/1, 48/1 or 24000/1001: %08x/%08x\n", // lu
-                            m_ADesc.SampleRate.Numerator, m_ADesc.SampleRate.Denominator);
+      DefaultLogSink().Error("PCM file EditRate is not a supported value: %d/%d\n", // lu
+                            m_ADesc.EditRate.Numerator, m_ADesc.EditRate.Denominator);
 
       // oh, they gave us the audio sampling rate instead, assume 24/1
-      if ( m_ADesc.SampleRate == SampleRate_48k )
+      if ( m_ADesc.EditRate == SampleRate_48k )
        {
-         DefaultLogSink().Warn("adjusting SampleRate to 24/1\n"); 
-         m_ADesc.SampleRate.Numerator = 24;
-         m_ADesc.SampleRate.Denominator = 1;
+         DefaultLogSink().Warn("adjusting EditRate to 24/1\n"); 
+         m_ADesc.EditRate = EditRate_24;
        }
       else
        {
@@ -266,7 +270,8 @@ ASDCP::PCM::MXFReader::h__Reader::ReadFrame(ui32_t FrameNum, FrameBuffer& FrameB
   if ( ! m_File.IsOpen() )
     return RESULT_INIT;
 
-  return ReadEKLVFrame(FrameNum, FrameBuf, Dict::ul(MDD_WAVEssence), Ctx, HMAC);
+  assert(m_Dict);
+  return ReadEKLVFrame(FrameNum, FrameBuf, m_Dict->ul(MDD_WAVEssence), Ctx, HMAC);
 }
 
 //------------------------------------------------------------------------------------------
@@ -290,7 +295,7 @@ ASDCP::PCM::FrameBuffer::Dump(FILE* stream, ui32_t dump_len) const
 
 ASDCP::PCM::MXFReader::MXFReader()
 {
-  m_Reader = new h__Reader;
+  m_Reader = new h__Reader(DefaultCompositeDict());
 }
 
 
@@ -368,20 +373,33 @@ ASDCP::PCM::MXFReader::DumpIndex(FILE* stream) const
     m_Reader->m_FooterPart.Dump(stream);
 }
 
+//
+ASDCP::Result_t
+ASDCP::PCM::MXFReader::Close() const
+{
+  if ( m_Reader && m_Reader->m_File.IsOpen() )
+    {
+      m_Reader->Close();
+      return RESULT_OK;
+    }
+
+  return RESULT_INIT;
+}
+
 
 //------------------------------------------------------------------------------------------
 
 //
 class ASDCP::PCM::MXFWriter::h__Writer : public ASDCP::h__Writer
 {
+  ASDCP_NO_COPY_CONSTRUCT(h__Writer);
+  h__Writer();
+
 public:
   AudioDescriptor m_ADesc;
   byte_t          m_EssenceUL[SMPTE_UL_LENGTH];
-
-
-  ASDCP_NO_COPY_CONSTRUCT(h__Writer);
   
-  h__Writer(){
+  h__Writer(const Dictionary& d) : ASDCP::h__Writer(d) {
     memset(m_EssenceUL, 0, SMPTE_UL_LENGTH);
   }
 
@@ -408,7 +426,7 @@ ASDCP::PCM::MXFWriter::h__Writer::OpenWrite(const char* filename, ui32_t HeaderS
   if ( ASDCP_SUCCESS(result) )
     {
       m_HeaderSize = HeaderSize;
-      m_EssenceDescriptor = new WaveAudioDescriptor;
+      m_EssenceDescriptor = new WaveAudioDescriptor(m_Dict);
       result = m_State.Goto_INIT();
     }
 
@@ -423,12 +441,16 @@ ASDCP::PCM::MXFWriter::h__Writer::SetSourceStream(const AudioDescriptor& ADesc)
   if ( ! m_State.Test_INIT() )
     return RESULT_STATE;
 
-  if ( ADesc.SampleRate != EditRate_24
-       && ADesc.SampleRate != EditRate_48
-       && ADesc.SampleRate != EditRate_23_98 )
+  if ( ADesc.EditRate != EditRate_24
+       && ADesc.EditRate != EditRate_25
+       && ADesc.EditRate != EditRate_30
+       && ADesc.EditRate != EditRate_48
+       && ADesc.EditRate != EditRate_50
+       && ADesc.EditRate != EditRate_60
+       && ADesc.EditRate != EditRate_23_98 )
     {
-      DefaultLogSink().Error("AudioDescriptor.SampleRate is not 24/1, 48/1 or 24000/1001: %d/%d\n",
-                            ADesc.SampleRate.Numerator, ADesc.SampleRate.Denominator);
+      DefaultLogSink().Error("AudioDescriptor.EditRate is not a supported value: %d/%d\n",
+                            ADesc.EditRate.Numerator, ADesc.EditRate.Denominator);
       return RESULT_RAW_FORMAT;
     }
 
@@ -439,21 +461,27 @@ ASDCP::PCM::MXFWriter::h__Writer::SetSourceStream(const AudioDescriptor& ADesc)
       return RESULT_RAW_FORMAT;
     }
 
+  assert(m_Dict);
   m_ADesc = ADesc;
+
   Result_t result = PCM_ADesc_to_MD(m_ADesc, (WaveAudioDescriptor*)m_EssenceDescriptor);
   
-  if ( ASDCP_SUCCESS(result) )
-      result = WriteMXFHeader(PCM_PACKAGE_LABEL, UL(Dict::ul(MDD_WAVWrapping)),
-                             SOUND_DEF_LABEL,   UL(Dict::ul(MDD_SoundDataDef)),
-                             m_ADesc.SampleRate, 24 /* TCFrameRate */, calc_CBR_frame_size(m_Info, m_ADesc));
-
   if ( ASDCP_SUCCESS(result) )
     {
-      memcpy(m_EssenceUL, Dict::ul(MDD_WAVEssence), SMPTE_UL_LENGTH);
+      memcpy(m_EssenceUL, m_Dict->ul(MDD_WAVEssence), SMPTE_UL_LENGTH);
       m_EssenceUL[SMPTE_UL_LENGTH-1] = 1; // first (and only) essence container
       result = m_State.Goto_READY();
     }
 
+  if ( ASDCP_SUCCESS(result) )
+    {
+      ui32_t TCFrameRate = ( m_ADesc.EditRate == EditRate_23_98  ) ? 24 : m_ADesc.EditRate.Numerator;
+      
+      result = WriteMXFHeader(PCM_PACKAGE_LABEL, UL(m_Dict->ul(MDD_WAVWrapping)),
+                             SOUND_DEF_LABEL, UL(m_EssenceUL), UL(m_Dict->ul(MDD_SoundDataDef)),
+                             m_ADesc.EditRate, TCFrameRate, calc_CBR_frame_size(m_Info, m_ADesc));
+    }
+
   return result;
 }
 
@@ -512,7 +540,11 @@ ASDCP::Result_t
 ASDCP::PCM::MXFWriter::OpenWrite(const char* filename, const WriterInfo& Info,
                                 const AudioDescriptor& ADesc, ui32_t HeaderSize)
 {
-  m_Writer = new h__Writer;
+  if ( Info.LabelSetType == LS_MXF_SMPTE )
+    m_Writer = new h__Writer(DefaultSMPTEDict());
+  else
+    m_Writer = new h__Writer(DefaultInteropDict());
+
   m_Writer->m_Info = Info;
   
   Result_t result = m_Writer->OpenWrite(filename, HeaderSize);