complex_handler.h

Go to the documentation of this file.

#ifndef PYTHON_FRONTEND_COMPLEX_HANDLER_H
#define PYTHON_FRONTEND_COMPLEX_HANDLER_H
 
#include <nlohmann/json.hpp>
#include <util/irep/expr.h>
#include <string>
#include <unordered_map>
 
class python_converter;
class contextt;
class type_handler;
class symbolt;

class complex_handler
{
public:
  complex_handler(
    python_converter &converter,
    contextt &symbol_table,
    type_handler &type_handler_ref);

  exprt handle_binary_op(
    const std::string &op,
    const exprt &lhs,
    const exprt &rhs,
    const nlohmann::json &element) const;

  exprt handle_unary_op(const std::string &op, const exprt &operand) const;

  exprt handle_attribute(const nlohmann::json &element) const;

  exprt
  handle_attribute_access(const exprt &obj, const std::string &attr) const;

  exprt handle_abs(const exprt &z) const;

  exprt handle_cmath_log(
    const std::string &func_name,
    const nlohmann::json &call,
    const nlohmann::json &args,
    const nlohmann::json &keywords) const;

  exprt normalize_numeric_expr(const exprt &value) const;
 
private:
  python_converter &converter_;
  contextt &symbol_table_;
  type_handler &type_handler_;

  mutable std::unordered_map<std::string, const symbolt *> symbol_cache_;

  void clear_cache() const;

  const symbolt *find_cached_symbol(const std::string &id) const;
 
  // ---- shared IEEE / complex arithmetic helpers ----

  exprt complex_mul(const exprt &x, const exprt &y) const;

  exprt complex_div(
    const exprt &x,
    const exprt &y,
    const nlohmann::json &loc_source) const;

  exprt complex_log(const exprt &z, const nlohmann::json &loc_source) const;

  exprt complex_exp(const exprt &z, const nlohmann::json &loc_source) const;

  exprt
  promote_int_arith_to_double(const exprt &input_expr, std::size_t depth) const;
};
 
#endif // PYTHON_FRONTEND_COMPLEX_HANDLER_H