00001 // @HEADER 00002 // ************************************************************************ 00003 // 00004 // Phalanx: A Partial Differential Equation Field Evaluation 00005 // Kernel for Flexible Management of Complex Dependency Chains 00006 // Copyright (2008) Sandia Corporation 00007 // 00008 // Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation, 00009 // the U.S. Government retains certain rights in this software. 00010 // 00011 // This library is free software; you can redistribute it and/or modify 00012 // it under the terms of the GNU Lesser General Public License as 00013 // published by the Free Software Foundation; either version 2.1 of the 00014 // License, or (at your option) any later version. 00015 // 00016 // This library is distributed in the hope that it will be useful, but 00017 // WITHOUT ANY WARRANTY; without even the implied warranty of 00018 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 00019 // Lesser General Public License for more details. 00020 // 00021 // You should have received a copy of the GNU Lesser General Public 00022 // License along with this library; if not, write to the Free Software 00023 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 00024 // USA 00025 // 00026 // Questions? Contact Roger Pawlowski (rppawlo@sandia.gov), Sandia 00027 // National Laboratories. 00028 // 00029 // ************************************************************************ 00030 // @HEADER 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_Field.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::Field<ScalarT> source; 00061 PHX::Field<ScalarT> density; 00062 PHX::Field<ScalarT> temp; 00063 00064 std::size_t data_layout_size; 00065 }; 00066 00067 #include "Evaluator_NonlinearSource_Def.hpp" 00068 00069 #endif