Merge master.
[dcpomatic.git] / src / lib / dcp_video_frame.cc
index 89db5761a1bb56f04a85e0ade660ef3a57ecd9ff..2f597522c3c4401fb17c476d6d2a7ffc20223450 100644 (file)
 #include "dcp_video_frame.h"
 #include "lut.h"
 #include "config.h"
-#include "options.h"
 #include "exceptions.h"
 #include "server.h"
 #include "util.h"
 #include "scaler.h"
 #include "image.h"
 #include "log.h"
-#include "subtitle.h"
+
+#include "i18n.h"
 
 using std::string;
 using std::stringstream;
@@ -65,35 +65,21 @@ 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 DCP's intrinsic duration.
- *  @param fps Frames per second of the Film's source.
- *  @param pp FFmpeg post-processing string to use.
+ *  @param f Index of the frame within the DCP.
  *  @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<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
+       shared_ptr<const Image> image, int f, int dcp_fps, int clut, int bw, shared_ptr<Log> l
        )
-       : _input (yuv)
-       , _subtitle (sub)
-       , _out_size (out)
-       , _padding (p)
-       , _subtitle_offset (subtitle_offset)
-       , _subtitle_scale (subtitle_scale)
-       , _scaler (s)
+       : _image (image)
        , _frame (f)
-       , _frames_per_second (DCPFrameRate(fps).frames_per_second)
-       , _post_process (pp)
+       , _frames_per_second (dcp_fps)
        , _colour_lut (clut)
        , _j2k_bandwidth (bw)
        , _log (l)
-       , _image (0)
+       , _opj_image (0)
        , _parameters (0)
        , _cinfo (0)
        , _cio (0)
@@ -108,8 +94,8 @@ DCPVideoFrame::create_openjpeg_container ()
        for (int i = 0; i < 3; ++i) {
                _cmptparm[i].dx = 1;
                _cmptparm[i].dy = 1;
-               _cmptparm[i].w = _out_size.width;
-               _cmptparm[i].h = _out_size.height;
+               _cmptparm[i].w = _image->size().width;
+               _cmptparm[i].h = _image->size().height;
                _cmptparm[i].x0 = 0;
                _cmptparm[i].y0 = 0;
                _cmptparm[i].prec = 12;
@@ -117,21 +103,21 @@ DCPVideoFrame::create_openjpeg_container ()
                _cmptparm[i].sgnd = 0;
        }
 
-       _image = opj_image_create (3, &_cmptparm[0], CLRSPC_SRGB);
-       if (_image == 0) {
-               throw EncodeError ("could not create libopenjpeg image");
+       _opj_image = opj_image_create (3, &_cmptparm[0], CLRSPC_SRGB);
+       if (_opj_image == 0) {
+               throw EncodeError (N_("could not create libopenjpeg image"));
        }
 
-       _image->x0 = 0;
-       _image->y0 = 0;
-       _image->x1 = _out_size.width;
-       _image->y1 = _out_size.height;
+       _opj_image->x0 = 0;
+       _opj_image->y0 = 0;
+       _opj_image->x1 = _image->size().width;
+       _opj_image->y1 = _image->size().height;
 }
 
 DCPVideoFrame::~DCPVideoFrame ()
 {
-       if (_image) {
-               opj_image_destroy (_image);
+       if (_opj_image) {
+               opj_image_destroy (_opj_image);
        }
 
        if (_cio) {
@@ -155,23 +141,6 @@ DCPVideoFrame::~DCPVideoFrame ()
 shared_ptr<EncodedData>
 DCPVideoFrame::encode_locally ()
 {
-       if (!_post_process.empty ()) {
-               _input = _input->post_process (_post_process, true);
-       }
-       
-       shared_ptr<Image> prepared = _input->scale_and_convert_to_rgb (_out_size, _padding, _scaler, true);
-
-       if (_subtitle) {
-               Rect tx = subtitle_transformed_area (
-                       float (_out_size.width) / _input->size().width,
-                       float (_out_size.height) / _input->size().height,
-                       _subtitle->area(), _subtitle_offset, _subtitle_scale
-                       );
-
-               shared_ptr<Image> im = _subtitle->image()->scale (tx.size(), _scaler, true);
-               prepared->alpha_blend (im, tx.position());
-       }
-
        create_openjpeg_container ();
 
        struct {
@@ -185,9 +154,9 @@ DCPVideoFrame::encode_locally ()
        /* Copy our RGB into the openjpeg container, converting to XYZ in the process */
 
        int jn = 0;
-       for (int y = 0; y < _out_size.height; ++y) {
-               uint8_t* p = prepared->data()[0] + y * prepared->stride()[0];
-               for (int x = 0; x < _out_size.width; ++x) {
+       for (int y = 0; y < _image->size().height; ++y) {
+               uint8_t* p = _image->data()[0] + y * _image->stride()[0];
+               for (int x = 0; x < _image->size().width; ++x) {
 
                        /* In gamma LUT (converting 8-bit input to 12-bit) */
                        s.r = lut_in[_colour_lut][*p++ << 4];
@@ -213,9 +182,9 @@ DCPVideoFrame::encode_locally ()
                        d.z = d.z * DCI_COEFFICENT * (DCI_LUT_SIZE - 1);
                        
                        /* Out gamma LUT */
-                       _image->comps[0].data[jn] = lut_out[LO_DCI][(int) d.x];
-                       _image->comps[1].data[jn] = lut_out[LO_DCI][(int) d.y];
-                       _image->comps[2].data[jn] = lut_out[LO_DCI][(int) d.z];
+                       _opj_image->comps[0].data[jn] = lut_out[LO_DCI][(int) d.x];
+                       _opj_image->comps[1].data[jn] = lut_out[LO_DCI][(int) d.y];
+                       _opj_image->comps[2].data[jn] = lut_out[LO_DCI][(int) d.z];
 
                        ++jn;
                }
@@ -265,7 +234,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 */
@@ -273,31 +242,31 @@ DCPVideoFrame::encode_locally ()
        
        /* set max image */
        _parameters->max_comp_size = max_comp_size;
-       _parameters->tcp_rates[0] = ((float) (3 * _image->comps[0].w * _image->comps[0].h * _image->comps[0].prec)) / (max_cs_len * 8);
+       _parameters->tcp_rates[0] = ((float) (3 * _opj_image->comps[0].w * _opj_image->comps[0].h * _opj_image->comps[0].prec)) / (max_cs_len * 8);
 
        /* 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) */
        _cinfo->event_mgr = 0;
 
        /* Setup the encoder parameters using the current image and user parameters */
-       opj_setup_encoder (_cinfo, _parameters, _image);
+       opj_setup_encoder (_cinfo, _parameters, _opj_image);
 
        _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);
+       int const r = opj_encode (_cinfo, _cio, _opj_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)));
 }
@@ -316,59 +285,33 @@ 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 " << _colour_lut << "\n"
-         << "j2k_bandwidth " << _j2k_bandwidth << "\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_("encode please\n")
+         << N_("width ") << _image->size().width << N_("\n")
+         << N_("height ") << _image->size().height << N_("\n")
+         << N_("frame ") << _frame << N_("\n")
+         << N_("frames_per_second ") << _frames_per_second << N_("\n")
+         << N_("colour_lut ") << _colour_lut << N_("\n")
+         << N_("j2k_bandwidth ") << _j2k_bandwidth << N_("\n");
 
        _log->log (String::compose (
-                          "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]
+                          N_("Sending to remote; pixel format %1, components %2, lines (%3,%4,%5), line sizes (%6,%7,%8)"),
+                          _image->pixel_format(), _image->components(),
+                          _image->lines(0), _image->lines(1), _image->lines(2),
+                          _image->line_size()[0], _image->line_size()[1], _image->line_size()[2]
                           ));
-       
-       socket->write ((uint8_t *) s.str().c_str(), s.str().length() + 1, 30);
 
-       _input->write_to_socket (socket);
-       if (_subtitle) {
-               _subtitle->image()->write_to_socket (socket);
-       }
+       socket->write (s.str().length() + 1);
+       socket->write ((uint8_t *) s.str().c_str(), s.str().length() + 1);
 
-       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)));
+       _image->write_to_socket (socket);
 
-       /* now read the rest */
-       socket->read_definite_and_consume (e->data(), e->size(), 30);
+       shared_ptr<EncodedData> e (new RemotelyEncodedData (socket->read_uint32 ()));
+       socket->read (e->data(), e->size());
 
-       _log->log (String::compose ("Finished remotely-encoded frame %1", _frame));
+       _log->log (String::compose (N_("Finished remotely-encoded frame %1"), _frame));
        
        return e;
 }
@@ -385,9 +328,9 @@ EncodedData::EncodedData (string file)
        _size = boost::filesystem::file_size (file);
        _data = new uint8_t[_size];
 
-       FILE* f = fopen (file.c_str(), "rb");
+       FILE* f = fopen (file.c_str(), N_("rb"));
        if (!f) {
-               throw FileError ("could not open file for reading", file);
+               throw FileError (_("could not open file for reading"), file);
        }
        
        fread (_data, 1, _size, f);
@@ -401,33 +344,35 @@ EncodedData::~EncodedData ()
 }
 
 /** Write this data to a J2K file.
- *  @param opt Options.
- *  @param frame DCP Frame index.
+ *  @param Film Film.
+ *  @param frame DCP frame index.
  */
 void
 EncodedData::write (shared_ptr<const Film> film, int frame) const
 {
-       string const tmp_j2k = film->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 = film->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 = film->hash_out_path (frame, false);
-       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.
@@ -436,10 +381,8 @@ EncodedData::write (shared_ptr<const Film> film, int frame) const
 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);
 }
 
 LocallyEncodedData::LocallyEncodedData (uint8_t* d, int s)