|
8 | 8 | parser = argparse.ArgumentParser(description='Make Starlight configuration', |
9 | 9 | formatter_class=argparse.ArgumentDefaultsHelpFormatter) |
10 | 10 |
|
11 | | -parser.add_argument('--pZ', type=int, default='82', |
12 | | - help='Projectile charge') |
| 11 | +parser.add_argument('--collType',default='PbPb', choices=['PbPb', 'pPb', 'Pbp', 'pp', 'OO', 'pO', 'Op'], |
| 12 | + help='Colission system') |
13 | 13 |
|
14 | | -parser.add_argument('--pA', type=int, default='208', |
15 | | - help='Projectile atomic number') |
16 | | - |
17 | | -parser.add_argument('--tZ', type=int, default='82', |
18 | | - help='Target charge') |
19 | | - |
20 | | -parser.add_argument('--tA', type=int, default='208', |
21 | | - help='Target atomic number') |
22 | | - |
23 | 14 | parser.add_argument('--eCM', type=float, default='5360', |
24 | 15 | help='Centre-of-mass energy') |
25 | 16 |
|
|
36 | 27 |
|
37 | 28 | args = parser.parse_args() |
38 | 29 |
|
| 30 | +if 'PbPb' in args.collType: |
| 31 | + pZ = 82 |
| 32 | + pA = 208 |
| 33 | + tZ = 82 |
| 34 | + tA = 208 |
| 35 | +if 'pPb' in args.collType: |
| 36 | + pZ = 1 |
| 37 | + pA = 1 |
| 38 | + tZ = 82 |
| 39 | + tA = 208 |
| 40 | +if 'Pbp' in args.collType: |
| 41 | + pZ = 82 |
| 42 | + pA = 208 |
| 43 | + tZ = 1 |
| 44 | + tA = 1 |
| 45 | +if 'pp' in args.collType: |
| 46 | + pZ = 1 |
| 47 | + pA = 1 |
| 48 | + tZ = 1 |
| 49 | + tA = 1 |
| 50 | +if 'OO' in args.collType: |
| 51 | + pZ = 8 |
| 52 | + pA = 16 |
| 53 | + tZ = 8 |
| 54 | + tA = 16 |
| 55 | +if 'pO' in args.collType: |
| 56 | + pZ = 1 |
| 57 | + pA = 1 |
| 58 | + tZ = 8 |
| 59 | + tA = 16 |
| 60 | +if 'Op' in args.collType: |
| 61 | + pZ = 8 |
| 62 | + pA = 16 |
| 63 | + tZ = 1 |
| 64 | + tA = 1 |
| 65 | + |
39 | 66 | ### open output file |
40 | 67 | fout = open(args.output, 'w') |
41 | 68 |
|
42 | 69 | ### Generator |
43 | 70 | fout.write('[GeneratorExternal] \n') |
44 | 71 | if 'Psi2sToMuPi' in args.process or 'Psi2sToElPi' in args.process or 'RhoPrime' in args.process or 'OmegaTo3Pi' in args.process or 'JpsiToElRad' in args.process : |
45 | 72 | fout.write('fileName = ${O2DPG_ROOT}/MC/config/PWGUD/external/generator/GeneratorStarlightToEvtGen.C \n') |
46 | | - fout.write('funcName = GeneratorStarlightToEvtGen("%s", %f, %d, %d, %d, %d) \n' % (args.process,args.eCM ,args.pZ ,args.pA,args.tZ,args.tA)) |
| 73 | + fout.write('funcName = GeneratorStarlightToEvtGen("%s", %f, %d, %d, %d, %d) \n' % (args.process,args.eCM ,pZ,pA,tZ,tA)) |
47 | 74 | else: |
48 | 75 | fout.write('fileName = ${O2DPG_ROOT}/MC/config/PWGUD/external/generator/GeneratorStarlight.C \n') |
49 | | - fout.write('funcName = GeneratorStarlight("%s", %f, %d, %d, %d, %d) \n' % (args.process,args.eCM ,args.pZ ,args.pA,args.tZ,args.tA)) |
| 76 | + fout.write('funcName = GeneratorStarlight("%s", %f, %d, %d, %d, %d) \n' % (args.process,args.eCM ,pZ,pA,tZ,tA)) |
50 | 77 |
|
51 | 78 | ###Trigger |
52 | 79 | fout.write('[TriggerExternal] \n') |
|
0 commit comments