From b3f41eebeb220427cdbb58c2b3255f1085c23000 Mon Sep 17 00:00:00 2001
From: YoungHui Park <ypark234@wisc.edu>
Date: Fri, 11 Dec 2020 03:34:07 -0600
Subject: [PATCH] Add missing surface source flag on DagMC surface

---
 src/dagmc.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/dagmc.cpp b/src/dagmc.cpp
index 362f9a6e48..2c20f6f94c 100644
--- a/src/dagmc.cpp
+++ b/src/dagmc.cpp
@@ -2,6 +2,7 @@
 
 #include "openmc/cell.h"
 #include "openmc/constants.h"
+#include "openmc/container_util.h"
 #include "openmc/error.h"
 #include "openmc/file_utils.h"
 #include "openmc/geometry.h"
@@ -296,6 +297,10 @@ void load_dagmc_geometry()
     s->id_ = model::DAG->id_by_index(2, s->dag_index_);
     s->dagmc_ptr_ = model::DAG;
 
+    if (contains(settings::src_write_surf_id, s->id_)) {
+      s->surf_src_ = true;
+    }
+
     // set BCs
     std::string bc_value = DMD.get_surface_property("boundary", surf_handle);
     to_lower(bc_value);