polserver/pol-core/bscript/compiler/optimizer/BinaryOperatorWithStringOptimizer.cpp
turleypol c0bfd2f820
Escript Optimizer more types and operator (#794)
* compile time optimization:
int with doubles and strings
doubles with ints and strings
strings with ints and doubles

* bool with other types, more unary ops, float branch optimizer

* more tests
fixed bool to dbl compare

* output cleanup
more tests

* optimize string values in if statements, optimize ternary operator

* optimize elvis, addes missing files, code cleanup

* use array to keep unoptimized if branch in funcexpr tests

* missing include

* better readable testdata

* removed file

* optimize while and dowhile loops if predicate is a compile time known
value

* cleaner variant of loop optimization?

* added ConstantPredicateLoop Node used by the optimizer for constant loop
predicates
optimize repeat until loop
change tests to run the loops more then once to be sure they work
correctly

* test break/continue with label for constant-loop

* docs
2025-07-10 22:32:29 +02:00

101 lines
2.4 KiB
C++

#include "BinaryOperatorWithStringOptimizer.h"
#include "bscript/bobject.h"
#include "bscript/compiler/ast/BinaryOperator.h"
#include "bscript/compiler/ast/BooleanValue.h"
#include "bscript/compiler/ast/FloatValue.h"
#include "bscript/compiler/ast/IntegerValue.h"
#include "bscript/compiler/ast/StringValue.h"
namespace Pol::Bscript::Compiler
{
BinaryOperatorWithStringOptimizer::BinaryOperatorWithStringOptimizer( StringValue& lhs,
BinaryOperator& op )
: lhs( lhs ), op( op )
{
}
void BinaryOperatorWithStringOptimizer::visit_children( Node& ) {}
void BinaryOperatorWithStringOptimizer::visit_string_value( StringValue& rhs )
{
std::string value;
switch ( op.token_id )
{
case TOK_ADD:
{
value = lhs.value + rhs.value;
break;
}
default:
return;
}
optimized_result = std::make_unique<StringValue>( op.source_location, value );
}
void BinaryOperatorWithStringOptimizer::visit_float_value( FloatValue& rhs )
{
std::string value;
switch ( op.token_id )
{
case TOK_ADD:
{
value = fmt::format( "{}{}", lhs.value, Bscript::Double::double_to_string( rhs.value ) );
break;
}
default:
return;
}
optimized_result = std::make_unique<StringValue>( op.source_location, value );
}
void BinaryOperatorWithStringOptimizer::visit_integer_value( IntegerValue& rhs )
{
std::string value;
switch ( op.token_id )
{
case TOK_ADD:
{
value = fmt::format( "{}{}", lhs.value, rhs.value );
break;
}
default:
return;
}
optimized_result = std::make_unique<StringValue>( op.source_location, value );
}
void BinaryOperatorWithStringOptimizer::visit_boolean_value( BooleanValue& rhs )
{
auto setString = [&]( std::string&& val )
{ optimized_result = std::make_unique<StringValue>( op.source_location, val ); };
auto setInt = [&]( int val )
{ optimized_result = std::make_unique<IntegerValue>( op.source_location, val ); };
switch ( op.token_id )
{
case TOK_ADD:
setString( fmt::format( "{}{}", lhs.value, rhs.value ) );
break;
case TOK_EQUAL:
setInt( !lhs.value.empty() == rhs.value );
break;
case TOK_NEQ:
setInt( !lhs.value.empty() != rhs.value );
break;
case TOK_OR:
setInt( !lhs.value.empty() || rhs.value );
break;
case TOK_AND:
setInt( !lhs.value.empty() && rhs.value );
break;
default:
return;
}
}
} // namespace Pol::Bscript::Compiler