37c9ca6206429d2a864e1f21e11c0c1f48805905
[dcpomatic.git] / src / lib / make_dcp_job.cc
1 /*
2     Copyright (C) 2012 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 /** @file src/make_dcp_job.cc
21  *  @brief A job to create DCPs.
22  */
23
24 #include <iostream>
25 #include <boost/filesystem.hpp>
26 #include <libdcp/dcp.h>
27 #include <libdcp/picture_asset.h>
28 #include <libdcp/sound_asset.h>
29 #include <libdcp/reel.h>
30 extern "C" {
31 #include <libavutil/pixdesc.h>
32 }
33 #include "make_dcp_job.h"
34 #include "dcp_content_type.h"
35 #include "exceptions.h"
36 #include "options.h"
37 #include "imagemagick_decoder.h"
38 #include "film.h"
39
40 using std::string;
41 using std::cout;
42 using boost::shared_ptr;
43
44 /** @param f Film we are making the DCP for.
45  *  @param o Options.
46  */
47 MakeDCPJob::MakeDCPJob (shared_ptr<Film> f, shared_ptr<const EncodeOptions> o, shared_ptr<Job> req)
48         : Job (f, req)
49         , _opt (o)
50 {
51         
52 }
53
54 string
55 MakeDCPJob::name () const
56 {
57         return String::compose ("Make DCP for %1", _film->name());
58 }
59
60 /** @param f DCP frame index */
61 string
62 MakeDCPJob::j2c_path (int f, int offset) const
63 {
64         SourceFrame const s = ((f + offset) * dcp_frame_rate(_film->frames_per_second()).skip) + _film->dcp_trim_start();
65         return _opt->frame_out_path (s, false);
66 }
67
68 string
69 MakeDCPJob::wav_path (libdcp::Channel c) const
70 {
71         return _opt->multichannel_audio_out_path (int (c), false);
72 }
73
74 void
75 MakeDCPJob::run ()
76 {
77         if (!_film->dcp_length()) {
78                 throw EncodeError ("cannot make a DCP when the source length is not known");
79         }
80
81         descend (0.9);
82         
83         string const dcp_path = _film->dir (_film->dcp_name());
84
85         /* Remove any old DCP */
86         boost::filesystem::remove_all (dcp_path);
87
88         DCPFrameRate const dfr = dcp_frame_rate (_film->frames_per_second ());
89
90         int frames = 0;
91         switch (_film->content_type ()) {
92         case VIDEO:
93                 /* Source frames -> DCP frames */
94                 frames = _film->dcp_length().get() / dfr.skip;
95                 break;
96         case STILL:
97                 frames = _film->still_duration() * 24;
98                 break;
99         }
100
101         libdcp::DCP dcp (_film->dir (_film->dcp_name()));
102         dcp.Progress.connect (boost::bind (&MakeDCPJob::dcp_progress, this, _1));
103
104         shared_ptr<libdcp::CPL> cpl (
105                 new libdcp::CPL (_film->dir (_film->dcp_name()), _film->dcp_name(), _film->dcp_content_type()->libdcp_kind (), frames, dfr.frames_per_second)
106                 );
107         
108         dcp.add_cpl (cpl);
109
110         int frames_per_reel = 0;
111         if (_film->reel_size()) {
112                 frames_per_reel = (_film->reel_size().get() / (_film->j2k_bandwidth() / 8)) * dfr.frames_per_second;
113         } else {
114                 frames_per_reel = frames;
115         }
116
117         int frames_done = 0;
118         int reel = 0;
119
120         while (frames_done < frames) {
121
122                 descend (float (frames_per_reel) / frames);
123
124                 int this_time = std::min (frames_per_reel, (frames - frames_done));
125
126                 descend (0.8);
127
128                 shared_ptr<libdcp::MonoPictureAsset> pa (
129                         new libdcp::MonoPictureAsset (
130                                 boost::bind (&MakeDCPJob::j2c_path, this, _1, frames_done),
131                                 _film->dir (_film->dcp_name()),
132                                 String::compose ("video_%1.mxf", reel),
133                                 &dcp.Progress,
134                                 dfr.frames_per_second,
135                                 this_time,
136                                 _opt->out_size.width,
137                                 _opt->out_size.height,
138                                 _film->encrypted()
139                                 )
140                         );
141         
142                 ascend ();
143                 
144                 shared_ptr<libdcp::SoundAsset> sa;
145                 
146                 if (_film->audio_channels() > 0) {
147                         descend (0.1);
148                         sa.reset (
149                                 new libdcp::SoundAsset (
150                                         boost::bind (&MakeDCPJob::wav_path, this, _1),
151                                         _film->dir (_film->dcp_name()),
152                                         String::compose ("audio_%1.mxf", reel),
153                                         &dcp.Progress,
154                                         dfr.frames_per_second,
155                                         this_time,
156                                         frames_done,
157                                         dcp_audio_channels (_film->audio_channels()),
158                                         _film->encrypted()
159                                         )
160                                 );
161                         ascend ();
162                 }
163
164                 descend (0.1);
165                 cpl->add_reel (shared_ptr<libdcp::Reel> (new libdcp::Reel (pa, sa, shared_ptr<libdcp::SubtitleAsset> ())));
166                 ascend ();
167                 
168                 frames_done += frames_per_reel;
169                 ++reel;
170
171                 ascend ();
172         }
173
174         ascend ();
175
176         descend (0.1);
177         dcp.write_xml ();
178         ascend ();
179                 
180         set_progress (1);
181         set_state (FINISHED_OK);
182 }
183
184 void
185 MakeDCPJob::dcp_progress (float p)
186 {
187         set_progress (p);
188 }