AVSValue ExpWhileLoop::Evaluate(IScriptEnvironment* env)
{
AVSValue result;
IScriptEnvironment2 *env2 = static_cast<IScriptEnvironment2*>(env);
env2->GetVar("last", &result);
AVSValue cond;
do {
cond = condition->Evaluate(env);
if (!cond.IsBool())
env->ThrowError("while: condition must be boolean (true/false)");
if (!cond.AsBool())
break;
if (body)
{
try
{
result = body->Evaluate(env);
if (result.IsClip())
env->SetVar("last", result);
}
catch(const BreakStmtException&)
{
break;
}
}
}
while (true);
return result;
}
bool Write::DoEval( IScriptEnvironment* env) {
bool keep_this_line = true;
int i;
AVSValue expr;
AVSValue result;
for (i=0; i<arrsize; i++) {
expr = arglist[i].expression;
if ( (linecheck==1) && (i==0)) {
try {
result = env->Invoke("Eval",expr);
if (!result.AsBool(true)) {
keep_this_line = false;
break;
}
} catch (AvisynthError) {
// env->ThrowError("Write: Can't eval linecheck expression!"); // results in KEEPING the line
}
} else {
try {
result = env->Invoke("Eval",expr);
result = env->Invoke("string",result); //convert all results to a string
arglist[i].string = result.AsString(EMPTY);
} catch (AvisynthError error) {
arglist[i].string = env->SaveString(error.msg);
}
}
}
return keep_this_line;
}
void ConditionalReader::SetRange(int start_frame, int stop_frame, AVSValue v) {
int i;
start_frame = std::max(std::min(start_frame, vi.num_frames-1), 0);
stop_frame = std::max(std::min(stop_frame, vi.num_frames-1), 0);
int p;
float q;
bool r;
switch (mode) {
case MODE_INT:
p = v.AsInt();
for (i = start_frame; i <= stop_frame; i++) {
intVal[i] = p;
}
break;
case MODE_FLOAT:
q = (float)v.AsFloat();
for (i = start_frame; i <= stop_frame; i++) {
floatVal[i] = q;
}
break;
case MODE_BOOL:
r = v.AsBool();
for (i = start_frame; i <= stop_frame; i++) {
boolVal[i] = r;
}
break;
}
}
AVSValue ExpConditional::Evaluate(IScriptEnvironment* env)
{
AVSValue cond = If->Evaluate(env);
if (!cond.IsBool())
env->ThrowError("Evaluate: left of `?' must be boolean (true/false)");
return (cond.AsBool() ? Then : Else)->Evaluate(env);
}
int __stdcall dimzon_avs_getvariable_b(SafeStruct* pstr, const char* name, bool* result)
{
try
{
pstr->err[0] = 0;
try
{
AVSValue var = pstr->env->GetVar(name);
if(var.Defined())
{
if(!var.IsBool())
{
strncpy_s(pstr->err, ERRMSG_LEN, "AviSynthWrapper: Requested variable is not Boolean!", _TRUNCATE);
return AVS_VARWRNGTYPE;
}
*result = var.AsBool();
return 0;
}
return AVS_VARNDEFINED;
}
catch(AvisynthError err)
{
strncpy_s(pstr->err, ERRMSG_LEN, err.msg, _TRUNCATE);
return AVS_GERROR;
}
}
catch(IScriptEnvironment::NotFound)
{
return AVS_VARNFOUND;
}
}
AVSValue ExpNot::Evaluate(IScriptEnvironment* env)
{
AVSValue x = e->Evaluate(env);
if (x.IsBool())
return !x.AsBool();
else {
env->ThrowError("Evaluate: operand of `!' must be boolean (true/false)");
return 0;
}
}
AVSValue ExpAnd::Evaluate(IScriptEnvironment* env)
{
AVSValue x = a->Evaluate(env);
if (!x.IsBool())
env->ThrowError("Evaluate: left operand of && must be boolean (true/false)");
if (!x.AsBool())
return x;
AVSValue y = b->Evaluate(env);
if (!y.IsBool())
env->ThrowError("Evaluate: right operand of && must be boolean (true/false)");
return y;
}
AVSValue ExpEqual::Evaluate(IScriptEnvironment* env)
{
AVSValue x = a->Evaluate(env);
AVSValue y = b->Evaluate(env);
if (x.IsBool() && y.IsBool()) {
return x.AsBool() == y.AsBool();
}
else if (x.IsInt() && y.IsInt()) {
return x.AsInt() == y.AsInt();
}
else if (x.IsFloat() && y.IsFloat()) {
return x.AsFloat() == y.AsFloat();
}
else if (x.IsClip() && y.IsClip()) {
return x.AsClip() == y.AsClip();
}
else if (x.IsString() && y.IsString()) {
return !lstrcmpi(x.AsString(), y.AsString());
}
else {
env->ThrowError("Evaluate: operands of `==' and `!=' must be comparable");
return 0;
}
}
void ConditionalReader::SetFrame(int framenumber, AVSValue v) {
if (framenumber < 0 || framenumber > vi.num_frames-1 )
return;
switch (mode) {
case MODE_INT:
intVal[framenumber] = v.AsInt();
break;
case MODE_FLOAT:
floatVal[framenumber] = (float)v.AsFloat();
break;
case MODE_BOOL:
boolVal[framenumber] = v.AsBool();
break;
}
}
AVSValueStruct CAsifScriptEnvironment::Invoke(const char* command) {
AVSValueStruct ir;
ir.arraysize = 0;
ir.errors = false;
ir.returnvalue.ival = 0;
ir.type = 0;
std::vector<const char *> tempargnames(args.size());
for (size_t i = 0; i < tempargnames.size(); i++)
tempargnames[i] = (argnames[i].c_str() == "" ? nullptr : argnames[i].c_str());
try {
AVSValue ret = envse->Invoke(command, AVSValue(args.data(), args.size()), tempargnames.data());
if (ret.IsClip()) {
ir.returnvalue.cval = new CAsifClip(ret.AsClip(), envse);
ir.type = 1;
} else if (ret.IsBool()) {
ir.returnvalue.bval = ret.AsBool();
ir.type = 2;
} else if (ret.IsInt()) {
ir.returnvalue.ival = ret.AsInt();
ir.type = 3;
} else if (ret.IsFloat()) {
ir.returnvalue.fval = (float)ret.AsFloat();
ir.type = 4;
} else if (ret.IsString()) {
ir.returnvalue.sval = ret.AsString();
ir.type = 5;
} else if (ret.IsArray()) {
// ir.returnvalue.aval=ret.
ir.arraysize = ret.ArraySize();
ir.type = 6;
}
} catch (AvisynthError &e) {
ir.type = 100;
ir.returnvalue.sval = e.msg;
ir.errors = 1;
}
ResetArgs();
return ir;
}
int __stdcall dimzon_avs_isfuncexists(SafeStruct* pstr, const char *func)
{
pstr->err[0] = 0;
try
{
AVSValue var = pstr->env->FunctionExists(func);
if (var.IsBool())
{
return (var.AsBool()) ? 0 : AVS_VARNFOUND;
}
return AVS_VARWRNGTYPE;
}
catch(AvisynthError err)
{
strncpy_s(pstr->err, ERRMSG_LEN, err.msg, _TRUNCATE);
return AVS_GERROR;
}
}
void ConditionalReader::SetFrame(int framenumber, AVSValue v) {
if ((framenumber+offset) < 0 || (framenumber+offset) > vi.num_frames-1 )
return;
switch (mode) {
case MODE_INT:
intVal[framenumber+offset] = v.AsInt();
break;
case MODE_FLOAT:
floatVal[framenumber+offset] = v.AsFloatf();
break;
case MODE_BOOL:
boolVal[framenumber+offset] = v.AsBool();
break;
case MODE_STRING:
stringVal[framenumber+offset] = v.AsString("");
break;
}
}
AVSValue ExpBlockConditional::Evaluate(IScriptEnvironment* env)
{
AVSValue result;
IScriptEnvironment2 *env2 = static_cast<IScriptEnvironment2*>(env);
env2->GetVar("last", &result);
AVSValue cond = If->Evaluate(env);
if (!cond.IsBool())
env->ThrowError("if: condition must be boolean (true/false)");
if (cond.AsBool())
{
if (Then) // note: "Then" can also be NULL if its block is empty
result = Then->Evaluate(env);
}
else if (Else) // note: "Else" can also be NULL if its block is empty
result = Else->Evaluate(env);
if (result.IsClip())
env->SetVar("last", result);
return result;
}
void ConditionalReader::SetRange(int start_frame, int stop_frame, AVSValue v) {
int i;
start_frame = max(start_frame+offset, 0);
stop_frame = min(stop_frame+offset, vi.num_frames-1);
int p;
float q;
bool r;
const char* s;
switch (mode) {
case MODE_INT:
p = v.AsInt();
for (i = start_frame; i <= stop_frame; i++) {
intVal[i] = p;
}
break;
case MODE_FLOAT:
q = v.AsFloatf();
for (i = start_frame; i <= stop_frame; i++) {
floatVal[i] = q;
}
break;
case MODE_BOOL:
r = v.AsBool();
for (i = start_frame; i <= stop_frame; i++) {
boolVal[i] = r;
}
break;
case MODE_STRING:
s = v.AsString("");
for (i = start_frame; i <= stop_frame; i++) {
stringVal[i] = s;
}
break;
}
}
int __stdcall dimzon_avs_invoke(SafeStruct* pstr, char *func, char **arg, int len, AVSDLLVideoInfo *vi, float* func_out)
{
try
{
*func_out = -FLT_MAX;
pstr->err[0] = 0;
const int N = 10;
int actual_len = 0;
AVSValue args[N] = { };
if (len == 0) args[0] = 0;
else if (len > N) len = N;
for(int i = 0; i < len; i++)
{
if (strlen(arg[i]) > 0)
{
string lower = arg[i];
bool was_letters = false;
bool was_digits = false;
bool was_spaces = false;
//Слишком длинные значения - точно текст
for (unsigned int n = 0; n < lower.size() && lower.size() <= 10; n++)
{
lower[n] = tolower(lower[n]);
if (!was_letters && isalpha(lower[n])) was_letters = true;
if (!was_digits && isdigit(lower[n])) was_digits = true;
if (!was_spaces && isspace(lower[n])) was_spaces = true;
}
if (i == 0 && was_letters && !was_digits && !was_spaces && lower.compare("last") == 0)
{
//Clip (last)
if(!pstr->clp) throw AvisynthError("AviSynthWrapper: The \"last\" clip was requested, but it doesn't exist!");
args[actual_len] = pstr->clp; //pstr->res->AsClip();
actual_len += 1;
//pstr->clp; pstr->res->AsClip(); //С обработкой после прошлых вызовов Invoke
//pstr->env->GetVar("last").AsClip(); //"Чистый" выход скрипта
}
else if (was_letters && !was_digits && !was_spaces && lower.compare("true") == 0)
{
//Bool (true)
args[actual_len] = true;
actual_len += 1;
}
else if (was_letters && !was_digits && !was_spaces && lower.compare("false") == 0)
{
//Bool (false)
args[actual_len] = false;
actual_len += 1;
}
else if (!was_letters && was_digits && !was_spaces && lower.find(".") != string::npos)
{
//Float (double..)
args[actual_len] = atof(arg[i]);
actual_len += 1;
}
else if (!was_letters && was_digits && !was_spaces)
{
//Integer
args[actual_len] = atoi(arg[i]);
actual_len += 1;
}
else
{
//String
args[actual_len] = arg[i];
actual_len += 1;
}
}
}
AVSValue res = pstr->env->Invoke(func, AVSValue(args, actual_len));
if (!res.IsClip())
{
//Вывод результата
if (res.IsBool())
{
if(!res.AsBool()) *func_out = 0;
else *func_out = FLT_MAX;
}
else if (res.IsInt()) *func_out = (float)res.AsInt();
else if (res.IsFloat()) *func_out = (float)res.AsFloat();
else if (res.IsString()) { *func_out = FLT_MAX; strncpy_s(pstr->err, ERRMSG_LEN, res.AsString(), _TRUNCATE); }
}
else
{
pstr->clp = res.AsClip();
VideoInfo inf = pstr->clp->GetVideoInfo();
if (vi != NULL)
{
vi->width = inf.width;
vi->height = inf.height;
vi->raten = inf.fps_numerator;
vi->rated = inf.fps_denominator;
vi->field_based = (inf.IsFieldBased()) ? 1 : 0;
vi->first_field = (inf.IsTFF()) ? 1 : (inf.IsBFF()) ? 2 : 0;
vi->num_frames = inf.num_frames;
//Или не меняем?
if (vi->pixel_type_orig == 0) vi->pixel_type_orig = inf.pixel_type;
if (vi->sample_type_orig == 0) vi->sample_type_orig = inf.sample_type;
vi->pixel_type = inf.pixel_type;
vi->sample_type = inf.sample_type;
vi->num_audio_samples = inf.num_audio_samples;
vi->audio_samples_per_second = inf.audio_samples_per_second;
vi->nchannels = inf.nchannels;
}
//Нужен ли нам вообще этот res?!
if(pstr->res) delete pstr->res;
pstr->res = new AVSValue(res);
pstr->err[0] = 0;
}
return 0;
}
catch(AvisynthError err)
{
strncpy_s(pstr->err, ERRMSG_LEN, err.msg, _TRUNCATE);
return AVS_GERROR;
}
catch(IScriptEnvironment::NotFound)
{
strncpy_s(pstr->err, ERRMSG_LEN, "AviSynthWrapper: Wrong function name or invalid parameters was passed to Invoke!", _TRUNCATE);
return AVS_VARNFOUND
}
}