interpreter/preprocessor/faust-0.9.47mr3/compiler/extended/fmodprim.cpp

   1 #include "xtended.hh"
   2 #include "Text.hh"
   3 #include <math.h>
   4
   5 #include "floats.hh"
   6
   7 class FmodPrim : public xtended
   8 {
   9
  10  public:
  11
  12         FmodPrim() : xtended("fmodf") {}
  13
  14         virtual unsigned int    arity () { return 2; }
  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]|args[1]);
  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 max(args[0], args[1]);
  29         }
  30
  31
  32         virtual Tree    computeSigOutput (const vector<Tree>& args) {
  33                 num n,m;
  34                 assert (args.size() == arity());
  35                 if (isNum(args[0],n) & isNum(args[1],m)) {
  36                         return tree(fmod(double(n), double(m)));
  37                 } else {
  38                         return tree(symbol(), args[0], args[1]);
  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("fmod$2($0,$1)", args[0], args[1], 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("$0\\pmod{$1}", args[0], args[1]);
  56         }
  57
  58 };
  59
  60
  61 xtended* gFmodPrim = new FmodPrim();
  62
  63