diff options
Diffstat (limited to 'src/picture_mxf.cc')
| -rw-r--r-- | src/picture_mxf.cc | 211 |
1 files changed, 211 insertions, 0 deletions
diff --git a/src/picture_mxf.cc b/src/picture_mxf.cc new file mode 100644 index 00000000..3221a88f --- /dev/null +++ b/src/picture_mxf.cc @@ -0,0 +1,211 @@ +/* + Copyright (C) 2012 Carl Hetherington <cth@carlh.net> + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program; if not, write to the Free Software + Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + +*/ + +/** @file src/picture_asset.cc + * @brief An asset made up of JPEG2000 files + */ + +#include <list> +#include <stdexcept> +#include <iostream> +#include <sstream> +#include <boost/filesystem.hpp> +#include <boost/lexical_cast.hpp> +#include <openjpeg.h> +#include <libxml++/nodes/element.h> +#include "AS_DCP.h" +#include "KM_fileio.h" +#include "picture_mxf.h" +#include "util.h" +#include "exceptions.h" +#include "xyz_frame.h" +#include "picture_mxf_writer.h" + +using std::string; +using std::ostream; +using std::list; +using std::vector; +using std::max; +using std::stringstream; +using std::pair; +using std::make_pair; +using std::istream; +using std::cout; +using boost::shared_ptr; +using boost::dynamic_pointer_cast; +using boost::lexical_cast; +using namespace dcp; + +PictureMXF::PictureMXF (boost::filesystem::path directory, boost::filesystem::path mxf_name) + : MXF (directory, mxf_name) +{ + +} + +void +PictureMXF::write_to_cpl (xmlpp::Element* node) const +{ + MXF::write_to_cpl (node); + + xmlpp::Node::NodeList c = node->get_children (); + xmlpp::Node::NodeList::iterator i = c.begin(); + while (i != c.end() && (*i)->get_name() != cpl_node_name ()) { + ++i; + } + + assert (i != c.end ()); + + (*i)->add_child ("FrameRate")->add_child_text (lexical_cast<string> (_edit_rate * edit_rate_factor ()) + " 1"); + if (_interop) { + stringstream s; + s << std::fixed << std::setprecision (2) << (float (_size.width) / _size.height); + (*i)->add_child ("ScreenAspectRatio")->add_child_text (s.str ()); + } else { + (*i)->add_child ("ScreenAspectRatio")->add_child_text (lexical_cast<string> (_size.width) + " " + lexical_cast<string> (_size.height)); + } +} + +bool +PictureMXF::descriptor_equals ( + ASDCP::JP2K::PictureDescriptor const & a, ASDCP::JP2K::PictureDescriptor const & b, boost::function<void (NoteType, string)> note + ) const +{ + if ( + a.EditRate != b.EditRate || + a.SampleRate != b.SampleRate || + a.StoredWidth != b.StoredWidth || + a.StoredHeight != b.StoredHeight || + a.AspectRatio != b.AspectRatio || + a.Rsize != b.Rsize || + a.Xsize != b.Xsize || + a.Ysize != b.Ysize || + a.XOsize != b.XOsize || + a.YOsize != b.YOsize || + a.XTsize != b.XTsize || + a.YTsize != b.YTsize || + a.XTOsize != b.XTOsize || + a.YTOsize != b.YTOsize || + a.Csize != b.Csize +// a.CodingStyleDefault != b.CodingStyleDefault || +// a.QuantizationDefault != b.QuantizationDefault + ) { + + note (ERROR, "video MXF picture descriptors differ"); + return false; + } + + if (a.ContainerDuration != b.ContainerDuration) { + note (ERROR, "video container durations differ"); + } + +// for (unsigned int j = 0; j < ASDCP::JP2K::MaxComponents; ++j) { +// if (a.ImageComponents[j] != b.ImageComponents[j]) { +// notes.pack_start ("video MXF picture descriptors differ"); +// } +// } + + return true; +} + +bool +PictureMXF::frame_buffer_equals ( + int frame, EqualityOptions opt, boost::function<void (NoteType, string)> note, + uint8_t const * data_A, unsigned int size_A, uint8_t const * data_B, unsigned int size_B + ) const +{ + if (size_A == size_B && memcmp (data_A, data_B, size_A) == 0) { + note (NOTE, "J2K identical"); + /* Easy result; the J2K data is identical */ + return true; + } + + /* Decompress the images to bitmaps */ + shared_ptr<XYZFrame> image_A = decompress_j2k (const_cast<uint8_t*> (data_A), size_A, 0); + shared_ptr<XYZFrame> image_B = decompress_j2k (const_cast<uint8_t*> (data_B), size_B, 0); + + /* Compare them */ + + vector<int> abs_diffs (image_A->size().width * image_A->size().height * 3); + int d = 0; + int max_diff = 0; + + for (int c = 0; c < 3; ++c) { + + if (image_A->size() != image_B->size()) { + note (ERROR, "image sizes for frame " + lexical_cast<string>(frame) + " differ"); + return false; + } + + int const pixels = image_A->size().width * image_A->size().height; + for (int j = 0; j < pixels; ++j) { + int const t = abs (image_A->data(c)[j] - image_B->data(c)[j]); + abs_diffs[d++] = t; + max_diff = max (max_diff, t); + } + } + + uint64_t total = 0; + for (vector<int>::iterator j = abs_diffs.begin(); j != abs_diffs.end(); ++j) { + total += *j; + } + + double const mean = double (total) / abs_diffs.size (); + + uint64_t total_squared_deviation = 0; + for (vector<int>::iterator j = abs_diffs.begin(); j != abs_diffs.end(); ++j) { + total_squared_deviation += pow (*j - mean, 2); + } + + double const std_dev = sqrt (double (total_squared_deviation) / abs_diffs.size()); + + note (NOTE, "mean difference " + lexical_cast<string> (mean) + ", deviation " + lexical_cast<string> (std_dev)); + + if (mean > opt.max_mean_pixel_error) { + note ( + ERROR, + "mean " + lexical_cast<string>(mean) + + " out of range " + lexical_cast<string>(opt.max_mean_pixel_error) + + " in frame " + lexical_cast<string>(frame) + ); + + return false; + } + + if (std_dev > opt.max_std_dev_pixel_error) { + note ( + ERROR, + "standard deviation " + lexical_cast<string>(std_dev) + + " out of range " + lexical_cast<string>(opt.max_std_dev_pixel_error) + + " in frame " + lexical_cast<string>(frame) + ); + + return false; + } + + return true; +} + +string +PictureMXF::key_type () const +{ + return "MDIK"; +} + + + |