Merge master and multifarious hackery.
[dcpomatic.git] / src / lib / dcp_video_frame.cc
index 3e58203ede9eecee86558416b169c8fed01e69ad..1c1838df79403f046f62511ec84bbe04273e4897 100644 (file)
@@ -47,7 +47,6 @@
 #include "dcp_video_frame.h"
 #include "lut.h"
 #include "config.h"
-#include "options.h"
 #include "exceptions.h"
 #include "server.h"
 #include "util.h"
 #include "log.h"
 #include "subtitle.h"
 
+#include "i18n.h"
+
 using std::string;
 using std::stringstream;
 using std::ofstream;
+using std::cout;
 using boost::shared_ptr;
+using libdcp::Size;
 
 /** Construct a DCP video frame.
  *  @param input Input image.
  *  @param out Required size of output, in pixels (including any padding).
  *  @param s Scaler to use.
  *  @param p Number of pixels of padding either side of the image.
- *  @param f Index of the frame within the Film.
- *  @param fps Frames per second of the Film.
+ *  @param f Index of the frame within the DCP.
+ *  @param fps Frames per second of the Film's source.
  *  @param pp FFmpeg post-processing string to use.
  *  @param clut Colour look-up table to use (see Config::colour_lut_index ())
  *  @param bw J2K bandwidth to use (see Config::j2k_bandwidth ())
  *  @param l Log to write to.
  */
 DCPVideoFrame::DCPVideoFrame (
-       shared_ptr<Image> yuv, shared_ptr<Subtitle> sub,
+       shared_ptr<const Image> yuv, shared_ptr<Subtitle> sub,
        Size out, int p, int subtitle_offset, float subtitle_scale,
-       Scaler const * s, int f, float fps, string pp, int clut, int bw, Log* l
+       Scaler const * s, int f, int dcp_fps, int clut, int bw, shared_ptr<Log> l
        )
        : _input (yuv)
        , _subtitle (sub)
@@ -86,10 +89,8 @@ DCPVideoFrame::DCPVideoFrame (
        , _subtitle_scale (subtitle_scale)
        , _scaler (s)
        , _frame (f)
-         /* we round here; not sure if this is right */
-       , _frames_per_second (rint (fps))
-       , _post_process (pp)
-       , _colour_lut_index (clut)
+       , _frames_per_second (dcp_fps)
+       , _colour_lut (clut)
        , _j2k_bandwidth (bw)
        , _log (l)
        , _image (0)
@@ -118,7 +119,7 @@ DCPVideoFrame::create_openjpeg_container ()
 
        _image = opj_image_create (3, &_cmptparm[0], CLRSPC_SRGB);
        if (_image == 0) {
-               throw EncodeError ("could not create libopenjpeg image");
+               throw EncodeError (N_("could not create libopenjpeg image"));
        }
 
        _image->x0 = 0;
@@ -154,11 +155,7 @@ DCPVideoFrame::~DCPVideoFrame ()
 shared_ptr<EncodedData>
 DCPVideoFrame::encode_locally ()
 {
-       if (!_post_process.empty ()) {
-               _input = _input->post_process (_post_process);
-       }
-       
-       shared_ptr<Image> prepared = _input->scale_and_convert_to_rgb (_out_size, _padding, _scaler);
+       shared_ptr<Image> prepared = _input->scale_and_convert_to_rgb (_out_size, _padding, _scaler, true);
 
        if (_subtitle) {
                Rect tx = subtitle_transformed_area (
@@ -167,7 +164,7 @@ DCPVideoFrame::encode_locally ()
                        _subtitle->area(), _subtitle_offset, _subtitle_scale
                        );
 
-               shared_ptr<Image> im = _subtitle->image()->scale (tx.size(), _scaler);
+               shared_ptr<Image> im = _subtitle->image()->scale (tx.size(), _scaler, true);
                prepared->alpha_blend (im, tx.position());
        }
 
@@ -189,22 +186,22 @@ DCPVideoFrame::encode_locally ()
                for (int x = 0; x < _out_size.width; ++x) {
 
                        /* In gamma LUT (converting 8-bit input to 12-bit) */
-                       s.r = lut_in[_colour_lut_index][*p++ << 4];
-                       s.g = lut_in[_colour_lut_index][*p++ << 4];
-                       s.b = lut_in[_colour_lut_index][*p++ << 4];
+                       s.r = lut_in[_colour_lut][*p++ << 4];
+                       s.g = lut_in[_colour_lut][*p++ << 4];
+                       s.b = lut_in[_colour_lut][*p++ << 4];
                        
                        /* RGB to XYZ Matrix */
-                       d.x = ((s.r * color_matrix[_colour_lut_index][0][0]) +
-                              (s.g * color_matrix[_colour_lut_index][0][1]) +
-                              (s.b * color_matrix[_colour_lut_index][0][2]));
+                       d.x = ((s.r * color_matrix[_colour_lut][0][0]) +
+                              (s.g * color_matrix[_colour_lut][0][1]) +
+                              (s.b * color_matrix[_colour_lut][0][2]));
                        
-                       d.y = ((s.r * color_matrix[_colour_lut_index][1][0]) +
-                              (s.g * color_matrix[_colour_lut_index][1][1]) +
-                              (s.b * color_matrix[_colour_lut_index][1][2]));
+                       d.y = ((s.r * color_matrix[_colour_lut][1][0]) +
+                              (s.g * color_matrix[_colour_lut][1][1]) +
+                              (s.b * color_matrix[_colour_lut][1][2]));
                        
-                       d.z = ((s.r * color_matrix[_colour_lut_index][2][0]) +
-                              (s.g * color_matrix[_colour_lut_index][2][1]) +
-                              (s.b * color_matrix[_colour_lut_index][2][2]));
+                       d.z = ((s.r * color_matrix[_colour_lut][2][0]) +
+                              (s.g * color_matrix[_colour_lut][2][1]) +
+                              (s.b * color_matrix[_colour_lut][2][2]));
                        
                        /* DCI companding */
                        d.x = d.x * DCI_COEFFICENT * (DCI_LUT_SIZE - 1);
@@ -264,7 +261,7 @@ DCPVideoFrame::encode_locally ()
        _parameters->tcp_numlayers++;
        _parameters->cp_disto_alloc = 1;
        _parameters->cp_rsiz = CINEMA2K;
-       _parameters->cp_comment = strdup ("DVD-o-matic");
+       _parameters->cp_comment = strdup (N_("DCP-o-matic"));
        _parameters->cp_cinema = CINEMA2K_24;
 
        /* 3 components, so use MCT */
@@ -277,7 +274,7 @@ DCPVideoFrame::encode_locally ()
        /* get a J2K compressor handle */
        _cinfo = opj_create_compress (CODEC_J2K);
        if (_cinfo == 0) {
-               throw EncodeError ("could not create JPEG2000 encoder");
+               throw EncodeError (N_("could not create JPEG2000 encoder"));
        }
 
        /* Set event manager to null (openjpeg 1.3 bug) */
@@ -288,15 +285,15 @@ DCPVideoFrame::encode_locally ()
 
        _cio = opj_cio_open ((opj_common_ptr) _cinfo, 0, 0);
        if (_cio == 0) {
-               throw EncodeError ("could not open JPEG2000 stream");
+               throw EncodeError (N_("could not open JPEG2000 stream"));
        }
 
        int const r = opj_encode (_cinfo, _cio, _image, 0);
        if (r == 0) {
-               throw EncodeError ("JPEG2000 encoding failed");
+               throw EncodeError (N_("JPEG2000 encoding failed"));
        }
 
-       _log->log (String::compose ("Finished locally-encoded frame %1", _frame));
+       _log->log (String::compose (N_("Finished locally-encoded frame %1"), _frame));
        
        return shared_ptr<EncodedData> (new LocallyEncodedData (_cio->buffer, cio_tell (_cio)));
 }
@@ -315,90 +312,112 @@ DCPVideoFrame::encode_remotely (ServerDescription const * serv)
 
        shared_ptr<Socket> socket (new Socket);
 
-       socket->connect (*endpoint_iterator, 30);
+       socket->connect (*endpoint_iterator);
 
        stringstream s;
-       s << "encode please\n"
-         << "input_width " << _input->size().width << "\n"
-         << "input_height " << _input->size().height << "\n"
-         << "input_pixel_format " << _input->pixel_format() << "\n"
-         << "output_width " << _out_size.width << "\n"
-         << "output_height " << _out_size.height << "\n"
-         << "padding " <<  _padding << "\n"
-         << "subtitle_offset " << _subtitle_offset << "\n"
-         << "subtitle_scale " << _subtitle_scale << "\n"
-         << "scaler " << _scaler->id () << "\n"
-         << "frame " << _frame << "\n"
-         << "frames_per_second " << _frames_per_second << "\n";
-
-       if (!_post_process.empty()) {
-               s << "post_process " << _post_process << "\n";
-       }
-       
-       s << "colour_lut " << Config::instance()->colour_lut_index () << "\n"
-         << "j2k_bandwidth " << Config::instance()->j2k_bandwidth () << "\n";
+       s << N_("encode please\n")
+         << N_("input_width ") << _input->size().width << N_("\n")
+         << N_("input_height ") << _input->size().height << N_("\n")
+         << N_("input_pixel_format ") << _input->pixel_format() << N_("\n")
+         << N_("output_width ") << _out_size.width << N_("\n")
+         << N_("output_height ") << _out_size.height << N_("\n")
+         << N_("padding ") <<  _padding << N_("\n")
+         << N_("subtitle_offset ") << _subtitle_offset << N_("\n")
+         << N_("subtitle_scale ") << _subtitle_scale << N_("\n")
+         << N_("scaler ") << _scaler->id () << N_("\n")
+         << N_("frame ") << _frame << N_("\n")
+         << N_("frames_per_second ") << _frames_per_second << N_("\n");
+
+       s << N_("colour_lut ") << _colour_lut << N_("\n")
+         << N_("j2k_bandwidth ") << _j2k_bandwidth << N_("\n");
 
        if (_subtitle) {
-               s << "subtitle_x " << _subtitle->position().x << "\n"
-                 << "subtitle_y " << _subtitle->position().y << "\n"
-                 << "subtitle_width " << _subtitle->image()->size().width << "\n"
-                 << "subtitle_height " << _subtitle->image()->size().height << "\n";
+               s << N_("subtitle_x ") << _subtitle->position().x << N_("\n")
+                 << N_("subtitle_y ") << _subtitle->position().y << N_("\n")
+                 << N_("subtitle_width ") << _subtitle->image()->size().width << N_("\n")
+                 << N_("subtitle_height ") << _subtitle->image()->size().height << N_("\n");
        }
 
        _log->log (String::compose (
-                          "Sending to remote; pixel format %1, components %2, line sizes (%3,%4,%5)",
+                          N_("Sending to remote; pixel format %1, components %2, lines (%3,%4,%5), line sizes (%6,%7,%8)"),
                           _input->pixel_format(), _input->components(),
+                          _input->lines(0), _input->lines(1), _input->lines(2),
                           _input->line_size()[0], _input->line_size()[1], _input->line_size()[2]
                           ));
-       
-       socket->write ((uint8_t *) s.str().c_str(), s.str().length() + 1, 30);
+
+       socket->write (s.str().length() + 1);
+       socket->write ((uint8_t *) s.str().c_str(), s.str().length() + 1);
 
        _input->write_to_socket (socket);
        if (_subtitle) {
                _subtitle->image()->write_to_socket (socket);
        }
 
-       char buffer[32];
-       socket->read_indefinite ((uint8_t *) buffer, sizeof (buffer), 30);
-       socket->consume (strlen (buffer) + 1);
-       shared_ptr<EncodedData> e (new RemotelyEncodedData (atoi (buffer)));
+       shared_ptr<EncodedData> e (new RemotelyEncodedData (socket->read_uint32 ()));
+       socket->read (e->data(), e->size());
 
-       /* now read the rest */
-       socket->read_definite_and_consume (e->data(), e->size(), 30);
-
-       _log->log (String::compose ("Finished remotely-encoded frame %1", _frame));
+       _log->log (String::compose (N_("Finished remotely-encoded frame %1"), _frame));
        
        return e;
 }
 
+EncodedData::EncodedData (int s)
+       : _data (new uint8_t[s])
+       , _size (s)
+{
+
+}
+
+EncodedData::EncodedData (string file)
+{
+       _size = boost::filesystem::file_size (file);
+       _data = new uint8_t[_size];
+
+       FILE* f = fopen (file.c_str(), N_("rb"));
+       if (!f) {
+               throw FileError (_("could not open file for reading"), file);
+       }
+       
+       fread (_data, 1, _size, f);
+       fclose (f);
+}
+
+
+EncodedData::~EncodedData ()
+{
+       delete[] _data;
+}
+
 /** Write this data to a J2K file.
- *  @param opt Options.
- *  @param frame Frame index.
+ *  @param Film Film.
+ *  @param frame DCP frame index.
  */
 void
-EncodedData::write (shared_ptr<const Options> opt, int frame)
+EncodedData::write (shared_ptr<const Film> film, int frame) const
 {
-       string const tmp_j2k = opt->frame_out_path (frame, true);
+       string const tmp_j2c = film->j2c_path (frame, true);
 
-       FILE* f = fopen (tmp_j2k.c_str (), "wb");
+       FILE* f = fopen (tmp_j2c.c_str (), N_("wb"));
        
        if (!f) {
-               throw WriteFileError (tmp_j2k, errno);
+               throw WriteFileError (tmp_j2c, errno);
        }
 
        fwrite (_data, 1, _size, f);
        fclose (f);
 
-       string const real_j2k = opt->frame_out_path (frame, false);
+       string const real_j2c = film->j2c_path (frame, false);
 
        /* Rename the file from foo.j2c.tmp to foo.j2c now that it is complete */
-       boost::filesystem::rename (tmp_j2k, real_j2k);
+       boost::filesystem::rename (tmp_j2c, real_j2c);
+}
 
-       /* Write a file containing the hash */
-       string const hash = real_j2k + ".md5";
-       ofstream h (hash.c_str());
-       h << md5_digest (_data, _size) << "\n";
-       h.close ();
+void
+EncodedData::write_info (shared_ptr<const Film> film, int frame, libdcp::FrameInfo fin) const
+{
+       string const info = film->info_path (frame);
+       ofstream h (info.c_str());
+       fin.write (h);
 }
 
 /** Send this data to a socket.
@@ -407,20 +426,19 @@ EncodedData::write (shared_ptr<const Options> opt, int frame)
 void
 EncodedData::send (shared_ptr<Socket> socket)
 {
-       stringstream s;
-       s << _size;
-       socket->write ((uint8_t *) s.str().c_str(), s.str().length() + 1, 30);
-       socket->write (_data, _size, 30);
+       socket->write (_size);
+       socket->write (_data, _size);
 }
 
-/** @param s Size of data in bytes */
-RemotelyEncodedData::RemotelyEncodedData (int s)
-       : EncodedData (new uint8_t[s], s)
+LocallyEncodedData::LocallyEncodedData (uint8_t* d, int s)
+       : EncodedData (s)
 {
-
+       memcpy (_data, d, s);
 }
 
-RemotelyEncodedData::~RemotelyEncodedData ()
+/** @param s Size of data in bytes */
+RemotelyEncodedData::RemotelyEncodedData (int s)
+       : EncodedData (s)
 {
-       delete[] _data;
+
 }