Some work on making KDMs in Film
[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 <boost/filesystem.hpp>
25 #include <libdcp/dcp.h>
26 #include <libdcp/picture_asset.h>
27 #include <libdcp/sound_asset.h>
28 #include <libdcp/reel.h>
29 extern "C" {
30 #include <libavutil/pixdesc.h>
31 }
32 #include "make_dcp_job.h"
33 #include "dcp_content_type.h"
34 #include "exceptions.h"
35 #include "options.h"
36 #include "imagemagick_decoder.h"
37 #include "film.h"
38
39 using std::string;
40 using boost::shared_ptr;
41
42 /** @param f Film we are making the DCP for.
43  *  @param o Options.
44  */
45 MakeDCPJob::MakeDCPJob (shared_ptr<Film> f, shared_ptr<const EncodeOptions> o, shared_ptr<Job> req)
46         : Job (f, req)
47         , _opt (o)
48 {
49         
50 }
51
52 string
53 MakeDCPJob::name () const
54 {
55         return String::compose ("Make DCP for %1", _film->name());
56 }
57
58 /** @param f DCP frame index */
59 string
60 MakeDCPJob::j2c_path (int f) const
61 {
62         SourceFrame const s = (f * dcp_frame_rate(_film->frames_per_second()).skip) + _film->dcp_trim_start();
63         return _opt->frame_out_path (s, false);
64 }
65
66 string
67 MakeDCPJob::wav_path (libdcp::Channel c) const
68 {
69         return _opt->multichannel_audio_out_path (int (c), false);
70 }
71
72 void
73 MakeDCPJob::run ()
74 {
75         if (!_film->dcp_length()) {
76                 throw EncodeError ("cannot make a DCP when the source length is not known");
77         }
78         
79         string const dcp_path = _film->dir (_film->dcp_name());
80
81         /* Remove any old DCP */
82         boost::filesystem::remove_all (dcp_path);
83
84         DCPFrameRate const dfr = dcp_frame_rate (_film->frames_per_second ());
85
86         int frames = 0;
87         switch (_film->content_type ()) {
88         case VIDEO:
89                 /* Source frames -> DCP frames */
90                 frames = _film->dcp_length().get() / dfr.skip;
91                 break;
92         case STILL:
93                 frames = _film->still_duration() * 24;
94                 break;
95         }
96
97         libdcp::DCP dcp (_film->dir (_film->dcp_name()));
98         dcp.Progress.connect (boost::bind (&MakeDCPJob::dcp_progress, this, _1));
99
100         shared_ptr<libdcp::CPL> cpl (
101                 new libdcp::CPL (_film->dir (_film->dcp_name()), _film->dcp_name(), _film->dcp_content_type()->libdcp_kind (), frames, dfr.frames_per_second)
102                 );
103         
104         dcp.add_cpl (cpl);
105
106         descend (0.8);
107         shared_ptr<libdcp::MonoPictureAsset> pa (
108                 new libdcp::MonoPictureAsset (
109                         boost::bind (&MakeDCPJob::j2c_path, this, _1),
110                         _film->dir (_film->dcp_name()),
111                         "video.mxf",
112                         &dcp.Progress,
113                         dfr.frames_per_second,
114                         frames,
115                         _opt->out_size.width,
116                         _opt->out_size.height,
117                         _film->encrypted ()
118                         )
119                 );
120         
121         ascend ();
122
123         shared_ptr<libdcp::SoundAsset> sa;
124
125         if (_film->audio_channels() > 0) {
126                 descend (0.1);
127                 sa.reset (
128                         new libdcp::SoundAsset (
129                                 boost::bind (&MakeDCPJob::wav_path, this, _1),
130                                 _film->dir (_film->dcp_name()),
131                                 "audio.mxf",
132                                 &dcp.Progress,
133                                 dfr.frames_per_second,
134                                 frames,
135                                 dcp_audio_channels (_film->audio_channels()),
136                                 _film->encrypted()
137                                 )
138                         );
139                 ascend ();
140         }
141
142         descend (0.1);
143         cpl->add_reel (shared_ptr<libdcp::Reel> (new libdcp::Reel (pa, sa, shared_ptr<libdcp::SubtitleAsset> ())));
144         dcp.write_xml ();
145         ascend ();
146
147         set_progress (1);
148         set_state (FINISHED_OK);
149 }
150
151 void
152 MakeDCPJob::dcp_progress (float p)
153 {
154         set_progress (p);
155 }