mia/Aegisub
1
0
Fork 1
Code Issues Pull requests Projects Releases Wiki Activity
Aegisub/core/automation.cpp
Niels Martin Hansen e75de69783 Patches by Azzy: 
- Search for both lua and lua50 libraries (fixes building on Debian/Ubuntu)
- Fix GCC preprocessor failing on version.cpp
Also forgot changelog for last commit

Originally committed to SVN as r541.
2006-10-25 10:33:02 +00:00

1665 lines
49 KiB
C++
Raw Blame History

// Copyright (c) 2005, 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
//
// Website: http://aegisub.cellosoft.com
// Contact: mailto:zeratul@cellosoft.com
//
#include <wx/file.h>
#include <wx/gdicmn.h>
#include <wx/dcmemory.h>
#include <wx/filename.h>
#include <wx/tokenzr.h>
#include "automation.h"
#include "ass_file.h"
#include "ass_entry.h"
#include "ass_dialogue.h"
#include "ass_style.h"
#include "options.h"
#include "string_codec.h"
#include "vfr.h"
#ifdef __WINDOWS__
#include <windows.h>
#include <wchar.h>
#else
#include <ft2build.h>
#include FT_FREETYPE_H
#endif
extern "C" {
#ifdef HAVE_LUA50_LUA_H
#include <lua50/lualib.h>
#include <lua50/lauxlib.h>
#else
#include <lualib.h>
#include <lauxlib.h>
#endif
}
int L_callfunc(lua_State *L, int nargs, int nresults);
void L_settable(lua_State *L, int table, wxString &key, lua_Number val);
void L_settable(lua_State *L, int table, wxString &key, wxString val);
void L_settable_bool(lua_State *L, int table, wxString &key, bool val);
void L_settable(lua_State *L, int table, const char *key, lua_Number val);
void L_settable(lua_State *L, int table, const char *key, wxString val);
void L_settable_bool(lua_State *L, int table, const char *key, bool val);
void L_settable_kara(lua_State *L, int table, int index, int duration, wxString &kind, wxString &text, wxString &text_stripped);
// these two assume the table to get from is on the top of the stack
lua_Number L_gettableN(lua_State *L, const char *key);
wxString L_gettableS(lua_State *L, const char *key);
bool L_gettableB(lua_State *L, const char *key);
namespace AutomationHelper {
// helper functions for the scripts
// expect a pointer to the automation script object to be on the private stack
/*
Helper function helper...
Get the AutomationScript object associated with a Lua state.
*/
AutomationScript *GetScriptObject(lua_State *L)
{
lua_pushstring(L, "aegisub");
lua_rawget(L, LUA_REGISTRYINDEX);
AutomationScript *s = (AutomationScript*)(lua_touserdata(L, -1));
if (!s) {
lua_pushstring(L, "Unable to retrieve AutomationScript object from the registry. This should never happen!");
lua_error(L); // never returns
}
lua_pop(L, 1);
return s;
}
/*
"Debug hook" function, used for checking if the interpreter has been asked to cancel.
If it has, a Lua error is reported.
*/
void hookfunc(lua_State *L, lua_Debug *ar)
{
AutomationScript *script = GetScriptObject(L);
if (script->force_cancel) {
if (ar->currentline < 0) {
lua_pushstring(L, "Script forcibly terminated at an unknown line");
} else {
lua_pushstring(L, "Script forcibly terminated at line ");
lua_pushnumber(L, ar->currentline);
lua_concat(L, 2);
}
lua_error(L);
}
}
/*
function aegisub.output_debug(text)
Output text to a debug console.
@text
String. The text to output.
Returns: nothing.
*/
int output_debug(lua_State *L)
{
AutomationScript *script = GetScriptObject(L);
// check we were passed a string
if (lua_gettop(L) < 1) {
// idiot user (nothing on the stack)
lua_pushstring(L, "output_debug called with no arguments");
lua_error(L); // never returns
} else if (!lua_isstring(L, -1)) {
// idiot user (didn't pass a string)
lua_pushstring(L, "output_debug called with non string-compatible argument");
lua_error(L); // never returns
}
script->OutputDebugString(wxString(lua_tostring(L, 1), wxConvUTF8), true);
return 0;
}
/*
function aegisub.set_status(text)
Sets the current status-message. (Used for progress-reporting.)
@text
String. The status message.
Returns: nothing.
*/
int set_status(lua_State *L)
{
AutomationScript *script = GetScriptObject(L);
// check we were passed a string
if (lua_gettop(L) < 1) {
// idiot user (nothing on the stack)
lua_pushstring(L, "output_debug called with no arguments");
lua_error(L); // never returns
} else if (!lua_isstring(L, -1)) {
// idiot user (didn't pass a string)
lua_pushstring(L, "output_debug called with non string-compatible argument");
lua_error(L); // never returns
}
script->OutputDebugString(wxString(lua_tostring(L, 1), wxConvUTF8), false);
return 0;
}
/*
function aegisub.colorstring_to_rgb(colorstring)
Convert an ASS color-string to a set of RGB values.
@colorstring
String. The color-string to convert.
Returns: Four values, all numbers, being the color components in this
order: Red, Green, Blue, Alpha-channel
*/
int colorstring_to_rgb(lua_State *L)
{
if (lua_gettop(L) < 1) {
lua_pushstring(L, "colorstring_to_rgb called without arguments");
lua_error(L);
}
if (!lua_isstring(L, 1)) {
lua_pushstring(L, "colorstring_to_rgb requires a string type argument");
lua_error(L);
}
wxString colorstring(lua_tostring(L, -1), wxConvUTF8);
lua_pop(L, 1);
AssColor rgb;
rgb.ParseASS(colorstring);
lua_pushnumber(L, rgb.r);
lua_pushnumber(L, rgb.g);
lua_pushnumber(L, rgb.b);
lua_pushnumber(L, rgb.a);
return 4;
}
/*
function aegisub.report_progress(percent)
Report the progress of the processing.
@percent
Number. How much of the data have been processed so far.
Returns: nothing.
*/
int report_progress(lua_State *L)
{
AutomationScript *script = GetScriptObject(L);
// check we were passed a string
if (lua_gettop(L) < 1) {
// idiot user (nothing on the stack)
lua_pushstring(L, "report_progress called with no arguments");
lua_error(L); // never returns
} else if (!lua_isnumber(L, -1)) {
// idiot user (didn't pass a string)
lua_pushstring(L, "report_progress requires a numeric argument");
lua_error(L); // never returns
}
lua_Number p = lua_tonumber(L, -1);
if (p < 0) p = 0;
if (p > 100) p = 100;
p = (p+100)/3;
script->ReportProgress(p);
return 0;
}
/*
function aegisub.text_extents(style, text)
Calculate the on-screen pixel size of the given text using the given style.
@style
Table. A single style definition like those passed to process_lines.
@text
String. The text to calculate the extents for. This should not contain
formatting codes, as they will be treated as part of the text.
Returns 4 values:
1: Number. Width of the text, in pixels.
2: Number. Height of the text, in pixels.
3: Number. Descent of the text, in pixels.
4: Number. External leading for the text, in pixels.
*/
int text_extents(lua_State *L)
{
// vars for the result
int resx = 0, resy = 0, resd = 0, resl = 0;
// get the input
// no error checking for the moment
wxString intext(lua_tostring(L, -1), wxConvUTF8);
// leave only style table
lua_settop(L, -2);
// read out the relevant parts of style
wxString fontname(L_gettableS(L, "fontname"));
double fontsize = L_gettableN(L, "fontsize");
bool bold = L_gettableB(L, "bold");
bool italic = L_gettableB(L, "italic");
bool underline = L_gettableB(L, "underline");
bool strikeout = L_gettableB(L, "strikeout");
double scale_x = L_gettableN(L, "scale_x");
double scale_y = L_gettableN(L, "scale_y");
int spacing = (int)L_gettableN(L, "spacing");
int charset = (int)L_gettableN(L, "encoding");
wxLogDebug(_T("text_extents for: %s:%f:%d%d%d%d:%f:%f:%d:%d"), fontname.c_str(), fontsize, bold, italic, underline, strikeout, scale_x, scale_y, spacing, charset);
#ifdef WIN32
HDC thedc = CreateCompatibleDC(0);
if (!thedc) return 0;
SetMapMode(thedc, MM_TEXT);
HDC dczero = GetDC(0);
fontsize = -MulDiv((int)(fontsize+0.5), GetDeviceCaps(dczero, LOGPIXELSY), 72);
ReleaseDC(0, dczero);
LOGFONT lf;
ZeroMemory(&lf, sizeof(lf));
lf.lfHeight = fontsize;
lf.lfWeight = bold ? FW_BOLD : FW_NORMAL;
lf.lfItalic = italic;
lf.lfUnderline = underline;
lf.lfStrikeOut = strikeout;
lf.lfCharSet = charset;
lf.lfOutPrecision = OUT_TT_PRECIS;
lf.lfClipPrecision = CLIP_DEFAULT_PRECIS;
lf.lfQuality = ANTIALIASED_QUALITY;
lf.lfPitchAndFamily = DEFAULT_PITCH|FF_DONTCARE;
wcsncpy(lf.lfFaceName, fontname.wc_str(), 32);
HFONT thefont = CreateFontIndirect(&lf);
if (!thefont) return 0;
SelectObject(thedc, thefont);
SIZE sz;
size_t thetextlen = intext.length();
const wchar_t *thetext = intext.wc_str();
if (spacing) {
resx = 0;
for (unsigned int i = 0; i < thetextlen; i++) {
GetTextExtentPoint32(thedc, &thetext[i], 1, &sz);
resx += sz.cx + spacing;
resy = sz.cy;
}
} else {
GetTextExtentPoint32(thedc, thetext, thetextlen, &sz);
resx = sz.cx;
resy = sz.cy;
}
// HACKISH FIX! This seems to work, but why? It shouldn't be needed?!?
fontsize = L_gettableN(L, "fontsize");
resx = (int)(resx * fontsize/resy + 0.5);
resy = (int)(fontsize + 0.5);
TEXTMETRIC tm;
GetTextMetrics(thedc, &tm);
resd = tm.tmDescent;
resl = tm.tmExternalLeading;
DeleteObject(thedc);
DeleteObject(thefont);
#else // not WIN32
wxMemoryDC thedc;
// fix fontsize to be 72 DPI
fontsize = -FT_MulDiv((int)(fontsize+0.5), 72, thedc.GetPPI().y);
// now try to get a font!
// use the font list to get some caching... (chance is the script will need the same font very often)
// USING wxTheFontList SEEMS TO CAUSE BAD LEAKS!
//wxFont *thefont = wxTheFontList->FindOrCreateFont(
wxFont thefont(
fontsize,
wxFONTFAMILY_DEFAULT,
italic ? wxFONTSTYLE_ITALIC : wxFONTSTYLE_NORMAL,
bold ? wxFONTWEIGHT_BOLD : wxFONTWEIGHT_NORMAL,
underline,
fontname,
wxFONTENCODING_SYSTEM);
thedc.SetFont(thefont);
if (spacing) {
// If there's inter-character spacing, kerning info must not be used, so calculate width per character
for (unsigned int i = 0; i < intext.length(); i++) {
int a, b, c, d;
thedc.GetTextExtent(intext[i], &a, &b, &c, &d);
resx += a + spacing;
resy = b > resy ? b : resy;
resd = c > resd ? c : resd;
resl = d > resl ? d : resl;
}
} else {
// If the inter-character spacing should be zero, kerning info can (and must) be used, so calculate everything in one go
thedc.GetTextExtent(intext, &resx, &resy, &resd, &resl);
}
#endif // WIN32
// Compensate for scaling
resx = (int)(scale_x / 100 * resx + 0.5);
resy = (int)(scale_y / 100 * resy + 0.5);
resd = (int)(scale_y / 100 * resd + 0.5);
resl = (int)(scale_y / 100 * resl + 0.5);
lua_pushnumber(L, resx);
lua_pushnumber(L, resy);
lua_pushnumber(L, resd);
lua_pushnumber(L, resl);
return 4;
}
/*
function aegisub.frame_from_ms(ms)
Return the video frame-number for the given time.
@ms
Number. Time in miliseconds to get the frame number for.
Returns: A number, the frame numer. If there is no framerate data, returns
nil.
*/
int frame_from_ms(lua_State *L)
{
int ms = (int)lua_tonumber(L, -1);
lua_pop(L, 1);
if (VFR_Output.IsLoaded()) {
lua_pushnumber(L, VFR_Output.GetFrameAtTime(ms, true));
return 1;
} else {
lua_pushnil(L);
return 1;
}
}
/*
function aegisub.ms_from_frame(frame)
Returns the start-time for the given video frame-number.
@frame
Number. Frame-number to get start-time from.
Returns: A number, the start-time of the frame. If there is no framerate
data, returns nil.
*/
int ms_from_frame(lua_State *L)
{
int frame = (int)lua_tonumber(L, -1);
lua_pop(L, 1);
if (VFR_Output.IsLoaded()) {
lua_pushnumber(L, VFR_Output.GetTimeAtFrame(frame, true));
return 1;
} else {
lua_pushnil(L);
return 1;
}
}
/*
function include(filename)
@filename
String. Name of the file to include.
Returns: Depends on the script included.
*/
int include(lua_State *L)
{
AutomationScript *script = GetScriptObject(L);
if (!lua_isstring(L, 1)) {
lua_pushstring(L, "First argument to the include function must be a string.");
lua_error(L);
}
wxString fnames(lua_tostring(L, 1), wxConvUTF8);
wxFileName fname(fnames);
if (fname.GetDirCount() == 0) {
// filename only
fname = script->include_path.FindAbsoluteValidPath(fnames);
} else if (fname.IsRelative()) {
// relative path
wxFileName sfname(script->filename);
fname.MakeAbsolute(sfname.GetPath(true));
} else {
// absolute path, invalid
lua_pushstring(L, "Filename passed to include seems to have an absolute path, which is not allowed.");
lua_error(L);
}
if (!fname.IsOk() || !fname.FileExists()) {
{
// need to make a new scope here, so the char buffer can go out of scope before lua_error() makes a longjmp
wxCharBuffer errmsg = wxString::Format(_T("The file could not be included, not found. \"%s\""), fnames.c_str()).mb_str(wxConvUTF8);
lua_pushstring(L, errmsg.data());
}
lua_error(L);
}
AutomationScriptFile *sfile;
sfile = AutomationScriptFile::CreateFromFile(fname.GetFullPath());
wxCharBuffer fnamebuf = fname.GetFullName().mb_str(wxConvUTF8);
switch (luaL_loadbuffer(L, sfile->scriptdata, sfile->scriptlen, fnamebuf.data())) {
// FIXME: these should be made into lua_error() things instead... probably
case 0:
// success!
break;
case LUA_ERRSYNTAX:
throw AutomationError(wxString::Format(_T("Lua syntax error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
case LUA_ERRMEM:
throw AutomationError(wxString::Format(_T("Lua memory allocation error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
default:
throw AutomationError(wxString::Format(_T("Lua unknown error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
}
delete sfile;
// top of stack before the call (correct for the function itself being on stack)
int pretop = lua_gettop(L)-1;
// call the loaded script
lua_call(L, 0, LUA_MULTRET);
// calculate the number of results the script produced
return lua_gettop(L)-pretop;
}
}
/*
Call a Lua function without risking killing the entire program
Just throw a C++ exception instead :)
Really just a thin wrapper around lua_pcall
*/
inline int L_callfunc(lua_State *L, int nargs, int nresults)
{
int res = lua_pcall(L, nargs, nresults, 0);
switch (res) {
case LUA_ERRRUN:
throw AutomationError(wxString::Format(_T("Lua runtime error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
case LUA_ERRMEM:
throw AutomationError(wxString::Format(_T("Lua memory allocation error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
case LUA_ERRERR:
// shouldn't happen as an error handling function isn't being used
throw AutomationError(wxString::Format(_T("Lua error calling error handler: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
default:
// success!
return res;
}
}
inline void L_settable(lua_State *L, int table, wxString &key, lua_Number val)
{
L_settable(L, table, key.mb_str(wxConvUTF8), val);
}
inline void L_settable(lua_State *L, int table, wxString &key, wxString val)
{
//wxLogMessage(_T("Wrapping adding of string at index '%s': %s"), key, val);
L_settable(L, table, key.mb_str(wxConvUTF8), val);
}
inline void L_settable_bool(lua_State *L, int table, wxString &key, bool val)
{
L_settable_bool(L, table, key.mb_str(wxConvUTF8), val);
}
inline void L_settable(lua_State *L, int table, const char *key, lua_Number val)
{
lua_pushstring(L, key);
lua_pushnumber(L, val);
if (table > 0 || table < -100) {
lua_settable(L, table);
} else {
lua_settable(L, table-2);
}
}
inline void L_settable(lua_State *L, int table, const char *key, wxString val)
{
//wxLogMessage(_T("Adding string at index '%s': %s"), wxString(key, wxConvUTF8), val);
lua_pushstring(L, key);
lua_pushstring(L, val.mb_str(wxConvUTF8));
if (table > 0 || table < -100) {
lua_settable(L, table);
} else {
lua_settable(L, table-2);
}
}
inline void L_settable_bool(lua_State *L, int table, const char *key, bool val)
{
lua_pushstring(L, key);
lua_pushboolean(L, val?1:0);
if (table > 0 || table < -100) {
lua_settable(L, table);
} else {
lua_settable(L, table-2);
}
}
inline void L_settable_kara(lua_State *L, int table, int index, int duration, wxString &kind, wxString &text, wxString &text_stripped)
{
lua_newtable(L);
L_settable(L, -1, "duration", duration);
L_settable(L, -1, "kind", kind);
L_settable(L, -1, "text", text);
L_settable(L, -1, "text_stripped", text_stripped);
if (table > 0 || table < -100) {
lua_rawseti(L, table, index);
} else {
lua_rawseti(L, table-1, index);
}
}
lua_Number L_gettableN(lua_State *L, const char *key)
{
lua_pushstring(L, key);
lua_gettable(L, -2);
lua_Number res = lua_tonumber(L, -1);
lua_settop(L, -2);
return res;
}
wxString L_gettableS(lua_State *L, const char *key)
{
lua_pushstring(L, key);
lua_gettable(L, -2);
wxString res(lua_tostring(L, -1), wxConvUTF8);
lua_settop(L, -2);
return res;
}
bool L_gettableB(lua_State *L, const char *key)
{
lua_pushstring(L, key);
lua_gettable(L, -2);
bool res = lua_toboolean(L, -1) != 0;
lua_settop(L, -2);
return res;
}
AutomationError::AutomationError(wxString msg)
: message(msg)
{
// nothing to do here...
}
wxString AutomationScriptConfiguration::serialize()
{
wxString result;
for (std::vector<AutomationScriptConfigurationOption>::iterator opt = options.begin(); opt != options.end(); opt++) {
switch (opt->kind) {
case COK_TEXT:
case COK_STYLE:
result << wxString::Format(_T("%s:%s|"), opt->name.c_str(), inline_string_encode(opt->value.stringval).c_str());
break;
case COK_INT:
result << wxString::Format(_T("%s:%d|"), opt->name.c_str(), opt->value.intval);
break;
case COK_FLOAT:
result << wxString::Format(_T("%s:%e|"), opt->name.c_str(), opt->value.floatval);
break;
case COK_BOOL:
result << wxString::Format(_T("%s:%d|"), opt->name.c_str(), opt->value.boolval?1:0);
break;
case COK_COLOUR:
result << wxString::Format(_T("%s:%s|"), opt->name.c_str(), opt->value.colourval.GetASSFormatted(false).c_str());
break;
default:
// The rest aren't stored
break;
}
}
if (result.Last() == _T('|'))
result.RemoveLast();
return result;
}
void AutomationScriptConfiguration::unserialize(wxString &settings)
{
//wxLogMessage(_T("Unserializing config string: %s"), settings);
wxStringTokenizer toker(settings, _T("|"), wxTOKEN_STRTOK);
while (toker.HasMoreTokens()) {
// get the parts of this setting
wxString setting = toker.GetNextToken();
//wxLogMessage(_T("Got token: %s"), setting);
wxString optname = setting.BeforeFirst(_T(':'));
wxString optval = setting.AfterFirst(_T(':'));
//wxLogMessage(_T("Split into: \"%s\" and \"%s\""), optname, optval);
// find the setting in the list loaded from the script
std::vector<AutomationScriptConfigurationOption>::iterator opt = options.begin();
while (opt != options.end() && opt->name != optname)
opt ++;
if (opt != options.end()) {
//wxLogMessage(_T("Found the option!"));
// ok, found the option!
switch (opt->kind) {
case COK_TEXT:
case COK_STYLE:
opt->value.stringval = inline_string_decode(optval);
//wxLogMessage(_T("Decoded string to: %s"), opt->value.stringval);
break;
case COK_INT:
{
long n;
optval.ToLong(&n, 10);
opt->value.intval = n;
}
break;
case COK_FLOAT:
optval.ToDouble(&opt->value.floatval);
break;
case COK_BOOL:
opt->value.boolval = optval == _T("1");
break;
case COK_COLOUR:
opt->value.colourval.ParseASS(optval);
break;
}
}
}
}
void AutomationScriptConfiguration::load_from_lua(lua_State *L)
{
//wxLogMessage(_T("Loading configuration options from script"));
present = false;
if (!lua_istable(L, -1)) {
return;
}
//wxLogMessage(_T("The script does have config options, good!"));
int i = 1;
while (true) {
// get an element from the array
//wxLogMessage(_T("Getting option %d (stacktop is %d)"), i, lua_gettop(L));
lua_pushnumber(L, i);
lua_gettable(L, -2);
// check if it was a table
if (!lua_istable(L, -1)) {
//wxLogMessage(_T("Damn! Not an option... breaking out (actual type was %d)"), lua_type(L, -1));
lua_pop(L, 1);
break;
}
//wxLogMessage(_T("Yay! It was an option, adding another blank option thing to the list"));
// add a new config option and fill it
{
AutomationScriptConfigurationOption opt;
options.push_back(opt);
}
AutomationScriptConfigurationOption &opt = options.back();
// get the "kind"
lua_pushstring(L, "kind");
lua_gettable(L, -2);
if (lua_isstring(L, -1)) {
// use C standard lib functions here, as it's probably faster than messing around with unicode
// lua is known to always properly null-terminate strings, and the strings are known to be pure ascii
const char *kind = lua_tostring(L, -1);
if (strcmp(kind, "label") == 0) {
opt.kind = COK_LABEL;
} else if (strcmp(kind, "text") == 0) {
opt.kind = COK_TEXT;
} else if (strcmp(kind, "int") == 0) {
opt.kind = COK_INT;
} else if (strcmp(kind, "float") == 0) {
opt.kind = COK_FLOAT;
} else if (strcmp(kind, "bool") == 0) {
opt.kind = COK_BOOL;
} else if (strcmp(kind, "colour") == 0) {
opt.kind = COK_COLOUR;
} else if (strcmp(kind, "style") == 0) {
opt.kind = COK_STYLE;
} else {
opt.kind = COK_INVALID;
}
} else {
opt.kind = COK_INVALID;
}
//wxLogMessage(_T("Got kind: %d"), opt.kind);
// remove "kind" string from stack again
lua_pop(L, 1);
// no need to check for rest if this one is already deemed invalid
if (opt.kind != COK_INVALID) {
// name
lua_pushstring(L, "name");
lua_gettable(L, -2);
if (lua_isstring(L, -1)) {
opt.name = wxString(lua_tostring(L, -1), wxConvUTF8);
lua_pop(L, 1);
} else {
lua_pop(L, 1);
// no name means invalid option
opt.kind = COK_INVALID;
goto continue_invalid_option;
}
//wxLogMessage(_T("Got name: %s"), opt.name);
// label
lua_pushstring(L, "label");
lua_gettable(L, -2);
if (lua_isstring(L, -1)) {
opt.label = wxString(lua_tostring(L, -1), wxConvUTF8);
lua_pop(L, 1);
} else {
lua_pop(L, 1);
// label is also required
opt.kind = COK_INVALID;
goto continue_invalid_option;
}
assert(opt.kind != COK_INVALID);
// hint
lua_pushstring(L, "hint");
lua_gettable(L, -2);
if (lua_isstring(L, -1)) {
opt.hint = wxString(lua_tostring(L, -1), wxConvUTF8);
} else {
opt.hint = _T("");
}
lua_pop(L, 1);
// min
lua_pushstring(L, "min");
lua_gettable(L, -2);
if (lua_isnumber(L, -1)) {
opt.min.isset = true;
opt.min.floatval = lua_tonumber(L, -1);
opt.min.intval = (int)opt.min.floatval;
} else {
opt.min.isset = false;
}
lua_pop(L, 1);
// max
lua_pushstring(L, "max");
lua_gettable(L, -2);
if (lua_isnumber(L, -1)) {
opt.max.isset = true;
opt.max.floatval = lua_tonumber(L, -1);
opt.max.intval = (int)opt.max.floatval;
} else {
opt.max.isset = false;
}
lua_pop(L, 1);
// default (this is going to kill me)
lua_pushstring(L, "default");
lua_gettable(L, -2);
switch (opt.kind) {
case COK_LABEL:
// nothing to do, nothing expected
break;
case COK_TEXT:
case COK_STYLE:
// expect it to be a string
if (lua_isstring(L, -1)) {
opt.default_val.stringval = wxString(lua_tostring(L, -1), wxConvUTF8);
} else {
// not a string, baaaad scripter
opt.kind = COK_INVALID;
}
break;
case COK_INT:
case COK_FLOAT:
// expect it to be a number
if (lua_isnumber(L, -1)) {
opt.default_val.floatval = lua_tonumber(L, -1);
opt.default_val.intval = (int)opt.default_val.floatval;
} else {
opt.kind = COK_INVALID;
}
break;
case COK_BOOL:
// expect it to be a bool
if (lua_isboolean(L, -1)) {
opt.default_val.boolval = lua_toboolean(L, -1)!=0;
} else {
opt.kind = COK_INVALID;
}
break;
case COK_COLOUR:
// expect it to be a ass hex colour formatted string
if (lua_isstring(L, -1)) {
opt.default_val.stringval = wxString(lua_tostring(L, -1), wxConvUTF8);
opt.default_val.colourval.ParseASS(opt.default_val.stringval); // and hope this goes well!
} else {
opt.kind = COK_INVALID;
}
break;
}
opt.value = opt.default_val;
lua_pop(L, 1);
}
// so we successfully got an option added, so at least there is a configuration present now
present = true;
continue_invalid_option:
// clean up and prepare for next iteration
lua_pop(L, 1);
i++;
}
}
void AutomationScriptConfiguration::store_to_lua(lua_State *L)
{
// we'll always need a new table, no matter what
lua_newtable(L);
//wxLogMessage(_T("Created table for configuration data (top=%d)"), lua_gettop(L));
for (std::vector<AutomationScriptConfigurationOption>::iterator opt = options.begin(); opt != options.end(); opt++) {
//wxLogMessage(_T("Storing option named '%s' (top=%d)"), opt->name, lua_gettop(L));
switch (opt->kind) {
case COK_INVALID:
case COK_LABEL:
//wxLogMessage(_T("Nothing to store"));
break;
case COK_TEXT:
case COK_STYLE:
//wxLogMessage(_T("Storing string value"));
L_settable(L, -1, opt->name, opt->value.stringval);
break;
case COK_INT:
//wxLogMessage(_T("Storing int value"));
L_settable(L, -1, opt->name, opt->value.intval);
break;
case COK_FLOAT:
//wxLogMessage(_T("Storing float value"));
L_settable(L, -1, opt->name, opt->value.floatval);
break;
case COK_BOOL:
//wxLogMessage(_T("Storing bool value"));
L_settable_bool(L, -1, opt->name, opt->value.boolval);
break;
case COK_COLOUR:
//wxLogMessage(_T("Storing colourvalue"));
L_settable(L, -1, opt->name, opt->value.colourval.GetASSFormatted(false, false));
break;
default:
//wxLogMessage(_T("Felt into default handler?!?"));
break;
}
}
//wxLogMessage(_T("Finished storing configuration data (top=%d)"), lua_gettop(L));
}
AutomationScript::AutomationScript(AutomationScriptFile *script)
{
force_cancel = false;
filename = script->filename;
progress_reporter = 0;
debug_reporter = 0;
// get a lua object
L = lua_open();
// put a pointer to this object in the registry index
lua_pushstring(L, "aegisub");
lua_pushlightuserdata(L, this);
lua_rawset(L, LUA_REGISTRYINDEX);
// set up the cancelling hook, call every 100 instructions
lua_sethook(L, AutomationHelper::hookfunc, LUA_MASKCOUNT, 100);
// provide some standard libraries
luaopen_base(L);
luaopen_string(L);
luaopen_table(L);
luaopen_math(L);
#ifdef _DEBUG
luaopen_debug(L);
#endif
// but no I/O, OS or debug facilities, those aren't safe
// disable the dofile() function
lua_pushstring(L, "dofile");
lua_pushnil(L);
lua_settable(L, LUA_GLOBALSINDEX);
// create an include function better suited aegisub later
// the path object is needed for this
include_path.EnsureFileAccessible(script->filename);
{
wxStringTokenizer toker(Options.AsText(_T("Automation Include Path")), _T("|"), false);
while (toker.HasMoreTokens()) {
wxFileName path(toker.GetNextToken());
if (!path.IsOk()) continue;
if (path.IsRelative()) continue;
if (!path.DirExists()) continue;
if (include_path.Member(path.GetLongPath())) continue;
include_path.Add(path.GetLongPath());
}
}
// add the include function to the global environment
lua_pushstring(L, "include");
lua_pushcfunction(L, AutomationHelper::include);
lua_settable(L, LUA_GLOBALSINDEX);
// create the "aegisub" table and fill it with some function pointers
// create the table and add it to the global environment
lua_pushstring(L, "aegisub");
lua_newtable(L);
lua_settable(L, LUA_GLOBALSINDEX);
// get it back onto the stack
lua_pushstring(L, "aegisub");
lua_gettable(L, LUA_GLOBALSINDEX);
// get the index of the table on the stack
int tabid = lua_gettop(L);
// now register some functions!
lua_pushstring(L, "output_debug");
lua_pushcfunction(L, AutomationHelper::output_debug);
lua_settable(L, tabid);
lua_pushstring(L, "set_status");
lua_pushcfunction(L, AutomationHelper::set_status);
lua_settable(L, tabid);
lua_pushstring(L, "report_progress");
lua_pushcfunction(L, AutomationHelper::report_progress);
lua_settable(L, tabid);
lua_pushstring(L, "colorstring_to_rgb");
lua_pushcfunction(L, AutomationHelper::colorstring_to_rgb);
lua_settable(L, tabid);
lua_pushstring(L, "text_extents");
lua_pushcfunction(L, AutomationHelper::text_extents);
lua_settable(L, tabid);
lua_pushstring(L, "frame_from_ms");
lua_pushcfunction(L, AutomationHelper::frame_from_ms);
lua_settable(L, tabid);
lua_pushstring(L, "ms_from_frame");
lua_pushcfunction(L, AutomationHelper::ms_from_frame);
lua_settable(L, tabid);
// and no more need for that tabid
lua_settop(L, tabid-1);
// ok, finally the environment is set up!
// now try to load the script
{
switch (luaL_loadbuffer(L, script->scriptdata, script->scriptlen, "user script")) {
case 0:
// success!
break;
case LUA_ERRSYNTAX:
throw AutomationError(wxString::Format(_T("Lua syntax error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
case LUA_ERRMEM:
throw AutomationError(wxString::Format(_T("Lua memory allocation error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
default:
throw AutomationError(wxString::Format(_T("Lua unknown error: %s"), wxString(lua_tostring(L, -1), wxConvUTF8).c_str()));
break;
}
}
// it's loaded and is now a function at the top of the stack
// it doesn't take any arguments and doesn't return anything
// so let's try executing it by calling!
L_callfunc(L, 0, 0);
// so, the script should be loaded
// now try to get the script data!
// first the version
lua_pushstring(L, "version");
lua_gettable(L, LUA_GLOBALSINDEX);
if (!lua_isnumber(L, -1)) {
throw AutomationError(wxString(_T("Script error: 'version' value not found or not a number")));
}
version = lua_tonumber(L, -1);
lua_settop(L, -2);
if (version < 3 || version > 4) {
// invalid version
throw AutomationError(wxString(_T("Script error: 'version' value invalid for this version of Automation")));
}
// kind
lua_pushstring(L, "kind");
lua_gettable(L, LUA_GLOBALSINDEX);
if (!lua_isstring(L, -1)) {
throw AutomationError(wxString(_T("Script error: 'kind' value not found or not a string")));
}
kind = wxString(lua_tostring(L, -1), wxConvUTF8);
lua_settop(L, -2);
// name
lua_pushstring(L, "name");
lua_gettable(L, LUA_GLOBALSINDEX);
if (!lua_isstring(L, -1)) {
throw AutomationError(wxString(_T("Script error: 'name' value not found or not a string")));
}
name = wxString(lua_tostring(L, -1), wxConvUTF8);
lua_settop(L, -2);
// description (optional)
lua_pushstring(L, "description");
lua_gettable(L, LUA_GLOBALSINDEX);
if (lua_isstring(L, -1)) {
description = wxString(lua_tostring(L, -1), wxConvUTF8);
lua_settop(L, -2);
} else {
description = _T("");
}
// process_lines (just check if it's there, no need to save it anywhere)
lua_pushstring(L, "process_lines");
lua_gettable(L, LUA_GLOBALSINDEX);
if (!lua_isfunction(L, -1)) {
throw AutomationError(wxString(_T("Script error: No 'process_lines' function provided")));
}
lua_settop(L, -2);
// configuration (let the config object do all the loading)
lua_pushstring(L, "configuration");
lua_gettable(L, LUA_GLOBALSINDEX);
//wxLogMessage(_T("Calling configuration.load_from_lua()"));
configuration.load_from_lua(L);
lua_settop(L, -2);
// done!
}
AutomationScript::~AutomationScript()
{
lua_close(L);
}
void AutomationScript::OutputDebugString(wxString str, bool isdebug)
{
//wxLogMessage(_T("automation message: ") + str);
if (debug_reporter) {
debug_reporter(str, isdebug, this, debug_target);
}
return;
}
void AutomationScript::ReportProgress(float progress)
{
//wxLogMessage(wxString::Format(_T("automation progress: %.1f%%"), progress));
if (progress_reporter) {
progress_reporter(progress, this, progress_target);
}
return;
}
int AutomationScript::L_panicfunc(lua_State *L)
{
wxLogError(_T("Lua produced an error. Attempting to recover."));
longjmp(AutomationHelper::GetScriptObject(L)->panicjmp, lua_gettop(L));
}
void AutomationScript::process_lines(AssFile *input)
{
// prepare for panic...
if (int ret = setjmp(panicjmp)) {
wxLogError(wxString::Format(_T("Returned out of Lua environment. Size of stack before: %d"), ret));
#ifdef _DEBUG
wxLogError(
#else
wxLogFatalError(
#endif
_T("Due to an internal error in the Lua engine, the internal state of Aegisub might have become inconsistent ")
_T("and cannot continue. If you can reproduce this error, please report it to the developers."));
lua_close(L);
return;
} else {
lua_atpanic(L, AutomationScript::L_panicfunc);
}
// start by generating lua representations of the data...
// maybe it's safest to start by making plenty of space on the stack
if (!lua_checkstack(L, 100)) {
throw AutomationError(wxString(_T("Lua error: Unable to allocate stack space")));
}
OutputDebugString(wxString(_T("Preparing subtitle data")));
// first put the function itself on the stack
lua_pushstring(L, "process_lines");
lua_gettable(L, LUA_GLOBALSINDEX);
// now put the three arguments on the stack
// first argument: the metadata table
lua_newtable(L);
L_settable(L, -1, "res_x", input->GetScriptInfoAsInt(_T("PlayResX")));
L_settable(L, -1, "res_y", input->GetScriptInfoAsInt(_T("PlayResY")));
// second and third arguments: styles and events tables
lua_newtable(L);
int styletab = lua_gettop(L);
lua_newtable(L);
int eventtab = lua_gettop(L);
int numstyles = 0, numevents = 0;
// fill the styles and events tables
int processed_lines = 1;
for (std::list<AssEntry*>::iterator i = input->Line.begin(); i != input->Line.end(); i++, processed_lines++) {
AssEntry *e = *i;
if (!e->Valid) continue;
if (e->GetType() == ENTRY_STYLE) {
AssStyle *style = e->GetAsStyle(e);
// gonna need a table to put the style data into
lua_newtable(L);
// put the table into index N in the style table
lua_pushvalue(L, -1);
lua_rawseti(L, styletab, numstyles);
// and put it into its named index
lua_pushstring(L, style->name.mb_str(wxConvUTF8));
lua_pushvalue(L, -2);
lua_settable(L, styletab);
// so now the table is regged and stuff, put some data into it
L_settable (L, -1, "name", style->name);
L_settable (L, -1, "fontname", style->font);
L_settable (L, -1, "fontsize", style->fontsize);
L_settable (L, -1, "color1", style->primary.GetASSFormatted(true, true));
L_settable (L, -1, "color2", style->secondary.GetASSFormatted(true, true));
L_settable (L, -1, "color3", style->outline.GetASSFormatted(true, true));
L_settable (L, -1, "color4", style->shadow.GetASSFormatted(true, true));
L_settable_bool(L, -1, "bold", style->bold);
L_settable_bool(L, -1, "italic", style->italic);
L_settable_bool(L, -1, "underline", style->underline);
L_settable_bool(L, -1, "strikeout", style->strikeout);
L_settable (L, -1, "scale_x", style->scalex);
L_settable (L, -1, "scale_y", style->scaley);
L_settable (L, -1, "spacing", style->spacing);
L_settable (L, -1, "angle", style->angle);
L_settable (L, -1, "borderstyle", style->borderstyle);
L_settable (L, -1, "outline", style->outline_w);
L_settable (L, -1, "shadow", style->shadow_w);
L_settable (L, -1, "align", style->alignment);
L_settable (L, -1, "margin_l", style->MarginL);
L_settable (L, -1, "margin_r", style->MarginR);
L_settable (L, -1, "margin_v", style->MarginV);
L_settable (L, -1, "encoding", style->encoding);
// and get that table off the stack again
lua_settop(L, -2);
numstyles++;
} else if (e->group == _T("[Events]")) {
if (e->GetType() != ENTRY_DIALOGUE) {
// not a dialogue/comment event
// start checking for a blank line
wxString entryData = e->GetEntryData();
if (entryData.IsEmpty()) {
lua_newtable(L);
L_settable(L, -1, "kind", wxString(_T("blank")));
} else if (entryData[0] == _T(';')) {
// semicolon comment
lua_newtable(L);
L_settable(L, -1, "kind", wxString(_T("scomment")));
L_settable(L, -1, "text", entryData.Mid(1));
} else {
// not a blank line and not a semicolon comment
// just skip...
continue;
}
} else {
// ok, so it is a dialogue/comment event
// massive handling :(
lua_newtable(L);
assert(e->GetType() == ENTRY_DIALOGUE);
AssDialogue *dia = e->GetAsDialogue(e);
// kind of line
if (dia->Comment) {
L_settable(L, -1, "kind", wxString(_T("comment")));
} else {
L_settable(L, -1, "kind", wxString(_T("dialogue")));
}
L_settable(L, -1, "layer", dia->Layer);
L_settable(L, -1, "start_time", dia->Start.GetMS()/10);
L_settable(L, -1, "end_time", dia->End.GetMS()/10);
L_settable(L, -1, "style", dia->Style);
L_settable(L, -1, "name", dia->Actor);
L_settable(L, -1, "margin_l", dia->MarginL);
L_settable(L, -1, "margin_r", dia->MarginR);
L_settable(L, -1, "margin_v", dia->MarginV);
L_settable(L, -1, "effect", dia->Effect);
L_settable(L, -1, "text", dia->Text);
// so that's the easy part
// now for the stripped text and *ugh* the karaoke!
// prepare for stripped text
wxString text_stripped = _T("");
L_settable(L, -1, "text_stripped", 0); // dummy item
// prepare karaoke table
lua_newtable(L);
lua_pushstring(L, "karaoke");
lua_pushvalue(L, -2);
lua_settable(L, -4);
// now the top of the stack is the karaoke table, and it's present in the dialogue table
int kcount = 0;
int kdur = 0;
wxString kkind = _T("");
wxString ktext = _T("");
wxString ktext_stripped = _T("");
dia->ParseASSTags();
for (std::vector<AssDialogueBlock*>::iterator block = dia->Blocks.begin(); block != dia->Blocks.end(); block++) {
switch ((*block)->type) {
case BLOCK_BASE:
throw wxString(_T("BLOCK_BASE found processing dialogue blocks. This should never happen."));
case BLOCK_PLAIN:
ktext += (*block)->text;
ktext_stripped += (*block)->text;
text_stripped += (*block)->text;
break;
case BLOCK_DRAWING:
ktext += (*block)->text;
break;
case BLOCK_OVERRIDE: {
bool brackets_open = false;
std::vector<AssOverrideTag*> &tags = (*block)->GetAsOverride(*block)->Tags;
for (std::vector<AssOverrideTag*>::iterator tag = tags.begin(); tag != tags.end(); tag++) {
if (!(*tag)->Name.Mid(0,2).CmpNoCase(_T("\\k")) && (*tag)->IsValid()) {
// it's a karaoke tag
if (brackets_open) {
ktext += _T("}");
brackets_open = false;
}
L_settable_kara(L, -1, kcount, kdur, kkind, ktext, ktext_stripped);
kcount++;
kdur = (*tag)->Params[0]->AsInt(); // no error checking; this should always be int
kkind = (*tag)->Name.Mid(1);
ktext = _T("");
ktext_stripped = _T("");
} else {
// it's something else
// don't care if it's a valid tag or not
if (!brackets_open) {
ktext += _T("{");
brackets_open = true;
}
ktext += (*tag)->ToString();
}
}
if (brackets_open) {
ktext += _T("}");
}
break;}
}
}
dia->ClearBlocks();
// add the final karaoke block to the table
// (even if there's no karaoke in the line, there's always at least one karaoke block)
// even if the line ends in {\k10} with no text after, an empty block should still be inserted
// (otherwise data are lost)
L_settable_kara(L, -1, kcount, kdur, kkind, ktext, ktext_stripped);
kcount++;
L_settable(L, -1, "n", kcount); // number of syllables in the karaoke
lua_settop(L, -2); // remove karaoke table from the stack again
L_settable(L, -1, "text_stripped", text_stripped); // store the real stripped text
}
// now the entry table has been created and placed on top of the stack
// now all that's missing it to insert it into the event table
lua_rawseti(L, eventtab, numevents);
numevents++;
} else {
// not really a line type automation needs to take care of... ignore it
}
ReportProgress(100.0f * processed_lines / input->Line.size() / 3);
}
// finally add the counter elements to the styles and events tables
lua_pushnumber(L, numstyles);
lua_rawseti(L, styletab, -1);
L_settable(L, eventtab, "n", numevents);
// and let the config object create a table for the @config argument
//wxLogMessage(_T("Calling configuration.store_to_lua()"));
configuration.store_to_lua(L);
// so now the metadata, styles and events tables are filled with data
// ready to call the processing function!
OutputDebugString(wxString(_T("Running script for processing")));
ReportProgress(100.0f/3);
L_callfunc(L, 4, 1);
ReportProgress(200.0f/3);
OutputDebugString(wxString(_T("Reading back data from script")));
// phew, survived the call =)
// time to read back the results
wxLogDebug(_T("Returned from Lua script call"));
if (!lua_istable(L, -1)) {
throw AutomationError(wxString(_T("The script function did not return a table as expected. Unable to process results. (Nothing was changed.)")));
}
// but start by removing all events
{
std::list<AssEntry*>::iterator cur, next;
next = input->Line.begin();
while (next != input->Line.end()) {
cur = next++;
if ((*cur)->group == _T("[Events]")) {
wxString temp = (*cur)->GetEntryData();
if (temp == _T("[Events]")) {
// skip the section header
continue;
}
if ((*cur)->GetType() != ENTRY_DIALOGUE && temp.Mid(0,1) != _T(";") && temp.Trim() != _T("")) {
// skip non-dialogue non-semicolon comment lines (such as Format)
continue;
}
delete (*cur);
input->Line.erase(cur);
}
}
}
wxLogDebug(_T("Finished removing old events from subtitles"));
// so anyway, there is a single table on the stack now
// that table contains a lot of events...
// and it ought to contain an "n" key as well, telling how many events
// but be lenient, and don't expect one to be there, but rather count from zero and let it be nil-terminated
// if the "n" key is there, use it as a progress indicator hint, though
int output_line_count;
lua_pushstring(L, "n");
lua_gettable(L, -2);
if (lua_isnumber(L, -1)) {
output_line_count = (int) lua_tonumber(L, -1);
} else {
// assume number of output lines == number of input lines
output_line_count = processed_lines;
}
lua_settop(L, -2);
wxLogDebug(_T("Retrieved number of lines in result: %d"), output_line_count);
// loop through the stack and report back the type of each element
wxLogDebug(_T("Size of Lua stack: %d"), lua_gettop(L));
for (int si = lua_gettop(L); si > 0; si--) {
wxString type = wxString(lua_typename(L, lua_type(L, si)), wxConvUTF8);
wxLogDebug(_T("Stack index %d, type %s"), si, type.c_str());
}
wxLogDebug(_T("Stack dump finished"));
int outline = 0;
int faketime = input->Line.back()->StartMS;
// If there's nothing at index 0, start at index 1 instead, to support both zero and one based indexing
lua_pushnumber(L, outline);
lua_gettable(L, -2);
if (!lua_istable(L, -1)) {
outline++;
output_line_count++;
}
lua_pop(L, 1);
while (lua_pushnumber(L, outline), lua_gettable(L, -2), lua_istable(L, -1)) {
// top of the stack is a table, hopefully with an AssEntry in it
wxLogDebug(_T("Processing output line %d"), outline);
// start by getting the kind
lua_pushstring(L, "kind");
lua_gettable(L, -2);
if (!lua_isstring(L, -1)) {
OutputDebugString(wxString::Format(_T("The output data at index %d is mising a valid 'kind' field, and has been skipped"), outline));
lua_settop(L, -2);
} else {
wxString kind = wxString(lua_tostring(L, -1), wxConvUTF8).Lower();
// remove "kind" from stack again
lua_settop(L, -2);
wxLogDebug(_T("Kind of line: %s"), kind.c_str());
if (kind == _T("dialogue") || kind == _T("comment")) {
lua_pushstring(L, "layer");
lua_gettable(L, -2);
lua_pushstring(L, "start_time");
lua_gettable(L, -3);
lua_pushstring(L, "end_time");
lua_gettable(L, -4);
lua_pushstring(L, "style");
lua_gettable(L, -5);
lua_pushstring(L, "name");
lua_gettable(L, -6);
lua_pushstring(L, "margin_l");
lua_gettable(L, -7);
lua_pushstring(L, "margin_r");
lua_gettable(L, -8);
lua_pushstring(L, "margin_v");
lua_gettable(L, -9);
lua_pushstring(L, "effect");
lua_gettable(L, -10);
lua_pushstring(L, "text");
lua_gettable(L, -11);
wxLogDebug(_T("Read out all fields for dialogue event"));
if (lua_isnumber(L, -10) && lua_isnumber(L, -9) && lua_isnumber(L, -8) &&
lua_isstring(L, -7) && lua_isstring(L, -6) && lua_isnumber(L, -5) &&
lua_isnumber(L, -4) && lua_isnumber(L, -3) && lua_isstring(L, -2) &&
lua_isstring(L, -1))
{
AssDialogue *e = new AssDialogue();
e->Layer = (int)lua_tonumber(L, -10);
e->Start.SetMS(10*(int)lua_tonumber(L, -9));
e->End.SetMS(10*(int)lua_tonumber(L, -8));
e->Style = wxString(lua_tostring(L, -7), wxConvUTF8);
e->Actor = wxString(lua_tostring(L, -6), wxConvUTF8);
e->MarginL = (int)lua_tonumber(L, -5);
e->MarginR = (int)lua_tonumber(L, -4);
e->MarginV = (int)lua_tonumber(L, -3);
e->Effect = wxString(lua_tostring(L, -2), wxConvUTF8);
e->Text = wxString(lua_tostring(L, -1), wxConvUTF8);
e->Comment = kind == _T("comment");
lua_settop(L, -11);
e->StartMS = e->Start.GetMS();
//e->ParseASSTags();
e->UpdateData();
input->Line.push_back(e);
wxLogDebug(_T("Produced new dialogue event in output subs"));
} else {
OutputDebugString(wxString::Format(_T("The output data at index %d (kind '%s') has one or more missing/invalid fields, and has been skipped"), outline, kind.c_str()));
}
} else if (kind == _T("scomment")) {
lua_pushstring(L, "text");
lua_gettable(L, -2);
if (lua_isstring(L, -1)) {
wxString text(lua_tostring(L, -1), wxConvUTF8);
lua_settop(L, -2);
AssEntry *e = new AssEntry(wxString(_T(";")) + text);
e->StartMS = faketime;
input->Line.push_back(e);
wxLogDebug(_T("Produced new semicolon comment in output subs"));
} else {
OutputDebugString(wxString::Format(_T("The output data at index %d (kind 'scomment') is missing a valid 'text' field, and has been skipped"), outline));
}
} else if (kind == _T("blank")) {
AssEntry *e = new AssEntry(_T(""));
e->StartMS = faketime;
input->Line.push_back(e);
wxLogDebug(_T("Produced new blank line in output subs"));
} else {
OutputDebugString(wxString::Format(_T("The output data at index %d has an invalid value in the 'kind' field, and has been skipped"), outline));
}
}
// remove table again
lua_settop(L, -2);
// progress report
if (outline >= output_line_count) {
ReportProgress(99.9f);
} else {
ReportProgress((200.0f + 100.0f*outline/output_line_count) / 3);
}
outline++;
wxLogDebug(_T("Size of Lua stack: %d"), lua_gettop(L));
}
ReportProgress(100);
OutputDebugString(wxString(_T("Script execution complete")));
return;
}
AutomationScriptFile::~AutomationScriptFile()
{
// if file had a bom
if (utf8bom)
scriptdata -= 3;
delete scriptdata;
}
AutomationScriptFile *AutomationScriptFile::CreateFromString(wxString &script)
{
wxCharBuffer rawscript = script.mb_str(wxConvUTF8);
AutomationScriptFile *res = new AutomationScriptFile();
res->scriptlen = strlen(rawscript);
res->scriptdata = new char[res->scriptlen];
res->filename = _T("");
return res;
}
AutomationScriptFile *AutomationScriptFile::CreateFromFile(wxString filename)
{
wxFile file(filename);
if (!file.IsOpened()) {
return 0;
}
// prepare variables
AutomationScriptFile *res = new AutomationScriptFile();
res->filename = filename;
res->scriptlen = file.Length();
res->scriptdata = new char[res->scriptlen];
// read from file
file.Read(res->scriptdata, res->scriptlen);
// check for UTF-8 BOM
res->utf8bom = ((res->scriptdata[0]&0xFF) == 0xEF) && ((res->scriptdata[1]&0xFF) == 0xBB) && ((res->scriptdata[2]&0xFF) == 0xBF);
if (res->utf8bom) {
// skip it if found
res->scriptdata += 3;
res->scriptlen -= 3;
}
return res;
}
Reference in a new issue View git blame Copy permalink