src/picture_asset_writer.h

   1 /*
   2     Copyright (C) 2012-2013 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 <stdint.h>
  21 #include <string>
  22 #include <fstream>
  23 #include <boost/shared_ptr.hpp>
  24 #include <boost/utility.hpp>
  25 #include "metadata.h"
  26 #include "types.h"
  27
  28 namespace libdcp {
  29
  30 class PictureAsset;
  31
  32 struct FrameInfo
  33 {
  34         FrameInfo (uint64_t o, uint64_t s, std::string h)
  35                 : offset (o)
  36                 , size (s)
  37                 , hash (h)
  38         {}
  39
  40         FrameInfo (std::istream& s);
  41
  42         void write (std::ostream& s);
  43
  44         uint64_t offset;
  45         uint64_t size;
  46         std::string hash;
  47 };
  48
  49 class PictureAssetWriter : public boost::noncopyable
  50 {
  51 public:
  52         virtual ~PictureAssetWriter () {}
  53         virtual FrameInfo write (uint8_t *, int) = 0;
  54         virtual void finalize () = 0;
  55         virtual void fake_write (int) = 0;
  56
  57 protected:
  58         template <class P, class Q>
  59         friend void start (PictureAssetWriter *, boost::shared_ptr<P>, Q *, uint8_t *, int);
  60
  61         PictureAssetWriter (PictureAsset *, bool, bool, MXFMetadata const &);
  62
  63         PictureAsset* _asset;
  64
  65         /** Number of picture frames written to the asset so far.  For stereo assets
  66          *  this will be incremented for each eye (i.e. there will be twice the number
  67          *  of frames as in a mono asset).
  68          */
  69         int _frames_written;
  70         bool _started;
  71         /** true if finalize() has been called */
  72         bool _finalized;
  73         bool _overwrite;
  74         bool _interop;
  75         MXFMetadata _metadata;
  76 };
  77
  78 /** A helper class for writing to MonoPictureAssets progressively (i.e. writing frame-by-frame,
  79  *  rather than giving libdcp all the frames in one go).
  80  *
  81  *  Objects of this class can only be created with MonoPictureAsset::start_write().
  82  *
  83  *  Frames can be written to the MonoPictureAsset by calling write() with a JPEG2000 image
  84  *  (a verbatim .j2c file).  finalize() must be called after the last frame has been written.
  85  *  The action of finalize() can't be done in MonoPictureAssetWriter's destructor as it may
  86  *  throw an exception.
  87  */
  88 class MonoPictureAssetWriter : public PictureAssetWriter
  89 {
  90 public:
  91         FrameInfo write (uint8_t *, int);
  92         void fake_write (int size);
  93         void finalize ();
  94
  95 private:
  96         friend class MonoPictureAsset;
  97
  98         MonoPictureAssetWriter (PictureAsset *, bool, bool, MXFMetadata const &);
  99         void start (uint8_t *, int);
 100
 101         /* do this with an opaque pointer so we don't have to include
 102            ASDCP headers
 103         */
 104
 105         struct ASDCPState;
 106         boost::shared_ptr<ASDCPState> _state;
 107 };
 108
 109 /** A helper class for writing to StereoPictureAssets progressively (i.e. writing frame-by-frame,
 110  *  rather than giving libdcp all the frames in one go).
 111  *
 112  *  Objects of this class can only be created with StereoPictureAsset::start_write().
 113  *
 114  *  Frames can be written to the MonoPictureAsset by calling write() with a JPEG2000 image
 115  *  (a verbatim .j2c file).  finalize() must be called after the last frame has been written.
 116  *  The action of finalize() can't be done in MonoPictureAssetWriter's destructor as it may
 117  *  throw an exception.
 118  */
 119 class StereoPictureAssetWriter : public PictureAssetWriter
 120 {
 121 public:
 122         FrameInfo write (uint8_t *, int);
 123         void fake_write (int size);
 124         void finalize ();
 125
 126 private:
 127         friend class StereoPictureAsset;
 128
 129         StereoPictureAssetWriter (PictureAsset *, bool, bool, MXFMetadata const &);
 130         void start (uint8_t *, int);
 131
 132         /* do this with an opaque pointer so we don't have to include
 133            ASDCP headers
 134         */
 135
 136         struct ASDCPState;
 137         boost::shared_ptr<ASDCPState> _state;
 138
 139         libdcp::Eye _next_eye;
 140 };
 141
 142 }