mirror of
https://github.com/polserver/polserver
synced 2026-08-13 08:23:08 -04:00
1463b76c7c
* all except pol * pol and includes under defines * something weird has happened * remove own folder from include searchpath * fixed remaining, adapted include style for external headers * missing include
136 lines
3.2 KiB
C++
136 lines
3.2 KiB
C++
#include "bscript/compiler/optimizer/UnaryOperatorOptimizer.h"
|
|
|
|
#include "bscript/compiler/ast/BooleanValue.h"
|
|
#include "bscript/compiler/ast/FloatValue.h"
|
|
#include "bscript/compiler/ast/IntegerValue.h"
|
|
#include "bscript/compiler/ast/MemberAccess.h"
|
|
#include "bscript/compiler/ast/MemberAssignmentByOperator.h"
|
|
#include "bscript/compiler/ast/UnaryOperator.h"
|
|
#include "bscript/compiler/ast/UninitializedValue.h"
|
|
|
|
namespace Pol::Bscript::Compiler
|
|
{
|
|
UnaryOperatorOptimizer::UnaryOperatorOptimizer( UnaryOperator& unary_operator )
|
|
: unary_operator( unary_operator )
|
|
{
|
|
}
|
|
|
|
std::unique_ptr<Expression> UnaryOperatorOptimizer::optimize()
|
|
{
|
|
unary_operator.operand().accept( *this );
|
|
return std::move( optimized_result );
|
|
}
|
|
|
|
void UnaryOperatorOptimizer::visit_children( Node& ) {}
|
|
|
|
void UnaryOperatorOptimizer::visit_boolean_value( BooleanValue& bv )
|
|
{
|
|
bool value;
|
|
|
|
switch ( unary_operator.token_id )
|
|
{
|
|
case TOK_LOG_NOT:
|
|
value = !bv.value;
|
|
break;
|
|
|
|
default:
|
|
return;
|
|
}
|
|
|
|
// Logical-not returns 1/0 as BLong, ie. `!false` == `1`
|
|
optimized_result = std::make_unique<IntegerValue>( bv.source_location, value );
|
|
}
|
|
|
|
void UnaryOperatorOptimizer::visit_float_value( FloatValue& fv )
|
|
{
|
|
double value;
|
|
|
|
switch ( unary_operator.token_id )
|
|
{
|
|
case TOK_LOG_NOT:
|
|
value = fv.value == 0.0;
|
|
break;
|
|
case TOK_UNMINUS:
|
|
value = -fv.value;
|
|
break;
|
|
case TOK_UNPLUSPLUS:
|
|
value = fv.value + 1;
|
|
break;
|
|
case TOK_UNMINUSMINUS:
|
|
value = fv.value - 1;
|
|
break;
|
|
|
|
default:
|
|
return;
|
|
}
|
|
|
|
optimized_result = std::make_unique<FloatValue>( fv.source_location, value );
|
|
}
|
|
|
|
void UnaryOperatorOptimizer::visit_integer_value( IntegerValue& iv )
|
|
{
|
|
int value;
|
|
|
|
switch ( unary_operator.token_id )
|
|
{
|
|
case TOK_UNMINUS:
|
|
value = -iv.value;
|
|
break;
|
|
case TOK_LOG_NOT:
|
|
value = !iv.value;
|
|
break;
|
|
case TOK_BITWISE_NOT:
|
|
value = static_cast<int>( ~static_cast<unsigned>( iv.value ) );
|
|
break;
|
|
case TOK_UNPLUSPLUS:
|
|
value = iv.value + 1;
|
|
break;
|
|
case TOK_UNMINUSMINUS:
|
|
value = iv.value - 1;
|
|
break;
|
|
|
|
default:
|
|
return;
|
|
}
|
|
|
|
optimized_result = std::make_unique<IntegerValue>( iv.source_location, value );
|
|
}
|
|
|
|
void UnaryOperatorOptimizer::visit_member_access( MemberAccess& gm )
|
|
{
|
|
if ( !gm.known_member )
|
|
return;
|
|
|
|
BTokenId new_token_id;
|
|
switch ( unary_operator.token_id )
|
|
{
|
|
case TOK_UNPLUSPLUS:
|
|
new_token_id = INS_SET_MEMBER_ID_UNPLUSPLUS;
|
|
break;
|
|
case TOK_UNMINUSMINUS:
|
|
new_token_id = INS_SET_MEMBER_ID_UNMINUSMINUS;
|
|
break;
|
|
case TOK_UNPLUSPLUS_POST:
|
|
new_token_id = INS_SET_MEMBER_ID_UNPLUSPLUS_POST;
|
|
break;
|
|
case TOK_UNMINUSMINUS_POST:
|
|
new_token_id = INS_SET_MEMBER_ID_UNMINUSMINUS_POST;
|
|
break;
|
|
default:
|
|
return;
|
|
}
|
|
|
|
bool consume = false;
|
|
auto entity = gm.take_entity();
|
|
optimized_result = std::make_unique<MemberAssignmentByOperator>(
|
|
gm.source_location, consume, std::move( entity ), gm.name, new_token_id, *gm.known_member );
|
|
}
|
|
|
|
void UnaryOperatorOptimizer::visit_uninitialized_value( UninitializedValue& uninit )
|
|
{
|
|
if ( unary_operator.token_id == TOK_LOG_NOT )
|
|
{
|
|
optimized_result = std::make_unique<IntegerValue>( uninit.source_location, 1 );
|
|
}
|
|
}
|
|
} // namespace Pol::Bscript::Compiler
|