src/sound_mxf.cc

   1 /*
   2     Copyright (C) 2012-2014 Carl Hetherington <cth@carlh.net>
   3
   4     This program is free software; you can redistribute it and/or modify
   5     it under the terms of the GNU General Public License as published by
   6     the Free Software Foundation; either version 2 of the License, or
   7     (at your option) any later version.
   8
   9     This program is distributed in the hope that it will be useful,
  10     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12     GNU General Public License for more details.
  13
  14     You should have received a copy of the GNU General Public License
  15     along with this program; if not, write to the Free Software
  16     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17
  18 */
  19
  20 /** @file  src/sound_mxf.cc
  21  *  @brief SoundMXF class.
  22  */
  23
  24 #include "sound_mxf.h"
  25 #include "util.h"
  26 #include "exceptions.h"
  27 #include "sound_frame.h"
  28 #include "sound_mxf_writer.h"
  29 #include "compose.hpp"
  30 #include "KM_fileio.h"
  31 #include "AS_DCP.h"
  32 #include <libxml++/nodes/element.h>
  33 #include <boost/filesystem.hpp>
  34 #include <iostream>
  35 #include <stdexcept>
  36
  37 using std::string;
  38 using std::stringstream;
  39 using std::ostream;
  40 using std::vector;
  41 using std::list;
  42 using boost::shared_ptr;
  43 using namespace dcp;
  44
  45 SoundMXF::SoundMXF (boost::filesystem::path file)
  46         : MXF (file)
  47         , _intrinsic_duration (0)
  48         , _channels (0)
  49         , _sampling_rate (0)
  50 {
  51         ASDCP::PCM::MXFReader reader;
  52         Kumu::Result_t r = reader.OpenRead (file.string().c_str());
  53         if (ASDCP_FAILURE (r)) {
  54                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file.string(), r));
  55         }
  56
  57         ASDCP::PCM::AudioDescriptor desc;
  58         if (ASDCP_FAILURE (reader.FillAudioDescriptor (desc))) {
  59                 boost::throw_exception (DCPReadError ("could not read audio MXF information"));
  60         }
  61
  62         _sampling_rate = desc.AudioSamplingRate.Numerator / desc.AudioSamplingRate.Denominator;
  63         _channels = desc.ChannelCount;
  64         _edit_rate = Fraction (desc.EditRate.Numerator, desc.EditRate.Denominator);
  65
  66         _intrinsic_duration = desc.ContainerDuration;
  67
  68         ASDCP::WriterInfo info;
  69         if (ASDCP_FAILURE (reader.FillWriterInfo (info))) {
  70                 boost::throw_exception (DCPReadError ("could not read audio MXF information"));
  71         }
  72
  73         read_writer_info (info);
  74
  75 }
  76
  77 SoundMXF::SoundMXF (Fraction edit_rate, int sampling_rate, int channels)
  78         : _edit_rate (edit_rate)
  79         , _intrinsic_duration (0)
  80         , _channels (channels)
  81         , _sampling_rate (sampling_rate)
  82 {
  83
  84 }
  85
  86 bool
  87 SoundMXF::equals (shared_ptr<const Asset> other, EqualityOptions opt, NoteHandler note) const
  88 {
  89         if (!MXF::equals (other, opt, note)) {
  90                 return false;
  91         }
  92
  93         ASDCP::PCM::MXFReader reader_A;
  94         Kumu::Result_t r = reader_A.OpenRead (file().string().c_str());
  95         if (ASDCP_FAILURE (r)) {
  96                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
  97         }
  98
  99         ASDCP::PCM::MXFReader reader_B;
 100         r = reader_B.OpenRead (other->file().string().c_str());
 101         if (ASDCP_FAILURE (r)) {
 102                 boost::throw_exception (MXFFileError ("could not open MXF file for reading", file().string(), r));
 103         }
 104
 105         ASDCP::PCM::AudioDescriptor desc_A;
 106         if (ASDCP_FAILURE (reader_A.FillAudioDescriptor (desc_A))) {
 107                 boost::throw_exception (DCPReadError ("could not read audio MXF information"));
 108         }
 109         ASDCP::PCM::AudioDescriptor desc_B;
 110         if (ASDCP_FAILURE (reader_B.FillAudioDescriptor (desc_B))) {
 111                 boost::throw_exception (DCPReadError ("could not read audio MXF information"));
 112         }
 113
 114         if (
 115                 desc_A.EditRate != desc_B.EditRate ||
 116                 desc_A.AudioSamplingRate != desc_B.AudioSamplingRate ||
 117                 desc_A.Locked != desc_B.Locked ||
 118                 desc_A.ChannelCount != desc_B.ChannelCount ||
 119                 desc_A.QuantizationBits != desc_B.QuantizationBits ||
 120                 desc_A.BlockAlign != desc_B.BlockAlign ||
 121                 desc_A.AvgBps != desc_B.AvgBps ||
 122                 desc_A.LinkedTrackID != desc_B.LinkedTrackID ||
 123                 desc_A.ContainerDuration != desc_B.ContainerDuration
 124 //              desc_A.ChannelFormat != desc_B.ChannelFormat ||
 125                 ) {
 126
 127                 note (DCP_ERROR, "audio MXF picture descriptors differ");
 128                 return false;
 129         }
 130
 131         ASDCP::PCM::FrameBuffer buffer_A (1 * Kumu::Megabyte);
 132         ASDCP::PCM::FrameBuffer buffer_B (1 * Kumu::Megabyte);
 133
 134         for (int i = 0; i < _intrinsic_duration; ++i) {
 135                 if (ASDCP_FAILURE (reader_A.ReadFrame (i, buffer_A))) {
 136                         boost::throw_exception (DCPReadError ("could not read audio frame"));
 137                 }
 138
 139                 if (ASDCP_FAILURE (reader_B.ReadFrame (i, buffer_B))) {
 140                         boost::throw_exception (DCPReadError ("could not read audio frame"));
 141                 }
 142
 143                 if (buffer_A.Size() != buffer_B.Size()) {
 144                         note (DCP_ERROR, String::compose ("sizes of audio data for frame %1 differ", i));
 145                         return false;
 146                 }
 147
 148                 if (memcmp (buffer_A.RoData(), buffer_B.RoData(), buffer_A.Size()) != 0) {
 149                         for (uint32_t i = 0; i < buffer_A.Size(); ++i) {
 150                                 int const d = abs (buffer_A.RoData()[i] - buffer_B.RoData()[i]);
 151                                 if (d > opt.max_audio_sample_error) {
 152                                         note (DCP_ERROR, String::compose ("PCM data difference of %1", d));
 153                                         return false;
 154                                 }
 155                         }
 156                 }
 157         }
 158
 159         return true;
 160 }
 161
 162 shared_ptr<const SoundFrame>
 163 SoundMXF::get_frame (int n) const
 164 {
 165         /* XXX: should add on entry point here? */
 166         return shared_ptr<const SoundFrame> (new SoundFrame (file(), n, _decryption_context));
 167 }
 168
 169 shared_ptr<SoundMXFWriter>
 170 SoundMXF::start_write (boost::filesystem::path file, Standard standard)
 171 {
 172         /* XXX: can't we use a shared_ptr here? */
 173         return shared_ptr<SoundMXFWriter> (new SoundMXFWriter (this, file, standard));
 174 }