Added NEST-like skew to NR

FlamTeam · Mar 19, 2024 · cd30459 · cd30459
1 parent cead6ae
commit cd30459
Show file tree

Hide file tree

Showing 5 changed files with 534 additions and 155 deletions.
diff --git a/flamedisx/block_source.py b/flamedisx/block_source.py
@@ -361,6 +361,7 @@ def _differential_rate(self, data_tensor, ptensor):
         already_stepped = ()  # Avoid double-multiplying to account for stepping
 
         for b in self.model_blocks:
+            # print('block: ',b)
             b_dims = b.dimensions
             # These are the the dimensions we will do variable stepping over
             scaling_dims = b.dimensions + tuple([bonus_dimension[0] for

diff --git a/flamedisx/dd_migdal/dd_migdal_blocks/energy_spectra.py b/flamedisx/dd_migdal/dd_migdal_blocks/energy_spectra.py
@@ -52,13 +52,13 @@ def _compute(self, data_tensor, ptensor, *, energy_first):
 
         spectrum *= self.source.mu_before_efficiencies()
 
-        spectrum *= tf.repeat(self.gimme('get_r_dt_diff_rate',
+        spectrum *= tf.repeat(self.gimme('get_r_dt_diff_rate', ### turn off to test s1s2 diff rates alone! TODO
                                          data_tensor=data_tensor,
                                          ptensor=ptensor)[:, o],
                               tf.shape(self.energies_first),
                               axis=1)
 
-        spectrum *= tf.repeat(self.gimme('get_S2Width_diff_rate',
+        spectrum *= tf.repeat(self.gimme('get_S2Width_diff_rate',  ### turn off to test s1s2 diff rates alone! TODO
                                          data_tensor=data_tensor,
                                          ptensor=ptensor)[:, o],
                               tf.shape(self.energies_first),