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
  27 namespace libdcp {
  28
  29 class MonoPictureAsset;
  30 class StereoPictureAsset;
  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 FrameInfo write (uint8_t *, int) = 0;
  53         virtual void fake_write (int) = 0;
  54         virtual void finalize () = 0;
  55
  56 protected:
  57
  58         PictureAssetWriter (bool, MXFMetadata const &);
  59         virtual void start (uint8_t *, int) = 0;
  60
  61         /** Number of picture frames written to the asset so far */
  62         int _frames_written;
  63         bool _started;
  64         /** true if finalize() has been called */
  65         bool _finalized;
  66         bool _overwrite;
  67         MXFMetadata _metadata;
  68 };
  69
  70 /** A helper class for writing to MonoPictureAssets progressively (i.e. writing frame-by-frame,
  71  *  rather than giving libdcp all the frames in one go).
  72  *
  73  *  Objects of this class can only be created with MonoPictureAsset::start_write().
  74  *
  75  *  Frames can be written to the MonoPictureAsset by calling write() with a JPEG2000 image
  76  *  (a verbatim .j2 file).  finalize() must be called after the last frame has been written.
  77  *  The action of finalize() can't be done in MonoPictureAssetWriter's destructor as it may
  78  *  throw an exception.
  79  */
  80 class MonoPictureAssetWriter : public PictureAssetWriter
  81 {
  82 public:
  83         FrameInfo write (uint8_t* data, int size);
  84         void fake_write (int size);
  85         void finalize ();
  86
  87 private:
  88         friend class MonoPictureAsset;
  89
  90         MonoPictureAssetWriter (MonoPictureAsset *, bool, MXFMetadata const &);
  91         void start (uint8_t *, int);
  92
  93         /* do this with an opaque pointer so we don't have to include
  94            ASDCP headers
  95         */
  96
  97         struct ASDCPState;
  98         boost::shared_ptr<ASDCPState> _state;
  99
 100         MonoPictureAsset* _asset;
 101 };
 102
 103 class StereoPictureAssetWriter : public PictureAssetWriter
 104 {
 105 public:
 106         FrameInfo write (uint8_t* data, int size);
 107         void fake_write (int size);
 108         void finalize ();
 109
 110 private:
 111         friend class StereoPictureAsset;
 112
 113         StereoPictureAssetWriter (StereoPictureAsset *, bool, MXFMetadata const &);
 114         void start (uint8_t *, int);
 115
 116         /* do this with an opaque pointer so we don't have to include
 117            ASDCP headers
 118         */
 119
 120         struct ASDCPState;
 121         boost::shared_ptr<ASDCPState> _state;
 122
 123         StereoPictureAsset* _asset;
 124 };
 125
 126 }