void
Albany::IossSTKMeshStruct::setFieldAndBulkData(
const Teuchos::RCP<const Epetra_Comm>& comm,
const Teuchos::RCP<Teuchos::ParameterList>& params,
const unsigned int neq_,
const AbstractFieldContainer::FieldContainerRequirements& req,
const Teuchos::RCP<Albany::StateInfoStruct>& sis,
const unsigned int worksetSize)
{
this->SetupFieldData(comm, neq_, req, sis, worksetSize);
*out << "IOSS-STK: number of node sets = " << nsPartVec.size() << std::endl;
*out << "IOSS-STK: number of side sets = " << ssPartVec.size() << std::endl;
metaData->commit();
// Restart index to read solution from exodus file.
int index = params->get("Restart Index",-1); // Default to no restart
double res_time = params->get<double>("Restart Time",-1.0); // Default to no restart
Ioss::Region *region = mesh_data->m_input_region;
/*
* The following code block reads a single mesh on PE 0, then distributes the mesh across
* the other processors. stk_rebalance is used, which requires Zoltan
*
* This code is only compiled if ALBANY_MPI and ALBANY_ZOLTAN are true
*/
#ifdef ALBANY_ZOLTAN // rebalance needs Zoltan
if(useSerialMesh){
bulkData->modification_begin();
if(comm->MyPID() == 0){ // read in the mesh on PE 0
stk::io::process_mesh_bulk_data(region, *bulkData);
// Read solution from exodus file.
if (index >= 0) { // User has specified a time step to restart at
*out << "Restart Index set, reading solution index : " << index << std::endl;
stk::io::input_mesh_fields(region, *bulkData, index);
m_restartDataTime = region->get_state_time(index);
m_hasRestartSolution = true;
}
else if (res_time >= 0) { // User has specified a time to restart at
*out << "Restart solution time set, reading solution time : " << res_time << std::endl;
stk::io::input_mesh_fields(region, *bulkData, res_time);
m_restartDataTime = res_time;
m_hasRestartSolution = true;
}
else {
*out << "Neither restart index or time are set. Not reading solution data from exodus file"<< std::endl;
}
}
bulkData->modification_end();
} // End UseSerialMesh - reading mesh on PE 0
else
#endif
/*
* The following code block reads a single mesh when Albany is compiled serially, or a
* Nemspread fileset if ALBANY_MPI is true.
*
*/
{ // running in Serial or Parallel read from Nemspread files
stk::io::populate_bulk_data(*bulkData, *mesh_data);
if (!usePamgen) {
// Read solution from exodus file.
if (index >= 0) { // User has specified a time step to restart at
*out << "Restart Index set, reading solution index : " << index << std::endl;
stk::io::process_input_request(*mesh_data, *bulkData, index);
m_restartDataTime = region->get_state_time(index);
m_hasRestartSolution = true;
}
else if (res_time >= 0) { // User has specified a time to restart at
*out << "Restart solution time set, reading solution time : " << res_time << std::endl;
stk::io::process_input_request(*mesh_data, *bulkData, res_time);
m_restartDataTime = res_time;
m_hasRestartSolution = true;
}
else {
*out << "Restart Index not set. Not reading solution from exodus ("
<< index << ")"<< std::endl;
}
}
bulkData->modification_end();
} // End Parallel Read - or running in serial
if(m_hasRestartSolution){
Teuchos::Array<std::string> default_field;
default_field.push_back("solution");
Teuchos::Array<std::string> restart_fields =
params->get<Teuchos::Array<std::string> >("Restart Fields", default_field);
// Get the fields to be used for restart
// See what state data was initialized from the stk::io request
// This should be propagated into stk::io
const Ioss::ElementBlockContainer& elem_blocks = region->get_element_blocks();
/*
// Uncomment to print what fields are in the exodus file
Ioss::NameList exo_fld_names;
elem_blocks[0]->field_describe(&exo_fld_names);
for(std::size_t i = 0; i < exo_fld_names.size(); i++){
*out << "Found field \"" << exo_fld_names[i] << "\" in exodus file" << std::endl;
}
*/
for (std::size_t i=0; i<sis->size(); i++) {
Albany::StateStruct& st = *((*sis)[i]);
if(elem_blocks[0]->field_exists(st.name))
for(std::size_t j = 0; j < restart_fields.size(); j++)
if(boost::iequals(st.name, restart_fields[j])){
*out << "Restarting from field \"" << st.name << "\" found in exodus file." << std::endl;
st.restartDataAvailable = true;
break;
}
}
}
// coordinates_field = metaData->get_field<VectorFieldType>(std::string("coordinates"));
//#ifdef ALBANY_FELIX
// surfaceHeight_field = metaData->get_field<ScalarFieldType>(std::string("surface height"));
//#endif
// Refine the mesh before starting the simulation if indicated
uniformRefineMesh(comm);
// Rebalance the mesh before starting the simulation if indicated
rebalanceInitialMesh(comm);
// Build additional mesh connectivity needed for mesh fracture (if indicated)
computeAddlConnectivity();
}
