LightFlavorRatios/unbinnedMaxLikelihood_fitter/createfilelist_2D.py

0001 import os
0002 import sys
0003 import math
0004
0005 # granular
0006 # bin_pt = [0.0, 0.5, 1.0, 1.5, 2.0, 8.0]
0007 # bin_eta = [-1.0, -0.75, -0.5, -0.25, 0.0, 0.25, 0.5, 0.75, 1.0]
0008 # bin_phi = [-3.15, -2.5, -2.0, -1.5, -1.0, -0.5, 0.0, 0.5, 1.0, 1.5, 2.0, 2.5, 3.15]
0009
0010 # coarse
0011 # bin_pt = [1.5, 8.0]
0012 # bin_eta = [-1.0, -0.5, 0.0, 0.5, 1.0]
0013 # bin_phi = [-3.15, -2.0, -1.0, 0.0, 1.0, 2.0, 3.15]
0014
0015 # for lambda
0016 bin_pt = [0.5, 0.6, 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.8, 2.1, 2.4, 2.7, 3.0, 4.0]
0017 # bin_pt = [0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.8, 2.1]
0018 # bin_pt = [1.8, 2.1]
0019 bin_phi = [-3.15, -2.0, -1.0, 0.0, 1.0, 2.0, 3.15]
0020
0021 # for kshort
0022 # bin_pt = [0.5, 0.6, 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3, 1.4, 1.5, 1.8, 2.1, 2.4, 2.7, 3.0, 4.0]
0023 # bin_eta = [-1.0, -0.5, 0.0, 0.5, 1.0]
0024 # bin_phi = [-3.15, -2.0, -1.0, 0.0, 1.0, 2.0, 3.15]
0025
0026 queuetext = ""
0027 idx = 0
0028 # for ipt in range(len(bin_pt)-1):
0029 #     print ("pt bin %d: %f - %f" % (ipt, bin_pt[ipt], bin_pt[ipt+1]))
0030 #     cut_pt = "K_S0_pT>={:.2f}&&K_S0_pT<{:.2f}".format(bin_pt[ipt], bin_pt[ipt+1])
0031 #     for ieta in range(len(bin_eta)-1):
0032 #         print ("eta bin %d: %f - %f" % (ieta, bin_eta[ieta], bin_eta[ieta+1]))
0033 #         cut_eta = "K_S0_pseudorapidity>={:.2f}&&K_S0_pseudorapidity<{:.2f}".format(bin_eta[ieta], bin_eta[ieta+1])
0034 #         for iphi in range(len(bin_phi)-1):
0035 #             print ("phi bin %d: %f - %f" % (iphi, bin_phi[iphi], bin_phi[iphi+1]))
0036 #             cut_phi = "K_S0_phi>={:.2f}&&K_S0_phi<{:.2f}".format(bin_phi[iphi], bin_phi[iphi+1])
0037 #             cut = "{}&&{}&&{}".format(cut_pt, cut_eta, cut_phi)
0038 #             queuetext += "{}\n".format(cut)
0039 #             idx += 1
0040
0041 for ipt in range(len(bin_pt)-1):
0042     print ("pt bin %d: %f - %f" % (ipt, bin_pt[ipt], bin_pt[ipt+1]))
0043     cut_pt = "Lambda0_pT>={:.2f}&&Lambda0_pT<{:.2f}".format(bin_pt[ipt], bin_pt[ipt+1])
0044     # only do phi bin for bins pt > 0.8 and pt < 2.7 bins
0045     if bin_pt[ipt] > 0.6 and bin_pt[ipt+1] < 2.4:
0046        for iphi in range(len(bin_phi)-1):
0047             print ("phi bin %d: %f - %f" % (iphi, bin_phi[iphi], bin_phi[iphi+1]))
0048             cut_phi = "Lambda0_phi>={:.2f}&&Lambda0_phi<{:.2f}".format(bin_phi[iphi], bin_phi[iphi+1])
0049             cut = "{}&&{}".format(cut_pt, cut_phi)
0050             queuetext += "{}\n".format(cut)
0051             idx += 1
0052     else:
0053         cut = "{}".format(cut_pt)
0054         queuetext += "{}\n".format(cut)
0055         idx += 1
0056
0057 # for ipt in range(len(bin_pt)-1):
0058 #     print ("pt bin %d: %f - %f" % (ipt, bin_pt[ipt], bin_pt[ipt+1]))
0059 #     cut_pt = "K_S0_pT>={:.2f}&&K_S0_pT<{:.2f}".format(bin_pt[ipt], bin_pt[ipt+1])
0060
0061 #     if bin_pt[ipt] > 0. and bin_pt[ipt+1] < 10.0:
0062 #        for iphi in range(len(bin_phi)-1):
0063 #             print ("phi bin %d: %f - %f" % (iphi, bin_phi[iphi], bin_phi[iphi+1]))
0064 #             cut_phi = "K_S0_phi>={:.2f}&&K_S0_phi<{:.2f}".format(bin_phi[iphi], bin_phi[iphi+1])
0065 #             cut = "{}&&{}".format(cut_pt, cut_phi)
0066 #             queuetext += "{}\n".format(cut)
0067 #             idx += 1
0068 #     else:
0069 #         cut = "{}".format(cut_pt)
0070 #         queuetext += "{}\n".format(cut)
0071 #         idx += 1
0072
0073 with open("./queue_lambda.list", "w") as f:
0074 # with open("./queue_kshort.list", "w") as f:
0075     f.write(queuetext)
0076 f.close()