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
00032 #ifndef PHX_EXAMPLE_VP_NONLINEAR_SOURCE_HPP
00033 #define PHX_EXAMPLE_VP_NONLINEAR_SOURCE_HPP
00034
00035 #include "Phalanx_ConfigDefs.hpp"
00036 #include "Phalanx_Evaluator_WithBaseImpl.hpp"
00037 #include "Phalanx_Evaluator_Derived.hpp"
00038 #include "Phalanx_MDField.hpp"
00039
00040 template<typename EvalT, typename Traits>
00041 class NonlinearSource : public PHX::EvaluatorWithBaseImpl<Traits>,
00042 public PHX::EvaluatorDerived<EvalT, Traits> {
00043
00044 public:
00045
00046 NonlinearSource(const Teuchos::ParameterList& p);
00047
00048 void postRegistrationSetup(PHX::FieldManager<Traits>& vm);
00049
00050 void evaluateFields(typename Traits::EvalData d);
00051
00052 void preEvaluate(typename Traits::PreEvalData d);
00053
00054 void postEvaluate(typename Traits::PostEvalData d);
00055
00056 private:
00057
00058 typedef typename EvalT::ScalarT ScalarT;
00059
00060 PHX::MDField<ScalarT,PHX::NaturalOrder,Cell,Point> source;
00061 PHX::MDField<ScalarT,PHX::NaturalOrder,Cell,Point> density;
00062 PHX::MDField<ScalarT,PHX::NaturalOrder,Cell,Point> temp;
00063
00064 std::size_t data_layout_size;
00065 };
00066
00067 #include "Evaluator_NonlinearSource_Def.hpp"
00068
00069 #endif