X-Git-Url: https://gitweb.ps.run/toc/blobdiff_plain/45409c781a9e35df68c43b1e2f028d30bf90c0a0..71a20a4f3d4e5f5278f7d004af710af89dfd7ebc:/src/repr.h

diff --git a/src/repr.h b/src/repr.h
index ce176ef..47a78d7 100644
--- a/src/repr.h
+++ b/src/repr.h
@@ -14,142 +14,253 @@ struct Body;
 struct Function;
 struct Struct;
 struct Program;
-struct CallExpr;
-struct LiteralExpr;
-struct VariableExpr;
+struct FuncExpr;
+struct LitExpr;
+struct IdentifierExpr;
+struct AccessExpr;
 struct BracketsExpr;
-struct OperatorExpr;
+struct UnaryOperatorExpr;
+struct BinaryOperatorExpr;
+struct TernaryOperatorExpr;
 struct DotExpr;
+struct ParenExpr;
 struct Expr;
 struct IfStmt;
+struct SwitchStmt;
+struct ForStmt;
 struct WhileStmt;
-struct ReturnStmt;
 struct AssignStmt;
+struct ReturnStmt;
 struct Stmt;
 
+enum class TypeModifierType
+{
+  Pointer, Array
+};
+
+struct TypeModifier
+{
+  TypeModifierType type;
 
-struct Type {
+  bool _staticArray;
+  int _arraySize;
+};
+
+struct Type
+{
   std::string name;
+  std::vector<TypeModifier> modifiers;
 };
 
-struct Variable {
+struct Variable
+{
   std::string name;
   Type type;
 };
 
-struct Body {
+struct Body
+{
   std::vector<Variable> variables;
   std::vector<Stmt> statements;
 };
 
-struct Function {
+struct Function
+{
+  Type returnType;
   std::string name;
   std::vector<Variable> parameters;
   Body body;
 };
 
-struct Struct {
+struct Struct
+{
   std::string name;
   std::vector<Variable> members;
   std::vector<Function> methods;
 };
 
-struct Program {
+struct Program
+{
   std::vector<Variable> variables;
   std::vector<Struct> structs;
   std::vector<Function> functions;
 };
 
-enum class ExprType {
-  Call, Literal, Variable, Brackets, Operator, Dot
+enum class ExprType
+{
+  Func, Lit, Identifier, Brackets, UnaryOperator, BinaryOperator, TernaryOperator, Dot
 };
 
-struct CallExpr {
-  Function function;
+struct FuncExpr
+{
+  std::string functionName;
   std::vector<Expr> arguments;
 };
 
-struct LiteralExpr {
-  int i;
+enum class LitType
+{
+  Int, Decimal, String, Bool
 };
 
-struct VariableExpr {
+struct LitExpr
+{
+  LitType type;
+
+  int _int;
+  double _decimal;
+  std::string _string;
+  bool _bool;
+};
+
+// TODO: accessExpr
+struct IdentifierExpr
+{
   std::string name;
 };
 
-struct BracketsExpr {
-  BracketsExpr() {}
-  BracketsExpr(const BracketsExpr &) {}
-  BracketsExpr & operator=(const BracketsExpr &) {return *this;};
-  std::unique_ptr<Expr> lexpr;
-  std::unique_ptr<Expr> rexpr;
+struct BracketsExpr
+{
+  std::shared_ptr<Expr> lexpr;
+  std::shared_ptr<Expr> rexpr;
 };
 
-enum class OperatorType {
-  Plus, Minus, Multiply, Divide,
-  Equals, NotEquals,
-  LessThan, GreaterThan
+enum class UnaryOperatorType
+{
+  Plus, Minus, IncrementPre, DecrementPre, IncrementPost, DecrementPost,
+  LogicalNot, BitwiseNot, Dereference, AddressOf,
+  COUNT
 };
 
-struct OperatorExpr {
-  OperatorExpr() {}
-  OperatorExpr(const OperatorExpr &) {}
-  OperatorExpr & operator=(const OperatorExpr &) {return *this;};
-  std::unique_ptr<Expr> lexpr;
-  std::unique_ptr<Expr> rexpr;
-  OperatorType type;
+enum class BinaryOperatorType
+{
+  Plus, Minus, Multiply, Divide, Modulo, BitwiseAnd, BitwiseOr, BitwiseXor, LessThan, GreaterThan,
+  LeftShift, RightShift, LogicalAnd, LogicalOr, Equals, NotEquals, LessThanEquals, GreaterThanEquals, BitwiseAndEquals, BitwiseOrEquals, BitwiseXorEquals,
+  PlusEquals, MinusEquals, MultiplyEquals, DivideEquals, ModuloEquals,
+  LeftShiftEquals, RightShiftEquals,
+  COUNT
 };
+static std::string UnaryOperatorTypeStrings[] =
+{
+  "+", "-", "++", "--", "++", "--", "!", "~", "*", "&" };
 
-struct DotExpr {
-  DotExpr() {}
-  DotExpr(const DotExpr &) {}
-  DotExpr & operator=(const DotExpr &) {return *this;};
-  std::unique_ptr<Expr> lexpr;
-  std::string name;
+static std::string BinaryOperatorTypeStrings[] =
+{
+  "+", "-", "*", "/", "%", "&", "|", "^", "<", ">",
+  "<<",">>","&&","||","==","!=","<=",">=","&=","|=","^=",
+  "+=","-=","*=","/=","%=",
+  "<<=",">>=" };
+
+struct UnaryOperatorExpr
+{
+  UnaryOperatorType type;
+  std::shared_ptr<Expr> expr;
+};
+
+struct BinaryOperatorExpr
+{
+  BinaryOperatorType type;
+  std::shared_ptr<Expr> lexpr;
+  std::shared_ptr<Expr> rexpr;
+};
+
+struct TernaryOperatorExpr
+{
+  std::shared_ptr<Expr> lexpr;
+  std::shared_ptr<Expr> rexprTrue;
+  std::shared_ptr<Expr> rexprFalse;
 };
 
-struct Expr {
+struct DotExpr
+{
+  std::shared_ptr<Expr> expr;
+  IdentifierExpr ident;
+};
+
+// TODO: paren expr
+struct Expr
+{
   ExprType type;
 
-  CallExpr     _call;
-  LiteralExpr  _literal;
-  VariableExpr _variable;
-  BracketsExpr _brackets;
-  OperatorExpr _operator;
-  DotExpr      _dot;
+  bool parenthesized;
+
+  FuncExpr            _func;
+  LitExpr             _lit;
+  IdentifierExpr      _identifier;
+  BracketsExpr        _brackets;
+  UnaryOperatorExpr   _unaryOperator;
+  BinaryOperatorExpr  _binaryOperator;
+  TernaryOperatorExpr _ternaryOperator;
+  DotExpr             _dot;
 };
 
-enum class StmtType {
-  If, While, Return, Assign, Expr
+enum class StmtType
+{
+  If, Switch, For, While, Assign, Return, Expr
 };
 
-struct IfStmt {
+struct ElseStmt
+{
+  bool _if;
+  std::shared_ptr<Expr> expr;
+  Body body;
+};
+struct IfStmt
+{
   Expr condition;
   Body body;
+  std::vector<ElseStmt> elses;
+};
+
+struct SwitchCase
+{
+  std::shared_ptr<Expr> expr;
+  Body body;
 };
 
-struct WhileStmt {
+struct SwitchStmt
+{
+  IdentifierExpr ident;
+  std::vector<SwitchCase> cases;
+};
+
+// TODO: int i = 0 (var decl)
+struct ForStmt
+{
+  std::string varName;
+  std::shared_ptr<Expr> initValue;
+  std::shared_ptr<Expr> condition;
+  std::shared_ptr<Expr> action;
+  Body body;
+};
+
+struct WhileStmt
+{
   Expr condition;
   Body body;
 };
 
-struct ReturnStmt {
+struct AssignStmt
+{
+  std::string name;
   Expr expr;
 };
 
-struct AssignStmt {
-  Expr lexpr;
-  Expr rexpr;
+struct ReturnStmt
+{
+  Expr expr;
 };
 
-struct Stmt {
+struct Stmt
+{
   StmtType type;
   
-  IfStmt     _if;
-  WhileStmt  _while;
-  ReturnStmt _return;
-  AssignStmt _assign;
-  Expr       _expr;
+  IfStmt      _if;
+  SwitchStmt  _switch;
+  ForStmt     _for;
+  WhileStmt   _while;
+  AssignStmt  _assign;
+  ReturnStmt  _return;
+  Expr        _expr;
 };