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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
#include "xtended.hh"
#include "Text.hh"
#include <math.h>
#include "floats.hh"
class SqrtPrim : public xtended
{
public:
SqrtPrim() : xtended("sqrt") {}
virtual unsigned int arity () { return 1; }
virtual bool needCache () { return true; }
virtual Type infereSigType (const vector<Type>& args)
{
assert (args.size() == 1);
Type t = args[0];
interval i = t->getInterval();
if (i.valid && i.lo >=0) {
return castInterval(floatCast(t), interval(sqrt(i.lo), sqrt(i.hi)));
} else {
return castInterval(floatCast(t), interval());
}
}
virtual void sigVisit (Tree sig, sigvisitor* visitor) {}
virtual int infereSigOrder (const vector<int>& args) {
return args[0];
}
virtual Tree computeSigOutput (const vector<Tree>& args) {
// verifier les simplifications
num n;
if (isNum(args[0],n)) {
return tree(sqrt(double(n)));
} else {
return tree(symbol(), args[0]);
}
}
virtual string generateCode (Klass* klass, const vector<string>& args, const vector<Type>& types)
{
assert (args.size() == arity());
assert (types.size() == arity());
return subst("sqrt$1($0)", args[0], isuffix());
}
virtual string generateLateq (Lateq* lateq, const vector<string>& args, const vector<Type>& types)
{
assert (args.size() == arity());
assert (types.size() == arity());
return subst("\\sqrt{$0}", args[0]);
}
};
xtended* gSqrtPrim = new SqrtPrim();
|
