#pragma once

#include <cmath>

#include "nasal_new_ast.h"
#include "ast_visitor.h"

class optimizer:public ast_visitor {
private:
    void const_string(binary_operator*);
    void const_number(binary_operator*);

public:
    bool visit_binary_operator(binary_operator*) override;
    bool visit_unary_operator(unary_operator*) override;

public:
    void do_optimization(code_block*);
};

// 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;
//         case ast_bitor:  res=i32(vec[0].num())|i32(vec[1].num()); break;
//         case ast_bitxor: res=i32(vec[0].num())^i32(vec[1].num()); break;
//         case ast_bitand: res=i32(vec[0].num())&i32(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 && root.type()!=ast_bitor &&
//        root.type()!=ast_bitxor && root.type()!=ast_bitand) {
//         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);
//     }
// }