packagerenderer.C

Go to the documentation of this file.

#include "arender.h"
#include "asset.h"
#include "auto.h"
#include "brender.h"
#include "cache.h"
#include "clip.h"
#include "cwindow.h"
#include "cwindowgui.h"
#include "edit.h"
#include "edits.h"
#include "edl.h"
#include "edlsession.h"
#include "errorbox.h"
#include "file.h"
#include "filesystem.h"
#include "indexfile.h"
#include "language.h"
#include "mwindow.h"
#include "mwindowgui.h"
#include "packagerenderer.h"
#include "playabletracks.h"
#include "playbackconfig.h"
#include "pluginserver.h"
#include "preferences.h"
#include "render.h"
#include "renderengine.h"
#include "renderfarmfsserver.h"
#include "sighandler.h"
#include "tracks.h"
#include "transportque.h"
#include "vedit.h"
#include "vframe.h"
#include "videodevice.h"
#include "vrender.h"







RenderPackage::RenderPackage()
{
        audio_start = 0;
        audio_end = 0;
        video_start = 0;
        video_end = 0;
        path[0] = 0;
        done = 0;
        use_brender = 0;
}

RenderPackage::~RenderPackage()
{
}








// Used by RenderFarm and in the future, Render, to do packages.
PackageRenderer::PackageRenderer()
{
        command = 0;
        audio_cache = 0;
        video_cache = 0;
        aconfig = 0;
        vconfig = 0;
}

PackageRenderer::~PackageRenderer()
{
        delete command;
        delete audio_cache;
        delete video_cache;
        delete vconfig;
}

int PackageRenderer::initialize(MWindow *mwindow,
                EDL *edl, 
                Preferences *preferences, 
                Asset *default_asset,
                ArrayList<PluginServer*> *plugindb)
{
        int result = 0;

        this->mwindow = mwindow;
        this->edl = edl;
        this->preferences = preferences;
        this->default_asset = default_asset;
        this->plugindb = plugindb;


//printf("PackageRenderer::initialize %d\n", preferences->processors);
        command = new TransportCommand;
        command->command = NORMAL_FWD;
        command->get_edl()->copy_all(edl);
        command->change_type = CHANGE_ALL;
        command->set_playback_range(edl);

        default_asset->frame_rate = command->get_edl()->session->frame_rate;
        default_asset->sample_rate = command->get_edl()->session->sample_rate;
        default_asset->aspect_ratio = (double)command->get_edl()->session->aspect_w /
                command->get_edl()->session->aspect_h;
        result = Render::check_asset(edl, *default_asset);

        audio_cache = new CICache(command->get_edl(), preferences, plugindb);
        video_cache = new CICache(command->get_edl(), preferences, plugindb);

        PlaybackConfig *config = command->get_edl()->session->playback_config;
        aconfig = new AudioOutConfig(0);
        vconfig = new VideoOutConfig;
//      playback_config = new PlaybackConfig(PLAYBACK_LOCALHOST, 0);
        for(int i = 0; i < MAX_CHANNELS; i++)
        {
                vconfig->do_channel[i] = (i < command->get_edl()->session->video_channels);
        }


        return result;
}

void PackageRenderer::create_output()
{
        FileSystem fs;
        asset = new Asset(*default_asset);





// Tag output paths for VFS here.
//      if(!mwindow && preferences->renderfarm_vfs && preferences->use_renderfarm)
        if(!get_master() && preferences->renderfarm_vfs && preferences->use_renderfarm)
                sprintf(asset->path, RENDERFARM_FS_PREFIX "%s", package->path);
        else
                strcpy(asset->path, package->path);


        

        file = new File;

        file->set_processors(preferences->processors);

        result = file->open_file(preferences, 
                                        asset, 
                                        0, 
                                        1, 
                                        command->get_edl()->session->sample_rate, 
                                        command->get_edl()->session->frame_rate);
//printf("PackageRenderer::create_output 10 %d\n", result);

        if(result && mwindow)
        {
// open failed
                char string[BCTEXTLEN];
                sprintf(string, _("Couldn't open %s"), asset->path);
                ErrorBox error(PROGRAM_NAME ": Error",
                        mwindow->gui->get_abs_cursor_x(1),
                        mwindow->gui->get_abs_cursor_y(1));
                error.create_objects(string);
                error.run_window();
        }
        else
        if(mwindow)
        {
                mwindow->sighandler->push_file(file);
                IndexFile::delete_index(preferences, asset);
        }
//printf("PackageRenderer::create_output 100 %d\n", result);
}

void PackageRenderer::create_engine()
{
        int current_achannel = 0, current_vchannel = 0;
        audio_read_length = command->get_edl()->session->sample_rate;

        aconfig->fragment_size = audio_read_length;


        render_engine = new RenderEngine(0,
                preferences,
                command,
                0,
                plugindb,
                0);
        render_engine->set_acache(audio_cache);
        render_engine->set_vcache(video_cache);
        render_engine->arm_command(command, current_achannel, current_vchannel);

        if(package->use_brender)
        {
                audio_preroll = Units::to_int64((double)preferences->brender_preroll /
                        default_asset->frame_rate *
                        default_asset->sample_rate);
                video_preroll = preferences->brender_preroll;
        }
        else
        {
                audio_preroll = Units::to_int64(preferences->render_preroll * 
                        default_asset->sample_rate);
                video_preroll = Units::to_int64(preferences->render_preroll * 
                        default_asset->frame_rate);
        }
        audio_position = package->audio_start - audio_preroll;
        video_position = package->video_start - video_preroll;




// Create output buffers
        if(asset->audio_data)
        {
                file->start_audio_thread(audio_read_length, 
                        preferences->processors > 1 ? 2 : 1);
        }


        if(asset->video_data)
        {
                compressed_output = new VFrame;
// The write length needs to correlate with the processor count because
// it is passed to the file handler which usually processes frames simultaneously.
                video_write_length = preferences->processors;
                video_write_position = 0;
                direct_frame_copying = 0;


//printf("PackageRenderer::create_engine 1\n");
                file->start_video_thread(video_write_length,
                        command->get_edl()->session->color_model,
                        preferences->processors > 1 ? 2 : 1,
                        0);


                if(mwindow)
                {
                        video_device = new VideoDevice;
                        video_device->open_output(vconfig, 
                                command->get_edl()->session->frame_rate, 
                                command->get_edl()->session->output_w, 
                                command->get_edl()->session->output_h, 
                                mwindow->cwindow->gui->canvas,
                                0);
                        video_device->start_playback();
                }
        }


        playable_tracks = new PlayableTracks(render_engine, 
                video_position, 
                TRACK_VIDEO,
                1);

}




void PackageRenderer::do_audio()
{
//printf("PackageRenderer::do_audio 1\n");
// Do audio data
        if(asset->audio_data)
        {
                audio_output = file->get_audio_buffer();
// Zero unused channels in output vector
                for(int i = 0; i < MAX_CHANNELS; i++)
                        audio_output_ptr[i] = (i < asset->channels) ? 
                                audio_output[i] : 
                                0;




// Call render engine
                result = render_engine->arender->process_buffer(audio_output_ptr, 
                        audio_read_length, 
                        audio_position,
                        0);

//printf("PackageRenderer::do_audio 3\n");


// Fix buffers for preroll
                int64_t output_length = audio_read_length;
                if(audio_preroll > 0)
                {
                        if(audio_preroll >= output_length)
                                output_length = 0;
                        else
                        {
                                output_length -= audio_preroll;
                                for(int i = 0; i < MAX_CHANNELS; i++)
                                {
                                        if(audio_output_ptr[i])
                                                for(int j = 0; j < output_length; j++)
                                                {
                                                        audio_output_ptr[i][j] = audio_output_ptr[i][j + audio_read_length - output_length];
                                                }
                                }
                        }
//printf("PackageRenderer::do_audio 4\n");

                        audio_preroll -= audio_read_length;
                }

// Must perform writes even if 0 length so get_audio_buffer doesn't block
                result |= file->write_audio_buffer(output_length);
        }

        audio_position += audio_read_length;
//printf("PackageRenderer::do_audio 5\n");
}


void PackageRenderer::do_video()
{
// Do video data
        if(asset->video_data)
        {
// get the absolute video position from the audio position
                int64_t video_end = video_position + video_read_length;

                if(video_end > package->video_end)
                        video_end = package->video_end;

                while(video_position < video_end && !result)
                {
// Try to copy the compressed frame directly from the input to output files
//printf("PackageRenderer::do_video 2 video_position=%ld\n", video_position);
                        if(direct_frame_copy(command->get_edl(), 
                                video_position, 
                                file, 
                                result))
                        {
// Direct frame copy failed.
// Switch back to background compression
                                if(direct_frame_copying)
                                {
                                        file->start_video_thread(video_write_length, 
                                                command->get_edl()->session->color_model,
                                                preferences->processors > 1 ? 2 : 1,
                                                0);
                                        direct_frame_copying = 0;
                                }

// Try to use the rendering engine to write the frame.
// Get a buffer for background writing.



                                if(video_write_position == 0)
                                        video_output = file->get_video_buffer();





// Construct layered output buffer
                                for(int i = 0; i < MAX_CHANNELS; i++)
                                        video_output_ptr[i] = 
                                                (i < asset->layers) ? 
                                                        video_output[i][video_write_position] : 
                                                        0;

                                if(!result)
                                        result = render_engine->vrender->process_buffer(
                                                video_output_ptr, 
                                                video_position, 
                                                0);



                                if(!result && 
                                        mwindow && 
                                        video_device->output_visible())
                                {
// Vector for video device
                                        VFrame *preview_output[MAX_CHANNELS];

                                        video_device->new_output_buffers(preview_output,
                                                command->get_edl()->session->color_model);

                                        for(int i = 0; i < MAX_CHANNELS; i++)
                                                if(preview_output[i])
                                                        preview_output[i]->copy_from(video_output_ptr[i]);
                                        video_device->write_buffer(preview_output, 
                                                command->get_edl());
                                }



// Don't write to file
                                if(video_preroll && !result)
                                {
                                        video_preroll--;
// Keep the write position at 0 until ready to write real frames
                                        result = file->write_video_buffer(0);
                                        video_write_position = 0;
                                }
                                else
                                if(!result)
                                {
//printf("PackageRenderer::do_video 7\n");
// Set background rendering parameters
//                                      if(package->use_brender)
//                                      {
// Allow us to skip sections of the output file by setting the frame number.
// Used by background render and render farm.
                                                video_output_ptr[0]->set_number(video_position);
//printf("PackageRenderer::do_video 8 %p %lld\n", video_output_ptr[0], video_position);
//                                      }
                                        video_write_position++;

                                        if(video_write_position >= video_write_length)
                                        {
//printf("PackageRenderer::do_video 9\n");
                                                result = file->write_video_buffer(video_write_position);
// Update the brender map after writing the files.
                                                if(package->use_brender)
                                                {
//printf("PackageRenderer::do_video 10\n");
                                                        for(int i = 0; i < video_write_position && !result; i++)
                                                        {
                                                                result = set_video_map(video_position + 1 - video_write_position + i, 
                                                                        BRender::RENDERED);
                                                        }
//printf("PackageRenderer::do_video 11 %d\n", result);
                                                }
                                                video_write_position = 0;
                                        }
                                }


                        }

                        video_position++;
                        if(!result && get_result()) result = 1;
                        if(!result && progress_cancelled()) result = 1;
                }
        }
        else
        {
                video_position += video_read_length;
        }
}


void PackageRenderer::stop_engine()
{
        delete render_engine;
        delete playable_tracks;
}


void PackageRenderer::stop_output()
{
        int error = 0;
        if(asset->audio_data)
        {
// stop file I/O
                file->stop_audio_thread();
        }

        if(asset->video_data)
        {
                delete compressed_output;
                if(video_write_position)
                        file->write_video_buffer(video_write_position);
                if(package->use_brender)
                {
                        for(int i = 0; i < video_write_position && !error; i++)
                        {
                                error = set_video_map(video_position - video_write_position + i, 
                                        BRender::RENDERED);
                        }
                }
                video_write_position = 0;       
                if(!error) file->stop_video_thread();
                if(mwindow)
                {
                        video_device->stop_playback();
                        video_device->close_all();
                        delete video_device;
                }
        }
}


void PackageRenderer::close_output()
{
        if(mwindow)
                mwindow->sighandler->pull_file(file);
        file->close_file();
        delete file;
        delete asset;
}

// Aborts and returns 1 if an error is encountered.
int PackageRenderer::render_package(RenderPackage *package)
{
        int audio_done = 0;
        int video_done = 0;
        int samples_rendered = 0;


        result = 0;
        this->package = package;

// printf(
// "PackageRenderer::render_package: audio s=%lld l=%lld video s=%lld l=%lld\n",
//      package->audio_start, 
//      package->audio_end - package->audio_start, 
//      package->video_start, 
//      package->video_end - package->video_start);


        create_output();

        if(!asset->video_data) video_done = 1;
        if(!asset->audio_data) audio_done = 1;

// Create render engine
        if(!result)
        {
                create_engine();

//printf("PackageRenderer::render_package 5 %d\n", result);

// Main loop
                while((!audio_done || !video_done) && !result)
                {
                        int need_audio = 0, need_video = 0;




// Calculate lengths to process.  Audio fragment is constant.
                        if(!audio_done)
                        {
                                if(audio_position + audio_read_length >= package->audio_end)
                                {
                                        audio_done = 1;
                                        audio_read_length = package->audio_end - audio_position;
                                }

                                samples_rendered = audio_read_length;
                                need_audio = 1;
                        }

//printf("PackageRenderer::render_package 6 %d\n", samples_rendered);

                        if(!video_done)
                        {
                                if(audio_done)
                                {
                                        video_read_length = package->video_end - video_position;
// Packetize video length so progress gets updated
                                        video_read_length = (int)MIN(asset->frame_rate, video_read_length);
                                        video_read_length = MAX(video_read_length, 30);
                                }
                                else
// Guide video with audio
                                {
                                        video_read_length = Units::to_int64(
                                                (double)(audio_position + audio_read_length) / 
                                                asset->sample_rate * 
                                                asset->frame_rate) - 
                                                video_position;
                                }

// Clamp length
                                if(video_position + video_read_length >= package->video_end)
                                {
                                        video_done = 1;
                                        video_read_length = package->video_end - video_position;
                                }

// Calculate samples rendered for progress bar.
                                if(audio_done)
                                        samples_rendered = Units::round((double)video_read_length /
                                                asset->frame_rate *
                                                asset->sample_rate);

                                need_video = 1;
                        }

//printf("PackageRenderer::render_package 1 %d %lld %lld\n", result, audio_read_length, video_read_length);
                        if(need_video && !result) do_video();
//printf("PackageRenderer::render_package 7 %d %d\n", result, samples_rendered);
                        if(need_audio && !result) do_audio();


                        if(!result) set_progress(samples_rendered);





                        if(!result && progress_cancelled()) result = 1;

// printf("PackageRenderer::render_package 10 %d %d %d %d\n", 
// audio_read_length, video_read_length, samples_rendered, result);
                        if(result) 
                                set_result(result);
                        else
                                result = get_result();
                }

//printf("PackageRenderer::render_package 20\n");
                stop_engine();
//printf("PackageRenderer::render_package 30\n");

                stop_output();
//printf("PackageRenderer::render_package 40\n");


        }



//printf("PackageRenderer::render_package 50\n");
        close_output();
//printf("PackageRenderer::render_package 60\n");


        set_result(result);
//printf("PackageRenderer::render_package 70\n");



        return result;
}








// Try to copy the compressed frame directly from the input to output files
// Return 1 on failure and 0 on success
int PackageRenderer::direct_frame_copy(EDL *edl, 
        int64_t &video_position, 
        File *file,
        int &error)
{
        Track *playable_track;
        Edit *playable_edit;
        int64_t frame_size;

//printf("Render::direct_frame_copy 1\n");
        if(direct_copy_possible(edl, 
                video_position, 
                playable_track, 
                playable_edit, 
                file))
        {
// Switch to direct copying
                if(!direct_frame_copying)
                {
                        if(video_write_position)
                        {
                                error |= file->write_video_buffer(video_write_position);
                                video_write_position = 0;
                        }
                        file->stop_video_thread();
                        direct_frame_copying = 1;
                }
//printf("Render::direct_frame_copy 2\n");

                if(!package->use_brender)
                        error |= ((VEdit*)playable_edit)->read_frame(compressed_output, 
                                video_position,
                                PLAY_FORWARD,
                                video_cache,
                                1,
                                0);


                if(!error && video_preroll > 0)
                {
                        video_preroll--;
                }
                else
                if(!error)
                {
                        if(package->use_brender)
                        {
//printf("PackageRenderer::direct_frame_copy 1\n");
                                error = set_video_map(video_position, BRender::SCANNED);
//printf("PackageRenderer::direct_frame_copy 10 %d\n", error);
                        }
                        else
                        {
                                VFrame ***temp_output = new VFrame**[1];
                                temp_output[0] = new VFrame*[1];
                                temp_output[0][0] = compressed_output;
                                error = file->write_frames(temp_output, 1);
                                delete temp_output[0];
                                delete temp_output;
                        }
                }
                return 0;
        }
        else
                return 1;
}

int PackageRenderer::direct_copy_possible(EDL *edl,
                                int64_t current_position, 
                                Track* playable_track,  // The one track which is playable
                                Edit* &playable_edit, // The edit which is playing
                                File *file)   // Output file
{
        int result = 1;
        int total_playable_tracks = 0;
        Track* current_track;
        Patch* current_patch;
        Auto* current_auto;
        int temp;

// Number of playable tracks must equal 1
        for(current_track = edl->tracks->first;
                current_track && result; 
                current_track = current_track->next)
        {
                if(current_track->data_type == TRACK_VIDEO)
                {
                        if(playable_tracks->is_playable(current_track, current_position, 1))
                        {
                                playable_track = current_track;
                                total_playable_tracks++;
                        }
                }
        }

//printf("Render::direct_copy_possible 1 %d\n", result);
        if(total_playable_tracks != 1) result = 0;
//printf("Render::direct_copy_possible 2 %d\n", result);

// Edit must have a source file
        if(result)
        {
//printf("Render::direct_copy_possible 3 %d\n", result);
                playable_edit = playable_track->edits->get_playable_edit(current_position, 1);
//printf("Render::direct_copy_possible 4 %d %p\n", result, playable_edit);
                if(!playable_edit)
                        result = 0;
        }

// Source file must be able to copy to destination file.
// Source file must be same size as project output.
        if(result)
        {
//printf("Render::direct_copy_possible 5 %d\n", result);
                if(!file->can_copy_from(playable_edit, 
                        current_position + playable_track->nudge,
                        edl->session->output_w, 
                        edl->session->output_h))
                        result = 0;
        }
//printf("Render::direct_copy_possible 6 %d\n", result);

// Test conditions mutual between vrender.C and this.
        if(result && 
                !playable_track->direct_copy_possible(current_position, PLAY_FORWARD, 1))
                result = 0;
//printf("Render::direct_copy_possible 7 %d\n", result);

        return result;
}









int PackageRenderer::get_master()
{
        return 0;
}

// Get result status from server
int PackageRenderer::get_result()
{
        return 0;
}

void PackageRenderer::set_result(int value)
{
}

void PackageRenderer::set_progress(int64_t value)
{
}       

int PackageRenderer::set_video_map(int64_t position, int value)
{
}

int PackageRenderer::progress_cancelled()
{
        return 0;
}       

---