code_generation.h

Go to the documentation of this file.

 
#ifndef __GINAC_CODE_GENERATION_H__
#define __GINAC_CODE_GENERATION_H__
 
#include <iostream>
#include <string>
 
#include "ginac/ex.h"
#include "ginac/lst.h"
 
namespace GiNaC {
 
  class subex_type {
  public:
    enum {
      add        = 0x00000001,
      mul        = 0x00000002,
      power      = 0x00000004,
      function   = 0x00000008,
      all        = 0xFFFFFFFF
    };
  };
 
  class assign_lst {
 
  public:
    assign_lst(const std::string & T, const std::string & temp_var_base_name);
    assign_lst(const std::string & T, const std::string & temp_var_base_name, const lst & lhs, const lst & rhs);
 
  public:
    void append(const ex & a, const ex & b);
    void prepend(const ex & a, const ex & b);
    void replace_common_subex(unsigned flag_type = subex_type::all);
    void split_large_subex(size_t nmax);
    void print(std::ostream & os) const;
    ex get_temporary_variables() const;
 
  private:
    std::string itos(int arg) const;
    bool check_substitution_type(const ex & expr, unsigned flag_type) const;
    bool check_large_subex(const ex & expr, size_t max_nops, lst & subexpr_lst) const;
    void split_large_line(const ex & a, const ex & b, lst & new_lhs_lst, lst & new_rhs_lst, size_t max_nops);
 
  public:
    // I/O operators
    friend std::ostream & operator<< (std::ostream & os, const assign_lst & arg);
 
 
  private:
    ex lhs; 
    ex rhs;
 
    std::string T; 
    std::string temp_var_base_name;
 
    std::vector<symbol> temp_var;
  };
 
  std::ostream & operator<< (std::ostream & os, const assign_lst & arg);
 
} // namespace GiNaC
 
#endif // ndef __GINAC_CODE_GENERATION_H__