45 #include "Teuchos_Assert.hpp" 47 #include "Panzer_EquationSet_DefaultImpl.hpp" 56 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
58 const Teuchos::ParameterList& ic_block_closure_models,
60 const Teuchos::ParameterList& user_data,
61 const bool write_graphviz_file,
62 const std::string& graphviz_file_prefix,
65 std::vector<Teuchos::RCP<panzer::PhysicsBlock> >::const_iterator blkItr;
66 for (blkItr=physicsBlocks.begin();blkItr!=physicsBlocks.end();++blkItr) {
67 Teuchos::RCP<panzer::PhysicsBlock> pb = *blkItr;
68 std::string blockId = pb->elementBlockID();
71 Teuchos::RCP<PHX::FieldManager<panzer::Traits> > fm
75 std::string closure_model_name =
"";
76 if (ic_block_closure_models.isSublist(blockId))
77 closure_model_name = blockId;
78 else if (ic_block_closure_models.isSublist(
"Default"))
79 closure_model_name =
"Default";
81 TEUCHOS_TEST_FOR_EXCEPTION(
true, std::logic_error,
"Failed to find initial condition for element block \"" << blockId
82 <<
"\". You must provide an initial condition for each element block or set a default!" 83 << ic_block_closure_models);
86 pb->buildAndRegisterInitialConditionEvaluators(*fm, cm_factory, closure_model_name, ic_block_closure_models, lo_factory, user_data);
92 fm->postRegistrationSetup(setupData);
93 phx_ic_field_managers[blockId] = fm;
95 if (write_graphviz_file)
96 fm->writeGraphvizFile(graphviz_file_prefix+
"_IC_"+blockId);
103 Teuchos::RCP<panzer::LinearObjContainer> loc,
105 const double time_stamp)
120 localCounter->initialize();
127 itr!=phx_ic_field_managers.end();++itr) {
128 std::string blockId = itr->first;
129 Teuchos::RCP< PHX::FieldManager<panzer::Traits> > fm = itr->second;
135 for (std::size_t i = 0; i < w.size(); ++i) {
139 workset.
time = time_stamp;
146 summedGhostedCounter->initialize();
151 globalCounter->initialize();
167 Teuchos::RCP<LOC> loc2 = Teuchos::rcp_dynamic_cast<LOC>(itr->second);
168 if(loc2!=Teuchos::null) {
169 bool zeroVectorRows =
false;
184 template <
typename EvalT>
185 class EquationSet_IC :
public EquationSet_DefaultImpl<EvalT> {
191 EquationSet_IC(
const Teuchos::RCP<Teuchos::ParameterList>& params,
192 const int& default_integration_order,
193 const CellData& cell_data,
195 const bool build_transient_support);
200 const FieldLibrary& field_library,
201 const Teuchos::ParameterList& user_data)
const {}
206 template <
typename EvalT>
207 EquationSet_IC<EvalT>::
208 EquationSet_IC(
const Teuchos::RCP<Teuchos::ParameterList>& params,
209 const int& default_integration_order,
210 const CellData& cell_data,
212 const bool build_transient_support) :
213 EquationSet_DefaultImpl<EvalT>(params, default_integration_order, cell_data,
global_data, build_transient_support )
218 Teuchos::ParameterList valid_parameters_sublist;
219 valid_parameters_sublist.set(
"Basis Type",
"HGrad",
"Type of Basis to use");
220 valid_parameters_sublist.set(
"Basis Order",1,
"Order of the basis");
222 for(
auto itr=params->begin();itr!=params->end();++itr) {
224 const std::string
field = params->name(itr);
225 const Teuchos::ParameterEntry & entry = params->entry(itr);
228 if(!entry.isList())
continue;
230 Teuchos::ParameterList & basisPL = entry.getValue((Teuchos::ParameterList *) 0);
231 basisPL.validateParametersAndSetDefaults(valid_parameters_sublist);
233 std::string basis_type = basisPL.get<std::string>(
"Basis Type");
234 int basis_order = basisPL.get<
int>(
"Basis Order");
236 this->addDOF(
field,basis_type,basis_order,default_integration_order);
239 this->addClosureModel(
"");
248 class IC_EquationSetFactory :
public EquationSetFactory {
251 Teuchos::RCP<EquationSet_TemplateManager<Traits> >
252 buildEquationSet(
const Teuchos::RCP<Teuchos::ParameterList>& params,
253 const int& default_integration_order,
254 const CellData& cell_data,
256 const bool build_transient_support)
const 258 Teuchos::RCP<EquationSet_TemplateManager<Traits> > eq_set=
259 Teuchos::rcp(
new EquationSet_TemplateManager<Traits>);
268 std::string msg =
"Error - the \"Equation Set\" called \"" + params->get<std::string>(
"Type") +
269 "\" is not a valid equation set identifier. Please supply the correct factory.\n";
270 TEUCHOS_TEST_FOR_EXCEPTION(
true, std::logic_error, msg);
282 const std::map<std::string,std::vector<ICFieldDescriptor> > & block_ids_to_fields,
286 const Teuchos::ParameterList & ic_closure_models,
287 const Teuchos::ParameterList & user_data,
292 std::vector<Teuchos::RCP<PhysicsBlock> > physics_blocks;
295 std::map<std::string, Teuchos::RCP< PHX::FieldManager<Traits> > > phx_ic_field_managers;
303 "initial_condition_control_test",
304 phx_ic_field_managers);
315 buildICPhysicsBlocks(
const std::map<std::string,Teuchos::RCP<const shards::CellTopology> > & block_ids_to_cell_topo,
316 const std::map<std::string,std::vector<ICFieldDescriptor> > & block_ids_to_fields,
318 std::vector<Teuchos::RCP<PhysicsBlock> > & physics_blocks)
323 std::map<std::string,std::string> block_ids_to_physics_ids;
325 RCP<Teuchos::ParameterList> ipb = rcp(
new Teuchos::ParameterList);
327 for(
auto itr=block_ids_to_cell_topo.begin();itr!=block_ids_to_cell_topo.end();++itr) {
328 std::string eblock = itr->first;
329 RCP<const shards::CellTopology> ct = itr->second;
332 auto fds_itr = block_ids_to_fields.find(eblock);
333 TEUCHOS_ASSERT(fds_itr!=block_ids_to_fields.end());
335 const std::vector<ICFieldDescriptor> & fd_vec = fds_itr->second;
337 std::string physics_id =
"ic_"+eblock;
338 block_ids_to_physics_ids[eblock] = physics_id;
341 Teuchos::ParameterList & physics_block = ipb->sublist(physics_id).sublist(
"");
342 physics_block.set(
"Type",
"IC");
344 for(std::size_t i=0;i<fd_vec.size();i++) {
353 RCP<EquationSetFactory> eqset_factory = Teuchos::rcp(
new IC_EquationSetFactory);
357 block_ids_to_cell_topo,
358 ipb,1,workset_size,eqset_factory,gd,
false,physics_blocks);
void buildICPhysicsBlocks(const std::map< std::string, Teuchos::RCP< const shards::CellTopology > > &block_ids_to_cell_topo, const std::map< std::string, std::vector< ICFieldDescriptor > > &block_ids_to_fields, int workset_size, std::vector< Teuchos::RCP< PhysicsBlock > > &physics_blocks)
const_iterator begin() const
Teuchos::RCP< std::vector< panzer::Workset > > worksets_
std::string first_sensitivities_name
void addDataObject(const std::string &key, const Teuchos::RCP< GlobalEvaluationData > &ged)
std::unordered_map< std::string, Teuchos::RCP< GlobalEvaluationData > >::iterator iterator
void setupControlInitialCondition(const std::map< std::string, Teuchos::RCP< const shards::CellTopology > > &block_ids_to_cell_topo, const std::map< std::string, std::vector< ICFieldDescriptor > > &block_ids_to_fields, WorksetContainer &wkstContainer, const LinearObjFactory< Traits > &lof, const ClosureModelFactory_TemplateManager< Traits > &cm_factory, const Teuchos::ParameterList &ic_closure_models, const Teuchos::ParameterList &user_data, int workset_size, double t0, const Teuchos::RCP< Thyra::VectorBase< double > > &vec)
#define PANZER_BUILD_EQSET_OBJECTS(key, fType)
virtual void adjustForDirichletConditions(const LinearObjContainer &localBCRows, const LinearObjContainer &globalBCRows, LinearObjContainer &ghostedObjs, bool zeroVectorRows=false, bool adjustX=false) const =0
Class that provides access to worksets on each element block and side set.
PHX::MDField< ScalarT > vector
Teuchos::RCP< panzer::GlobalData > global_data
virtual Teuchos::RCP< LinearObjContainer > buildPrimitiveGhostedLinearObjContainer() const =0
virtual void initializeContainer(int, LinearObjContainer &loc) const =0
void buildPhysicsBlocks(const std::map< std::string, std::string > &block_ids_to_physics_ids, const std::map< std::string, Teuchos::RCP< const shards::CellTopology > > &block_ids_to_cell_topo, const Teuchos::RCP< Teuchos::ParameterList > &physics_blocks_plist, const int default_integration_order, const std::size_t workset_size, const Teuchos::RCP< const panzer::EquationSetFactory > &eqset_factory, const Teuchos::RCP< panzer::GlobalData > &global_data, const bool build_transient_support, std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const std::vector< std::string > &tangent_param_names=std::vector< std::string >())
Nonmember function for building the physics blocks from a Teuchos::ParameterList for a given list of ...
Teuchos::RCP< panzer::GlobalData > createGlobalData(bool build_default_os=true, int print_process=0)
Nonmember constructor.
virtual Teuchos::RCP< LinearObjContainer > buildLinearObjContainer() const =0
#define PANZER_DECLARE_EQSET_TEMPLATE_BUILDER(fClass, fType)
PHX::MDField< const ScalarT, Cell, IP > field
void setupInitialConditionFieldManagers(WorksetContainer &wkstContainer, const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &cm_factory, const Teuchos::ParameterList &ic_block_closure_models, const panzer::LinearObjFactory< panzer::Traits > &lo_factory, const Teuchos::ParameterList &user_data, const bool write_graphviz_file, const std::string &graphviz_file_prefix, std::map< std::string, Teuchos::RCP< PHX::FieldManager< panzer::Traits > > > &phx_ic_field_managers)
Builds PHX::FieldManager objects for inital conditions and registers evaluators.
GlobalEvaluationDataContainer gedc
virtual Teuchos::RCP< LinearObjContainer > buildPrimitiveLinearObjContainer() const =0
virtual void globalToGhostContainer(const LinearObjContainer &container, LinearObjContainer &ghostContainer, int) const =0
virtual void initializeGhostedContainer(int, LinearObjContainer &loc) const =0
virtual Teuchos::RCP< LinearObjContainer > buildGhostedLinearObjContainer() const =0
const_iterator end() const
void evaluateInitialCondition(WorksetContainer &wkstContainer, const std::map< std::string, Teuchos::RCP< PHX::FieldManager< panzer::Traits > > > &phx_ic_field_managers, Teuchos::RCP< panzer::LinearObjContainer > loc, const panzer::LinearObjFactory< panzer::Traits > &lo_factory, const double time_stamp)
Teuchos::RCP< std::vector< Workset > > getVolumeWorksets(const std::string &eBlock)
Access to volume worksets.
virtual void ghostToGlobalContainer(const LinearObjContainer &ghostContainer, LinearObjContainer &container, int) const =0