[PATCH] Add relation like constructor to ScalarCode

[Richard Guenther]

Hi!

This adds a relation functor required constructor to ScalarCode to be able
to use ScalarCode objects as relation functors.  It also updates
TestBasicRelations check to excercise this and also adds actual checking
to the test.

Tested partly (relation and scalarcode tests) on ia32-linux, ok?

Richard.


2004Apr02  Richard Guenther <address@hidden>

        * src/Evaluator/ScalarCode.h: add copy constructor with
        extra LHS argument for relation functor compatibility.
        src/Field/tests/TestBasicRelations.cpp: check using
        ScalarCode object as relation functor.

===== Evaluator/ScalarCode.h 1.5 vs edited =====
--- 1.5/r2/src/Evaluator/ScalarCode.h   Thu Oct 23 14:41:02 2003
+++ edited/Evaluator/ScalarCode.h       Fri Apr  2 13:29:26 2004
@@ -403,6 +403,14 @@
   {
   }

+  /// Constructor to allow ScalarCode being used as RelationFunctor
+
+  template <class LHS>
+  ScalarCode(const ScalarCode<Function>& sc, const LHS&)
+    : function_m(sc.function_m)
+  {
+  }
+
   template<class F>
   static inline bool checkValidity(const F& f, WrappedInt<false>)
   {
===== Field/tests/TestBasicRelations.cpp 1.1 vs edited =====
--- 1.1/r2/src/Field/tests/TestBasicRelations.cpp       Mon May 13 17:47:37 2002
+++ edited/Field/tests/TestBasicRelations.cpp   Fri Apr  2 13:37:14 2004
@@ -54,10 +54,31 @@
   }
 };

-void computePotentialEnergy(const Field_t &U, const Field_t &m, const Field_t 
&h)
-{
-  U = m * g * h;
-}
+template<int Dim>
+struct computePotentialEnergy {
+  computePotentialEnergy() {}
+  void scalarCodeInfo(ScalarCodeInfo& info) const
+  {
+    info.dimensions(Dim);
+    info.arguments(3);
+    info.write(0, true);
+    info.write(1, false);
+    info.write(2, false);
+    info.useGuards(0, false);
+    info.useGuards(1, false);
+    info.useGuards(2, false);
+    for (int i=0; i<Dim; ++i)
+    {
+      info.lowerExtent(i) = 0;
+      info.upperExtent(i) = 0;
+    }
+  }
+  template <class F1, class F2, class F3>
+  void operator()(const F1& U, const F2& m, const F3& h, const Loc<Dim>& loc) 
const
+  {
+    U(loc) = m(loc) * g * h(loc);
+  }
+};

 struct ComputeVelocity
 {
@@ -102,14 +123,16 @@

   Pooma::newRelation(Pooma::functionPtr(computeTotalEnergy), E, K, U);
   Pooma::newRelation(ComputeKineticEnergy(), K, m, v);
-  Pooma::newRelation(Pooma::functionPtr(computePotentialEnergy), U, m, h);
+  Pooma::newRelation(ScalarCode<computePotentialEnergy<2> >(), U, m, h);
   Pooma::newRelation(Pooma::memberPtr(obj, &ComputeVelocity::doit), v, p, m);

   tester.out() << E << std::endl;
+  tester.check("Total energy at h=4.0", all(E == 3.0*g*4.0 + 
0.5*3.0*pow(2.0/3.0, 2)));

   h = 0;

   tester.out() << E << std::endl;
+  tester.check("Total energy at h=0.0", all(E == 3.0*g*0.0 + 
0.5*3.0*pow(2.0/3.0, 2)));

   int ret = tester.results("TestBasicRelations");
   Pooma::finalize();