interpretor/preprocessor/faust-0.9.47mr3/compiler/extended/atanprim.cpp

   1 #include "xtended.hh"
   2 #include "Text.hh"
   3 #include <math.h>
   4
   5 #include "floats.hh"
   6
   7 class AtanPrim : public xtended
   8 {
   9
  10  public:
  11
  12         AtanPrim() : xtended("atan") {}
  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() == 1);
  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                 return args[0];
  28         }
  29
  30
  31         virtual Tree    computeSigOutput (const vector<Tree>& args) {
  32                 num n;
  33                 if (isNum(args[0],n)) {
  34                         return tree(atan(double(n)));
  35                 } else {
  36                         return tree(symbol(), args[0]);
  37                 }
  38         }
  39
  40         virtual string  generateCode (Klass* klass, const vector<string>& args, const vector<Type>& types)
  41         {
  42                 assert (args.size() == arity());
  43                 assert (types.size() == arity());
  44
  45         return subst("atan$1($0)", args[0], isuffix());
  46         }
  47
  48         virtual string  generateLateq (Lateq* lateq, const vector<string>& args, const vector<Type>& types)
  49         {
  50                 assert (args.size() == arity());
  51                 assert (types.size() == arity());
  52
  53         return subst("\\arctan\\left($0\\right)", args[0]);
  54         }
  55
  56 };
  57
  58
  59 xtended* gAtanPrim = new AtanPrim();
  60
  61