Bug fixed in preprocessor for new primitives.
[Faustine.git] / interpretor / preprocessor / faust-0.9.47mr3 / compiler / extended / floorprim.cpp
1 #include "xtended.hh"
2 #include "Text.hh"
3 #include <math.h>
4
5 #include "floats.hh"
6
7 class FloorPrim : public xtended
8 {
9
10 public:
11
12 FloorPrim() : xtended("floor") {}
13
14 virtual unsigned int arity () { return 1; }
15
16 virtual bool needCache () { return true; }
17
18 virtual Type infereSigType (const vector<Type>& args)
19 {
20 assert (args.size() == arity());
21 return floatCast(args[0]);
22 }
23
24 virtual void sigVisit (Tree sig, sigvisitor* visitor) {}
25
26 virtual int infereSigOrder (const vector<int>& args) {
27 assert (args.size() == arity());
28 return args[0];
29 }
30
31
32 virtual Tree computeSigOutput (const vector<Tree>& args) {
33 num n;
34 assert (args.size() == arity());
35 if (isNum(args[0],n)) {
36 return tree(floor(double(n)));
37 } else {
38 return tree(symbol(), args[0]);
39 }
40 }
41
42 virtual string generateCode (Klass* klass, const vector<string>& args, const vector<Type>& types)
43 {
44 assert (args.size() == arity());
45 assert (types.size() == arity());
46
47 return subst("floor$1($0)", args[0], isuffix());
48 }
49
50 virtual string generateLateq (Lateq* lateq, const vector<string>& args, const vector<Type>& types)
51 {
52 assert (args.size() == arity());
53 assert (types.size() == arity());
54
55 return subst("\\left\\lfloor {$0} \\right\\rfloor", args[0]);
56 }
57
58 };
59
60
61 xtended* gFloorPrim = new FloorPrim();
62
63