Fixing capabilities on continuous energy mode

mcdc/card.py

@@ -82,7 +82,7 @@ def reset(self):
             "ww_mesh": make_card_mesh(),
             "weight_roulette": False,
             "wr_threshold": 0.0,
-            "wr_chance": 1.0,
+            "wr_survive": 1.0,
             "iQMC": False,
             "iqmc": {
                 "generator": "sobol",

mcdc/constant.py

@@ -65,6 +65,11 @@
 XS_SCATTER = 1
 XS_CAPTURE = 2
 XS_FISSION = 3
-NU_TOTAL = 0
-NU_PROMPT = 1
-NU_DELAYED = 2
+XS_NU_FISSION = 4
+XS_NU_FISSION_PROMPT = 5
+XS_NU_FISSION_DELAYED = 6
+XS_NU_SCATTER = 7
+
+NU_FISSION = 0
+NU_FISSION_PROMPT = 1
+NU_FISSION_DELAYED = 2

mcdc/input_.py

@@ -1303,17 +1303,16 @@ def iQMC(
     card["iqmc"]["krylov_restart"] = krylov_restart
 
 
-def weight_roulette(chance, wr_threshold):
+def weight_roulette(w_threshold=0.2, w_survive=1.0):
     """
-    If neutron weight is below wr_threshold, then enter weight rouelette
-    technique. Neutron has 'chance' probability of having its weight increased
-    by factor of 1/CHANCE, and 1-CHANCE probability of terminating.
+    If neutron weight is below `w_threshold`, then enter weight rouelette
+    technique with survival weight `w_survive`.
 
     Parameters
     ----------
-    chance : probability of survival
-    wr_threshold : weight_roulette() is called on a particle
-                    if P['w'] <= wr_threshold
+    w_threshold : weight_roulette() is called on a particle if
+                  P['w'] <= wr_threshold
+    w_survive : weight survival
 
     Returns
     -------
@@ -1322,8 +1321,8 @@ def weight_roulette(chance, wr_threshold):
     """
     card = mcdc.input_deck.technique
     card["weight_roulette"] = True
-    card["wr_chance"] = chance
-    card["wr_threshold"] = wr_threshold
+    card["wr_threshold"] = w_threshold
+    card["wr_survive"] = w_survive
 
 
 def IC_generator(