45 #include "Teuchos_Assert.hpp" 47 #include "Panzer_EquationSet_DefaultImpl.hpp" 57 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
59 const Teuchos::ParameterList& ic_block_closure_models,
61 const Teuchos::ParameterList& user_data,
62 const bool write_graphviz_file,
63 const std::string& graphviz_file_prefix,
66 std::vector<Teuchos::RCP<panzer::PhysicsBlock> >::const_iterator blkItr;
67 for (blkItr=physicsBlocks.begin();blkItr!=physicsBlocks.end();++blkItr) {
68 Teuchos::RCP<panzer::PhysicsBlock> pb = *blkItr;
69 std::string blockId = pb->elementBlockID();
72 Teuchos::RCP<PHX::FieldManager<panzer::Traits> > fm
76 std::string closure_model_name =
"";
77 if (ic_block_closure_models.isSublist(blockId))
78 closure_model_name = blockId;
79 else if (ic_block_closure_models.isSublist(
"Default"))
80 closure_model_name =
"Default";
82 TEUCHOS_TEST_FOR_EXCEPTION(
true, std::logic_error,
"Failed to find initial condition for element block \"" << blockId
83 <<
"\". You must provide an initial condition for each element block or set a default!" 84 << ic_block_closure_models);
87 std::vector<bool> active_evaluation_types(Sacado::mpl::size<panzer::Traits::EvalTypes>::value,
false);
88 int residual_index = Sacado::mpl::find<panzer::Traits::EvalTypes,panzer::Traits::Residual>::value;
89 active_evaluation_types[residual_index] =
true;
90 pb->setActiveEvaluationTypes(active_evaluation_types);
93 pb->buildAndRegisterInitialConditionEvaluators(*fm, cm_factory, closure_model_name, ic_block_closure_models, lo_factory, user_data);
95 pb->activateAllEvaluationTypes();
104 for (
auto eval_type=fm->begin(); eval_type != fm->end(); ++eval_type,++i) {
105 if (active_evaluation_types[i])
106 eval_type->postRegistrationSetup(setupData,*fm,
false,
false,
nullptr);
109 phx_ic_field_managers[blockId] = fm;
111 if (write_graphviz_file)
112 fm->writeGraphvizFile(graphviz_file_prefix+
"_IC_"+blockId);
118 const std::vector<Teuchos::RCP<panzer::PhysicsBlock> >& physicsBlocks,
120 const Teuchos::ParameterList& closure_models,
121 const Teuchos::ParameterList& ic_block_closure_models,
123 const Teuchos::ParameterList& user_data,
124 const bool write_graphviz_file,
125 const std::string& graphviz_file_prefix,
128 std::vector<Teuchos::RCP<panzer::PhysicsBlock> >::const_iterator blkItr;
129 for (blkItr=physicsBlocks.begin();blkItr!=physicsBlocks.end();++blkItr) {
130 Teuchos::RCP<panzer::PhysicsBlock> pb = *blkItr;
131 std::string blockId = pb->elementBlockID();
134 Teuchos::RCP<PHX::FieldManager<panzer::Traits> > fm
138 std::string closure_model_name =
"";
139 if (ic_block_closure_models.isSublist(blockId))
140 closure_model_name = blockId;
141 else if (ic_block_closure_models.isSublist(
"Default"))
142 closure_model_name =
"Default";
144 TEUCHOS_TEST_FOR_EXCEPTION(
true, std::logic_error,
"Failed to find initial condition for element block \"" << blockId
145 <<
"\". You must provide an initial condition for each element block or set a default!" 146 << ic_block_closure_models);
149 std::vector<bool> active_evaluation_types(Sacado::mpl::size<panzer::Traits::EvalTypes>::value,
false);
150 int residual_index = Sacado::mpl::find<panzer::Traits::EvalTypes,panzer::Traits::Residual>::value;
151 active_evaluation_types[residual_index] =
true;
152 pb->setActiveEvaluationTypes(active_evaluation_types);
155 pb->buildAndRegisterClosureModelEvaluators(*fm,cm_factory,closure_models,user_data);
158 pb->buildAndRegisterInitialConditionEvaluators(*fm, cm_factory, closure_model_name, ic_block_closure_models, lo_factory, user_data);
160 pb->activateAllEvaluationTypes();
169 for (
auto eval_type=fm->begin(); eval_type != fm->end(); ++eval_type,++i) {
170 if (active_evaluation_types[i])
171 eval_type->postRegistrationSetup(setupData,*fm,
false,
false,
nullptr);
174 phx_ic_field_managers[blockId] = fm;
176 if (write_graphviz_file)
177 fm->writeGraphvizFile(graphviz_file_prefix+
"_IC_"+blockId);
184 Teuchos::RCP<panzer::LinearObjContainer> loc,
186 const double time_stamp)
201 localCounter->initialize();
203 ped.
gedc->addDataObject(
"Residual Scatter Container",ghostedloc);
204 ped.
gedc->addDataObject(
"Dirichlet Counter",localCounter);
208 itr!=phx_ic_field_managers.end();++itr) {
209 std::string blockId = itr->first;
210 Teuchos::RCP< PHX::FieldManager<panzer::Traits> > fm = itr->second;
216 std::vector<panzer::Workset>& w = *wkstContainer.
getWorksets(wd);
217 for (std::size_t i = 0; i < w.size(); ++i) {
221 workset.
time = time_stamp;
228 summedGhostedCounter->initialize();
233 globalCounter->initialize();
249 Teuchos::RCP<LOC> loc2 = Teuchos::rcp_dynamic_cast<LOC>(itr->second);
250 if(loc2!=Teuchos::null) {
251 bool zeroVectorRows =
false;
266 template <
typename EvalT>
267 class EquationSet_IC :
public EquationSet_DefaultImpl<EvalT> {
273 EquationSet_IC(
const Teuchos::RCP<Teuchos::ParameterList>& params,
274 const int& default_integration_order,
275 const CellData& cell_data,
276 const Teuchos::RCP<GlobalData>& global_data,
277 const bool build_transient_support);
282 const FieldLibrary& ,
283 const Teuchos::ParameterList& )
const {}
288 template <
typename EvalT>
289 EquationSet_IC<EvalT>::
290 EquationSet_IC(
const Teuchos::RCP<Teuchos::ParameterList>& params,
291 const int& default_integration_order,
292 const CellData& cell_data,
293 const Teuchos::RCP<GlobalData>& global_data,
294 const bool build_transient_support) :
295 EquationSet_DefaultImpl<EvalT>(params, default_integration_order, cell_data, global_data, build_transient_support )
300 Teuchos::ParameterList valid_parameters_sublist;
301 valid_parameters_sublist.set(
"Basis Type",
"HGrad",
"Type of Basis to use");
302 valid_parameters_sublist.set(
"Basis Order",1,
"Order of the basis");
304 for(
auto itr=params->begin();itr!=params->end();++itr) {
306 const std::string
field = params->name(itr);
307 const Teuchos::ParameterEntry & entry = params->entry(itr);
310 if(!entry.isList())
continue;
312 Teuchos::ParameterList & basisPL = entry.getValue((Teuchos::ParameterList *) 0);
313 basisPL.validateParametersAndSetDefaults(valid_parameters_sublist);
315 std::string basis_type = basisPL.get<std::string>(
"Basis Type");
316 int basis_order = basisPL.get<
int>(
"Basis Order");
318 this->addDOF(
field,basis_type,basis_order,default_integration_order);
321 this->addClosureModel(
"");
330 class IC_EquationSetFactory :
public EquationSetFactory {
333 Teuchos::RCP<EquationSet_TemplateManager<Traits> >
334 buildEquationSet(
const Teuchos::RCP<Teuchos::ParameterList>& params,
335 const int& default_integration_order,
336 const CellData& cell_data,
337 const Teuchos::RCP<GlobalData>& global_data,
338 const bool build_transient_support)
const 340 Teuchos::RCP<EquationSet_TemplateManager<Traits> > eq_set=
341 Teuchos::rcp(
new EquationSet_TemplateManager<Traits>);
350 std::string msg =
"Error - the \"Equation Set\" called \"" + params->get<std::string>(
"Type") +
351 "\" is not a valid equation set identifier. Please supply the correct factory.\n";
352 TEUCHOS_TEST_FOR_EXCEPTION(
true, std::logic_error, msg);
364 const std::map<std::string,std::vector<ICFieldDescriptor> > & block_ids_to_fields,
368 const Teuchos::ParameterList & ic_closure_models,
369 const Teuchos::ParameterList & user_data,
374 std::vector<Teuchos::RCP<PhysicsBlock> > physics_blocks;
377 std::map<std::string, Teuchos::RCP< PHX::FieldManager<Traits> > > phx_ic_field_managers;
385 "initial_condition_control_test",
386 phx_ic_field_managers);
397 buildICPhysicsBlocks(
const std::map<std::string,Teuchos::RCP<const shards::CellTopology> > & block_ids_to_cell_topo,
398 const std::map<std::string,std::vector<ICFieldDescriptor> > & block_ids_to_fields,
400 std::vector<Teuchos::RCP<PhysicsBlock> > & physics_blocks)
405 std::map<std::string,std::string> block_ids_to_physics_ids;
407 RCP<Teuchos::ParameterList> ipb = rcp(
new Teuchos::ParameterList);
409 for(
auto itr=block_ids_to_cell_topo.begin();itr!=block_ids_to_cell_topo.end();++itr) {
410 std::string eblock = itr->first;
411 RCP<const shards::CellTopology> ct = itr->second;
414 auto fds_itr = block_ids_to_fields.find(eblock);
415 TEUCHOS_ASSERT(fds_itr!=block_ids_to_fields.end());
417 const std::vector<ICFieldDescriptor> & fd_vec = fds_itr->second;
419 std::string physics_id =
"ic_"+eblock;
420 block_ids_to_physics_ids[eblock] = physics_id;
423 Teuchos::ParameterList & physics_block = ipb->sublist(physics_id).sublist(
"");
424 physics_block.set(
"Type",
"IC");
426 for(std::size_t i=0;i<fd_vec.size();i++) {
435 RCP<EquationSetFactory> eqset_factory = Teuchos::rcp(
new IC_EquationSetFactory);
439 block_ids_to_cell_topo,
440 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)
Teuchos::RCP< GlobalEvaluationDataContainer > gedc
const_iterator begin() const
Teuchos::RCP< const std::vector< Intrepid2::Orientation > > orientations_
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.
virtual Teuchos::RCP< LinearObjContainer > buildPrimitiveGhostedLinearObjContainer() const =0
std::string first_sensitivities_name
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.
Teuchos::RCP< std::vector< Workset > > getWorksets(const WorksetDescriptor &wd)
Access to volume worksets.
virtual Teuchos::RCP< LinearObjContainer > buildLinearObjContainer() const =0
#define PANZER_DECLARE_EQSET_TEMPLATE_BUILDER(fClass, fType)
void setupInitialConditionFieldManagers(WorksetContainer &wkstContainer, const std::vector< Teuchos::RCP< panzer::PhysicsBlock > > &physicsBlocks, const panzer::ClosureModelFactory_TemplateManager< panzer::Traits > &cm_factory, const Teuchos::ParameterList &closure_models, 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 closure models and registers evaluators...
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.
PHX::MDField< ScalarT, panzer::Cell, panzer::BASIS > field
A field to which we'll contribute, or in which we'll store, the result of computing this integral...
Teuchos::RCP< const std::vector< Intrepid2::Orientation > > getOrientations() const
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
WorksetDescriptor blockDescriptor(const std::string &eBlock)
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)
virtual void ghostToGlobalContainer(const LinearObjContainer &ghostContainer, LinearObjContainer &container, int) const =0
Teuchos::RCP< const std::vector< panzer::Workset > > worksets_