X-Git-Url: https://scm.cri.ensmp.fr/git/Faustine.git/blobdiff_plain/c7f552fd8888da2f0d8cfb228fe0f28d3df3a12c..b4b6f2ea75b9f0f3ca918f5b84016610bf7a4d4f:/interpretor/preprocessor/faust-0.9.47mr3/compiler/extended/sqrtprim.cpp

diff --git a/interpretor/preprocessor/faust-0.9.47mr3/compiler/extended/sqrtprim.cpp b/interpretor/preprocessor/faust-0.9.47mr3/compiler/extended/sqrtprim.cpp
new file mode 100644
index 0000000..e6fb640
--- /dev/null
+++ b/interpretor/preprocessor/faust-0.9.47mr3/compiler/extended/sqrtprim.cpp
@@ -0,0 +1,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();
+
+