420f5a8d219240ed726e617c9cab49febae7a9a5
[dcpomatic.git] / src / lib / transcode_job.cc
1 /*
2     Copyright (C) 2012-2015 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 /** @file src/transcode_job.cc
22  *  @brief A job which transcodes from one format to another.
23  */
24
25 #include "transcode_job.h"
26 #include "dcp_encoder.h"
27 #include "upload_job.h"
28 #include "job_manager.h"
29 #include "film.h"
30 #include "encoder.h"
31 #include "log.h"
32 #include "compose.hpp"
33 #include <iostream>
34 #include <iomanip>
35
36 #include "i18n.h"
37
38 #define LOG_GENERAL(...) _film->log()->log (String::compose (__VA_ARGS__), LogEntry::TYPE_GENERAL);
39 #define LOG_GENERAL_NC(...) _film->log()->log (__VA_ARGS__, LogEntry::TYPE_GENERAL);
40 #define LOG_ERROR_NC(...)   _film->log()->log (__VA_ARGS__, LogEntry::TYPE_ERROR);
41
42 using std::string;
43 using std::fixed;
44 using std::setprecision;
45 using std::cout;
46 using boost::shared_ptr;
47
48 /** @param film Film to use */
49 TranscodeJob::TranscodeJob (shared_ptr<const Film> film)
50         : Job (film)
51 {
52
53 }
54
55 string
56 TranscodeJob::name () const
57 {
58         return String::compose (_("Transcode %1"), _film->name());
59 }
60
61 string
62 TranscodeJob::json_name () const
63 {
64         return N_("transcode");
65 }
66
67 void
68 TranscodeJob::set_encoder (shared_ptr<Encoder> e)
69 {
70         _encoder = e;
71 }
72
73 void
74 TranscodeJob::run ()
75 {
76         try {
77                 struct timeval start;
78                 gettimeofday (&start, 0);
79                 LOG_GENERAL_NC (N_("Transcode job starting"));
80
81                 DCPOMATIC_ASSERT (_encoder);
82                 _encoder->go ();
83                 set_progress (1);
84                 set_state (FINISHED_OK);
85
86                 struct timeval finish;
87                 gettimeofday (&finish, 0);
88
89                 float fps = 0;
90                 if (finish.tv_sec != start.tv_sec) {
91                         fps = _encoder->frames_done() / (finish.tv_sec - start.tv_sec);
92                 }
93
94                 LOG_GENERAL (N_("Transcode job completed successfully: %1 fps"), fps);
95                 _encoder.reset ();
96
97                 /* XXX: this shouldn't be here */
98                 if (_film->upload_after_make_dcp ()) {
99                         shared_ptr<Job> job (new UploadJob (_film));
100                         JobManager::instance()->add (job);
101                 }
102
103         } catch (...) {
104                 _encoder.reset ();
105                 throw;
106         }
107 }
108
109 string
110 TranscodeJob::status () const
111 {
112         if (!_encoder) {
113                 return Job::status ();
114         }
115
116
117         char buffer[256];
118         if (finished() || _encoder->finishing()) {
119                 strncpy (buffer, Job::status().c_str(), 256);
120         } else {
121                 snprintf (
122                         buffer, sizeof(buffer), "%s; %" PRId64 "/%" PRId64 " frames",
123                         Job::status().c_str(),
124                         _encoder->frames_done(),
125                         _film->length().frames_round (_film->video_frame_rate ())
126                         );
127
128                 float const fps = _encoder->current_rate ();
129                 if (fps) {
130                         char fps_buffer[64];
131                         /// TRANSLATORS: fps here is an abbreviation for frames per second
132                         snprintf (fps_buffer, sizeof(fps_buffer), _("; %.1f fps"), fps);
133                         strncat (buffer, fps_buffer, strlen(buffer) - 1);
134                 }
135         }
136
137         return buffer;
138 }
139
140 /** @return Approximate remaining time in seconds */
141 int
142 TranscodeJob::remaining_time () const
143 {
144         /* _encoder might be destroyed by the job-runner thread */
145         shared_ptr<Encoder> e = _encoder;
146
147         if (!e || e->finishing()) {
148                 /* We aren't doing any actual encoding so just use the job's guess */
149                 return Job::remaining_time ();
150         }
151
152         /* We're encoding so guess based on the current encoding rate */
153
154         float fps = e->current_rate ();
155
156         if (fps == 0) {
157                 return 0;
158         }
159
160         /* Compute approximate proposed length here, as it's only here that we need it */
161         return (_film->length().frames_round (_film->video_frame_rate ()) - e->frames_done()) / fps;
162 }