time_structure.patch - MAUS - MICEmine

Feature #1872 » time_structure.patch

Rogers, Chris, 16 September 2016 12:16

     # Dictionary of variable to be set when using G4BL to generate particles. If
     # "get_magnet_currents_pa_cdb" is set to True magnet currents & proton absorber
     # thickness will be retrieved from the CDB for the run_number specified
     g4bl = {"run_number":2873,"q_1":1.066,"q_2":-1.332,"q_3":0.927,"d_1":-1.302,"d_2":-0.396,\
             "d_s":3.837,"particles_per_spill":0,"rotation_angle":0,"translation_z":1000.0,\
             "protonabsorberin":1,"proton_absorber_thickness":93,"proton_number":1E9,"proton_weight":1,\
             "particle_charge":'all',"file_path":'MAUS_ROOT_DIR/src/map/MapPyBeamlineSimulation/G4bl',\
         "get_magnet_currents_pa_cdb":False,"random_seed":1,
     g4bl = {
         "run_number":2873,
         "q_1":1.066,
         "q_2":-1.332,
         "q_3":0.927,
         "d_1":-1.302,
         "d_2":-0.396,
         "d_s":3.837,
         "particles_per_spill":0,
         "rotation_angle":0,
         "translation_z":1000.0,
         "protonabsorberin":1,
         "proton_absorber_thickness":93,
         "proton_number":1E9,
         "proton_weight":1,
         "particle_charge":'all',
         "file_path":'MAUS_ROOT_DIR/src/map/MapPyBeamlineSimulation/G4bl',
         "get_magnet_currents_pa_cdb":False,
         "random_seed":1,
         "seed_algorithm":"random_seed_and_spill_number",
         "time_structure":"uniform",
         "spill_gate":3e6, # 3e6 ns = 3 ms
+    }
     # Used by MapPyRemoveTracks.

             self.seed_algorithm = "use_random_seed"
             self.master_seed = 0
             self.random_seed = 0
             self.time_structure = "none"
             self.spill_gate = 0.
         def birth(self, json_configuration): #pylint: disable=R0912, R0915
             "birth doc string"
-...
                     print('Error: Number of protons must be greater than zero!')
                     good_birth = False
             except Exception: #pylint: disable=W0703
                 print("Error: proton_numer is not found in the config file!")
                 print("Error: proton_number is not found in the config file!")
                 good_birth = False
             try:
-...
                 self.seed_algorithm = "use_random_seed"
             try:
                 self.time_structure = config_doc["g4bl"]["time_structure"]
                 if self.time_structure not in ["uniform", "none"]:
                     raise ValueError("time structure "+self.time_structure+\
                                      " should be 'uniform' or 'none'")
                 if self.time_structure == "uniform":
                     self.spill_gate = float(config_doc["g4bl"]["spill_gate"])
                     if self.spill_gate < 0.:
                         raise ValueError("spill gate should be > 0.")
             except KeyError:
                 print config_doc["g4bl"]
                 print("Error: time_structure set up error in the config file!")
                 good_birth = False
             try:
                 self.random_seed = config_doc["g4bl"]["random_seed"]
                 if self.random_seed == -1:
                     self.random_seed = random.randint(1, 1000000)
-...
                     for i in range(self.queue.qsize()):
                         primary = (self.queue.get_nowait())
                         spill['mc_events'].append({"primary":primary})
                     self.apply_time_structure(spill)
                     with open('G4BLoutput.json', 'w') as outfile:
                         json.dump(spill, outfile)
                         #self.queue = 0
                         #self.queue = 0
                     return json.dumps(spill)
                 else:
                     spill = {}
-...
                         for i in range(self.particles_per_spill):
                             primary = (self.queue.get_nowait())
                             spill['mc_events'].append({"primary":primary})
                         self.apply_time_structure(spill)
                         with open('G4BLoutputsimulation'+str(i)+'.json', 'w') \
                                                                   as outfile:
                             json.dump(spill, outfile)
-...
                         for i in range(self.particles_per_spill):
                             primary = (self.queue.get_nowait())
                             spill['mc_events'].append({"primary":primary})
                         self.apply_time_structure(spill)
                         with open('G4BLoutputsimulation'+str(i)+'.json', 'w') \
                                                                   as outfile:
                             json.dump(spill, outfile)
-...
             else:
                 print("Warning: G4BL simulated zero output particles!")
                 raise IndexError("G4BL simulated zero output particles")
         def apply_time_structure(self, spill):
             """
             apply time structure corresponding to the MICE spill gate
             - if self.time_structure is uniform, add time uniformly between 0. and
               spill gate to the G4BL output time structure
             """
             if self.time_structure == "uniform":
                 for event in spill["mc_events"]:
                     numpy.random.seed(event["primary"]["random_seed"])
                     event["primary"]["time"] += \
                                           numpy.random.uniform(0., self.spill_gate)
         def death(self): #pylint: disable = R0201
             """

     adjusting number of protons from target.
     """ #pylint: disable = W0105
     TEST_1 = {
         "g4bl":{"q_1":1.16, "q_2":-1.45, "q_3":1.006, "d_1":-1.41,\
         "g4bl":{"q_1":1.16, "q_2":-1.45, "q_3":1.006, "d_1":-1.41, "d_2":-0.7,\
         "proton_absorber_thickness":11,'d_s':200,\
         "proton_number":500000000,'particles_per_spill':0,\
         "particle_charge":"all", 'rotation_angle':0, 'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,
         "proton_weight":1,
         "protonabsorberin":0,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
-...
         "proton_number":500000000,'particles_per_spill':200,\
         "particle_charge":"all", 'rotation_angle':0, 'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
-...
         "proton_number":500000000,'particles_per_spill':500,\
         "particle_charge":"all", 'rotation_angle':0, 'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
-...
         "proton_number":10000,'particles_per_spill':200,\
         "particle_charge":"all", 'rotation_angle':0, 'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
-...
         "proton_number":-10000,'particles_per_spill':200,\
         "particle_charge":"all", 'rotation_angle':0, 'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
-...
         "particle_charge":"1.21 jiggawatts", 'rotation_angle':0,\
         'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
+        }
+    }
-...
         "particle_charge":"negative", 'rotation_angle':0,\
         'translation_z':20000.0,\
         "random_seed":0,"run_number":0,"get_magnet_currents_pa_cdb":False,\
         "proton_weight":1
         "proton_weight":1,
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
     """
     Test random seed - seed from spill
     """ #pylint: disable = W0105
     TEST_8 = {
         "g4bl":{"q_1":1.16, "q_2":-1.45, "q_3":1.006, "d_1":-1.41, "d_2":1.,
-...
         "random_seed":-2,"run_number":0,"get_magnet_currents_pa_cdb":False,
         "proton_weight":1,
         "seed_algorithm":"random_seed_and_spill_number",
         "time_structure":"none",
         "spill_gate":3e6,
+        }
+    }
     """
     Test time structure - uniform
     """ #pylint: disable = W0105
     TEST_9 = {
         "g4bl":{"q_1":1.16, "q_2":-1.45, "q_3":1.006, "d_1":-1.41, "d_2":1.,
         "proton_absorber_thickness":11,'d_s':200,
         "proton_number":100,'particles_per_spill':1000,
         "particle_charge":"positive", 'rotation_angle':0,
         'translation_z':20000.0,
         "random_seed":-2,"run_number":0,"get_magnet_currents_pa_cdb":False,
         "proton_weight":1,
         "seed_algorithm":"random_seed_and_spill_number",
         "time_structure":"uniform",
         "spill_gate":3e6,
+        }
+    }
-...
             """ Initialise the class """
             self.particles = {}
             for i in range(self.number_of_entries):
                 self.particles['entry'+str(i)] = {}
                 self.particles['entry'+str(i)] = {
                   "position":{"x":0.0, "y":-0.0, "z":-0.0},
                   "momentum":{"x":0.0, "y":0.0, "z":1.0},
                   "particle_id":-13,
                   "energy":226.0, "time":0.0,
                   "random_seed":i,
                   "spin":{"x":0.0, "y":0.0, "z":1.0}
+                }
         number_of_entries = 10
-...
             self.assertNotEqual(self.g4bl.random_seed, random_seed)
             MapPyBeamlineSimulation.CallG4bl = CallG4bl
         def test_mappybeamline_time_structure(self):
             """ Test mappybeamline time structure"""
             MapPyBeamlineSimulation.CallG4bl = MockCallG4bl
             self.g4bl = MapPyBeamlineSimulation.MapPyBeamlineSimulation()
             self.g4bl.birth(json.dumps(TEST_8))
             spill = self.g4bl.process(json.dumps({"spill_number":0}))
             spill = json.loads(spill)
             for event in spill["mc_events"]:
                 self.assertAlmostEqual(event["primary"]["time"], 0.)
             self.g4bl.death()
             self.g4bl.birth(json.dumps(TEST_9))
             spill = self.g4bl.process(json.dumps({"spill_number":0}))
             spill = json.loads(spill)
             time_list = [event["primary"]["time"] for event in spill["mc_events"]]
             bin_1, bin_2 = 0, 0
             for time in time_list:
                 if time > 0.-1e-9 and time < 1.5e6:
                     bin_1 += 1
                 elif time >= 1.5e6 and time < 3.0e6+1e-9:
                     bin_2 += 1
             self.assertEqual(bin_1+bin_2, 1000)
             # should do KS test or something; this will do
             self.assertGreater(bin_1, 100)
             self.assertGreater(bin_2, 100)
             self.g4bl.death()
     if __name__ == "__main__":
         unittest.main()

(1-1/1)

Project

General

Profile

Feature #1872 » time_structure.patch