Aegisub/aegisub/src/dialog_resample.cpp

259 lines
8.6 KiB
C++
Raw Normal View History

// Copyright (c) 2012, Thomas Goyne <plorkyeran@aegisub.org>
2006-01-16 22:02:54 +01:00
//
// Permission to use, copy, modify, and distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice appear in all copies.
2006-01-16 22:02:54 +01:00
//
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
// ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
// ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
// OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
2006-01-16 22:02:54 +01:00
//
// $Id$
/// @file dialog_resample.cpp
/// @brief Resample Resolution dialogue box and logic
/// @ingroup tools_ui
///
2006-01-16 22:02:54 +01:00
#include "config.h"
#include "dialog_resample.h"
#ifndef AGI_PRE
#include <algorithm>
#include <tr1/functional>
#include <wx/checkbox.h>
#include <wx/sizer.h>
#include <wx/spinctrl.h>
#include <wx/statbox.h>
#include <wx/stattext.h>
#include <wx/valgen.h>
#endif
2006-01-16 22:02:54 +01:00
#include "ass_dialogue.h"
#include "ass_file.h"
2006-01-16 22:02:54 +01:00
#include "ass_override.h"
#include "ass_style.h"
#include "include/aegisub/context.h"
#include "help_button.h"
#include "libresrc/libresrc.h"
#include "utils.h"
#include "video_context.h"
2006-01-16 22:02:54 +01:00
enum {
LEFT = 0,
RIGHT = 1,
TOP = 2,
BOTTOM = 3
};
DialogResample::DialogResample(agi::Context *c, ResampleSettings &settings)
: wxDialog(c->parent, -1, _("Resample Resolution"))
, c(c)
2006-01-16 22:02:54 +01:00
{
SetIcon(BitmapToIcon(GETIMAGE(resample_toolbutton_24)));
memset(&settings, 0, sizeof(settings));
c->ass->GetResolution(settings.script_x, settings.script_y);
2006-01-16 22:02:54 +01:00
// Create all controls and set validators
for (size_t i = 0; i < 4; ++i) {
margin_ctrl[i] = new wxSpinCtrl(this, -1, "0", wxDefaultPosition, wxSize(50, -1), wxSP_ARROW_KEYS, -9999, 9999, 0);
margin_ctrl[i]->SetValidator(wxGenericValidator(&settings.margin[i]));
}
2006-01-16 22:02:54 +01:00
symmetrical = new wxCheckBox(this, -1, _("&Symmetrical"));
symmetrical->SetValue(true);
margin_ctrl[RIGHT]->Enable(false);
margin_ctrl[BOTTOM]->Enable(false);
res_x = new wxSpinCtrl(this, -1, "", wxDefaultPosition, wxSize(50, -1), wxSP_ARROW_KEYS, 1, INT_MAX);
res_y = new wxSpinCtrl(this, -1, "", wxDefaultPosition, wxSize(50, -1), wxSP_ARROW_KEYS, 1, INT_MAX);
res_x->SetValidator(wxGenericValidator(&settings.script_x));
res_y->SetValidator(wxGenericValidator(&settings.script_y));
wxButton *from_video = new wxButton(this, -1, _("From &video"));
from_video->Enable(c->videoController->IsLoaded());
wxCheckBox *change_ar = new wxCheckBox(this, -1, _("&Change aspect ratio"));
change_ar->SetValidator(wxGenericValidator(&settings.change_ar));
// Position the controls
wxSizer *margin_sizer = new wxGridSizer(3, 3, 5, 5);
margin_sizer->AddSpacer(1);
margin_sizer->Add(margin_ctrl[TOP], wxSizerFlags(1).Expand());
margin_sizer->AddSpacer(1);
margin_sizer->Add(margin_ctrl[LEFT], wxSizerFlags(1).Expand());
margin_sizer->Add(symmetrical, wxSizerFlags(1).Expand());
margin_sizer->Add(margin_ctrl[RIGHT], wxSizerFlags(1).Expand());
margin_sizer->AddSpacer(1);
margin_sizer->Add(margin_ctrl[BOTTOM], wxSizerFlags(1).Expand());
margin_sizer->AddSpacer(1);
wxSizer *margin_box = new wxStaticBoxSizer(wxVERTICAL, this, _("Margin offset"));
margin_box->Add(margin_sizer, wxSizerFlags(1).Expand().Border(wxBOTTOM));
wxSizer *res_sizer = new wxBoxSizer(wxHORIZONTAL);
res_sizer->Add(res_x, wxSizerFlags(1).Border(wxRIGHT));
res_sizer->Add(new wxStaticText(this, -1, _("x")), wxSizerFlags().Center().Border(wxRIGHT));
res_sizer->Add(res_y, wxSizerFlags(1).Border(wxRIGHT));
res_sizer->Add(from_video, wxSizerFlags(1));
wxSizer *res_box = new wxStaticBoxSizer(wxVERTICAL, this, _("Resolution"));
res_box->Add(res_sizer, wxSizerFlags(1).Expand().Border(wxBOTTOM));
res_box->Add(change_ar);
wxSizer *main_sizer = new wxBoxSizer(wxVERTICAL);
main_sizer->Add(margin_box, wxSizerFlags(1).Expand().Border());
main_sizer->Add(res_box, wxSizerFlags(0).Expand().Border());
main_sizer->Add(CreateStdDialogButtonSizer(wxOK | wxCANCEL | wxHELP), wxSizerFlags().Expand().Border(wxALL & ~wxTOP));
SetSizerAndFit(main_sizer);
CenterOnParent();
// Bind events
using std::tr1::bind;
Bind(wxEVT_COMMAND_BUTTON_CLICKED, bind(&HelpButton::OpenPage, "Resample resolution"), wxID_HELP);
from_video->Bind(wxEVT_COMMAND_BUTTON_CLICKED, &DialogResample::SetDestFromVideo, this);
symmetrical->Bind(wxEVT_COMMAND_CHECKBOX_CLICKED, &DialogResample::OnSymmetrical, this);
margin_ctrl[LEFT]->Bind(wxEVT_COMMAND_SPINCTRL_UPDATED, bind(&DialogResample::OnMarginChange, this, margin_ctrl[LEFT], margin_ctrl[RIGHT]));
margin_ctrl[TOP]->Bind(wxEVT_COMMAND_SPINCTRL_UPDATED, bind(&DialogResample::OnMarginChange, this, margin_ctrl[TOP], margin_ctrl[BOTTOM]));
}
void DialogResample::SetDestFromVideo(wxCommandEvent &) {
res_x->SetValue(c->videoController->GetWidth());
res_y->SetValue(c->videoController->GetHeight());
2006-01-16 22:02:54 +01:00
}
void DialogResample::OnSymmetrical(wxCommandEvent &) {
bool state = !symmetrical->IsChecked();
2006-01-16 22:02:54 +01:00
margin_ctrl[RIGHT]->Enable(state);
margin_ctrl[BOTTOM]->Enable(state);
if (!state) {
margin_ctrl[RIGHT]->SetValue(margin_ctrl[LEFT]->GetValue());
margin_ctrl[BOTTOM]->SetValue(margin_ctrl[TOP]->GetValue());
}
}
void DialogResample::OnMarginChange(wxSpinCtrl *src, wxSpinCtrl *dst) {
if (symmetrical->IsChecked())
dst->SetValue(src->GetValue());
}
namespace {
struct resample_state {
const int *margin;
double rx;
double ry;
double ar;
};
void resample_tags(wxString name, int n, AssOverrideParameter *cur, void *ud) {
resample_state *state = static_cast<resample_state *>(ud);
double resizer = 1.0;
int shift = 0;
switch (cur->classification) {
case PARCLASS_ABSOLUTE_SIZE:
resizer = state->ry;
break;
case PARCLASS_ABSOLUTE_POS_X:
resizer = state->rx;
shift = state->margin[LEFT];
break;
case PARCLASS_ABSOLUTE_POS_Y:
resizer = state->ry;
shift = state->margin[TOP];
break;
case PARCLASS_RELATIVE_SIZE_X:
resizer = state->ar;
break;
case PARCLASS_RELATIVE_SIZE_Y:
//resizer = ry;
break;
case PARCLASS_DRAWING: {
AssDialogueBlockDrawing block(cur->Get<wxString>(), 1);
block.TransformCoords(state->margin[LEFT], state->margin[TOP], state->rx, state->ry);
cur->Set(block.GetText());
return;
2006-01-16 22:02:54 +01:00
}
default:
return;
2006-01-16 22:02:54 +01:00
}
VariableDataType curType = cur->GetType();
if (curType == VARDATA_FLOAT)
cur->Set((cur->Get<double>() + shift) * resizer);
else if (curType == VARDATA_INT)
cur->Set<int>((cur->Get<int>() + shift) * resizer + 0.5);
2006-01-16 22:02:54 +01:00
}
void resample_line(resample_state *state, AssEntry *line) {
AssDialogue *diag = dynamic_cast<AssDialogue*>(line);
if (diag && !(diag->Comment && (diag->Effect.StartsWith("template") || diag->Effect.StartsWith("code")))) {
diag->ParseASSTags();
diag->ProcessParameters(resample_tags, state);
2006-01-16 22:02:54 +01:00
for (size_t i = 0; i < diag->Blocks.size(); ++i) {
if (AssDialogueBlockDrawing *block = dynamic_cast<AssDialogueBlockDrawing*>(diag->Blocks[i]))
block->TransformCoords(state->margin[LEFT], state->margin[TOP], state->rx, state->ry);
}
2006-01-16 22:02:54 +01:00
for (size_t i = 0; i < 4; ++i)
diag->Margin[i] = int((diag->Margin[i] + state->margin[i]) * (i < 2 ? state->rx : state->ry) + 0.5);
2006-01-16 22:02:54 +01:00
diag->UpdateText();
diag->ClearBlocks();
}
else if (AssStyle *style = dynamic_cast<AssStyle*>(line)) {
style->fontsize = int(style->fontsize * state->ry + 0.5);
style->outline_w *= state->ry;
style->shadow_w *= state->ry;
style->spacing *= state->rx;
style->scalex *= state->ar;
for (int i = 0; i < 4; i++)
style->Margin[i] = int((style->Margin[i] + state->margin[i]) * (i < 2 ? state->rx : state->ry) + 0.5);
style->UpdateData();
}
}
2006-01-16 22:02:54 +01:00
}
void ResampleResolution(AssFile *ass, ResampleSettings const& settings) {
int src_x, src_y;
ass->GetResolution(src_x, src_y);
2006-01-16 22:02:54 +01:00
// Add margins to original resolution
src_x += settings.margin[LEFT] + settings.margin[RIGHT];
src_y += settings.margin[TOP] + settings.margin[BOTTOM];
resample_state state = {
settings.margin,
double(settings.script_x) / double(src_x),
double(settings.script_y) / double(src_y),
settings.change_ar ? state.rx / state.ry : 1.0
};
for_each(ass->Line.begin(), ass->Line.end(), bind(resample_line, &state, std::tr1::placeholders::_1));
ass->SetScriptInfo("PlayResX", wxString::Format("%d", settings.script_x));
ass->SetScriptInfo("PlayResY", wxString::Format("%d", settings.script_y));
ass->Commit(_("resolution resampling"), AssFile::COMMIT_SCRIPTINFO | AssFile::COMMIT_DIAG_FULL);
}