--- /dev/null
+
+/*
+ * CINELERRA
+ * Copyright (C) 1997-2011 Adam Williams <broadcast at earthling dot net>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ *
+ */
+
+#include "asset.h"
+#include "clip.h"
+#include "confirmsave.h"
+#include "edl.h"
+#include "edlsession.h"
+#include "labels.h"
+#include "mutex.h"
+#include "mwindow.h"
+#include "packagedispatcher.h"
+#include "packagerenderer.h"
+#include "preferences.h"
+#include "render.h"
+#include "file.h"
+
+
+
+PackageDispatcher::PackageDispatcher()
+{
+ packages = 0;
+ package_lock = new Mutex("PackageDispatcher::package_lock");
+ packaging_engine = 0;
+}
+
+PackageDispatcher::~PackageDispatcher()
+{
+ if(packages)
+ {
+ for(int i = 0; i < total_packages; i++)
+ delete packages[i];
+ delete [] packages;
+ }
+ if (packaging_engine)
+ delete packaging_engine;
+ delete package_lock;
+}
+
+int PackageDispatcher::create_packages(MWindow *mwindow,
+ EDL *edl,
+ Preferences *preferences,
+ int strategy,
+ Asset *default_asset,
+ double total_start,
+ double total_end,
+ int test_overwrite)
+{
+ int result = 0;
+
+ this->mwindow = mwindow;
+ this->edl = edl;
+ this->preferences = preferences;
+ this->strategy = strategy;
+ this->default_asset = default_asset;
+ this->total_start = total_start;
+ this->total_end = total_end;
+
+ nodes = preferences->get_enabled_nodes();
+ audio_position = Units::to_int64(total_start * default_asset->sample_rate);
+ video_position = Units::to_int64(total_start * default_asset->frame_rate);
+ audio_end = Units::to_int64(total_end * default_asset->sample_rate);
+ video_end = Units::to_int64(total_end * default_asset->frame_rate);
+ current_package = 0;
+
+// sleep(1);
+// printf("PackageDispatcher::create_packages 1 %d %f %f\n",
+// video_end,
+// total_end,
+// default_asset->frame_rate);
+
+
+
+ if(strategy == SINGLE_PASS)
+ {
+ total_len = this->total_end - this->total_start;
+ package_len = total_len;
+ min_package_len = total_len;
+ total_packages = 1;
+ total_allocated = 1;
+ packages = new RenderPackage*[total_allocated];
+ packages[0] = new RenderPackage;
+ packages[0]->audio_start = audio_position;
+ packages[0]->audio_end = audio_end;
+ packages[0]->video_start = video_position;
+ packages[0]->video_end = video_end;
+ packages[0]->audio_do = default_asset->audio_data;
+ packages[0]->video_do = default_asset->video_data;
+ strcpy(packages[0]->path, default_asset->path);
+ }
+ else
+ if(strategy == SINGLE_PASS_FARM)
+ {
+ packaging_engine = File::new_packaging_engine(default_asset);
+ packaging_engine->create_packages_single_farm(
+ edl,
+ preferences,
+ default_asset,
+ total_start,
+ total_end);
+ }
+ else
+ if(strategy == FILE_PER_LABEL || strategy == FILE_PER_LABEL_FARM)
+ {
+ Label *label = edl->labels->first;
+ total_packages = 0;
+ packages = new RenderPackage*[edl->labels->total() + 2];
+
+ Render::get_starting_number(default_asset->path,
+ current_number,
+ number_start,
+ total_digits,
+ 2);
+
+ while(audio_position < audio_end)
+ {
+ RenderPackage *package =
+ packages[total_packages] =
+ new RenderPackage;
+ package->audio_start = audio_position;
+ package->video_start = video_position;
+ package->audio_do = default_asset->audio_data;
+ package->video_do = default_asset->video_data;
+
+
+ while(label &&
+ (label->position < (double)audio_position / default_asset->sample_rate ||
+ EQUIV(label->position, (double)audio_position / default_asset->sample_rate)))
+ {
+ label = label->next;
+ }
+
+ if(!label)
+ {
+ package->audio_end = Units::to_int64(total_end * default_asset->sample_rate);
+ package->video_end = Units::to_int64(total_end * default_asset->frame_rate);
+ }
+ else
+ {
+ package->audio_end = Units::to_int64(label->position * default_asset->sample_rate);
+ package->video_end = Units::to_int64(label->position * default_asset->frame_rate);
+ }
+
+ if(package->audio_end > audio_end)
+ {
+ package->audio_end = audio_end;
+ }
+
+ if(package->video_end > video_end)
+ {
+ package->video_end = video_end;
+ }
+
+ audio_position = package->audio_end;
+ video_position = package->video_end;
+// Create file number differently if image file sequence
+ Render::create_filename(package->path,
+ default_asset->path,
+ current_number,
+ total_digits,
+ number_start);
+ current_number++;
+
+ total_packages++;
+ }
+
+ total_allocated = total_packages;
+ }
+ else
+ if(strategy == BRENDER_FARM)
+ {
+ total_len = this->total_end - this->total_start;
+
+// Create packages as they're requested.
+ total_packages = 0;
+ total_allocated = 0;
+ packages = 0;
+
+ Render::get_starting_number(default_asset->path,
+ current_number,
+ number_start,
+ total_digits,
+ 6);
+
+// Master node only
+ if(preferences->renderfarm_nodes.total == 1)
+ {
+ package_len = total_len;
+ min_package_len = total_len;
+ }
+ else
+ {
+ package_len = preferences->brender_fragment /
+ edl->session->frame_rate;
+ min_package_len = 1.0 / edl->session->frame_rate;
+ }
+ }
+
+// Test existence of every output file.
+// Only if this isn't a background render or non interactive.
+ if(strategy != BRENDER_FARM &&
+ test_overwrite &&
+ mwindow)
+ {
+ ArrayList<char*> paths;
+ get_package_paths(&paths);
+ result = ConfirmSave::test_files(mwindow, &paths);
+ paths.remove_all_objects();
+ }
+
+ return result;
+}
+
+void PackageDispatcher::get_package_paths(ArrayList<char*> *path_list)
+{
+ if (strategy == SINGLE_PASS_FARM)
+ packaging_engine->get_package_paths(path_list);
+ else
+ {
+ for(int i = 0; i < total_allocated; i++)
+ path_list->append(strdup(packages[i]->path));
+ path_list->set_free();
+ }
+
+}
+
+RenderPackage* PackageDispatcher::get_package(double frames_per_second,
+ int client_number,
+ int use_local_rate)
+{
+ const int debug = 0;
+ package_lock->lock("PackageDispatcher::get_package");
+
+ if(debug) printf("PackageDispatcher::get_package %d %f %d %d\n",
+ __LINE__,
+ frames_per_second,
+ client_number,
+ use_local_rate);
+
+// Store new frames per second for the node
+ if(!EQUIV(frames_per_second, 0))
+ {
+ preferences->set_rate(frames_per_second, client_number);
+ if(mwindow) mwindow->preferences->copy_rates_from(preferences);
+ }
+ else
+// Use previous frames per second
+ {
+ frames_per_second = preferences->get_rate(client_number);
+ }
+
+ if(debug) printf("PackageDispatcher::get_package %d %f %d %d\n",
+ __LINE__,
+ frames_per_second,
+ client_number,
+ use_local_rate);
+
+ float avg_frames_per_second = preferences->get_avg_rate(use_local_rate);
+
+ RenderPackage *result = 0;
+//printf("PackageDispatcher::get_package 1 %d\n", strategy);
+ if(strategy == SINGLE_PASS ||
+ strategy == FILE_PER_LABEL ||
+ strategy == FILE_PER_LABEL_FARM)
+ {
+ if(current_package < total_packages)
+ {
+ result = packages[current_package];
+ current_package++;
+ }
+ }
+ else
+ if(strategy == SINGLE_PASS_FARM)
+ {
+ result = packaging_engine->get_package_single_farm(frames_per_second,
+ client_number,
+ use_local_rate);
+ }
+ else
+ if(strategy == BRENDER_FARM)
+ {
+//printf("Dispatcher::get_package 1 %d %d\n", video_position, video_end);
+ if(video_position < video_end)
+ {
+// Allocate new packages
+ if(total_packages == 0)
+ {
+ total_allocated = 256;
+ packages = new RenderPackage*[total_allocated];
+ }
+ else
+ if(total_packages >= total_allocated)
+ {
+ RenderPackage **old_packages = packages;
+ total_allocated *= 2;
+ packages = new RenderPackage*[total_allocated];
+ memcpy(packages,
+ old_packages,
+ total_packages * sizeof(RenderPackage*));
+ delete [] old_packages;
+ }
+
+// Calculate package.
+ result = packages[total_packages] = new RenderPackage;
+ double scaled_len;
+
+// No load balancing data exists
+ if(EQUIV(frames_per_second, 0) ||
+ EQUIV(avg_frames_per_second, 0))
+ {
+ scaled_len = package_len;
+ }
+ else
+// Load balancing data exists
+ {
+ scaled_len = package_len *
+ frames_per_second /
+ avg_frames_per_second;
+ }
+
+ scaled_len = MAX(scaled_len, min_package_len);
+
+// Always an image file sequence
+ result->audio_start = audio_position;
+ result->video_start = video_position;
+ result->audio_end = result->audio_start +
+ Units::to_int64(scaled_len * default_asset->sample_rate);
+ result->video_end = result->video_start +
+ Units::to_int64(scaled_len * default_asset->frame_rate);
+ if(result->video_end == result->video_start) result->video_end++;
+ audio_position = result->audio_end;
+ video_position = result->video_end;
+ result->audio_do = default_asset->audio_data;
+ result->video_do = default_asset->video_data;
+
+
+// The frame numbers are read from the vframe objects themselves.
+ Render::create_filename(result->path,
+ default_asset->path,
+ 0,
+ total_digits,
+ number_start);
+//printf("PackageDispatcher::get_package 2 %s\n", result->path);
+
+ current_number++;
+ total_packages++;
+ current_package++;
+ }
+ }
+
+ package_lock->unlock();
+
+ if(debug && result) printf("PackageDispatcher::get_package %d %ld\n", __LINE__, (long)(result->video_end - result->video_start));
+ return result;
+}
+
+
+ArrayList<Indexable*>* PackageDispatcher::get_asset_list()
+{
+ ArrayList<Indexable*> *assets = new ArrayList<Indexable*>;
+
+const int debug = 0;
+if(debug) printf("PackageDispatcher::get_asset_list %d\n", __LINE__);
+if(debug) default_asset->dump();
+ for(int i = 0; i < current_package; i++)
+ {
+ Asset *asset = new Asset;
+ asset->copy_from(default_asset, 1);
+ strcpy(asset->path, packages[i]->path);
+ asset->video_length = packages[i]->video_end - packages[i]->video_start;
+ asset->audio_length = packages[i]->audio_end - packages[i]->audio_start;
+ assets->append(asset);
+if(debug) printf("PackageDispatcher::get_asset_list %d\n", __LINE__);
+if(debug) asset->dump();
+ }
+
+ return assets;
+}
+
+int64_t PackageDispatcher::get_progress_max()
+{
+ if (strategy == SINGLE_PASS_FARM)
+ return packaging_engine->get_progress_max();
+ else
+ return Units::to_int64(default_asset->sample_rate *
+ (total_end - total_start)) +
+ Units::to_int64(preferences->render_preroll *
+ total_allocated *
+ default_asset->sample_rate);
+}
+
+int PackageDispatcher::get_total_packages()
+{
+ return total_allocated;
+}
+
+int PackageDispatcher::packages_are_done()
+{
+ if (packaging_engine)
+ return packaging_engine->packages_are_done();
+ return 0;
+}