Belle II Software  release-06-02-00
ImportStripEfficiency.py
1 #!/usr/bin/env python3
2 # -*- coding: utf-8 -*-
3 
4 
11 
12 # Import KLM strip efficiency payloads.
13 
14 import sys
15 from ROOT.Belle2 import KLMDatabaseImporter, KLMStripEfficiency, KLMChannelIndex
16 
17 mc = False
18 if (len(sys.argv) < 2):
19  print('Usage: basf2 ImportStripEfficiency.py [mc | input_file].')
20 elif (sys.argv[1] == 'mc'):
21  mc = True
22 else:
23  inputFile = sys.argv[1]
24 
25 dbImporter = KLMDatabaseImporter()
26 
27 stripEfficiency = KLMStripEfficiency()
28 
29 if (mc):
30  index = KLMChannelIndex()
31  index2 = KLMChannelIndex()
32  while (index != index2.end()):
33  channel = index.getKLMChannelNumber()
34  stripEfficiency.setEfficiency(channel, 1.0, 0.0)
35  index.increment()
36 
37  dbImporter.setIOV(0, 0, 0, -1)
38  dbImporter.importStripEfficiency(stripEfficiency)
39 
40  dbImporter.setIOV(1002, 0, 1002, -1)
41  dbImporter.importStripEfficiency(stripEfficiency)
42 
43  dbImporter.setIOV(1003, 0, 1003, -1)
44  dbImporter.importStripEfficiency(stripEfficiency)
45 
46 else:
47  dbImporter.loadStripEfficiency(stripEfficiency, inputFile)
48 
49  dbImporter.setIOV(0, 0, -1, -1)
50  dbImporter.importStripEfficiency(stripEfficiency)