Credit Andrew - fix bug in render farm usage when using in/out pointers or selection
[goodguy/cinelerra.git] / cinelerra-5.1 / cinelerra / packagingengine.C
1
2 /*
3  * CINELERRA
4  * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
5  *
6  * This program 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  * This program 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 this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  *
20  */
21
22 #include "packagingengine.h"
23 #include "preferences.h"
24 #include "edlsession.h"
25 #include "asset.h"
26 #include "render.h"
27 #include "clip.h"
28
29
30 // Default packaging engine implementation, simply split the range up, and consider client's speed
31
32 PackagingEngineDefault::PackagingEngineDefault()
33 {
34         packages = 0;
35 }
36
37 PackagingEngineDefault::~PackagingEngineDefault()
38 {
39         if(packages)
40         {
41                 for(int i = 0; i < total_packages; i++)
42                         delete packages[i];
43                 delete [] packages;
44         }
45 }
46
47
48 int PackagingEngineDefault::create_packages_single_farm(EDL *edl,
49                 Preferences *preferences, Asset *default_asset,
50                 double total_start, double total_end)
51 {
52         this->total_start = total_start;
53         this->total_end = total_end;
54
55         this->preferences = preferences;
56         this->default_asset = default_asset;
57         audio_position = Units::to_int64(total_start * default_asset->sample_rate);
58         video_position = Units::to_int64(total_start * default_asset->frame_rate);
59         audio_end = Units::to_int64(total_end * default_asset->sample_rate);
60         video_end = Units::to_int64(total_end * default_asset->frame_rate);
61         current_package = 0;
62         current_position = 0;
63
64         double total_len = total_end - total_start;
65         total_packages = preferences->renderfarm_job_count;
66         total_allocated = total_packages + preferences->get_enabled_nodes();
67         packages = new RenderPackage*[total_allocated];
68         package_len = total_len / total_packages;
69         min_package_len = 2.0 / edl->session->frame_rate;
70
71         int current_number;    // The number being injected into the filename.
72         int number_start;      // Character in the filename path at which the number begins
73         int total_digits;      // Total number of digits including padding the user specified.
74
75         Render::get_starting_number(default_asset->path,
76                 current_number, number_start, total_digits, 3);
77
78         for( int i=0; i<total_allocated; ++i ) {
79                 RenderPackage *package = packages[i] = new RenderPackage;
80
81 // Create file number differently if image file sequence
82                 Render::create_filename(package->path, default_asset->path,
83                         current_number, total_digits, number_start);
84                 current_number++;
85         }
86         return 0;
87 }
88
89 RenderPackage* PackagingEngineDefault::get_package_single_farm(double frames_per_second,
90                 int client_number, int use_local_rate)
91 {
92
93 //printf("PackageDispatcher::get_package %ld %ld %ld %ld\n", audio_position, video_position, audio_end, video_end);
94
95                 RenderPackage *result = 0;
96                 float avg_frames_per_second = preferences->get_avg_rate(use_local_rate);
97
98                 if(audio_position < audio_end ||
99                         video_position < video_end)
100                 {
101 // Last package
102                         double scaled_len;
103                         result = packages[current_package];
104                         result->audio_start = audio_position;
105                         result->video_start = video_position;
106                         result->video_do = default_asset->video_data;
107                         result->audio_do = default_asset->audio_data;
108
109                         if(current_package >= total_allocated - 1)
110                         {
111                                 result->audio_end = audio_end;
112                                 result->video_end = video_end;
113                                 audio_position = result->audio_end;
114                                 video_position = result->video_end;
115                         }
116                         else
117 // No useful speed data.  May get infinity for real fast jobs.
118                         if(frames_per_second > 0x7fffff || frames_per_second < 0 ||
119                                 EQUIV(frames_per_second, 0) ||
120                                 EQUIV(avg_frames_per_second, 0))
121                         {
122                                 scaled_len = MAX(package_len, min_package_len);
123
124                                 result->audio_end = audio_position +
125                                         Units::round(scaled_len * default_asset->sample_rate);
126                                 result->video_end = video_position +
127                                         Units::round(scaled_len * default_asset->frame_rate);
128
129 // If we get here without any useful speed data render the whole thing.
130                                 if(current_package >= total_packages - 1)
131                                 {
132                                         result->audio_end = audio_end;
133                                         result->video_end = video_end;
134                                 }
135                                 else
136                                 {
137                                         result->audio_end = MIN(audio_end, result->audio_end);
138                                         result->video_end = MIN(video_end, result->video_end);
139                                 }
140
141                                 audio_position = result->audio_end;
142                                 video_position = result->video_end;
143                         }
144                         else
145 // Useful speed data and future packages exist.  Scale the
146 // package size to fit the requestor.
147                         {
148                                 scaled_len = package_len *
149                                         frames_per_second /
150                                         avg_frames_per_second;
151                                 scaled_len = MAX(scaled_len, min_package_len);
152
153                                 result->audio_end = result->audio_start +
154                                         Units::to_int64(scaled_len * default_asset->sample_rate);
155                                 result->video_end = result->video_start +
156                                         Units::to_int64(scaled_len * default_asset->frame_rate);
157
158                                 result->audio_end = MIN(audio_end, result->audio_end);
159                                 result->video_end = MIN(video_end, result->video_end);
160
161                                 audio_position = result->audio_end;
162                                 video_position = result->video_end;
163
164 // Package size is no longer touched between total_packages and total_allocated
165                                 if(current_package < total_packages - 1)
166                                 {
167                                         package_len = (double)(audio_end - audio_position) /
168                                                 (double)default_asset->sample_rate /
169                                                 (double)(total_packages - current_package);
170                                 }
171
172                         }
173
174                         current_package++;
175 //printf("Dispatcher::get_package 50 %lld %lld %lld %lld\n",
176 // result->audio_start, result->video_start, result->audio_end, result->video_end);
177                 }
178                 return result;
179
180 }
181
182 void PackagingEngineDefault::get_package_paths(ArrayList<char*> *path_list)
183 {
184         for( int i=0; i<total_allocated; ++i ) {
185                 path_list->append(strdup(packages[i]->path));
186         }
187         path_list->set_free();
188 }
189
190 int PackagingEngineDefault::get_asset_list(ArrayList<Indexable *> &idxbls)
191 {
192         for( int i=0; i<current_package; ++i ) {
193                 Asset *asset = new Asset;
194                 asset->copy_from(default_asset, 1);
195                 strcpy(asset->path, packages[i]->path);
196                 asset->video_length = packages[i]->video_end - packages[i]->video_start;
197                 asset->audio_length = packages[i]->audio_end - packages[i]->audio_start;
198                 idxbls.append(asset);
199         }
200         return current_package;
201 }
202
203 int64_t PackagingEngineDefault::get_progress_max()
204 {
205         return Units::to_int64(default_asset->sample_rate * (total_end - total_start)) +
206                 Units::to_int64(preferences->render_preroll * 2 * default_asset->sample_rate);
207 }
208
209 int PackagingEngineDefault::packages_are_done()
210 {
211         return 0;
212 }
213