00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031 #include "SundanceUserDefOp.hpp"
00032 #include "SundanceUserDefOpElement.hpp"
00033 #include "SundanceUserDefFunctor.hpp"
00034 #include "SundanceUserDefFunctorElement.hpp"
00035 #include "PlayaTabs.hpp"
00036 #include "SundanceOut.hpp"
00037
00038 using namespace Sundance;
00039 using namespace Sundance;
00040
00041 using namespace Sundance;
00042 using namespace Teuchos;
00043
00044
00045 UserDefOp::UserDefOp(const Expr& args,
00046 const RCP<const UserDefFunctor>& functor)
00047 : ListExpr()
00048 {
00049 int nElems = functor->rangeDim();
00050 Array<RCP<ScalarExpr> > scalarArgs = getScalarArgs(args);
00051
00052 RCP<Sundance::Map<EvalContext, RCP<const UserDefOpCommonEvaluator> > >
00053 commonEvaluatorsMap = rcp(new Sundance::Map<EvalContext, RCP<const UserDefOpCommonEvaluator> >());
00054
00055 for (int i=0; i<nElems; i++)
00056 {
00057 RCP<UserDefFunctorElement> e
00058 = rcp(new UserDefFunctorElement(functor, i));
00059 Expr elem = new UserDefOpElement(scalarArgs, commonEvaluatorsMap, e);
00060 append(elem);
00061 }
00062 }
00063
00064
00065
00066 Array<RCP<ScalarExpr> > UserDefOp::getScalarArgs(const Expr& args)
00067 {
00068 Expr fargs = args.flatten();
00069 Array<RCP<ScalarExpr> > sargs(fargs.size());
00070
00071 for (int i=0; i<fargs.size(); i++)
00072 {
00073 sargs[i] = rcp_dynamic_cast<ScalarExpr>(fargs[i].ptr());
00074 }
00075 return sargs;
00076 }
00077