polserver/pol-core/bscript/compiler/codegen/InstructionGenerator.h
Kevin Eady 0ca7912fcd
Add support for conditional operator (#387)
* Grammar update

* AST generation

* Emit instructions

* Add initial tests

* Additional tests

* Rename to ConditionalOperator

* Add docs
2021-03-14 23:20:20 +01:00

89 lines
3.9 KiB
C++

#ifndef POLSERVER_INSTRUCTIONGENERATOR_H
#define POLSERVER_INSTRUCTIONGENERATOR_H
#include "bscript/compiler/ast/NodeVisitor.h"
#include <map>
#include <string>
namespace Pol::Bscript::Compiler
{
class FlowControlLabel;
class InstructionEmitter;
class SourceLocation;
class InstructionGenerator : public NodeVisitor
{
public:
InstructionGenerator( InstructionEmitter&,
std::map<std::string, FlowControlLabel>& user_function_labels,
bool in_function);
void generate( Node& );
void update_debug_location( const Node& );
void update_debug_location( const SourceLocation& );
void visit_array_initializer( ArrayInitializer& ) override;
void visit_basic_for_loop( BasicForLoop& loop ) override;
void visit_case_statement( CaseStatement& ) override;
void visit_cstyle_for_loop( CstyleForLoop& loop ) override;
void visit_binary_operator( BinaryOperator& ) override;
void visit_block( Block& ) override;
void visit_branch_selector( BranchSelector& ) override;
void visit_debug_statement_marker( DebugStatementMarker& ) override;
void visit_dictionary_entry( DictionaryEntry& ) override;
void visit_dictionary_initializer( DictionaryInitializer& ) override;
void visit_do_while_loop( DoWhileLoop& ) override;
void visit_element_access( ElementAccess& ) override;
void visit_element_assignment( ElementAssignment& ) override;
void visit_elvis_operator( ElvisOperator& elvis ) override;
void visit_error_initializer( ErrorInitializer& ) override;
void visit_exit_statement( ExitStatement& ) override;
void visit_float_value( FloatValue& ) override;
void visit_foreach_loop( ForeachLoop& ) override;
void visit_function_call( FunctionCall& ) override;
void visit_function_parameter_list( FunctionParameterList& ) override;
void visit_function_parameter_declaration( FunctionParameterDeclaration& ) override;
void visit_function_reference( FunctionReference& ) override;
void visit_identifier( Identifier& ) override;
void visit_if_then_else_statement( IfThenElseStatement& ) override;
void visit_integer_value( IntegerValue& ) override;
void visit_jump_statement( JumpStatement& ) override;
void visit_member_access( MemberAccess& member_access ) override;
void visit_member_assignment( MemberAssignment& ) override;
void visit_member_assignment_by_operator( MemberAssignmentByOperator& ) override;
void visit_method_call( MethodCall& member_call ) override;
void visit_program( Program& ) override;
void visit_program_parameter_declaration( ProgramParameterDeclaration& ) override;
void visit_repeat_until_loop( RepeatUntilLoop& repeat_until ) override;
void visit_return_statement( ReturnStatement& ) override;
void visit_string_value( StringValue& ) override;
void visit_struct_initializer( StructInitializer& ) override;
void visit_struct_member_initializer( StructMemberInitializer& ) override;
void visit_unary_operator( UnaryOperator& ) override;
void visit_uninitialized_value( UninitializedValue& ) override;
void visit_user_function( UserFunction& ) override;
void visit_value_consumer( ValueConsumer& ) override;
void visit_var_statement( VarStatement& ) override;
void visit_variable_assignment_statement( VariableAssignmentStatement& ) override;
void visit_while_loop( WhileLoop& ) override;
void visit_interpolate_string( InterpolateString& ) override;
void visit_format_expression( FormatExpression& ) override;
void visit_conditional_operator( ConditionalOperator& ) override;
private:
// There are two of these because sometimes when calling a method
// on InstructionEmitter, the variable name reads better as a noun,
// and sometimes it reads better as a verb.
InstructionEmitter& emitter;
InstructionEmitter& emit;
std::map<std::string, FlowControlLabel>& user_function_labels;
const bool in_function;
};
} // namespace Pol::Bscript::Compiler
#endif // POLSERVER_INSTRUCTIONGENERATOR_H