2b54eea0fc2d9db0efe5d368015d39daa2c41c70
[Faustine.git] / interpretor / faust-0.9.47mr3 / compiler / extended / absprim.cpp
1 #include "xtended.hh"
2 #include "Text.hh"
3 #include <math.h>
4 #include "sigtyperules.hh"
5
6 #include "floats.hh"
7
8 class AbsPrim : public xtended
9 {
10
11 public:
12
13 AbsPrim() : xtended("abs") {}
14
15 virtual unsigned int arity () { return 1; }
16
17 virtual bool needCache () { return true; }
18
19 virtual Type infereSigType (const vector<Type>& types)
20 {
21 assert (types.size() == arity());
22 Type t = types[0];
23 return castInterval(t, abs(t->getInterval()));
24 return t;
25 }
26
27 virtual void sigVisit (Tree sig, sigvisitor* visitor) {}
28
29 virtual int infereSigOrder (const vector<int>& args)
30 {
31 assert (args.size() == arity());
32 return args[0];
33 }
34
35
36 virtual Tree computeSigOutput (const vector<Tree>& args)
37 {
38 double f; int i;
39
40 assert (args.size() == arity());
41
42 if (isDouble(args[0]->node(),&f)) {
43 return tree(fabs(f));
44
45 } else if (isInt(args[0]->node(),&i)) {
46 return tree(abs(i));
47
48 } else {
49 return tree(symbol(), args[0]);
50 }
51 }
52
53 virtual string generateCode (Klass* klass, const vector<string>& args, const vector<Type>& types)
54 {
55 assert (args.size() == arity());
56 assert (types.size() == arity());
57
58 Type t = infereSigType(types);
59 if (t->nature() == kReal) {
60 return subst("fabs$1($0)", args[0], isuffix());
61 } else {
62 return subst("abs($0)", args[0]);
63 }
64 }
65
66 virtual string generateLateq (Lateq* lateq, const vector<string>& args, const vector<Type>& types)
67 {
68 assert (args.size() == arity());
69 assert (types.size() == arity());
70
71 Type t = infereSigType(types);
72 return subst("\\left\\lvert{$0}\\right\\rvert", args[0]);
73 }
74 };
75
76
77 xtended* gAbsPrim = new AbsPrim();
78
79