add two options to set a flat mixing, and the bunchspace

cms-sw · Dec 15, 2024 · ba7c1fd · ba7c1fd
1 parent 0b35058
commit ba7c1fd
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/Configuration/StandardSequences/python/Mixing.py b/Configuration/StandardSequences/python/Mixing.py
@@ -222,4 +222,18 @@ def defineMixing(dict):
     if 'F' in dict:
         commands.append('process.mix.input.fileNames = cms.untracked.vstring(%s)'%(repr(dict['F'])))
         dict.pop('F')
+    if 'BS' in dict:
+        bunch_space = dict['BS']
+        commands.append(f'process.mix.bunchspace = cms.int32({bunch_space})')
+        dict.pop('BS')
+    if 'Flat' in dict:
+        pu_min,pu_max=dict['Flat']
+        pu_x = list(range(pu_max+1))
+        pu_y = [0]*(pu_max+1)
+        prob=1./(pu_max+1-pu_min)
+        for pu in range(pu_min,pu_max+1):
+            pu_y[pu]=prob
+        commands.append(f'process.mix.input.nbPileupEvents.probFunctionVariable = cms.vint32({pu_x})')
+        commands.append(f'process.mix.input.nbPileupEvents.probValue = cms.vdouble({pu_y})')
+        dict.pop('Flat')
     return commands