Loading...
Searching...
No Matches
SymEngine::SimplifyVisitor Class Reference
+ Inheritance diagram for SymEngine::SimplifyVisitor:
+ Collaboration diagram for SymEngine::SimplifyVisitor:

Public Member Functions

void bvisit (const Mul &x)
 
void bvisit (const Pow &x)
 
void bvisit (const OneArgFunction &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 Basicapply (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)
 

Private Member Functions

std::pair< RCP< const Basic >, RCP< const Basic > > simplify_pow (const RCP< const Basic > &e, const RCP< const Basic > &b)
 

Additional Inherited Members

- Protected Attributes inherited from SymEngine::TransformVisitor
RCP< const Basicresult_
 

Detailed Description

Definition at line 11 of file simplify.h.

Constructor & Destructor Documentation

◆ SimplifyVisitor()

SymEngine::SimplifyVisitor::SimplifyVisitor ( )
inline

Definition at line 20 of file simplify.h.

20: BaseVisitor<SimplifyVisitor, TransformVisitor>() {}

Member Function Documentation

◆ bvisit() [1/11]

void SymEngine::TransformVisitor::bvisit ( const Add x)

Definition at line 247 of file visitor.cpp.

154{
155 vec_basic newargs;
156 for (const auto &a : x.get_args()) {
157 newargs.push_back(apply(a));
158 }
159 result_ = add(newargs);
160}
RCP< const Basic > add(const RCP< const Basic > &a, const RCP< const Basic > &b)
Adds two objects (safely).
Definition: add.cpp:425
T push_back(T... args)

◆ bvisit() [2/11]

void SymEngine::TransformVisitor::bvisit ( const Basic x)

Definition at line 246 of file visitor.cpp.

149{
150 result_ = x.rcp_from_this();
151}

◆ bvisit() [3/11]

void SymEngine::TransformVisitor::bvisit ( const Mul x)

Definition at line 248 of file visitor.cpp.

163{
164 vec_basic newargs;
165 for (const auto &a : x.get_args()) {
166 newargs.push_back(apply(a));
167 }
168 result_ = mul(newargs);
169}
RCP< const Basic > mul(const RCP< const Basic > &a, const RCP< const Basic > &b)
Multiplication.
Definition: mul.cpp:352

◆ bvisit() [4/11]

void SymEngine::SimplifyVisitor::bvisit ( const Mul x)

Definition at line 43 of file simplify.cpp.

44{
45 map_basic_basic map;
46 for (const auto &p : x.get_dict()) {
47 auto base = apply(p.first);
48 auto newpair = simplify_pow(p.second, base);
49 Mul::dict_add_term(map, newpair.first, newpair.second);
50 }
51 result_ = Mul::from_dict(x.get_coef(), std::move(map));
52}
static void dict_add_term(map_basic_basic &d, const RCP< const Basic > &exp, const RCP< const Basic > &t)
Add terms to dict.
Definition: mul.cpp:150
static RCP< const Basic > from_dict(const RCP< const Number > &coef, map_basic_basic &&d)
Create a Mul from a dict.
Definition: mul.cpp:115
T move(T... args)

◆ bvisit() [5/11]

void SymEngine::TransformVisitor::bvisit ( const MultiArgFunction x)

Definition at line 264 of file visitor.cpp.

194{
195 auto fargs = x.get_args();
196 vec_basic newargs;
197 for (const auto &a : fargs) {
198 newargs.push_back(apply(a));
199 }
200 auto nbarg = x.create(newargs);
201 result_ = nbarg;
202}

◆ bvisit() [6/11]

void SymEngine::TransformVisitor::bvisit ( const OneArgFunction x)

Definition at line 250 of file visitor.cpp.

183{
184 auto farg = x.get_arg();
185 auto newarg = apply(farg);
186 if (eq(*newarg, *farg)) {
187 result_ = x.rcp_from_this();
188 } else {
189 result_ = x.create(newarg);
190 }
191}
bool eq(const Basic &a, const Basic &b)
Checks equality for a and b
Definition: basic-inl.h:21

◆ bvisit() [7/11]

void SymEngine::SimplifyVisitor::bvisit ( const OneArgFunction x)

Definition at line 7 of file simplify.cpp.

8{
9 auto farg = x.get_arg();
10 auto newarg = apply(farg);
11 result_ = x.create(newarg);
12}

◆ bvisit() [8/11]

void SymEngine::TransformVisitor::bvisit ( const Piecewise x)

Definition at line 265 of file visitor.cpp.

205{
206 auto branch_cond_pairs = x.get_vec();
207 PiecewiseVec new_pairs;
208 for (const auto &branch_cond : branch_cond_pairs) {
209 auto branch = branch_cond.first;
210 auto cond = branch_cond.second;
211 auto new_branch = apply(branch);
212 auto new_cond = apply(cond);
213 if (!is_a_Boolean(*new_cond)) {
214 new_cond = Eq(new_cond, boolTrue);
215 }
216 new_pairs.push_back(
217 {new_branch, rcp_static_cast<const Boolean>(new_cond)});
218 }
219 result_ = piecewise(new_pairs);
220}
RCP< const Boolean > Eq(const RCP< const Basic > &lhs)
Returns the canonicalized Equality object from a single argument.
Definition: logic.cpp:653

◆ bvisit() [9/11]

void SymEngine::TransformVisitor::bvisit ( const Pow x)

Definition at line 249 of file visitor.cpp.

172{
173 auto base_ = x.get_base(), exp_ = x.get_exp();
174 auto newarg1 = apply(base_), newarg2 = apply(exp_);
175 if (base_ != newarg1 or exp_ != newarg2) {
176 result_ = pow(newarg1, newarg2);
177 } else {
178 result_ = x.rcp_from_this();
179 }
180}
T pow(T... args)

◆ bvisit() [10/11]

void SymEngine::SimplifyVisitor::bvisit ( const Pow x)

Definition at line 14 of file simplify.cpp.

15{
16 auto e = apply(x.get_exp());
17 auto base = apply(x.get_base());
18 auto pair = simplify_pow(e, base);
19 result_ = pow(pair.second, pair.first);
20}

◆ bvisit() [11/11]

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

Definition at line 253 of file visitor.h.

254 {
255 auto farg1 = x.get_arg1(), farg2 = x.get_arg2();
256 auto newarg1 = apply(farg1), newarg2 = apply(farg2);
257 if (farg1 != newarg1 or farg2 != newarg2) {
258 result_ = x.create(newarg1, newarg2);
259 } else {
260 result_ = x.rcp_from_this();
261 }
262 }

◆ simplify_pow()

std::pair< RCP< const Basic >, RCP< const Basic > > SymEngine::SimplifyVisitor::simplify_pow ( const RCP< const Basic > &  e,
const RCP< const Basic > &  b 
)
private

Definition at line 23 of file simplify.cpp.

25{
26 if (is_a<Csc>(*b) and eq(*e, *minus_one)) {
27 // csc(expr) ** -1 = sin(expr)
28 return std::make_pair(
29 one, sin(down_cast<const OneArgFunction &>(*b).get_arg()));
30 } else if (is_a<Sec>(*b) and eq(*e, *minus_one)) {
31 // sec(expr) ** -1 = cos(expr)
32 return std::make_pair(
33 one, cos(down_cast<const OneArgFunction &>(*b).get_arg()));
34 } else if (is_a<Cot>(*b) and eq(*e, *minus_one)) {
35 // cot(expr) ** -1 = tan(expr)
36 return std::make_pair(
37 one, tan(down_cast<const OneArgFunction &>(*b).get_arg()));
38 } else {
39 return std::make_pair(e, b);
40 }
41}
T make_pair(T... args)
RCP< const Basic > tan(const RCP< const Basic > &arg)
Canonicalize Tan:
Definition: functions.cpp:1007
RCP< const Basic > cos(const RCP< const Basic > &arg)
Canonicalize Cos:
Definition: functions.cpp:942
RCP< const Basic > sin(const RCP< const Basic > &arg)
Canonicalize Sin:
Definition: functions.cpp:874

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