polserver/pol-core/bscript/compiler/codegen/InstructionEmitter.h

#ifndef POLSERVER_INSTRUCTIONEMITTER_H
#define POLSERVER_INSTRUCTIONEMITTER_H

#include <map>
#include <memory>
#include <string>
#include <vector>

#ifndef OBJMEMBERS_H
#include "bscript/objmembers.h"
#endif
#ifndef OBJMETHODS_H
#include "bscript/objmethods.h"
#endif
#ifndef __TOKENS_H
#include "bscript/tokens.h"
#endif
#include "bscript/compiler/codegen/CodeEmitter.h"
#include "bscript/compiler/codegen/DataEmitter.h"
#include "bscript/compiler/model/SimpleTypes.h"
#include "bscript/compiler/representation/CompiledScript.h"
#include "bscript/compiler/representation/DebugStore.h"

namespace Pol::Bscript
{
class StoredToken;
}

namespace Pol::Bscript::Compiler
{
class CaseJumpDataBlock;
class CompiledScript;
class FlowControlLabel;
class LocalVariableScopeInfo;
class ModuleDeclarationRegistrar;
class ClassDeclaration;
class ClassDeclarationRegistrar;
class ModuleFunctionDeclaration;
class FunctionReferenceRegistrar;
class UserFunction;
class Node;
class Variable;
class SourceLocation;
class Report;

class InstructionEmitter
{
public:
  InstructionEmitter( CodeSection& code, DataSection& data, DebugStore& debug,
                      ExportedFunctions& exported_functions, ModuleDeclarationRegistrar&,
                      FunctionReferenceRegistrar&, ClassDeclarationRegistrar&, Report& );

  void initialize_data();

  void register_exported_function( FlowControlLabel& label, const std::string& name,
                                   unsigned arguments );
  void register_class_declaration( const ClassDeclaration&,
                                   std::map<std::string, FlowControlLabel>& );

  unsigned enter_debug_block( const LocalVariableScopeInfo& );
  void set_debug_block( unsigned );

  void access_variable( const Variable&, VariableIndex function_params_count,
                        VariableIndex function_capture_count );
  void array_append();
  void array_create();
  void array_declare();
  void assign();
  void assign_subscript_consume();
  void assign_subscript();
  void assign_multisubscript( unsigned indexes );
  void assign_variable( const Variable&, VariableIndex function_params_count,
                        VariableIndex function_capture_count );
  void basic_for_init( FlowControlLabel& );
  void basic_for_next( FlowControlLabel& );
  void binary_operator( BTokenId token_id );
  void check_mro( unsigned offset );
  void call_method( const std::string& name, unsigned argument_count );
  void call_method_id( MethodID method_id, unsigned argument_count );
  void call_modulefunc( const ModuleFunctionDeclaration& );
  void call_userfunc( FlowControlLabel& );
  void classinst_create( unsigned index );
  unsigned casejmp();
  unsigned case_dispatch_table( const CaseJumpDataBlock& );
  void consume();
  void ctrl_statementbegin( unsigned file_index, unsigned file_offset,
                            const std::string& source_text );

  // If `take`, the variable's value will be taken/moved from the top of ValueStack
  void declare_variable( const Variable&, VariableIndex function_capture_count, bool take );
  void dictionary_create();
  void dictionary_add_member();
  void error_create();
  void exit();
  void foreach_init( FlowControlLabel& );
  void foreach_step( FlowControlLabel& );
  void function_reference( const UserFunction&, FlowControlLabel& );
  void functor_create( const UserFunction&, FlowControlLabel& );
  void get_arg( const std::string& name );
  void get_member( const std::string& name );
  void get_member_id( MemberID );
  void jmp_always( FlowControlLabel& );
  void jmp_if_false( FlowControlLabel& );
  void jmp_if_true( FlowControlLabel& );
  void label( FlowControlLabel& );
  void leaveblock( unsigned local_vars_to_remove );
  void makelocal();
  void pop_param( const std::string& name );
  void pop_param_byref( const std::string& name );
  void progend();
  void return_from_user_function();
  // Emit the 'this' variable, always stored at local `this_offset`.
  void return_from_constructor_function( unsigned int this_offset );
  unsigned skip_if_true_else_consume();
  void set_member( const std::string& name );
  void set_member_id( MemberID member_id );
  void set_member_id_consume( MemberID member_id );
  void set_member_consume( const std::string& name );
  void set_member_by_operator( BTokenId, MemberID );
  void spread( bool spread_into );
  void struct_create();
  void struct_add_uninit_member( const std::string& name );
  void struct_add_member( const std::string& name );
  void subscript_single();
  void subscript_multiple( unsigned indexes );
  void unary_operator( BTokenId );
  void uninit();
  void unpack_sequence( unsigned count, unsigned rest_at );
  void unpack_indices( unsigned count, unsigned rest_at );
  void value( double );
  void value( int );
  void value( bool );
  void value( const std::string& );
  void regular_expression_value( const std::string& pattern, const std::string& flags );
  void interpolate_string( unsigned count );
  void format_expression();

  void debug_file_line( unsigned file_index, unsigned line_number );
  void debug_statementbegin();
  unsigned next_instruction_address();
  void debug_user_function( const std::string& name, unsigned first_address,
                            unsigned last_address );

  void patch_offset( unsigned index, unsigned offset );

  bool has_function_reference( const UserFunction& );

  void logical_jmp( FlowControlLabel& label, bool if_true );
  void logical_convert();

private:
  unsigned emit_data( const std::string& );
  unsigned emit_token( BTokenId id, BTokenType type, unsigned offset = 0 );
  unsigned append_token( StoredToken& );
  void register_with_label( FlowControlLabel&, unsigned offset );

  CodeEmitter code_emitter;
  DataEmitter data_emitter;
  DebugStore& debug;
  ExportedFunctions& exported_functions;
  ModuleDeclarationRegistrar& module_declaration_registrar;
  FunctionReferenceRegistrar& function_reference_registrar;
  ClassDeclarationRegistrar& class_declaration_registrar;

  DebugStore::InstructionInfo debug_instruction_info{};
  Report& report;
};
}  // namespace Pol::Bscript::Compiler

#endif  // POLSERVER_INSTRUCTIONEMITTER_H