// ======================================================================== void process_surface_entity(Ioss::SideSet *sset, stk::mesh::MetaData &meta, stk::mesh::EntityRank sset_rank) { assert(sset->type() == Ioss::SIDESET); Ioss::SideSet *fs = dynamic_cast<Ioss::SideSet *>(sset); assert(fs != NULL); const Ioss::SideBlockContainer& blocks = fs->get_side_blocks(); stk::io::default_part_processing(blocks, meta, sset_rank); stk::mesh::Part* const fs_part = meta.get_part(sset->name()); STKIORequire(fs_part != NULL); stk::mesh::Field<double, stk::mesh::ElementNode> *distribution_factors_field = NULL; bool surface_df_defined = false; // Has the surface df field been defined yet? int block_count = sset->block_count(); for (int i=0; i < block_count; i++) { Ioss::SideBlock *side_block = sset->get_block(i); if (stk::io::include_entity(side_block)) { stk::mesh::Part * const side_block_part = meta.get_part(side_block->name()); STKIORequire(side_block_part != NULL); meta.declare_part_subset(*fs_part, *side_block_part); const stk::mesh::EntityRank part_rank = side_block_part->primary_entity_rank(); if (side_block->field_exists("distribution_factors")) { if (!surface_df_defined) { std::string field_name = sset->name() + "_distribution_factors"; distribution_factors_field = &meta.declare_field<stk::mesh::Field<double, stk::mesh::ElementNode> >(static_cast<stk::topology::rank_t>(part_rank), field_name); stk::io::set_distribution_factor_field(*fs_part, *distribution_factors_field); surface_df_defined = true; } stk::io::set_distribution_factor_field(*side_block_part, *distribution_factors_field); int side_node_count = side_block->topology()->number_nodes(); stk::mesh::put_field(*distribution_factors_field, *side_block_part, side_node_count); } /** \todo IMPLEMENT truly handle fields... For this case we * are just defining a field for each transient field that is * present in the mesh... */ stk::io::define_io_fields(side_block, Ioss::Field::TRANSIENT, *side_block_part, part_rank); } } }
// ======================================================================== void process_elementblocks(Ioss::Region ®ion, stk::mesh::MetaData &meta) { const stk::mesh::EntityRank element_rank = stk::topology::ELEMENT_RANK; const Ioss::ElementBlockContainer& elem_blocks = region.get_element_blocks(); stk::io::default_part_processing(elem_blocks, meta, element_rank); // Parts were created above, now handle element block specific // information (topology, attributes, ...); for(Ioss::ElementBlockContainer::const_iterator it = elem_blocks.begin(); it != elem_blocks.end(); ++it) { Ioss::ElementBlock *entity = *it; if (stk::io::include_entity(entity)) { stk::mesh::Part* const part = meta.get_part(entity->name()); STKIORequire(part != NULL); const stk::mesh::EntityRank part_rank = part->primary_entity_rank(); // Element Block attributes (if any)... /** \todo IMPLEMENT truly handle attribute fields... For this * case we are just defining a field for each attribute field * that is present in the mesh... */ stk::io::define_io_fields(entity, Ioss::Field::ATTRIBUTE, *part, part_rank); /** \todo IMPLEMENT truly handle fields... For this case we * are just defining a field for each transient field that is * present in the mesh... */ stk::io::define_io_fields(entity, Ioss::Field::TRANSIENT, *part, part_rank); } } }
// ======================================================================== void process_nodesets(Ioss::Region ®ion, stk::mesh::MetaData &meta) { const Ioss::NodeSetContainer& node_sets = region.get_nodesets(); stk::io::default_part_processing(node_sets, meta, stk::topology::NODE_RANK); /** \todo REFACTOR should "distribution_factor" be a default field * that is automatically declared on all objects that it exists * on as is done in current framework? */ stk::mesh::Field<double> & distribution_factors_field = meta.declare_field<stk::mesh::Field<double> >(stk::topology::NODE_RANK, "distribution_factors"); /** \todo REFACTOR How to associate distribution_factors field * with the nodeset part if a node is a member of multiple * nodesets */ for(Ioss::NodeSetContainer::const_iterator it = node_sets.begin(); it != node_sets.end(); ++it) { Ioss::NodeSet *entity = *it; if (stk::io::include_entity(entity)) { stk::mesh::Part* const part = meta.get_part(entity->name()); STKIORequire(part != NULL); STKIORequire(entity->field_exists("distribution_factors")); stk::mesh::put_field(distribution_factors_field, *part); /** \todo IMPLEMENT truly handle fields... For this case we * are just defining a field for each transient field that is * present in the mesh... */ stk::io::define_io_fields(entity, Ioss::Field::TRANSIENT, *part, part->primary_entity_rank() ) ; } } }
void copy_part_supersets(const stk::mesh::Part &oldPart, stk::mesh::Part &newPart, stk::mesh::MetaData &newMeta) { stk::mesh::OrdinalVector oldSupersets = get_part_supersets(oldPart); for(stk::mesh::PartOrdinal partOrd : oldSupersets) newMeta.declare_part_subset(newMeta.get_part(partOrd), newPart); }