Differentiate materials in DAGMC universes

include/openmc/capi.h

@@ -29,6 +29,9 @@ int openmc_cell_set_temperature(
   int32_t index, double T, const int32_t* instance, bool set_contained = false);
 int openmc_cell_set_translation(int32_t index, const double xyz[]);
 int openmc_cell_set_rotation(int32_t index, const double rot[], size_t rot_len);
+int openmc_dagmc_universe_get_cell_ids(
+  int32_t univ_id, int32_t* ids, size_t* n);
+int openmc_dagmc_universe_get_num_cells(int32_t univ_id, size_t* n);
 int openmc_energy_filter_get_bins(
   int32_t index, const double** energies, size_t* n);
 int openmc_energy_filter_set_bins(

include/openmc/cell.h

@@ -320,7 +320,6 @@ class Cell {
   int32_t universe_;        //!< Universe # this cell is in
   int32_t fill_;            //!< Universe # filling this cell
   int32_t n_instances_ {0}; //!< Number of instances of this cell
-  GeometryType geom_type_;  //!< Geometric representation type (CSG, DAGMC)
 
   //! \brief Index corresponding to this cell in distribcell arrays
   int distribcell_index_ {C_NONE};
@@ -350,6 +349,13 @@ class Cell {
   vector<double> rotation_;
 
   vector<int32_t> offset_; //!< Distribcell offset table
+
+  // Accessors
+  const GeometryType& geom_type() const { return geom_type_; }
+  GeometryType& geom_type() { return geom_type_; }
+
+private:
+  GeometryType geom_type_; //!< Geometric representation type (CSG, DAGMC)
 };
 
 struct CellInstanceItem {

include/openmc/dagmc.h

@@ -133,6 +133,12 @@ class DAGUniverse : public Universe {
   void legacy_assign_material(
     std::string mat_string, std::unique_ptr<DAGCell>& c) const;
 
+  //! Assign a material overriding normal assignement to a cell
+  //! \param[in] key The material key to override
+  //! \param[in] c The OpenMC cell to which the material is assigned
+  void override_assign_material(std::string key, moab::EntityHandle vol_handle,
+    std::unique_ptr<DAGCell>& c) const;
+
   //! Return the index into the model cells vector for a given DAGMC volume
   //! handle in the universe
   //! \param[in] vol MOAB handle to the DAGMC volume set
@@ -184,6 +190,12 @@ class DAGUniverse : public Universe {
                              //!< generate new material IDs for the universe
   bool has_graveyard_; //!< Indicates if the DAGMC geometry has a "graveyard"
                        //!< volume
+  std::map<std::string, vector<std::string>>
+    instance_material_overrides; ///!< Map of material overrides
+                                 ///!< keys correspond to the material name
+                                 ///!< or id
+                                 ///!< values are a list of materials used
+                                 ///!< git for the override
 };
 
 //==============================================================================

include/openmc/string_utils.h

@@ -19,6 +19,7 @@ void to_lower(std::string& str);
 int word_count(const std::string& str);
 
 vector<std::string> split(const std::string& in);
+vector<std::string> split(const std::string& in, char delim);
 
 bool ends_with(const std::string& value, const std::string& ending);
 

include/openmc/surface.h

@@ -38,7 +38,6 @@ class Surface {
   int id_;                           //!< Unique ID
   std::string name_;                 //!< User-defined name
   unique_ptr<BoundaryCondition> bc_; //!< Boundary condition
-  GeometryType geom_type_;           //!< Geometry type indicator (CSG or DAGMC)
   bool surf_source_ {false}; //!< Activate source banking for the surface?
 
   explicit Surface(pugi::xml_node surf_node);
@@ -91,6 +90,13 @@ class Surface {
   //! Get the BoundingBox for this surface.
   virtual BoundingBox bounding_box(bool /*pos_side*/) const { return {}; }
 
+  // Accessors
+  const GeometryType& geom_type() const { return geom_type_; }
+  GeometryType& geom_type() { return geom_type_; }
+
+private:
+  GeometryType geom_type_; //!< Geometry type indicator (CSG or DAGMC)
+
 protected:
   virtual void to_hdf5_inner(hid_t group_id) const = 0;
 };

openmc/__init__.py

@@ -15,6 +15,7 @@
 from openmc.weight_windows import *
 from openmc.surface import *
 from openmc.universe import *
+from openmc.dagmc import *
 from openmc.source import *
 from openmc.settings import *
 from openmc.lattice import *

openmc/cell.py

@@ -780,3 +780,4 @@ def from_xml_element(cls, elem, surfaces, materials, get_universe):
         univ_id = int(get_text(elem, 'universe', 0))
         get_universe(univ_id).add_cell(c)
         return c
+