123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761 |
- /*
- * ERMInterpreter.cpp, part of VCMI engine
- *
- * Authors: listed in file AUTHORS in main folder
- *
- * License: GNU General Public License v2.0 or later
- * Full text of license available in license.txt file, in main folder
- *
- */
- #include "StdInc.h"
- #include "ERMInterpreter.h"
- #include <cctype>
- namespace spirit = boost::spirit;
- using ::scripting::ContextBase;
- using namespace ::VERMInterpreter;
- typedef int TUnusedType;
- namespace ERMConverter
- {
- //console printer
- using namespace ERM;
- static const std::map<std::string, std::string> CMP_OPERATION =
- {
- {"<", "<"},
- {">", ">"},
- {">=", ">="},
- {"=>", ">="},
- {"<=", "<="},
- {"=<", "<="},
- {"=", "=="},
- {"<>", "~="},
- {"><", "~="},
- };
- struct Variable
- {
- std::string name = "";
- std::string macro = "";
- int index = 0;
- Variable(const std::string & name_, int index_)
- {
- name = name_;
- index = index_;
- }
- Variable(const std::string & macro_)
- {
- macro = macro_;
- }
- bool isEmpty() const
- {
- return (name == "") && (macro == "");
- }
- bool isMacro() const
- {
- return (name == "") && (macro != "");
- }
- bool isSpecial() const
- {
- return (name.size() > 0) && (name[0] == 'd');
- }
- std::string str() const
- {
- if(isEmpty())
- {
- return std::to_string(index);
- }
- else if(isMacro())
- {
- return boost::to_string(boost::format("M['%s']") % macro);
- }
- else if(isSpecial() && (name.size() == 1))
- {
- boost::format fmt;
- fmt.parse("{'d', %d}");
- fmt % index;
- return fmt.str();
- }
- else if(isSpecial() && (name.size() != 1))
- {
- std::string ret;
- {
- boost::format fmt;
- if(index == 0)
- {
- fmt.parse("Q['%s']");
- fmt % name[name.size()-1];
- }
- else
- {
- fmt.parse("%s['%d']");
- fmt % name[name.size()-1] % index;
- }
- ret = fmt.str();
- }
- for(int i = ((int) name.size())-2; i > 0; i--)
- {
- boost::format fmt("%s[tostring(%s)]");
- fmt % name[i] % ret;
- ret = fmt.str();
- }
- {
- boost::format fmt;
- fmt.parse("{'d', %s}");
- fmt % ret;
- return fmt.str();
- }
- }
- else
- {
- std::string ret;
- {
- boost::format fmt;
- if(index == 0)
- {
- fmt.parse("Q['%s']");
- fmt % name[name.size()-1];
- }
- else
- {
- fmt.parse("%s['%d']");
- fmt % name[name.size()-1] % index;
- }
- ret = fmt.str();
- }
- for(int i = ((int) name.size())-2; i >= 0; i--)
- {
- boost::format fmt("%s[tostring(%s)]");
- fmt % name[i] % ret;
- ret = fmt.str();
- }
- return ret;
- }
- }
- };
- struct LVL2IexpToVar : boost::static_visitor<Variable>
- {
- LVL2IexpToVar() = default;
- Variable operator()(const TVarExpNotMacro & val) const
- {
- if(val.val.is_initialized())
- return Variable(val.varsym, val.val.get());
- else
- return Variable(val.varsym, 0);
- }
- Variable operator()(const TMacroUsage & val) const
- {
- return Variable(val.macro);
- }
- };
- struct LVL1IexpToVar : boost::static_visitor<Variable>
- {
- LVL1IexpToVar() = default;
- Variable operator()(int const & constant) const
- {
- return Variable("", constant);
- }
- Variable operator()(const TVarExp & var) const
- {
- return boost::apply_visitor(LVL2IexpToVar(), var);
- }
- };
- struct Condition : public boost::static_visitor<std::string>
- {
- Condition()
- {}
- std::string operator()(const TComparison & cmp) const
- {
- Variable lhs = boost::apply_visitor(LVL1IexpToVar(), cmp.lhs);
- Variable rhs = boost::apply_visitor(LVL1IexpToVar(), cmp.rhs);
- auto sign = CMP_OPERATION.find(cmp.compSign);
- if(sign == std::end(CMP_OPERATION))
- throw EScriptExecError(std::string("Wrong comparison sign: ") + cmp.compSign);
- boost::format fmt("(%s %s %s)");
- fmt % lhs.str() % sign->second % rhs.str();
- return fmt.str();
- }
- std::string operator()(int const & flag) const
- {
- return boost::to_string(boost::format("F['%d']") % flag);
- }
- };
- struct ParamIO
- {
- ParamIO() = default;
- std::string name = "";
- bool isInput = false;
- bool semi = false;
- std::string semiCmpSign = "";
- };
- struct Converter : public boost::static_visitor<>
- {
- mutable std::ostream * out;
- Converter(std::ostream * out_)
- : out(out_)
- {}
- protected:
- void put(const std::string & text) const
- {
- (*out) << text;
- }
- void putLine(const std::string & line) const
- {
- (*out) << line << std::endl;
- }
- void endLine() const
- {
- (*out) << std::endl;
- }
- };
- struct GetBodyOption : public boost::static_visitor<std::string>
- {
- GetBodyOption()
- {}
- virtual std::string operator()(const TVarConcatString & cmp) const
- {
- throw EScriptExecError("String concatenation not allowed in this receiver");
- }
- virtual std::string operator()(const TStringConstant & cmp) const
- {
- throw EScriptExecError("String constant not allowed in this receiver");
- }
- virtual std::string operator()(const TCurriedString & cmp) const
- {
- throw EScriptExecError("Curried string not allowed in this receiver");
- }
- virtual std::string operator()(const TSemiCompare & cmp) const
- {
- throw EScriptExecError("Semi comparison not allowed in this receiver");
- }
- virtual std::string operator()(const TMacroDef & cmp) const
- {
- throw EScriptExecError("Macro definition not allowed in this receiver");
- }
- virtual std::string operator()(const TIexp & cmp) const
- {
- throw EScriptExecError("i-expression not allowed in this receiver");
- }
- virtual std::string operator()(const TVarpExp & cmp) const
- {
- throw EScriptExecError("Varp expression not allowed in this receiver");
- }
- virtual std::string operator()(const spirit::unused_type & cmp) const
- {
- throw EScriptExecError("\'Nothing\' not allowed in this receiver");
- }
- };
- struct BodyOption : public boost::static_visitor<ParamIO>
- {
- ParamIO operator()(const TVarConcatString & cmp) const
- {
- throw EScriptExecError(std::string("String concatenation not allowed in this receiver|")+cmp.string.str+"|");
- }
- ParamIO operator()(const TStringConstant & cmp) const
- {
- boost::format fmt("[===[%s]===]");
- fmt % cmp.str;
- ParamIO ret;
- ret.isInput = true;
- ret.name = fmt.str();
- return ret;
- }
- ParamIO operator()(const TCurriedString & cmp) const
- {
- throw EScriptExecError("Curried string not allowed in this receiver");
- }
- ParamIO operator()(const TSemiCompare & cmp) const
- {
- ParamIO ret;
- ret.isInput = false;
- ret.semi = true;
- ret.semiCmpSign = cmp.compSign;
- ret.name = (boost::apply_visitor(LVL1IexpToVar(), cmp.rhs)).str();
- return ret;
- }
- ParamIO operator()(const TMacroDef & cmp) const
- {
- throw EScriptExecError("Macro definition not allowed in this receiver");
- }
- ParamIO operator()(const TIexp & cmp) const
- {
- ParamIO ret;
- ret.isInput = true;
- ret.name = (boost::apply_visitor(LVL1IexpToVar(), cmp)).str();;
- return ret;
- }
- ParamIO operator()(const TVarpExp & cmp) const
- {
- ParamIO ret;
- ret.isInput = false;
- ret.name = (boost::apply_visitor(LVL2IexpToVar(), cmp.var)).str();
- return ret;
- }
- ParamIO operator()(const spirit::unused_type & cmp) const
- {
- throw EScriptExecError("\'Nothing\' not allowed in this receiver");
- }
- };
- struct Receiver : public Converter
- {
- Receiver(std::ostream * out_)
- : Converter(out_)
- {}
- virtual void operator()(const TVRLogic & trig) const
- {
- throw EInterpreterError("VR logic is not allowed in this receiver!");
- }
- virtual void operator()(const TVRArithmetic & trig) const
- {
- throw EInterpreterError("VR arithmetic is not allowed in this receiver!");
- }
- virtual void operator()(const TNormalBodyOption & trig) const
- {
- throw EInterpreterError("Normal body is not allowed in this receiver!");
- }
- };
- struct GenericReceiver : public Receiver
- {
- std::string name;
- bool specialSemiCompare = false;
- GenericReceiver(std::ostream * out_, const std::string & name_, bool specialSemiCompare_)
- : Receiver(out_),
- name(name_),
- specialSemiCompare(specialSemiCompare_)
- {}
- using Receiver::operator();
- void operator()(const TNormalBodyOption & trig) const override
- {
- std::string outParams;
- std::string inParams;
- std::string semiCompareDecl;
- std::vector<std::string> compares;
- bool hasOutput = false;
- bool hasSemiCompare = false;
- std::vector<ParamIO> optionParams;
- if(trig.params.is_initialized())
- {
- for(auto & p : trig.params.get())
- optionParams.push_back(boost::apply_visitor(BodyOption(), p));
- }
- int idx = 1;
- int fidx = 1;
- for(const ParamIO & p : optionParams)
- {
- if(p.isInput)
- {
- if(outParams.empty())
- outParams = "_";
- else
- outParams += ", _";
- inParams += ", ";
- inParams += p.name;
- }
- else if(p.semi)
- {
- hasOutput = true;
- hasSemiCompare = true;
- std::string tempVar = std::string("s")+std::to_string(idx);
- if(semiCompareDecl.empty())
- {
- semiCompareDecl = "local "+tempVar;
- }
- else
- {
- semiCompareDecl += ", ";
- semiCompareDecl += tempVar;
- }
- if(outParams.empty())
- {
- outParams = tempVar;
- }
- else
- {
- outParams += ", ";
- outParams += tempVar;
- }
- inParams += ", nil";
- auto sign = CMP_OPERATION.find(p.semiCmpSign);
- if(sign == std::end(CMP_OPERATION))
- throw EScriptExecError(std::string("Wrong comparison sign: ") + p.semiCmpSign);
- boost::format cmpFmt("F[%d] = (%s %s %s)");
- cmpFmt % fidx % p.name % sign->second % tempVar;
- compares.push_back(cmpFmt.str());
- fidx++;
- }
- else
- {
- hasOutput = true;
- if(outParams.empty())
- {
- outParams = p.name;
- }
- else
- {
- outParams += ", ";
- outParams += p.name;
- }
- inParams += ", nil";
- }
- idx++;
- }
- if(hasSemiCompare)
- {
- putLine(semiCompareDecl);
- }
- boost::format callFormat;
- if(hasOutput)
- {
- callFormat.parse("%s = %s:%s(x%s)");
- callFormat % outParams;
- }
- else
- {
- callFormat.parse("%s:%s(x%s)");
- }
- callFormat % name;
- callFormat % trig.optionCode;
- callFormat % inParams;
- putLine(callFormat.str());
- for(auto & str : compares)
- putLine(str);
- }
- };
- struct FU : public Receiver
- {
- Variable v;
- FU(std::ostream * out_, const ERM::TIexp & tid)
- : Receiver(out_),
- v(boost::apply_visitor(LVL1IexpToVar(), tid))
- {
- }
- FU(std::ostream * out_)
- : Receiver(out_),
- v("", 0)
- {
- }
- using Receiver::operator();
- void operator()(const TNormalBodyOption & trig) const override
- {
- switch(trig.optionCode)
- {
- case 'E':
- {
- putLine("do return end");
- }
- break;
- default:
- throw EInterpreterError("Unknown opcode in FU receiver");
- break;
- }
- }
- };
- struct MC_S : public GetBodyOption
- {
- MC_S()
- {}
- using GetBodyOption::operator();
- std::string operator()(const TMacroDef & cmp) const override
- {
- return cmp.macro;
- }
- };
- struct MC : public Receiver
- {
- Variable v;
- MC(std::ostream * out_, const ERM::TIexp & tid)
- : Receiver(out_),
- v(boost::apply_visitor(LVL1IexpToVar(), tid))
- {
- }
- MC(std::ostream * out_)
- : Receiver(out_),
- v("", 0)
- {
- }
- using Receiver::operator();
- void operator()(const TNormalBodyOption & option) const override
- {
- switch(option.optionCode)
- {
- case 'S':
- {
- if(option.params.is_initialized())
- {
- for(auto & p : option.params.get())
- {
- std::string macroName = boost::apply_visitor(MC_S(), p);
- boost::format callFormat;
- if(v.isEmpty())
- {
- callFormat.parse("ERM:addMacro('%s', 'v', '%s')");
- callFormat % macroName % macroName;
- }
- else
- {
- callFormat.parse("ERM:addMacro('%s', '%s', '%d')");
- callFormat % macroName % v.name % v.index;
- }
- putLine(callFormat.str());
- }
- }
- }
- break;
- default:
- throw EInterpreterError("Unknown opcode in MC receiver");
- break;
- }
- }
- };
- struct VR_S : public GetBodyOption
- {
- VR_S()
- {}
- using GetBodyOption::operator();
- std::string operator()(const TIexp & cmp) const override
- {
- auto v = boost::apply_visitor(LVL1IexpToVar(), cmp);
- return v.str();
- }
- std::string operator()(const TStringConstant & cmp) const override
- {
- boost::format fmt("[===[%s]===]");
- fmt % cmp.str;
- return fmt.str();
- }
- };
- struct VR_H : public GetBodyOption
- {
- VR_H()
- {}
- using GetBodyOption::operator();
- std::string operator()(const TIexp & cmp) const override
- {
- Variable p = boost::apply_visitor(LVL1IexpToVar(), cmp);
- if(p.index <= 0)
- throw EScriptExecError("VR:H requires flag index");
- if(p.name != "")
- throw EScriptExecError("VR:H accept only flag index");
- boost::format fmt("'%d'");
- fmt % p.index;
- return fmt.str();
- }
- };
- struct VR_X : public GetBodyOption
- {
- VR_X()
- {
- }
- using GetBodyOption::operator();
- std::string operator()(const TIexp & cmp) const override
- {
- Variable p = boost::apply_visitor(LVL1IexpToVar(), cmp);
- return p.str();
- }
- };
- struct VR : public Receiver
- {
- Variable v;
- VR(std::ostream * out_, const ERM::TIexp & tid)
- : Receiver(out_),
- v(boost::apply_visitor(LVL1IexpToVar(), tid))
- {
- }
- using Receiver::operator();
- void operator()(const TVRLogic & trig) const override
- {
- Variable rhs = boost::apply_visitor(LVL1IexpToVar(), trig.var);
- std::string opcode;
- switch (trig.opcode)
- {
- case '&':
- opcode = "bit.band";
- break;
- case '|':
- opcode = "bit.bor";
- break;
- default:
- throw EInterpreterError("Wrong opcode in VR logic expression!");
- break;
- }
- boost::format fmt("%s = %s(%s, %s)");
- fmt % v.str() % opcode % v.str() % rhs.str();
- putLine(fmt.str());
- }
- void operator()(const TVRArithmetic & trig) const override
- {
- Variable rhs = boost::apply_visitor(LVL1IexpToVar(), trig.rhs);
- std::string opcode;
- switch (trig.opcode)
- {
- case '+':
- opcode = v.name[0] == 'z' ? ".." : "+";
- break;
- case '-':
- case '*':
- case '%':
- opcode = trig.opcode;
- break;
- case ':':
- opcode = "/";
- break;
- default:
- throw EInterpreterError("Wrong opcode in VR arithmetic!");
- break;
- }
- boost::format fmt("%s = %s %s %s");
- fmt % v.str() % v.str() % opcode % rhs.str();
- putLine(fmt.str());
- }
- void operator()(const TNormalBodyOption & trig) const override
- {
- switch(trig.optionCode)
- {
- case 'C': //setting/checking v vars
- {
- if(v.index <= 0)
- throw EScriptExecError("VR:C requires indexed variable");
- std::vector<ParamIO> optionParams;
- if(trig.params.is_initialized())
- {
- for(auto & p : trig.params.get())
- optionParams.push_back(boost::apply_visitor(BodyOption(), p));
- }
- auto index = v.index;
- for(auto & p : optionParams)
- {
- boost::format fmt;
- if(p.isInput)
- fmt.parse("%s['%d'] = %s") % v.name % index % p.name;
- else
- fmt.parse("%s = %s['%d']") % p.name % v.name % index;
- putLine(fmt.str());
- index++;
- }
- }
- break;
- case 'H': //checking if string is empty
- {
- if(!trig.params.is_initialized() || trig.params.get().size() != 1)
- throw EScriptExecError("VR:H option takes exactly 1 parameter!");
- std::string opt = boost::apply_visitor(VR_H(), trig.params.get()[0]);
- boost::format fmt("ERM.VR(%s):H(%s)");
- fmt % v.str() % opt;
- putLine(fmt.str());
- }
- break;
- case 'U':
- {
- if(!trig.params.is_initialized() || trig.params.get().size() != 1)
- throw EScriptExecError("VR:H/U need 1 parameter!");
- std::string opt = boost::apply_visitor(VR_S(), trig.params.get()[0]);
- boost::format fmt("ERM.VR(%s):%c(%s)");
- fmt % v.str() % (trig.optionCode) % opt;
- putLine(fmt.str());
- }
- break;
- case 'M': //string operations
- {
- if(!trig.params.is_initialized() || trig.params.get().size() < 2)
- throw EScriptExecError("VR:M needs at least 2 parameters!");
- std::string opt = boost::apply_visitor(VR_X(), trig.params.get()[0]);
- int paramIndex = 1;
- if(opt == "3")
- {
- boost::format fmt("%s = ERM.VR(%s):M3(");
- fmt % v.str() % v.str();
- put(fmt.str());
- }
- else
- {
- auto target = boost::apply_visitor(VR_X(), trig.params.get()[paramIndex++]);
- boost::format fmt("%s = ERM.VR(%s):M%s(");
- fmt % target % v.str() % opt;
- put(fmt.str());
- }
-
- for(int i = paramIndex; i < trig.params.get().size(); i++)
- {
- opt = boost::apply_visitor(VR_X(), trig.params.get()[i]);
- if(i > paramIndex) put(",");
- put(opt);
- }
-
- putLine(")");
- }
- break;
- case 'X': //bit xor
- {
- if(!trig.params.is_initialized() || trig.params.get().size() != 1)
- throw EScriptExecError("VR:X option takes exactly 1 parameter!");
- std::string opt = boost::apply_visitor(VR_X(), trig.params.get()[0]);
- boost::format fmt("%s = bit.bxor(%s, %s)");
- fmt % v.str() % v.str() % opt;putLine(fmt.str());
- }
- break;
- case 'R': //random variables
- {
- //TODO
- putLine("--VR:R not implemented");
- }
- break;
- case 'S': //setting variable
- {
- if(!trig.params.is_initialized() || trig.params.get().size() != 1)
- throw EScriptExecError("VR:S option takes exactly 1 parameter!");
- std::string opt = boost::apply_visitor(VR_S(), trig.params.get()[0]);
- put(v.str());
- put(" = ");
- put(opt);
- endLine();
- }
- break;
- case 'T': //random variables
- {
- //TODO
- putLine("--VR:T not implemented");
- }
- break;
- case 'V': //convert string to value
- {
- if(!trig.params.is_initialized() || trig.params.get().size() != 1)
- throw EScriptExecError("VR:V option takes exactly 1 parameter!");
- std::string opt = boost::apply_visitor(VR_X(), trig.params.get()[0]);
- boost::format fmt("%s = tostring(%s)");
- fmt % v.str() % opt;
- putLine(fmt.str());
- }
- break;
- default:
- throw EScriptExecError("Wrong VR receiver option!");
- break;
- }
- }
- };
- struct ERMExp : public Converter
- {
- ERMExp(std::ostream * out_)
- : Converter(out_)
- {}
- template <typename Visitor>
- void performBody(const boost::optional<ERM::Tbody> & body, const Visitor & visitor) const
- {
- if(body.is_initialized())
- {
- const ERM::Tbody & bo = body.get();
- for(int g=0; g<bo.size(); ++g)
- {
- boost::apply_visitor(visitor, bo[g]);
- }
- }
- }
- void convert(const std::string & name, const boost::optional<Tidentifier> & identifier, const boost::optional<Tbody> & body) const
- {
- if(name == "VR")
- {
- if(!identifier.is_initialized())
- throw EScriptExecError("VR receiver requires arguments");
- ERM::Tidentifier tid = identifier.get();
- if(tid.size() != 1)
- throw EScriptExecError("VR receiver takes exactly 1 argument");
- performBody(body, VR(out, tid[0]));
- }
- else if(name == "re")
- {
- if(!identifier.is_initialized())
- throw EScriptExecError("re receiver requires arguments");
- ERM::Tidentifier tid = identifier.get();
- auto argc = tid.size();
- if(argc > 0)
- {
- std::string loopCounter = (boost::apply_visitor(LVL1IexpToVar(), tid.at(0))).str();
- std::string startVal = argc > 1 ? (boost::apply_visitor(LVL1IexpToVar(), tid.at(1))).str() : loopCounter;
- std::string stopVal = argc > 2 ? (boost::apply_visitor(LVL1IexpToVar(), tid.at(2))).str() : loopCounter;
- std::string increment = argc > 3 ? (boost::apply_visitor(LVL1IexpToVar(), tid.at(3))).str() : "1";
- boost::format fmt("for __iter = %s, %s, %s do");
- fmt % startVal % stopVal % increment;
- putLine(fmt.str());
- fmt.parse("%s = __iter");
- fmt % loopCounter;
- putLine(fmt.str());
- }
- else
- {
- throw EScriptExecError("re receiver requires arguments");
- }
- }
- else if(name == "FU" && !identifier.is_initialized())
- {
- performBody(body, FU(out)); //assume FU:E
- }
- else if(name == "MC")
- {
- if(identifier.is_initialized())
- {
- ERM::Tidentifier tid = identifier.get();
- if(tid.size() != 1)
- throw EScriptExecError("MC receiver takes no more than 1 argument");
- performBody(body, MC(out, tid[0]));
- }
- else
- {
- performBody(body, MC(out));
- }
- }
- else
- {
- std::vector<std::string> identifiers;
- if(identifier.is_initialized())
- {
- for(const auto & id : identifier.get())
- {
- Variable v = boost::apply_visitor(LVL1IexpToVar(), id);
- if(v.isSpecial())
- throw ELineProblem("Special variable syntax ('d') is not allowed in receiver identifier");
- identifiers.push_back(v.str());
- }
- }
- std::string params;
- for(auto iter = std::begin(identifiers); iter != std::end(identifiers); ++iter)
- {
- if(!params.empty())
- params += ", ";
- params += *iter;
- }
- if(body.is_initialized())
- {
- const ERM::Tbody & bo = body.get();
- if(bo.size() == 1)
- {
- boost::format fmt("ERM.%s(%s)");
- fmt % name;
- fmt % params;
- GenericReceiver receiver(out, fmt.str(), (name == "DO"));
- bo[0].apply_visitor(receiver);
- }
- else
- {
- putLine("do");
- boost::format fmt("local %s = ERM.%s(%s)");
- fmt % name;
- fmt % name;
- fmt % params;
- putLine(fmt.str());
- performBody(body, GenericReceiver(out, name, (name=="DO") ));
- putLine("end");
- }
- }
- else
- {
- //is it an error?
- logMod->warn("ERM receiver '%s %s' w/o body", name, params);
- }
- }
- }
- void convertConditionInner(const Tcondition & cond, char op) const
- {
- std::string lhs = boost::apply_visitor(Condition(), cond.cond);
- if(cond.ctype != '/')
- op = cond.ctype;
- switch (op)
- {
- case '&':
- put(" and ");
- break;
- case '|':
- put(" or ");
- break;
- default:
- throw EInterpreterProblem(std::string("Wrong condition connection (") + cond.ctype + ") !");
- break;
- }
- put(lhs);
- if(cond.rhs.is_initialized())
- {
- switch (op)
- {
- case '&':
- case '|':
- break;
- default:
- throw EInterpreterProblem(std::string("Wrong condition connection (") + cond.ctype + ") !");
- break;
- }
- convertConditionInner(cond.rhs.get().get(), op);
- }
- }
- void convertCondition(const Tcondition & cond) const
- {
- //&c1/c2/c3|c4/c5/c6 -> (c1 & c2 & c3) | c4 | c5 | c6
- std::string lhs = boost::apply_visitor(Condition(), cond.cond);
- put("if ");
- put(lhs);
- if(cond.rhs.is_initialized())
- {
- switch (cond.ctype)
- {
- case '&':
- case '|':
- break;
- default:
- throw EInterpreterProblem(std::string("Wrong condition connection (") + cond.ctype + ") !");
- break;
- }
- convertConditionInner(cond.rhs.get().get(), cond.ctype);
- }
- putLine(" then ");
- }
- void convertReceiverOrInstruction(const boost::optional<Tcondition> & condition,
- const std::string & name,
- const boost::optional<Tidentifier> & identifier,
- const boost::optional<Tbody> & body) const
- {
- if(name=="if")
- {
- if(condition.is_initialized())
- convertCondition(condition.get());
- else
- putLine("if true then");
- }
- else if(name=="el")
- {
- putLine("else");
- }
- else if(name=="en")
- {
- putLine("end");
- }
- else
- {
- if(condition.is_initialized())
- {
- convertCondition(condition.get());
- convert(name, identifier, body);
- putLine("end");
- }
- else
- {
- convert(name, identifier, body);
- }
- }
- }
- void operator()(const Ttrigger & trig) const
- {
- throw EInterpreterError("Triggers cannot be executed!");
- }
- void operator()(const TPostTrigger & trig) const
- {
- throw EInterpreterError("Post-triggers cannot be executed!");
- }
- void operator()(const Tinstruction & trig) const
- {
- convertReceiverOrInstruction(trig.condition, trig.name, trig.identifier, boost::make_optional(trig.body));
- }
- void operator()(const Treceiver & trig) const
- {
- convertReceiverOrInstruction(trig.condition, trig.name, trig.identifier, trig.body);
- }
- };
- struct Command : public Converter
- {
- Command(std::ostream * out_)
- : Converter(out_)
- {}
- void operator()(const Tcommand & cmd) const
- {
- boost::apply_visitor(ERMExp(out), cmd.cmd);
- }
- void operator()(const std::string & comment) const
- {
- (*out) << "-- " << comment;
- endLine();
- }
- void operator()(spirit::unused_type const &) const
- {
- }
- };
- struct TLiteralEval : public boost::static_visitor<std::string>
- {
- std::string operator()(char const & val)
- {
- return "{\"'\",'"+ std::to_string(val) +"'}";
- }
- std::string operator()(double const & val)
- {
- return std::to_string(val);
- }
- std::string operator()(int const & val)
- {
- return std::to_string(val);
- }
- std::string operator()(const std::string & val)
- {
- return "{\"'\",[===[" + val + "]===]}";
- }
- };
- struct VOptionEval : public Converter
- {
- VOptionEval(std::ostream * out_)
- : Converter(out_)
- {}
- void operator()(VNIL const & opt) const
- {
- (*out) << "{}";
- }
- void operator()(VNode const & opt) const;
- void operator()(VSymbol const & opt) const
- {
- (*out) << "\"" << opt.text << "\"";
- }
- void operator()(TLiteral const & opt) const
- {
- TLiteralEval tmp;
- (*out) << boost::apply_visitor(tmp, opt);
- }
- void operator()(ERM::Tcommand const & opt) const
- {
- //this is how FP works, evaluation == producing side effects
- //TODO: can we evaluate to smth more useful?
- //???
- throw EVermScriptExecError("Using ERM options in VERM expression is not (yet) allowed");
- // boost::apply_visitor(ERMExp(out), opt.cmd);
- }
- };
- void VOptionEval::operator()(VNode const& opt) const
- {
- VNode tmpn(opt);
- (*out) << "{";
- for(VOption & op : tmpn.children)
- {
- boost::apply_visitor(VOptionEval(out), op);
- (*out) << ",";
- }
- (*out) << "}";
- }
- struct Line : public Converter
- {
- Line(std::ostream * out_)
- : Converter(out_)
- {}
- void operator()(TVExp const & cmd) const
- {
- put("VERM:E");
- VNode line(cmd);
- VOptionEval eval(out);
- eval(line);
- endLine();
- }
- void operator()(TERMline const & cmd) const
- {
- boost::apply_visitor(Command(out), cmd);
- }
- };
- void convertInstructions(std::ostream & out, ERMInterpreter * owner)
- {
- out << "local function instructions()" << std::endl;
- out << "local e, x, y = {}, {}, {}" << std::endl;
- Line lineConverter(&out);
- for(const LinePointer & lp : owner->instructions)
- {
- ERM::TLine & line = owner->retrieveLine(lp);
- boost::apply_visitor(lineConverter, line);
- }
- out << "end" << std::endl;
- }
- void convertFunctions(std::ostream & out, ERMInterpreter * owner, const std::vector<VERMInterpreter::Trigger> & triggers)
- {
- Line lineConverter(&out);
- for(const VERMInterpreter::Trigger & trigger : triggers)
- {
- ERM::TLine & firstLine = owner->retrieveLine(trigger.line);
- const ERM::TTriggerBase & trig = ERMInterpreter::retrieveTrigger(firstLine);
- //TODO: condition
- out << "ERM:addTrigger({" << std::endl;
- if(!trig.identifier.is_initialized())
- throw EInterpreterError("Function must have identifier");
- ERM::Tidentifier tid = trig.identifier.get();
- if(tid.size() == 0)
- throw EInterpreterError("Function must have identifier");
- Variable v = boost::apply_visitor(LVL1IexpToVar(), tid[0]);
- if(v.isSpecial())
- throw ELineProblem("Special variable syntax ('d') is not allowed in function definition");
- out << "id = {" << v.str() << "}," << std::endl;
- out << "name = 'FU'," << std::endl;
- out << "fn = function (e, y, x)" << std::endl;
- out << "local _" << std::endl;
- LinePointer lp = trigger.line;
- ++lp;
- for(; lp.isValid(); ++lp)
- {
- ERM::TLine curLine = owner->retrieveLine(lp);
- if(owner->isATrigger(curLine))
- break;
- boost::apply_visitor(lineConverter, curLine);
- }
- out << "end," << std::endl;
- out << "})" << std::endl;
- }
- }
- void convertTriggers(std::ostream & out, ERMInterpreter * owner, const VERMInterpreter::TriggerType & type, const std::vector<VERMInterpreter::Trigger> & triggers)
- {
- Line lineConverter(&out);
- for(const VERMInterpreter::Trigger & trigger : triggers)
- {
- ERM::TLine & firstLine = owner->retrieveLine(trigger.line);
- const ERM::TTriggerBase & trig = ERMInterpreter::retrieveTrigger(firstLine);
- //TODO: condition
- out << "ERM:addTrigger({" << std::endl;
- std::vector<std::string> identifiers;
- if(trig.identifier.is_initialized())
- {
- for(const auto & id : trig.identifier.get())
- {
- Variable v = boost::apply_visitor(LVL1IexpToVar(), id);
- if(v.isSpecial())
- throw ELineProblem("Special variable syntax ('d') is not allowed in trigger definition");
- identifiers.push_back(v.str());
- }
- }
- out << "id = {";
- for(const auto & id : identifiers)
- out << "'" << id << "',";
- out << "}," << std::endl;
- out << "name = '" << trig.name << "'," << std::endl;
- out << "fn = function (e, y)" << std::endl;
- out << "local _" << std::endl;
- LinePointer lp = trigger.line;
- ++lp;
- for(; lp.isValid(); ++lp)
- {
- ERM::TLine curLine = owner->retrieveLine(lp);
- if(owner->isATrigger(curLine))
- break;
- boost::apply_visitor(lineConverter, curLine);
- }
- out << "end," << std::endl;
- out << "})" << std::endl;
- }
- }
- }
- struct ScriptScanner : boost::static_visitor<>
- {
- ERMInterpreter * interpreter;
- LinePointer lp;
- ScriptScanner(ERMInterpreter * interpr, const LinePointer & _lp) : interpreter(interpr), lp(_lp)
- {}
- void operator()(TVExp const& cmd) const
- {
- //
- }
- void operator()(TERMline const& cmd) const
- {
- if(cmd.which() == 0) //TCommand
- {
- Tcommand tcmd = boost::get<Tcommand>(cmd);
- switch (tcmd.cmd.which())
- {
- case 0: //trigger
- {
- Trigger trig;
- trig.line = lp;
- interpreter->triggers[ TriggerType(boost::get<ERM::Ttrigger>(tcmd.cmd).name) ].push_back(trig);
- }
- break;
- case 1: //instruction
- {
- interpreter->instructions.push_back(lp);
- }
- break;
- case 3: //post trigger
- {
- Trigger trig;
- trig.line = lp;
- interpreter->postTriggers[ TriggerType(boost::get<ERM::TPostTrigger>(tcmd.cmd).name) ].push_back(trig);
- }
- break;
- default:
- break;
- }
- }
- }
- };
- ERMInterpreter::ERMInterpreter(vstd::CLoggerBase * logger_)
- : logger(logger_)
- {
- }
- ERMInterpreter::~ERMInterpreter()
- {
- }
- bool ERMInterpreter::isATrigger( const ERM::TLine & line )
- {
- switch(line.which())
- {
- case 0: //v-exp
- {
- TVExp vexp = boost::get<TVExp>(line);
- if(vexp.children.size() == 0)
- return false;
- switch (getExpType(vexp.children[0]))
- {
- case SYMBOL:
- return false;
- break;
- case TCMD:
- return isCMDATrigger( boost::get<ERM::Tcommand>(vexp.children[0]) );
- break;
- default:
- return false;
- break;
- }
- }
- break;
- case 1: //erm
- {
- TERMline ermline = boost::get<TERMline>(line);
- switch(ermline.which())
- {
- case 0: //tcmd
- return isCMDATrigger( boost::get<ERM::Tcommand>(ermline) );
- break;
- default:
- return false;
- break;
- }
- }
- break;
- default:
- assert(0); //it should never happen
- break;
- }
- assert(0);
- return false;
- }
- ERM::EVOtions ERMInterpreter::getExpType(const ERM::TVOption & opt)
- {
- //MAINTENANCE: keep it correct!
- return static_cast<ERM::EVOtions>(opt.which());
- }
- bool ERMInterpreter::isCMDATrigger(const ERM::Tcommand & cmd)
- {
- switch (cmd.cmd.which())
- {
- case 0: //trigger
- case 3: //post trigger
- return true;
- break;
- default:
- return false;
- break;
- }
- }
- ERM::TLine & ERMInterpreter::retrieveLine(const LinePointer & linePtr)
- {
- return scripts.find(linePtr)->second;
- }
- ERM::TTriggerBase & ERMInterpreter::retrieveTrigger(ERM::TLine & line)
- {
- if(line.which() == 1)
- {
- ERM::TERMline &tl = boost::get<ERM::TERMline>(line);
- if(tl.which() == 0)
- {
- ERM::Tcommand &tcm = boost::get<ERM::Tcommand>(tl);
- if(tcm.cmd.which() == 0)
- {
- return boost::get<ERM::Ttrigger>(tcm.cmd);
- }
- else if(tcm.cmd.which() == 3)
- {
- return boost::get<ERM::TPostTrigger>(tcm.cmd);
- }
- throw ELineProblem("Given line is not a trigger!");
- }
- throw ELineProblem("Given line is not a command!");
- }
- throw ELineProblem("Given line is not an ERM trigger!");
- }
- std::string ERMInterpreter::loadScript(const std::string & name, const std::string & source)
- {
- CERMPreprocessor preproc(source);
- const bool isVERM = preproc.version == CERMPreprocessor::Version::VERM;
- ERMParser ep;
- std::vector<LineInfo> buf = ep.parseFile(preproc);
- for(int g=0; g<buf.size(); ++g)
- scripts[LinePointer(static_cast<int>(buf.size()), g, buf[g].realLineNum)] = buf[g].tl;
- for(auto p : scripts)
- boost::apply_visitor(ScriptScanner(this, p.first), p.second);
- std::stringstream out;
- out << "local ERM = require(\"core:erm\")" << std::endl;
- if(isVERM)
- {
- out << "local VERM = require(\"core:verm\")" << std::endl;
- }
- out << "local _" << std::endl;
- out << "local v, w, z, F, M, Q = ERM.v, ERM.w, ERM.z, ERM.F, ERM.M, ERM.Q" << std::endl;
- ERMConverter::convertInstructions(out, this);
- for(const auto & p : triggers)
- {
- const VERMInterpreter::TriggerType & tt = p.first;
- if(tt.type == VERMInterpreter::TriggerType::FU)
- {
- ERMConverter::convertFunctions(out, this, p.second);
- }
- else
- {
- ERMConverter::convertTriggers(out, this, tt, p.second);
- }
- }
- for(const auto & p : postTriggers)
- ;//TODO:postTriggers
- out << "ERM:callInstructions(instructions)" << std::endl;
- return out.str();
- }
- namespace VERMInterpreter
- {
- VOption convertToVOption(const ERM::TVOption & tvo)
- {
- return boost::apply_visitor(OptionConverterVisitor(), tvo);
- }
- VNode::VNode( const ERM::TVExp & exp )
- {
- for(int i=0; i<exp.children.size(); ++i)
- {
- children.push_back(convertToVOption(exp.children[i]));
- }
- processModifierList(exp.modifier, false);
- }
- VNode::VNode( const VOption & first, const VOptionList & rest ) /*merges given arguments into [a, rest] */
- {
- setVnode(first, rest);
- }
- VNode::VNode( const VOptionList & cdren ) : children(cdren)
- {}
- VNode::VNode( const ERM::TSymbol & sym )
- {
- children.car() = VSymbol(sym.sym);
- processModifierList(sym.symModifier, true);
- }
- void VNode::setVnode( const VOption & first, const VOptionList & rest )
- {
- children.car() = first;
- children.cdr() = rest;
- }
- void VNode::processModifierList( const std::vector<TVModifier> & modifierList, bool asSymbol )
- {
- for(int g=0; g<modifierList.size(); ++g)
- {
- if(asSymbol)
- {
- children.resize(children.size()+1);
- for(int i=children.size()-1; i >0; i--)
- {
- children[i] = children[i-1];
- }
- }
- else
- {
- children.cdr() = VNode(children);
- }
- if(modifierList[g] == "`")
- {
- children.car() = VSymbol("`");
- }
- else if(modifierList[g] == ",!")
- {
- children.car() = VSymbol("comma-unlist");
- }
- else if(modifierList[g] == ",")
- {
- children.car() = VSymbol(",");
- }
- else if(modifierList[g] == "#'")
- {
- children.car() = VSymbol("get-func");
- }
- else if(modifierList[g] == "'")
- {
- children.car() = VSymbol("'");
- }
- else
- throw EInterpreterError("Incorrect value of modifier!");
- }
- }
- VermTreeIterator & VermTreeIterator::operator=( const VOption & opt )
- {
- switch (state)
- {
- case CAR:
- if(parent->size() <= basePos)
- parent->push_back(opt);
- else
- (*parent)[basePos] = opt;
- break;
- case NORM:
- parent->resize(basePos+1);
- (*parent)[basePos] = opt;
- break;
- default://should never happen
- break;
- }
- return *this;
- }
- VermTreeIterator & VermTreeIterator::operator=( const std::vector<VOption> & opt )
- {
- switch (state)
- {
- case CAR:
- //TODO: implement me
- break;
- case NORM:
- parent->resize(basePos+1);
- parent->insert(parent->begin()+basePos, opt.begin(), opt.end());
- break;
- default://should never happen
- break;
- }
- return *this;
- }
- VermTreeIterator & VermTreeIterator::operator=( const VOptionList & opt )
- {
- return *this = opt;
- }
- VOption & VermTreeIterator::getAsItem()
- {
- if(state == CAR)
- return (*parent)[basePos];
- else
- throw EInterpreterError("iterator is not in car state, cannot get as list");
- }
- size_t VermTreeIterator::size() const
- {
- return parent->size() - basePos;
- }
- VERMInterpreter::VOptionList VermTreeIterator::getAsList()
- {
- VOptionList ret;
- for(int g = basePos; g<parent->size(); ++g)
- {
- ret.push_back((*parent)[g]);
- }
- return ret;
- }
- VOption OptionConverterVisitor::operator()( ERM::TVExp const& cmd ) const
- {
- return VNode(cmd);
- }
- VOption OptionConverterVisitor::operator()( ERM::TSymbol const& cmd ) const
- {
- if(cmd.symModifier.size() == 0)
- return VSymbol(cmd.sym);
- else
- return VNode(cmd);
- }
- VOption OptionConverterVisitor::operator()( char const& cmd ) const
- {
- return TLiteral(cmd);
- }
- VOption OptionConverterVisitor::operator()( double const& cmd ) const
- {
- return TLiteral(cmd);
- }
- VOption OptionConverterVisitor::operator()(int const& cmd) const
- {
- return TLiteral(cmd);
- }
- VOption OptionConverterVisitor::operator()(ERM::Tcommand const& cmd) const
- {
- return cmd;
- }
- VOption OptionConverterVisitor::operator()( ERM::TStringConstant const& cmd ) const
- {
- return TLiteral(cmd.str);
- }
- VermTreeIterator VOptionList::cdr()
- {
- VermTreeIterator ret(*this);
- ret.basePos = 1;
- return ret;
- }
- VermTreeIterator VOptionList::car()
- {
- VermTreeIterator ret(*this);
- ret.state = VermTreeIterator::CAR;
- return ret;
- }
- }
|