69 lines
1.9 KiB
C++
69 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include <cmath>
|
|
|
|
void const_str(ast& root)
|
|
{
|
|
auto& vec=root.child();
|
|
root.set_str(vec[0].str()+vec[1].str());
|
|
root.child().clear();
|
|
root.set_type(ast_str);
|
|
}
|
|
|
|
void const_num(ast& root)
|
|
{
|
|
auto& vec=root.child();
|
|
f64 res=0;
|
|
switch(root.type())
|
|
{
|
|
case ast_add: res=vec[0].num()+vec[1].num(); break;
|
|
case ast_sub: res=vec[0].num()-vec[1].num(); break;
|
|
case ast_mult:res=vec[0].num()*vec[1].num(); break;
|
|
case ast_div: res=vec[0].num()/vec[1].num(); break;
|
|
case ast_less:res=vec[0].num()<vec[1].num(); break;
|
|
case ast_leq: res=vec[0].num()<=vec[1].num();break;
|
|
case ast_grt: res=vec[0].num()>vec[1].num(); break;
|
|
case ast_geq: res=vec[0].num()>=vec[1].num();break;
|
|
}
|
|
// inf and nan will cause number hashmap error in codegen
|
|
if(std::isinf(res) || std::isnan(res))
|
|
return;
|
|
root.set_num(res);
|
|
root.child().clear();
|
|
root.set_type(ast_num);
|
|
}
|
|
|
|
void calc_const(ast& root)
|
|
{
|
|
auto& vec=root.child();
|
|
for(auto& i:vec)
|
|
calc_const(i);
|
|
if(vec.size()==1 && root.type()==ast_neg && vec[0].type()==ast_num)
|
|
{
|
|
f64 res=-vec[0].num();
|
|
root.set_num(res);
|
|
root.child().clear();
|
|
root.set_type(ast_num);
|
|
return;
|
|
}
|
|
if(vec.size()!=2)
|
|
return;
|
|
if(root.type()!=ast_add && root.type()!=ast_sub &&
|
|
root.type()!=ast_mult && root.type()!=ast_div &&
|
|
root.type()!=ast_link && root.type()!=ast_less &&
|
|
root.type()!=ast_leq && root.type()!=ast_grt &&
|
|
root.type()!=ast_geq)
|
|
return;
|
|
if(root.type()==ast_link &&
|
|
vec[0].type()==ast_str && vec[1].type()==ast_str)
|
|
const_str(root);
|
|
else if(root.type()!=ast_link &&
|
|
vec[0].type()==ast_num && vec[1].type()==ast_num)
|
|
const_num(root);
|
|
}
|
|
void optimize(ast& root)
|
|
{
|
|
for(auto& i:root.child())
|
|
calc_const(i);
|
|
}
|