forked from mia/Aegisub
325 lines
11 KiB
C++
325 lines
11 KiB
C++
// Copyright (c) 2006, Niels Martin Hansen
|
|
// 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 auto4_base.h
|
|
/// @see auto4_base.cpp
|
|
/// @ingroup scripting
|
|
///
|
|
|
|
|
|
#pragma once
|
|
|
|
|
|
#ifndef AGI_PRE
|
|
#include <list>
|
|
#include <vector>
|
|
|
|
#include <wx/dialog.h>
|
|
#include <wx/filename.h>
|
|
#include <wx/gauge.h>
|
|
#include <wx/sizer.h>
|
|
#include <wx/stattext.h>
|
|
#include <wx/string.h>
|
|
#include <wx/textctrl.h>
|
|
#include <wx/timer.h>
|
|
#endif
|
|
|
|
#include <libaegisub/background_runner.h>
|
|
#include <libaegisub/exception.h>
|
|
#include <libaegisub/scoped_ptr.h>
|
|
#include <libaegisub/signal.h>
|
|
|
|
#include "ass_export_filter.h"
|
|
|
|
class AssFile;
|
|
class AssStyle;
|
|
class DialogProgress;
|
|
class SubtitleFormat;
|
|
class wxWindow;
|
|
class wxDialog;
|
|
class wxStopWatch;
|
|
class wxPathList;
|
|
|
|
namespace agi { struct Context; }
|
|
namespace cmd { class Command; }
|
|
|
|
|
|
/// DOCME
|
|
namespace Automation4 {
|
|
DEFINE_BASE_EXCEPTION_NOINNER(AutomationError, agi::Exception)
|
|
DEFINE_SIMPLE_EXCEPTION_NOINNER(ScriptLoadError, AutomationError, "automation/load/generic")
|
|
DEFINE_SIMPLE_EXCEPTION_NOINNER(MacroRunError, AutomationError, "automation/macro/generic")
|
|
|
|
// Calculate the extents of a text string given a style
|
|
bool CalculateTextExtents(AssStyle *style, wxString const& text, double &width, double &height, double &descent, double &extlead);
|
|
|
|
class ScriptDialog;
|
|
|
|
class ExportFilter : public AssExportFilter {
|
|
ScriptDialog *config_dialog;
|
|
|
|
/// subclasses should implement this, producing a new ScriptDialog
|
|
virtual ScriptDialog* GenerateConfigDialog(wxWindow *parent, agi::Context *c) = 0;
|
|
|
|
protected:
|
|
wxString GetScriptSettingsIdentifier();
|
|
|
|
public:
|
|
ExportFilter(wxString const& name, wxString const& description, int priority);
|
|
virtual ~ExportFilter();
|
|
|
|
wxWindow* GetConfigDialogWindow(wxWindow *parent, agi::Context *c);
|
|
void LoadSettings(bool is_default, agi::Context *c);
|
|
|
|
// Subclasses must implement ProcessSubs from AssExportFilter
|
|
};
|
|
|
|
/// A "dialog" which actually generates a non-top-level window that is then
|
|
/// either inserted into a dialog or into the export filter configuration
|
|
/// panel
|
|
class ScriptDialog {
|
|
public:
|
|
virtual ~ScriptDialog() { }
|
|
|
|
/// Create a window with the given parent
|
|
virtual wxWindow *CreateWindow(wxWindow *parent) = 0;
|
|
|
|
/// Read the values of the controls in this dialog and cache them in
|
|
/// some way so that they can be accessed from a background thread
|
|
virtual void ReadBack() = 0;
|
|
|
|
/// Serialize the values of the controls in this dialog to a string
|
|
/// suitable for storage in the subtitle script
|
|
virtual wxString Serialise() { return ""; }
|
|
|
|
/// Restore the values of the controls in this dialog from a string
|
|
/// stored in the subtitle script
|
|
virtual void Unserialise(wxString const& serialised) { }
|
|
};
|
|
|
|
class ProgressSink;
|
|
|
|
class BackgroundScriptRunner {
|
|
agi::scoped_ptr<DialogProgress> impl;
|
|
|
|
void OnDialog(wxThreadEvent &evt);
|
|
void OnScriptDialog(wxThreadEvent &evt);
|
|
public:
|
|
void QueueEvent(wxEvent *evt);
|
|
wxWindow *GetParentWindow() const;
|
|
|
|
void Run(std::tr1::function<void(ProgressSink*)> task);
|
|
|
|
BackgroundScriptRunner(wxWindow *parent, wxString const& title);
|
|
~BackgroundScriptRunner();
|
|
};
|
|
|
|
/// A wrapper around agi::ProgressSink which adds the ability to open
|
|
/// dialogs on the GUI thread
|
|
class ProgressSink : public agi::ProgressSink {
|
|
agi::ProgressSink *impl;
|
|
BackgroundScriptRunner *bsr;
|
|
int trace_level;
|
|
public:
|
|
void SetIndeterminate() { impl->SetIndeterminate(); }
|
|
void SetTitle(std::string const& title) { impl->SetTitle(title); }
|
|
void SetMessage(std::string const& msg) { impl->SetMessage(msg); }
|
|
void SetProgress(int cur, int max) { impl->SetProgress(cur, max); }
|
|
void Log(std::string const& str) { impl->Log(str); }
|
|
bool IsCancelled() { return impl->IsCancelled(); }
|
|
|
|
/// Show the passed dialog on the GUI thread, blocking the calling
|
|
/// thread until it closes
|
|
void ShowDialog(ScriptDialog *config_dialog);
|
|
int ShowDialog(wxDialog *dialog);
|
|
wxWindow *GetParentWindow() const { return bsr->GetParentWindow(); }
|
|
|
|
/// Get the current automation trace level
|
|
int GetTraceLevel() const { return trace_level; }
|
|
|
|
ProgressSink(agi::ProgressSink *impl, BackgroundScriptRunner *bsr);
|
|
};
|
|
|
|
class Script {
|
|
wxString filename;
|
|
|
|
protected:
|
|
/// The automation include path, consisting of the user-specified paths
|
|
/// along with the script's path
|
|
wxPathList include_path;
|
|
Script(wxString const& filename);
|
|
|
|
public:
|
|
virtual ~Script() { }
|
|
|
|
/// Reload this script
|
|
virtual void Reload() = 0;
|
|
|
|
/// The script's file name with path
|
|
wxString GetFilename() const { return filename; }
|
|
/// The script's file name without path
|
|
wxString GetPrettyFilename() const { return wxFileName(filename).GetFullName(); }
|
|
/// The script's name. Not required to be unique.
|
|
virtual wxString GetName() const=0;
|
|
/// A short description of the script
|
|
virtual wxString GetDescription() const=0;
|
|
/// The author of the script
|
|
virtual wxString GetAuthor() const=0;
|
|
/// A version string that should not be used for anything but display
|
|
virtual wxString GetVersion() const=0;
|
|
/// Did the script load correctly?
|
|
virtual bool GetLoadedState() const=0;
|
|
|
|
/// Get a list of commands provided by this script
|
|
virtual std::vector<cmd::Command*> GetMacros() const=0;
|
|
/// Get a list of export filters provided by this script
|
|
virtual std::vector<ExportFilter*> GetFilters() const=0;
|
|
/// Get a list of subtitle formats provided by this script
|
|
virtual std::vector<SubtitleFormat*> GetFormats() const=0;
|
|
};
|
|
|
|
/// A manager of loaded automation scripts
|
|
class ScriptManager {
|
|
protected:
|
|
std::vector<Script*> scripts;
|
|
std::vector<cmd::Command*> macros;
|
|
|
|
agi::signal::Signal<> ScriptsChanged;
|
|
|
|
public:
|
|
/// Deletes all scripts managed
|
|
virtual ~ScriptManager();
|
|
/// Add a script to the manager. The ScriptManager takes ownership of the script and will automatically delete it.
|
|
void Add(Script *script);
|
|
/// Remove a script from the manager, and delete the Script object.
|
|
void Remove(Script *script);
|
|
/// Deletes all scripts managed
|
|
void RemoveAll();
|
|
/// Reload all scripts mananaged
|
|
virtual void Reload() = 0;
|
|
/// Reload a single managed script
|
|
virtual void Reload(Script *script);
|
|
|
|
/// Get all managed scripts (both loaded and invalid)
|
|
const std::vector<Script*>& GetScripts() const { return scripts; }
|
|
|
|
const std::vector<cmd::Command*>& GetMacros();
|
|
// No need to have getters for the other kinds of features, I think.
|
|
// They automatically register themselves in the relevant places.
|
|
|
|
DEFINE_SIGNAL_ADDERS(ScriptsChanged, AddScriptChangeListener)
|
|
};
|
|
|
|
/// Manager for scripts specified by a subtitle file
|
|
class LocalScriptManager : public ScriptManager {
|
|
std::list<agi::signal::Connection> slots;
|
|
agi::Context *context;
|
|
|
|
void OnSubtitlesSave();
|
|
public:
|
|
LocalScriptManager(agi::Context *context);
|
|
void Reload();
|
|
};
|
|
|
|
/// Manager for scripts in the autoload directory
|
|
class AutoloadScriptManager : public ScriptManager {
|
|
wxString path;
|
|
public:
|
|
AutoloadScriptManager(wxString const& path);
|
|
void Reload();
|
|
};
|
|
|
|
/// Both a base class for script factories and a manager of registered
|
|
/// script factories
|
|
class ScriptFactory {
|
|
/// Vector of loaded script engines
|
|
static std::vector<ScriptFactory*> *factories;
|
|
|
|
wxString engine_name;
|
|
wxString filename_pattern;
|
|
|
|
/// Load a file, or return NULL if the file is not in a supported
|
|
/// format. If the file is in a supported format but is invalid, a
|
|
/// script should be returned which returns false from IsLoaded and
|
|
/// an appropriate error message from GetDescription.
|
|
///
|
|
/// This is private as it should only ever be called through
|
|
/// CreateFromFile
|
|
virtual Script* Produce(wxString const& filename) const = 0;
|
|
|
|
protected:
|
|
ScriptFactory(wxString engine_name, wxString filename_pattern);
|
|
virtual ~ScriptFactory() { }
|
|
|
|
public:
|
|
/// Name of this automation engine
|
|
const wxString& GetEngineName() const { return engine_name; }
|
|
/// Extension which this engine supports
|
|
const wxString& GetFilenamePattern() const { return filename_pattern; }
|
|
|
|
/// Register an automation engine. Calling code retains ownership of pointer
|
|
static void Register(ScriptFactory *factory);
|
|
/// Unregister and delete an automation engine
|
|
static void Unregister(ScriptFactory *factory);
|
|
/// Is there an automation engine registered which can open the file?
|
|
static bool CanHandleScriptFormat(wxString const& filename);
|
|
|
|
/// Get the full wildcard string for all loaded engines
|
|
static wxString GetWildcardStr();
|
|
|
|
/// Load a script from a file
|
|
/// @param filename Script to load
|
|
/// @param log_errors Should load errors be displayed?
|
|
/// @return Always returns a valid Script, even if no engine could load the file
|
|
static Script* CreateFromFile(wxString const& filename, bool log_errors);
|
|
|
|
static const std::vector<ScriptFactory*>& GetFactories();
|
|
};
|
|
|
|
/// A script which represents a file not recognized by any registered
|
|
/// automation engines
|
|
class UnknownScript : public Script {
|
|
public:
|
|
UnknownScript(wxString const& filename);
|
|
|
|
void Reload() { }
|
|
|
|
wxString GetName() const { return wxFileName(GetFilename()).GetName(); }
|
|
wxString GetDescription() const { return _("File was not recognized as a script"); }
|
|
wxString GetAuthor() const { return ""; }
|
|
wxString GetVersion() const { return ""; }
|
|
bool GetLoadedState() const { return false; }
|
|
|
|
std::vector<cmd::Command*> GetMacros() const { return std::vector<cmd::Command*>(); }
|
|
std::vector<ExportFilter*> GetFilters() const { return std::vector<ExportFilter*>(); }
|
|
std::vector<SubtitleFormat*> GetFormats() const { return std::vector<SubtitleFormat*>(); }
|
|
};
|
|
}
|