From 2410a4dfbad72470c90bd8d0e6e92f01e774592b Mon Sep 17 00:00:00 2001
From: Patrick Shriwise <pshriwise@gmail.com>
Date: Wed, 18 Dec 2019 17:16:20 -0600
Subject: [PATCH] Updating how the libmesh solution set is handled. Also
 applying data for a single score, single nuclide tally.

---
 include/openmc/mesh.h |  1 -
 src/mesh.cpp          | 12 ++++--------
 2 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/include/openmc/mesh.h b/include/openmc/mesh.h
index f42a19490f..7c4a2dbdd0 100644
--- a/include/openmc/mesh.h
+++ b/include/openmc/mesh.h
@@ -521,7 +521,6 @@ private:
   std::unique_ptr<libMesh::Mesh> m_;
   std::unique_ptr<libMesh::PointLocatorBase> point_locator_;
   std::unique_ptr<libMesh::EquationSystems> equation_systems_;
-  std::unique_ptr<libMesh::NumericVector<libMesh::Real>> sol_;
   std::map<std::string, unsigned int> variable_map_;
   std::string eq_system_name_;
   libMesh::Elem* first_element_;
diff --git a/src/mesh.cpp b/src/mesh.cpp
index 849f264dbc..26e8ed1f28 100644
--- a/src/mesh.cpp
+++ b/src/mesh.cpp
@@ -2064,8 +2064,6 @@ LibMesh::LibMesh(pugi::xml_node node) : UnstructuredMeshBase(node) {
   libMesh::ExplicitSystem& eq_sys =
     equation_systems_->add_system<libMesh::ExplicitSystem>(eq_system_name_);
 
-  sol_ = eq_sys.current_local_solution->clone();
-
   point_locator_ = m_->sub_point_locator();
   point_locator_->enable_out_of_mesh_mode();
 
@@ -2154,17 +2152,15 @@ LibMesh::set_variable(const std::string& var_name, int bin, double value) {
   std::vector<libMesh::dof_id_type> dof_indices;
   dof_map.dof_indices(e, dof_indices, var_num);
   Ensures(dof_indices.size() == 1);
-  (*sol_).set(dof_indices[0], value);
+  (*eqn_sys.solution).set(dof_indices[0], value);
 }
 
 void LibMesh::write() const {
-  // update solution
-  auto& eqn_sys = equation_systems_->get_system(eq_system_name_);
-  *eqn_sys.solution = *sol_;
-
   libMesh::ExodusII_IO exo(*m_);
   std::set<std::string> systems_out = {eq_system_name_};
-  exo.write_discontinuous_exodusII("test.exo", *equation_systems_, &systems_out);
+  std::stringstream output_filename;
+  output_filename << "umesh_" << id_ << ".exo";
+  exo.write_discontinuous_exodusII(output_filename.str(), *equation_systems_, &systems_out);
 }