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_DEF_HPP
00033 #define PHX_EXAMPLE_VP_NONLINEAR_SOURCE_DEF_HPP
00034
00035
00036 template<typename EvalT, typename Traits> NonlinearSource<EvalT, Traits>::
00037 NonlinearSource(const Teuchos::ParameterList& p) :
00038 source("Nonlinear Source",
00039 p.get< Teuchos::RCP<PHX::DataLayout> >("Data Layout")),
00040 density("Density", p.get< Teuchos::RCP<PHX::DataLayout> >("Data Layout")),
00041 temp("Temperature", p.get< Teuchos::RCP<PHX::DataLayout> >("Data Layout"))
00042 {
00043 this->addEvaluatedField(source);
00044 this->addDependentField(density);
00045 this->addDependentField(temp);
00046
00047 this->setName("NonlinearSource");
00048 }
00049
00050
00051 template<typename EvalT, typename Traits>
00052 void NonlinearSource<EvalT, Traits>::
00053 postRegistrationSetup(PHX::FieldManager<Traits>& vm)
00054 {
00055
00056
00057
00058
00059
00060 vm.template getFieldData<ScalarT,EvalT>(source);
00061 vm.template getFieldData<ScalarT,EvalT>(density);
00062 vm.template getFieldData<ScalarT,EvalT>(temp);
00063
00064 data_layout_size = source.fieldTag().dataLayout().size();
00065 }
00066
00067
00068 template<typename EvalT, typename Traits>
00069 void NonlinearSource<EvalT, Traits>::
00070 evaluateFields(typename Traits::EvalData d)
00071 {
00072 std::size_t size = d.num_cells * data_layout_size;
00073
00074 for (std::size_t i = 0; i < size; ++i)
00075 source[i] = density[i] * temp[i] * temp[i];
00076 }
00077
00078
00079 template<typename EvalT, typename Traits>
00080 void NonlinearSource<EvalT, Traits>::
00081 preEvaluate(typename Traits::PreEvalData d)
00082 {
00083 using namespace std;
00084 cout << "In Source Pre Op" << endl;
00085 }
00086
00087
00088 template< typename EvalT, typename Traits>
00089 void NonlinearSource<EvalT, Traits>::
00090 postEvaluate(typename Traits::PostEvalData d)
00091 {
00092 using namespace std;
00093 cout << "In Source Post Op" << endl;
00094 }
00095
00096
00097
00098 #endif