src/lib/audio_analysis.cc

   1 /*
   2     Copyright (C) 2012-2015 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 #include "audio_analysis.h"
  21 #include "cross.h"
  22 #include "util.h"
  23 #include "raw_convert.h"
  24 #include <libxml++/libxml++.h>
  25 #include <boost/filesystem.hpp>
  26 #include <boost/foreach.hpp>
  27 #include <stdint.h>
  28 #include <cmath>
  29 #include <cstdio>
  30 #include <iostream>
  31 #include <inttypes.h>
  32
  33 using std::ostream;
  34 using std::istream;
  35 using std::string;
  36 using std::vector;
  37 using std::cout;
  38 using std::max;
  39 using std::list;
  40 using boost::shared_ptr;
  41
  42 AudioAnalysis::AudioAnalysis (int channels)
  43 {
  44         _data.resize (channels);
  45 }
  46
  47 AudioAnalysis::AudioAnalysis (boost::filesystem::path filename)
  48 {
  49         cxml::Document f ("AudioAnalysis");
  50         f.read_file (filename);
  51
  52         BOOST_FOREACH (cxml::NodePtr i, f.node_children ("Channel")) {
  53                 vector<AudioPoint> channel;
  54
  55                 BOOST_FOREACH (cxml::NodePtr j, i->node_children ("Point")) {
  56                         channel.push_back (AudioPoint (j));
  57                 }
  58
  59                 _data.push_back (channel);
  60         }
  61
  62         _peak = f.number_child<float> ("Peak");
  63         _peak_time = DCPTime (f.number_child<DCPTime::Type> ("PeakTime"));
  64         _analysis_gain = f.optional_number_child<double> ("AnalysisGain");
  65 }
  66
  67 void
  68 AudioAnalysis::add_point (int c, AudioPoint const & p)
  69 {
  70         DCPOMATIC_ASSERT (c < channels ());
  71         _data[c].push_back (p);
  72 }
  73
  74 AudioPoint
  75 AudioAnalysis::get_point (int c, int p) const
  76 {
  77         DCPOMATIC_ASSERT (p < points (c));
  78         return _data[c][p];
  79 }
  80
  81 int
  82 AudioAnalysis::channels () const
  83 {
  84         return _data.size ();
  85 }
  86
  87 int
  88 AudioAnalysis::points (int c) const
  89 {
  90         DCPOMATIC_ASSERT (c < channels ());
  91         return _data[c].size ();
  92 }
  93
  94 void
  95 AudioAnalysis::write (boost::filesystem::path filename)
  96 {
  97         shared_ptr<xmlpp::Document> doc (new xmlpp::Document);
  98         xmlpp::Element* root = doc->create_root_node ("AudioAnalysis");
  99
 100         BOOST_FOREACH (vector<AudioPoint>& i, _data) {
 101                 xmlpp::Element* channel = root->add_child ("Channel");
 102                 BOOST_FOREACH (AudioPoint& j, i) {
 103                         j.as_xml (channel->add_child ("Point"));
 104                 }
 105         }
 106
 107         if (_peak) {
 108                 root->add_child("Peak")->add_child_text (raw_convert<string> (_peak.get ()));
 109                 root->add_child("PeakTime")->add_child_text (raw_convert<string> (_peak_time.get().get ()));
 110         }
 111
 112         if (_analysis_gain) {
 113                 root->add_child("AnalysisGain")->add_child_text (raw_convert<string> (_analysis_gain.get ()));
 114         }
 115
 116         doc->write_to_file_formatted (filename.string ());
 117 }