forked from mia/Aegisub
Rather than everything having to separately commit changes to the ass and then tell the subs grid to notify various parts of Aegisub about the changes, committing the AssFile now triggers an event which objects listen for. AssFile::Commit now also has an argument to indicate what sorts of changes were made to the file. For now these types are very broad. Originally committed to SVN as r4901.
546 lines
20 KiB
C++
546 lines
20 KiB
C++
// Copyright (c) 2006, Rodrigo Braz Monteiro
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are met:
|
|
//
|
|
// * Redistributions of source code must retain the above copyright notice,
|
|
// this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above copyright notice,
|
|
// this list of conditions and the following disclaimer in the documentation
|
|
// and/or other materials provided with the distribution.
|
|
// * Neither the name of the Aegisub Group nor the names of its contributors
|
|
// may be used to endorse or promote products derived from this software
|
|
// without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
|
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
|
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
|
|
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
|
|
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
|
|
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
|
|
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
|
|
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
|
|
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
|
|
// POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
// Aegisub Project http://www.aegisub.org/
|
|
//
|
|
// $Id$
|
|
|
|
/// @file dialog_timing_processor.cpp
|
|
/// @brief Timing Post-processor dialogue box and logic
|
|
/// @ingroup tools_ui
|
|
///
|
|
|
|
#include "config.h"
|
|
|
|
#include "ass_dialogue.h"
|
|
#include "ass_file.h"
|
|
#include "ass_time.h"
|
|
#include "dialog_timing_processor.h"
|
|
#include "help_button.h"
|
|
#include "libresrc/libresrc.h"
|
|
#include "main.h"
|
|
#include "subs_grid.h"
|
|
#include "utils.h"
|
|
#include "validators.h"
|
|
#include "video_box.h"
|
|
#include "video_context.h"
|
|
#include "video_display.h"
|
|
|
|
/// Window IDs
|
|
enum {
|
|
CHECK_ENABLE_LEADIN = 1850,
|
|
CHECK_ENABLE_LEADOUT,
|
|
CHECK_ENABLE_KEYFRAME,
|
|
CHECK_ENABLE_ADJASCENT,
|
|
BUTTON_SELECT_ALL,
|
|
BUTTON_SELECT_NONE,
|
|
TIMING_STYLE_LIST
|
|
};
|
|
|
|
/// @brief Constructor
|
|
/// @param parent
|
|
/// @param _grid
|
|
///
|
|
DialogTimingProcessor::DialogTimingProcessor(wxWindow *parent,SubtitlesGrid *_grid)
|
|
: wxDialog(parent,-1,_("Timing Post-Processor"),wxDefaultPosition,wxSize(400,250),wxDEFAULT_DIALOG_STYLE)
|
|
{
|
|
SetIcon(BitmapToIcon(GETIMAGE(timing_processor_toolbutton_24)));
|
|
|
|
// Set variables
|
|
grid = _grid;
|
|
leadInTime = AegiIntegerToString(OPT_GET("Audio/Lead/IN")->GetInt());
|
|
leadOutTime = AegiIntegerToString(OPT_GET("Audio/Lead/OUT")->GetInt());
|
|
thresStartBefore = AegiIntegerToString(OPT_GET("Tool/Timing Post Processor/Threshold/Key Start Before")->GetInt());
|
|
thresStartAfter = AegiIntegerToString(OPT_GET("Tool/Timing Post Processor/Threshold/Key Start After")->GetInt());
|
|
thresEndBefore = AegiIntegerToString(OPT_GET("Tool/Timing Post Processor/Threshold/Key End Before")->GetInt());
|
|
thresEndAfter = AegiIntegerToString(OPT_GET("Tool/Timing Post Processor/Threshold/Key End After")->GetInt());
|
|
adjsThresTime = AegiIntegerToString(OPT_GET("Tool/Timing Post Processor/Threshold/Adjacent")->GetInt());
|
|
|
|
// Styles box
|
|
wxSizer *LeftSizer = new wxStaticBoxSizer(wxVERTICAL,this,_("Apply to styles"));
|
|
wxArrayString styles = grid->ass->GetStyles();
|
|
StyleList = new wxCheckListBox(this,TIMING_STYLE_LIST,wxDefaultPosition,wxSize(150,150),styles);
|
|
StyleList->SetToolTip(_("Select styles to process. Unchecked ones will be ignored."));
|
|
wxButton *all = new wxButton(this,BUTTON_SELECT_ALL,_("All"));
|
|
all->SetToolTip(_("Select all styles."));
|
|
wxButton *none = new wxButton(this,BUTTON_SELECT_NONE,_("None"));
|
|
none->SetToolTip(_("Deselect all styles."));
|
|
|
|
// Options box
|
|
wxSizer *optionsSizer = new wxStaticBoxSizer(wxHORIZONTAL,this,_("Options"));
|
|
onlySelection = new wxCheckBox(this,-1,_("Affect selection only"));
|
|
onlySelection->SetValue(OPT_GET("Tool/Timing Post Processor/Only Selection")->GetBool());
|
|
optionsSizer->Add(onlySelection,1,wxALL,0);
|
|
|
|
// Lead-in/out box
|
|
wxSizer *LeadSizer = new wxStaticBoxSizer(wxHORIZONTAL,this,_("Lead-in/Lead-out"));
|
|
hasLeadIn = new wxCheckBox(this,CHECK_ENABLE_LEADIN,_("Add lead in:"));
|
|
hasLeadIn->SetToolTip(_("Enable adding of lead-ins to lines."));
|
|
hasLeadIn->SetValue(OPT_GET("Tool/Timing Post Processor/Enable/Lead/IN")->GetBool());
|
|
leadIn = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(80,-1),0,NumValidator(&leadInTime));
|
|
leadIn->SetToolTip(_("Lead in to be added, in milliseconds."));
|
|
hasLeadOut = new wxCheckBox(this,CHECK_ENABLE_LEADOUT,_("Add lead out:"));
|
|
hasLeadOut->SetToolTip(_("Enable adding of lead-outs to lines."));
|
|
hasLeadOut->SetValue(OPT_GET("Tool/Timing Post Processor/Enable/Lead/OUT")->GetBool());
|
|
leadOut = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(80,-1),0,NumValidator(&leadOutTime));
|
|
leadOut->SetToolTip(_("Lead out to be added, in milliseconds."));
|
|
LeadSizer->Add(hasLeadIn,0,wxRIGHT|wxEXPAND,5);
|
|
LeadSizer->Add(leadIn,0,wxRIGHT|wxEXPAND,5);
|
|
LeadSizer->Add(hasLeadOut,0,wxRIGHT|wxEXPAND,5);
|
|
LeadSizer->Add(leadOut,0,wxRIGHT|wxEXPAND,0);
|
|
LeadSizer->AddStretchSpacer(1);
|
|
|
|
// Adjacent subs sizer
|
|
wxSizer *AdjacentSizer = new wxStaticBoxSizer(wxHORIZONTAL,this,_("Make adjacent subtitles continuous"));
|
|
adjsEnable = new wxCheckBox(this,CHECK_ENABLE_ADJASCENT,_("Enable"));
|
|
adjsEnable->SetToolTip(_("Enable snapping of subtitles together if they are within a certain distance of each other."));
|
|
adjsEnable->SetValue(OPT_GET("Tool/Timing Post Processor/Enable/Adjacent")->GetBool());
|
|
wxStaticText *adjsThresText = new wxStaticText(this,-1,_("Threshold:"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE);
|
|
adjacentThres = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(60,-1),0,NumValidator(&adjsThresTime));
|
|
adjacentThres->SetToolTip(_("Maximum difference between start and end time for two subtitles to be made continuous, in milliseconds."));
|
|
adjacentBias = new wxSlider(this,-1,MID(0,int(OPT_GET("Tool/Timing Post Processor/Adjacent Bias")->GetDouble()*100),100),0,100,wxDefaultPosition,wxSize(-1,20));
|
|
adjacentBias->SetToolTip(_("Sets how to set the adjoining of lines. If set totally to left, it will extend start time of the second line; if totally to right, it will extend the end time of the first line."));
|
|
AdjacentSizer->Add(adjsEnable,0,wxRIGHT|wxEXPAND,10);
|
|
AdjacentSizer->Add(adjsThresText,0,wxRIGHT|wxALIGN_CENTER,5);
|
|
AdjacentSizer->Add(adjacentThres,0,wxRIGHT|wxEXPAND,5);
|
|
AdjacentSizer->Add(new wxStaticText(this,-1,_("Bias: Start <- "),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE),0,wxALIGN_CENTER,0);
|
|
AdjacentSizer->Add(adjacentBias,1,wxEXPAND,0);
|
|
AdjacentSizer->Add(new wxStaticText(this,-1,_(" -> End"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE),0,wxALIGN_CENTER,0);
|
|
|
|
// Keyframes sizer
|
|
KeyframesSizer = new wxStaticBoxSizer(wxHORIZONTAL,this,_("Keyframe snapping"));
|
|
wxSizer *KeyframesFlexSizer = new wxFlexGridSizer(2,5,5,0);
|
|
keysEnable = new wxCheckBox(this,CHECK_ENABLE_KEYFRAME,_("Enable"));
|
|
keysEnable->SetToolTip(_("Enable snapping of subtitles to nearest keyframe, if distance is within threshold."));
|
|
keysEnable->SetValue(OPT_GET("Tool/Timing Post Processor/Enable/Keyframe")->GetBool());
|
|
wxStaticText *textStartBefore = new wxStaticText(this,-1,_("Starts before thres.:"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE);
|
|
keysStartBefore = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(60,-1),0,NumValidator(&thresStartBefore));
|
|
keysStartBefore->SetToolTip(_("Threshold for 'before start' distance, that is, how many frames a subtitle must start before a keyframe to snap to it."));
|
|
wxStaticText *textStartAfter = new wxStaticText(this,-1,_("Starts after thres.:"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE);
|
|
keysStartAfter = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(60,-1),0,NumValidator(&thresStartAfter));
|
|
keysStartAfter->SetToolTip(_("Threshold for 'after start' distance, that is, how many frames a subtitle must start after a keyframe to snap to it."));
|
|
wxStaticText *textEndBefore = new wxStaticText(this,-1,_("Ends before thres.:"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE);
|
|
keysEndBefore = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(60,-1),0,NumValidator(&thresEndBefore));
|
|
keysEndBefore->SetToolTip(_("Threshold for 'before end' distance, that is, how many frames a subtitle must end before a keyframe to snap to it."));
|
|
wxStaticText *textEndAfter = new wxStaticText(this,-1,_("Ends after thres.:"),wxDefaultPosition,wxDefaultSize,wxALIGN_CENTRE);
|
|
keysEndAfter = new wxTextCtrl(this,-1,_T(""),wxDefaultPosition,wxSize(60,-1),0,NumValidator(&thresEndAfter));
|
|
keysEndAfter->SetToolTip(_("Threshold for 'after end' distance, that is, how many frames a subtitle must end after a keyframe to snap to it."));
|
|
KeyframesFlexSizer->Add(keysEnable,0,wxRIGHT|wxEXPAND,10);
|
|
KeyframesFlexSizer->Add(textStartBefore,0,wxRIGHT|wxALIGN_CENTER,5);
|
|
KeyframesFlexSizer->Add(keysStartBefore,0,wxRIGHT|wxEXPAND,5);
|
|
KeyframesFlexSizer->Add(textStartAfter,0,wxRIGHT|wxALIGN_CENTER,5);
|
|
KeyframesFlexSizer->Add(keysStartAfter,0,wxRIGHT|wxEXPAND,0);
|
|
KeyframesFlexSizer->AddStretchSpacer(1);
|
|
KeyframesFlexSizer->Add(textEndBefore,0,wxRIGHT|wxALIGN_CENTER,5);
|
|
KeyframesFlexSizer->Add(keysEndBefore,0,wxRIGHT|wxEXPAND,5);
|
|
KeyframesFlexSizer->Add(textEndAfter,0,wxRIGHT|wxALIGN_CENTER,5);
|
|
KeyframesFlexSizer->Add(keysEndAfter,0,wxRIGHT|wxEXPAND,0);
|
|
KeyframesSizer->Add(KeyframesFlexSizer,0,wxEXPAND);
|
|
KeyframesSizer->AddStretchSpacer(1);
|
|
|
|
// Button sizer
|
|
wxStdDialogButtonSizer *ButtonSizer = new wxStdDialogButtonSizer();
|
|
ApplyButton = new wxButton(this,wxID_OK);
|
|
ButtonSizer->AddButton(ApplyButton);
|
|
ButtonSizer->AddButton(new wxButton(this,wxID_CANCEL));
|
|
ButtonSizer->AddButton(new HelpButton(this,_T("Timing Processor")));
|
|
ButtonSizer->Realize();
|
|
|
|
// Right Sizer
|
|
wxSizer *RightSizer = new wxBoxSizer(wxVERTICAL);
|
|
RightSizer->Add(optionsSizer,0,wxBOTTOM|wxEXPAND,5);
|
|
RightSizer->Add(LeadSizer,0,wxBOTTOM|wxEXPAND,5);
|
|
RightSizer->Add(AdjacentSizer,0,wxBOTTOM|wxEXPAND,5);
|
|
RightSizer->Add(KeyframesSizer,0,wxBOTTOM|wxEXPAND,5);
|
|
RightSizer->AddStretchSpacer(1);
|
|
RightSizer->Add(ButtonSizer,0,wxLEFT|wxRIGHT|wxBOTTOM|wxEXPAND,0);
|
|
|
|
// Style buttons sizer
|
|
wxSizer *StyleButtonsSizer = new wxBoxSizer(wxHORIZONTAL);
|
|
StyleButtonsSizer->Add(all,1,0,0);
|
|
StyleButtonsSizer->Add(none,1,0,0);
|
|
|
|
// Left sizer
|
|
size_t len = StyleList->GetCount();
|
|
for (size_t i=0;i<len;i++) {
|
|
StyleList->Check(i);
|
|
}
|
|
LeftSizer->Add(StyleList,1,wxBOTTOM|wxEXPAND,0);
|
|
LeftSizer->Add(StyleButtonsSizer,0,wxEXPAND,0);
|
|
|
|
// Top Sizer
|
|
wxSizer *TopSizer = new wxBoxSizer(wxHORIZONTAL);
|
|
TopSizer->Add(LeftSizer,0,wxRIGHT|wxEXPAND,5);
|
|
TopSizer->Add(RightSizer,1,wxALL|wxEXPAND,0);
|
|
|
|
// Main Sizer
|
|
wxSizer *MainSizer = new wxBoxSizer(wxVERTICAL);
|
|
MainSizer->Add(TopSizer,1,wxALL|wxEXPAND,5);
|
|
MainSizer->SetSizeHints(this);
|
|
SetSizer(MainSizer);
|
|
|
|
CenterOnParent();
|
|
|
|
UpdateControls();
|
|
}
|
|
|
|
/// @brief Update controls
|
|
///
|
|
void DialogTimingProcessor::UpdateControls() {
|
|
// Boxes
|
|
leadIn->Enable(hasLeadIn->IsChecked());
|
|
leadOut->Enable(hasLeadOut->IsChecked());
|
|
adjacentThres->Enable(adjsEnable->IsChecked());
|
|
adjacentBias->Enable(adjsEnable->IsChecked());
|
|
|
|
// Keyframes are only available if timecodes are loaded
|
|
bool keysAvailable = VideoContext::Get()->KeyFramesLoaded();
|
|
bool enableKeys = keysEnable->IsChecked() && keysAvailable;
|
|
keysStartBefore->Enable(enableKeys);
|
|
keysStartAfter->Enable(enableKeys);
|
|
keysEndBefore->Enable(enableKeys);
|
|
keysEndAfter->Enable(enableKeys);
|
|
if (!keysAvailable) {
|
|
keysEnable->SetValue(false);
|
|
keysEnable->Enable(false);
|
|
}
|
|
|
|
// Apply button
|
|
int checked = 0;
|
|
size_t len = StyleList->GetCount();
|
|
for (size_t i=0;i<len;i++) {
|
|
if (StyleList->IsChecked(i)) checked++;
|
|
}
|
|
ApplyButton->Enable(checked && (hasLeadIn->IsChecked() | hasLeadOut->IsChecked() | keysEnable->IsChecked() | adjsEnable->IsChecked()));
|
|
}
|
|
|
|
BEGIN_EVENT_TABLE(DialogTimingProcessor,wxDialog)
|
|
EVT_CHECKBOX(CHECK_ENABLE_LEADIN,DialogTimingProcessor::OnCheckBox)
|
|
EVT_CHECKBOX(CHECK_ENABLE_LEADOUT,DialogTimingProcessor::OnCheckBox)
|
|
EVT_CHECKBOX(CHECK_ENABLE_KEYFRAME,DialogTimingProcessor::OnCheckBox)
|
|
EVT_CHECKBOX(CHECK_ENABLE_ADJASCENT,DialogTimingProcessor::OnCheckBox)
|
|
EVT_CHECKLISTBOX(TIMING_STYLE_LIST,DialogTimingProcessor::OnCheckBox)
|
|
EVT_BUTTON(wxID_OK,DialogTimingProcessor::OnApply)
|
|
EVT_BUTTON(BUTTON_SELECT_ALL,DialogTimingProcessor::OnSelectAll)
|
|
EVT_BUTTON(BUTTON_SELECT_NONE,DialogTimingProcessor::OnSelectNone)
|
|
END_EVENT_TABLE()
|
|
|
|
void DialogTimingProcessor::OnCheckBox(wxCommandEvent &) {
|
|
UpdateControls();
|
|
}
|
|
|
|
void DialogTimingProcessor::OnSelectAll(wxCommandEvent &) {
|
|
size_t len = StyleList->GetCount();
|
|
for (size_t i=0;i<len;i++) {
|
|
StyleList->Check(i);
|
|
}
|
|
UpdateControls();
|
|
}
|
|
|
|
void DialogTimingProcessor::OnSelectNone(wxCommandEvent &) {
|
|
size_t len = StyleList->GetCount();
|
|
for (size_t i=0;i<len;i++) {
|
|
StyleList->Check(i,false);
|
|
}
|
|
UpdateControls();
|
|
}
|
|
|
|
void DialogTimingProcessor::OnApply(wxCommandEvent &) {
|
|
// Save settings
|
|
long temp = 0;
|
|
leadIn->GetValue().ToLong(&temp);
|
|
OPT_SET("Audio/Lead/IN")->SetInt(temp);
|
|
leadOut->GetValue().ToLong(&temp);
|
|
OPT_SET("Audio/Lead/OUT")->SetInt(temp);
|
|
keysStartBefore->GetValue().ToLong(&temp);
|
|
OPT_SET("Tool/Timing Post Processor/Threshold/Key Start Before")->SetInt(temp);
|
|
keysStartAfter->GetValue().ToLong(&temp);
|
|
OPT_SET("Tool/Timing Post Processor/Threshold/Key Start After")->SetInt(temp);
|
|
keysEndBefore->GetValue().ToLong(&temp);
|
|
OPT_SET("Tool/Timing Post Processor/Threshold/Key End Before")->SetInt(temp);
|
|
keysEndAfter->GetValue().ToLong(&temp);
|
|
OPT_SET("Tool/Timing Post Processor/Threshold/Key End After")->SetInt(temp);
|
|
adjacentThres->GetValue().ToLong(&temp);
|
|
OPT_SET("Tool/Timing Post Processor/Threshold/Adjacent")->SetInt(temp);
|
|
OPT_SET("Tool/Timing Post Processor/Adjacent Bias")->SetDouble(adjacentBias->GetValue() / 100.0);
|
|
OPT_SET("Tool/Timing Post Processor/Enable/Lead/IN")->SetBool(hasLeadIn->IsChecked());
|
|
OPT_SET("Tool/Timing Post Processor/Enable/Lead/OUT")->SetBool(hasLeadOut->IsChecked());
|
|
if (keysEnable->IsEnabled()) OPT_SET("Tool/Timing Post Processor/Enable/Keyframe")->SetBool(keysEnable->IsChecked());
|
|
OPT_SET("Tool/Timing Post Processor/Enable/Adjacent")->SetBool(adjsEnable->IsChecked());
|
|
OPT_SET("Tool/Timing Post Processor/Only Selection")->SetBool(onlySelection->IsChecked());
|
|
|
|
// Check if rows are valid
|
|
bool valid = true;
|
|
AssDialogue *tempDiag;
|
|
int i = 0;
|
|
for (std::list<AssEntry*>::iterator cur=grid->ass->Line.begin();cur!=grid->ass->Line.end();cur++) {
|
|
tempDiag = dynamic_cast<AssDialogue*>(*cur);
|
|
if (tempDiag) {
|
|
i++;
|
|
if (tempDiag->Start.GetMS() > tempDiag->End.GetMS()) {
|
|
valid = false;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (valid) Process();
|
|
else wxMessageBox(wxString::Format(_("One of the lines in the file (%i) has negative duration. Aborting."),i),_("Invalid script"),wxICON_ERROR|wxOK);
|
|
|
|
EndModal(0);
|
|
}
|
|
|
|
int DialogTimingProcessor::GetClosestKeyFrame(int frame) {
|
|
// Linear dumb search, not very efficient, but it doesn't really matter
|
|
int closest = 0;
|
|
size_t n = KeyFrames.size();
|
|
for (size_t i=0;i<n;i++) {
|
|
if (abs(KeyFrames[i]-frame) < abs(closest-frame)) {
|
|
closest = KeyFrames[i];
|
|
}
|
|
}
|
|
return closest;
|
|
}
|
|
|
|
/// @brief Check if style is listed
|
|
/// @param styleName
|
|
/// @return
|
|
///
|
|
bool DialogTimingProcessor::StyleOK(wxString styleName) {
|
|
size_t len = StyleList->GetCount();
|
|
for (size_t i=0;i<len;i++) {
|
|
if (StyleList->GetString(i) == styleName && StyleList->IsChecked(i)) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
/// @brief Sort dialogues
|
|
///
|
|
void DialogTimingProcessor::SortDialogues() {
|
|
// Copy from original to temporary list
|
|
std::list<AssDialogue*> temp;
|
|
AssDialogue *tempDiag;
|
|
int count = grid->GetRows();
|
|
for (int i=0;i<count;i++) {
|
|
tempDiag = grid->GetDialogue(i);
|
|
if (tempDiag && StyleOK(tempDiag->Style) && !tempDiag->Comment) {
|
|
if (!onlySelection->IsChecked() || grid->IsInSelection(i)) {
|
|
temp.push_back(tempDiag);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Sort temporary list
|
|
AssFile::Sort(temp);
|
|
|
|
// Copy temporary list to final vector
|
|
Sorted.clear();
|
|
for (std::list<AssDialogue*>::iterator cur=temp.begin();cur!=temp.end();cur++) {
|
|
Sorted.push_back(*cur);
|
|
}
|
|
}
|
|
|
|
/// @brief Gets sorted dialogue
|
|
/// @param n
|
|
/// @return
|
|
///
|
|
AssDialogue *DialogTimingProcessor::GetSortedDialogue(int n) {
|
|
try {
|
|
return Sorted.at(n);
|
|
}
|
|
catch (...) {
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
/// @brief Actually process subtitles
|
|
///
|
|
void DialogTimingProcessor::Process() {
|
|
SortDialogues();
|
|
int rows = Sorted.size();
|
|
|
|
// Options
|
|
long inVal = 0;
|
|
long outVal = 0;
|
|
leadIn->GetValue().ToLong(&inVal);
|
|
leadOut->GetValue().ToLong(&outVal);
|
|
bool addIn = hasLeadIn->IsChecked() && inVal;
|
|
bool addOut = hasLeadOut->IsChecked() && outVal;
|
|
|
|
// Add lead-in/out
|
|
if (addIn || addOut) {
|
|
// Variables
|
|
AssDialogue *cur;
|
|
AssDialogue *comp;
|
|
int start,end;
|
|
int startLead,endLead;
|
|
int compStart,compEnd;
|
|
|
|
// For each row
|
|
for (int i=0;i<rows;i++) {
|
|
// Get line and check if it's OK
|
|
cur = GetSortedDialogue(i);
|
|
|
|
// Set variables
|
|
start = cur->Start.GetMS();
|
|
end = cur->End.GetMS();
|
|
if (addIn) startLead = start - inVal;
|
|
else startLead = start;
|
|
if (addOut) endLead = end + outVal;
|
|
else endLead = end;
|
|
|
|
// Compare to every previous line (yay for O(n^2)!) to see if it's OK to add lead-in
|
|
if (addIn) {
|
|
for (int j=0;j<i;j++) {
|
|
comp = GetSortedDialogue(j);
|
|
|
|
// Check if they don't already collide (ignore it in that case)
|
|
if (cur->CollidesWith(comp)) continue;
|
|
|
|
// Get comparison times
|
|
compEnd = comp->End.GetMS();
|
|
|
|
// Limit lead-in if needed
|
|
if (compEnd > startLead) startLead = compEnd;
|
|
}
|
|
}
|
|
|
|
// Compare to every line to see how far can lead-out be extended
|
|
if (addOut) {
|
|
for (int j=i+1;j<rows;j++) {
|
|
comp = GetSortedDialogue(j);
|
|
|
|
// Check if they don't already collide (ignore it in that case)
|
|
if (cur->CollidesWith(comp)) continue;
|
|
|
|
// Get comparison times
|
|
compStart = comp->Start.GetMS();
|
|
|
|
// Limit lead-in if needed
|
|
if (compStart < endLead) endLead = compStart;
|
|
}
|
|
}
|
|
|
|
// Set times
|
|
cur->Start.SetMS(startLead);
|
|
cur->End.SetMS(endLead);
|
|
}
|
|
}
|
|
|
|
// Make adjacent
|
|
if (adjsEnable->IsChecked()) {
|
|
// Variables
|
|
AssDialogue *cur;
|
|
AssDialogue *prev = NULL;
|
|
int curStart,prevEnd;
|
|
long adjsThres = 0;
|
|
int dist;
|
|
|
|
// Get threshold
|
|
adjacentThres->GetValue().ToLong(&adjsThres);
|
|
|
|
// Get bias
|
|
float bias = adjacentBias->GetValue() / 100.0;
|
|
|
|
// For each row
|
|
for (int i=0;i<rows;i++) {
|
|
// Get line and check if it's OK
|
|
cur = GetSortedDialogue(i);
|
|
|
|
// Check if previous is OK
|
|
if (!prev) {
|
|
prev = cur;
|
|
continue;
|
|
}
|
|
|
|
// Check if they don't collide
|
|
if (cur->CollidesWith(prev)) continue;
|
|
|
|
// Compare distance
|
|
curStart = cur->Start.GetMS();
|
|
prevEnd = prev->End.GetMS();
|
|
dist = curStart-prevEnd;
|
|
if (dist > 0 && dist < adjsThres) {
|
|
int setPos = prevEnd+int(dist*bias);
|
|
cur->Start.SetMS(setPos);
|
|
prev->End.SetMS(setPos);
|
|
}
|
|
|
|
// Set previous
|
|
prev = cur;
|
|
}
|
|
}
|
|
|
|
// Keyframe snapping
|
|
if (keysEnable->IsChecked()) {
|
|
// Get keyframes
|
|
VideoContext *con = VideoContext::Get();
|
|
KeyFrames = con->GetKeyFrames();
|
|
KeyFrames.push_back(con->GetLength()-1);
|
|
|
|
// Variables
|
|
int startF,endF;
|
|
int closest;
|
|
AssDialogue *cur;
|
|
|
|
// Get variables
|
|
long beforeStart = 0;
|
|
long afterStart = 0;
|
|
long beforeEnd = 0;
|
|
long afterEnd = 0;
|
|
keysStartBefore->GetValue().ToLong(&beforeStart);
|
|
keysStartAfter->GetValue().ToLong(&afterStart);
|
|
keysEndBefore->GetValue().ToLong(&beforeEnd);
|
|
keysEndAfter->GetValue().ToLong(&afterEnd);
|
|
|
|
// For each row
|
|
for (int i=0;i<rows;i++) {
|
|
// Get line and check if it's OK
|
|
cur = GetSortedDialogue(i);
|
|
|
|
// Get start/end frames
|
|
startF = con->FrameAtTime(cur->Start.GetMS(),agi::vfr::START);
|
|
endF = con->FrameAtTime(cur->End.GetMS(),agi::vfr::END);
|
|
|
|
// Get closest for start
|
|
closest = GetClosestKeyFrame(startF);
|
|
if ((closest > startF && closest-startF <= beforeStart) || (closest < startF && startF-closest <= afterStart)) {
|
|
cur->Start.SetMS(con->TimeAtFrame(closest,agi::vfr::START));
|
|
}
|
|
|
|
// Get closest for end
|
|
closest = GetClosestKeyFrame(endF)-1;
|
|
if ((closest > endF && closest-endF <= beforeEnd) || (closest < endF && endF-closest <= afterEnd)) {
|
|
cur->End.SetMS(con->TimeAtFrame(closest,agi::vfr::END));
|
|
}
|
|
}
|
|
}
|
|
|
|
// Update grid
|
|
grid->ass->Commit(_("timing processor"), AssFile::COMMIT_TIMES);
|
|
}
|