SymEngine::RebuildVisitor Class Reference

Inheritance diagram for SymEngine::RebuildVisitor:

Collaboration diagram for SymEngine::RebuildVisitor:

Public Member Functions
	RebuildVisitor (umap_basic_basic &subs_, umap_basic_basic &opt_subs_, set_basic &to_eliminate_, set_basic &excluded_symbols_, vec_pair &replacements_)
RCP< const Basic >	apply (const RCP< const Basic > &orig_expr) override
RCP< const Basic >	next_symbol ()
void	bvisit (const FunctionSymbol &x)
void	bvisit (const Basic &x)
void	bvisit (const Add &x)
void	bvisit (const Mul &x)
void	bvisit (const Pow &x)
void	bvisit (const OneArgFunction &x)
template<class T >
void	bvisit (const TwoArgBasic< T > &x)
void	bvisit (const MultiArgFunction &x)
void	bvisit (const Piecewise &x)
Public Member Functions inherited from SymEngine::TransformVisitor
virtual RCP< const Basic >	apply (const RCP< const Basic > &x)
void	bvisit (const Basic &x)
void	bvisit (const Add &x)
void	bvisit (const Mul &x)
void	bvisit (const Pow &x)
void	bvisit (const OneArgFunction &x)
template<class T >
void	bvisit (const TwoArgBasic< T > &x)
void	bvisit (const MultiArgFunction &x)
void	bvisit (const Piecewise &x)

Data Fields
RCP< const Basic >	result_

Private Attributes
umap_basic_basic &	subs
umap_basic_basic &	opt_subs
set_basic &	to_eliminate
set_basic &	excluded_symbols
vec_pair &	replacements
unsigned	next_symbol_index = 0

Additional Inherited Members
Protected Attributes inherited from SymEngine::TransformVisitor
RCP< const Basic >	result_

Detailed Description

Definition at line 420 of file cse.cpp.

Constructor & Destructor Documentation

RebuildVisitor()

SymEngine::RebuildVisitor::RebuildVisitor ( umap_basic_basic &  subs_, umap_basic_basic &  opt_subs_, set_basic &  to_eliminate_, set_basic &  excluded_symbols_, vec_pair &  replacements_  )

inline

Definition at line 433 of file cse.cpp.

        : subs(subs_), opt_subs(opt_subs_), to_eliminate(to_eliminate_),
          excluded_symbols(excluded_symbols_), replacements(replacements_)
    {
    }

Member Function Documentation

apply()

RCP< const Basic > SymEngine::RebuildVisitor::apply ( const RCP< const Basic > &  orig_expr )

inlineoverridevirtual

Reimplemented from SymEngine::TransformVisitor.

Definition at line 440 of file cse.cpp.

    {
        RCP<const Basic> expr = orig_expr;
        if (is_a_Atom(*expr)) {
            return expr;
        }
 
        auto iter = subs.find(expr);
        if (iter != subs.end()) {
            return iter->second;
        }
        auto iter2 = opt_subs.find(expr);
        if (iter2 != opt_subs.end()) {
            expr = iter2->second;
        }
        expr->accept(*this);
        auto new_expr = result_;
        if (to_eliminate.find(orig_expr) != to_eliminate.end()) {
            auto sym = next_symbol();
            subs[orig_expr] = sym;
            replacements.push_back(
                std::pair<RCP<const Basic>, RCP<const Basic>>(sym, new_expr));
            return sym;
        }
        return new_expr;
    }

bvisit() [1/9]

void SymEngine::TransformVisitor::bvisit

(

const Add &

x

)

Definition at line 247 of file visitor.cpp.

{
    vec_basic newargs;
    for (const auto &a : x.get_args()) {
        newargs.push_back(apply(a));
    }
    result_ = add(newargs);
}

bvisit() [2/9]

void SymEngine::TransformVisitor::bvisit

(

const Basic &

x

)

Definition at line 246 of file visitor.cpp.

{
    result_ = x.rcp_from_this();
}

bvisit() [3/9]

void SymEngine::RebuildVisitor::bvisit ( const FunctionSymbol &  x )

inline

Definition at line 476 of file cse.cpp.

    {
        auto &fargs = x.get_vec();
        vec_basic newargs;
        for (const auto &a : fargs) {
            newargs.push_back(apply(a));
        }
        if (x.get_name() == "add") {
            result_ = add(newargs);
        } else if (x.get_name() == "mul") {
            result_ = mul(newargs);
        } else if (x.get_name() == "pow") {
            result_ = pow(newargs[0], newargs[1]);
        } else {
            result_ = x.create(newargs);
        }
    }

bvisit() [4/9]

void SymEngine::TransformVisitor::bvisit

(

const Mul &

x

)

Definition at line 248 of file visitor.cpp.

{
    vec_basic newargs;
    for (const auto &a : x.get_args()) {
        newargs.push_back(apply(a));
    }
    result_ = mul(newargs);
}

bvisit() [5/9]

void SymEngine::TransformVisitor::bvisit

(

const MultiArgFunction &

x

)

Definition at line 264 of file visitor.cpp.

{
    auto fargs = x.get_args();
    vec_basic newargs;
    for (const auto &a : fargs) {
        newargs.push_back(apply(a));
    }
    auto nbarg = x.create(newargs);
    result_ = nbarg;
}

bvisit() [6/9]

void SymEngine::TransformVisitor::bvisit

(

const OneArgFunction &

x

)

Definition at line 250 of file visitor.cpp.

{
    auto farg = x.get_arg();
    auto newarg = apply(farg);
    if (eq(*newarg, *farg)) {
        result_ = x.rcp_from_this();
    } else {
        result_ = x.create(newarg);
    }
}

bvisit() [7/9]

void SymEngine::TransformVisitor::bvisit

(

const Piecewise &

x

)

Definition at line 265 of file visitor.cpp.

{
    auto branch_cond_pairs = x.get_vec();
    PiecewiseVec new_pairs;
    for (const auto &branch_cond : branch_cond_pairs) {
        auto branch = branch_cond.first;
        auto cond = branch_cond.second;
        auto new_branch = apply(branch);
        auto new_cond = apply(cond);
        if (!is_a_Boolean(*new_cond)) {
            new_cond = Eq(new_cond, boolTrue);
        }
        new_pairs.push_back(
            {new_branch, rcp_static_cast<const Boolean>(new_cond)});
    }
    result_ = piecewise(new_pairs);
}

bvisit() [8/9]

void SymEngine::TransformVisitor::bvisit

(

const Pow &

x

)

Definition at line 249 of file visitor.cpp.

{
    auto base_ = x.get_base(), exp_ = x.get_exp();
    auto newarg1 = apply(base_), newarg2 = apply(exp_);
    if (base_ != newarg1 or exp_ != newarg2) {
        result_ = pow(newarg1, newarg2);
    } else {
        result_ = x.rcp_from_this();
    }
}

bvisit() [9/9]

template<class T >

void SymEngine::TransformVisitor::bvisit ( const TwoArgBasic< T > &  x )

inline

Definition at line 253 of file visitor.h.

    {
        auto farg1 = x.get_arg1(), farg2 = x.get_arg2();
        auto newarg1 = apply(farg1), newarg2 = apply(farg2);
        if (farg1 != newarg1 or farg2 != newarg2) {
            result_ = x.create(newarg1, newarg2);
        } else {
            result_ = x.rcp_from_this();
        }
    }

next_symbol()

RCP< const Basic > SymEngine::RebuildVisitor::next_symbol ( )

inline

Definition at line 466 of file cse.cpp.

    {
        RCP<const Basic> sym = symbol("x" + to_string(next_symbol_index));
        next_symbol_index++;
        if (excluded_symbols.find(sym) == excluded_symbols.end()) {
            return sym;
        } else {
            return next_symbol();
        }
    };

Field Documentation

excluded_symbols

set_basic& SymEngine::RebuildVisitor::excluded_symbols

private

Definition at line 426 of file cse.cpp.

next_symbol_index

unsigned SymEngine::RebuildVisitor::next_symbol_index = 0

private

Definition at line 428 of file cse.cpp.

opt_subs

umap_basic_basic& SymEngine::RebuildVisitor::opt_subs

private

Definition at line 424 of file cse.cpp.

replacements

vec_pair& SymEngine::RebuildVisitor::replacements

private

Definition at line 427 of file cse.cpp.

result_

RCP<const Basic> SymEngine::TransformVisitor::result_

Definition at line 239 of file visitor.h.

subs

umap_basic_basic& SymEngine::RebuildVisitor::subs

private

Definition at line 423 of file cse.cpp.

to_eliminate

set_basic& SymEngine::RebuildVisitor::to_eliminate

private

Definition at line 425 of file cse.cpp.

---

The documentation for this class was generated from the following file:

• /home/runner/work/symengine/symengine/symengine/cse.cpp