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_SCALAR_CONTAINER_HPP 00033 #define PHX_SCALAR_CONTAINER_HPP 00034 00035 #include "Teuchos_RCP.hpp" 00036 #include "Teuchos_ArrayRCP.hpp" 00037 #include "Phalanx_EvaluationContainer_Base.hpp" 00038 #include "Phalanx_FieldTag.hpp" 00039 #include "Phalanx_Evaluator.hpp" 00040 #include "Phalanx_DataContainer_TemplateManager.hpp" 00041 00042 namespace PHX { 00043 00048 template <typename EvalT, typename Traits> 00049 class EvaluationContainer : public PHX::EvaluationContainerBase<Traits> { 00050 00051 public: 00052 00053 EvaluationContainer(); 00054 00055 ~EvaluationContainer(); 00056 00058 void requireField(const PHX::FieldTag& f); 00059 00060 void 00061 registerEvaluator(const Teuchos::RCP<PHX::Evaluator<Traits> >& p); 00062 00063 template <typename DataT> 00064 Teuchos::ArrayRCP<DataT> getFieldData(const PHX::FieldTag& f); 00065 00066 void postRegistrationSetup(std::size_t max_num_cells, 00067 PHX::FieldManager<Traits>& fm); 00068 00069 void evaluateFields(typename Traits::EvalData d); 00070 00071 void preEvaluate(typename Traits::PreEvalData d); 00072 00073 void postEvaluate(typename Traits::PostEvalData d); 00074 00075 void print(std::ostream& os) const; 00076 00077 protected: 00078 00079 typedef PHX::DataContainer_TemplateManager<EvalT, Traits> DCTM; 00080 00081 PHX::DataContainer_TemplateManager<EvalT, Traits> 00082 data_container_template_manager_; 00083 00084 typename Traits::Allocator allocator_; 00085 00086 }; 00087 00088 } 00089 00090 #include "Phalanx_EvaluationContainer_Def.hpp" 00091 00092 #endif