1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
#include "Cleanup.hh"
#include "algorithms/keep_terms.hh"
using namespace cadabra;
keep_terms::keep_terms(const Kernel& k, Ex& tr, std::vector<int> terms)
: Algorithm(k, tr), terms_(terms)
{
}
bool keep_terms::can_apply(iterator it)
{
if(*it->name!="\\sum") return false;
return true;
}
Algorithm::result_t keep_terms::apply(iterator& it)
{
result_t res=result_t::l_no_action;
int count=0;
sibling_iterator walk=tr.begin(it);
while(walk!=tr.end(it)) {
if(std::find(terms_.begin(), terms_.end(), count)==terms_.end()) {
node_zero(walk);
res=result_t::l_applied;
}
++count;
++walk;
}
cleanup_dispatch(kernel, tr, it);
return res;
}
|
