PandAna.var.analysis_vars Namespace Reference

Functions
def	kLongestProng (tables)
	Nue Vars. More...
def	kMaxY (tables)
def	kNueCalibrationCorr (det, ismc, run)
def	kEMEnergy (tables)
def	kHadEnergy (tables)
def	kNueEnergy (tables)
def	kCosNumi (tables)
	Numu Vars. More...
def	kNumuMuE (tables)
def	kNumuHadE (tables)
def	NusScale (isRHC, det)
	Nus Vars. More...
def	kNusEnergy (tables)

Variables
	kCVNe = Var(lambda tables: tables['rec.sel.cvn2017']['nueid'])
	General Vars. More...
	kCVNm = Var(lambda tables: tables['rec.sel.cvn2017']['numuid'])
	kCVNnc = Var(lambda tables: tables['rec.sel.cvn2017']['ncid'])
	kNHit = Var(lambda tables: tables['rec.slc']['nhit'])
	kRHC = Var(lambda tables: tables['rec.spill']['isRHC'])
	kDetID = Var(lambda tables: tables['rec.hdr']['det'])
	kMeantime = Var(lambda tables: tables['rec.slc']['meantime'])
	kCVNe20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['nueid'])
	kCVNm20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['numuid'])
	kCVNnc20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['ncid'])
	kCVNcos20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['cosmicid'])
	kLongestProng = Var(kLongestProng)
	kDistAllTop = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngtop'])
	kDistAllBottom = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngbottom'])
	kDistAllWest = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngwest'])
	kDistAllEast = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngeast'])
	kDistAllBack = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngback'])
	kDistAllFront = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngfront'])
	kHitsPerPlane = Var(lambda tables: tables['rec.sel.nuecosrej']['hitsperplane'])
	kPtP = Var(lambda tables: tables['rec.sel.nuecosrej']['partptp'])
	kMaxY = Var(kMaxY)
	kSparsenessAsymm = Var(lambda tables: tables['rec.sel.nuecosrej']['sparsenessasymm'])
	kCaloE = Var(lambda tables: tables['rec.slc']['calE'])
	kEMEnergy = Var(kEMEnergy)
	kHadEnergy = Var(kHadEnergy)
	kNueEnergy = Var(kNueEnergy)
	kCosNumi = Var(kCosNumi)
	kNumuMuE = Var(kNumuMuE)
	kNumuHadE = Var(kNumuHadE)
	kNumuE = kNumuMuE+kNumuHadE
	kCCE = kNumuE
	kNusEnergy = Var(kNusEnergy)
	kClosestSlcTime = Var(lambda tables: tables['rec.slc']['closestslicetime'])
	kClosestSlcMinDist = Var(lambda tables: tables['rec.slc']['closestslicemindist'])
	kClosestSlcMinTop = Var(lambda tables: tables['rec.slc']['closestsliceminfromtop'])

Function Documentation

def PandAna.var.analysis_vars.kCosNumi

(

tables

)

Numu Vars.

Definition at line 169 of file analysis_vars.py.

References PandAna.var.analysis_vars.kCosNumi, PandAna.var.analysis_vars.kDetID, and sum.

def kCosNumi(tables):
    df = tables['rec.trk.kalman.tracks']
    # Primary kalman track only
    df = df[df['rec.trk.kalman.tracks_idx']==0]
    KalDir = df[['dir.x','dir.y', 'dir.z']]

    # Use separate beam dir for each detector
    KalDirFD = KalDir[kDetID(tables)==detector.kFD]
    KalDirND = KalDir[kDetID(tables)==detector.kND]

    CosFD = KalDirFD.mul(BeamDirFD, axis=1).sum(axis=1)
    CosND = KalDirND.mul(BeamDirND, axis=1).sum(axis=1)

    return pd.concat([CosFD, CosND])

def PandAna.var.analysis_vars.kEMEnergy

(

tables

)

Definition at line 73 of file analysis_vars.py.

References PandAna.var.analysis_vars.kEMEnergy, PandAna.var.analysis_vars.kLongestProng, PandAna.var.analysis_vars.kNueCalibrationCorr(), PandAna.var.analysis_vars.kRHC, and sum.

def kEMEnergy(tables):
    lng_png = kLongestProng(tables)
    isRHC = kRHC(tables)

    shwlid_df = tables['rec.vtx.elastic.fuzzyk.png.shwlid']
    prim_png = shwlid_df['calE'][(shwlid_df['rec.vtx.elastic.fuzzyk.png_idx']==0)]

  
    png_df = tables['rec.vtx.elastic.fuzzyk.png']
    cvn_png_df = tables['rec.vtx.elastic.fuzzyk.png.cvnpart']
    
    cvn_em_pid_df = cvn_png_df[['photonid',  \
                                'pizeroid',  \
                                'electronid']].sum(axis=1)
    cvn_had_pid_df = cvn_png_df[['pionid',   \
                                 'protonid', \
                                 'neutronid',\
                                 'otherid',  \
                                 'muonid']].sum(axis=1)

    cvn_em_calE = shwlid_df['calE'].where(                     \
                            (cvn_em_pid_df > 0) &              \
                            (cvn_em_pid_df >= cvn_had_pid_df), \
                            0).groupby(level=KL).agg(np.sum)

    if isRHC.agg(np.all):
      cvn_em_calE[cvn_em_calE == 0] = prim_png[cvn_em_calE == 0]
    else:
      cvn_em_calE[lng_png >= 500] = prim_png[lng_png >= 500]

    hdr_df = tables['rec.hdr']
    det = hdr_df['det']
    ismc = tables['rec.hdr']['ismc']
    runs = hdr_df.assign(run=hdr_df.index.get_level_values('run'))['run']
    df = pd.concat([runs, det, ismc], axis=1).dropna()
    scale = df.apply(lambda x: kNueCalibrationCorr(x['det'], x['ismc'], x['run']), axis=1, result_type='reduce')

    cvn_em_calE *= scale
    cvn_em_calE.name = 'calE'

    return cvn_em_calE

def PandAna.var.analysis_vars.kHadEnergy

(

tables

)

Definition at line 116 of file analysis_vars.py.

References PandAna.var.analysis_vars.kEMEnergy, PandAna.var.analysis_vars.kHadEnergy, and PandAna.var.analysis_vars.kNueCalibrationCorr().

def kHadEnergy(tables):
    EMEnergy = kEMEnergy(tables)
    
    hdr_df = tables['rec.hdr']
    det = hdr_df['det']
    ismc = tables['rec.hdr']['ismc']
    runs = hdr_df.assign(run=hdr_df.index.get_level_values('run'))['run']
    df = pd.concat([runs, det, ismc], axis=1).dropna()
    scale = df.apply(lambda x: kNueCalibrationCorr(x['det'], x['ismc'], x['run']), axis=1, result_type='reduce')
   
    calE = tables['rec.slc']['calE']*scale
    calE.name = 'calE'
   
    HadEnergy = calE - EMEnergy
    return HadEnergy.where(HadEnergy > 0, 0)

def PandAna.var.analysis_vars.kLongestProng

(

tables

)

Nue Vars.

Definition at line 41 of file analysis_vars.py.

References PandAna.var.analysis_vars.kLongestProng.

def kLongestProng(tables):
    df = tables['rec.vtx.elastic.fuzzyk.png']['len']
    return df.groupby(level=KL).agg(np.max)    

def PandAna.var.analysis_vars.kMaxY

(

tables

)

Definition at line 57 of file analysis_vars.py.

References PandAna.var.analysis_vars.kMaxY, and cet::sqlite.max().

def kMaxY(tables):
    df = tables['rec.vtx.elastic.fuzzyk.png.shwlid']
    df = df[['start.y','stop.y']].max(axis=1)
    return df.groupby(level=KL).agg(np.max)

def PandAna.var.analysis_vars.kNueCalibrationCorr	(		det,
			ismc,
			run
	)

Definition at line 67 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kEMEnergy(), and PandAna.var.analysis_vars.kHadEnergy().

def kNueCalibrationCorr(det, ismc, run):
  if (det != detector.kFD): return 1.
  if not ismc: return 0.9949
  if run < 20753: return 0.9949/0.9844
  return 1.

def PandAna.var.analysis_vars.kNueEnergy

(

tables

)

Definition at line 133 of file analysis_vars.py.

References PandAna.var.analysis_vars.kEMEnergy, PandAna.var.analysis_vars.kHadEnergy, PandAna.var.analysis_vars.kNueEnergy, and PandAna.var.analysis_vars.kRHC.

def kNueEnergy(tables):
    EMEnergy = kEMEnergy(tables)
    HadEnergy = kHadEnergy(tables)
    isRHC = kRHC(tables)

    p0 =  0.0
    p1 =  1.00756
    p2 =  1.07093
    p3 =  0.0
    p4 =  1.28608e-02
    p5 =  2.27129e-01
    norm = 0.0501206
    if isRHC.agg(np.all):
      p0 = 0.0
      p1 = 0.980479
      p2 = 1.45170
      p3 = 0.0
      p4 = -5.82609e-03
      p5 = -2.27599e-01
      norm = 0.001766
  
    NueEnergy = 1./(1+norm)*(HadEnergy*HadEnergy*p5 + \
                                     EMEnergy*EMEnergy*p4 +   \
                                     EMEnergy*HadEnergy*p3 +  \
                                     HadEnergy*p2 +              \
                                     EMEnergy*p1 + p0)
    return  NueEnergy.where((HadEnergy >= 0) & (EMEnergy >= 0), -5.)

def PandAna.var.analysis_vars.kNumuHadE

(

tables

)

Definition at line 220 of file analysis_vars.py.

References PandAna.utils.misc.GetPeriod(), PandAna.var.numuE_utils.GetSpline(), PandAna.var.analysis_vars.kDetID, PandAna.var.analysis_vars.kNumuHadE, and PandAna.var.analysis_vars.kRHC.

def kNumuHadE(tables):
  det = kDetID(tables)
  isRHC = kRHC(tables)

  hdr_df = tables['rec.hdr']
  runs = hdr_df.assign(run=hdr_df.index.get_level_values('run'))['run']
  
  hadvisE = tables['rec.energy.numu']['hadtrkE'] + tables['rec.energy.numu']['hadcalE']
  hadvisE.name = 'hadvisE'
  ntracks = (tables['rec.trk.kalman']['ntracks'] != 0)
  
  if (det == detector.kFD).agg(np.all) and not det.empty:
    ismc = tables['rec.hdr']['ismc']
    if not ismc.agg(np.all) and not ismc.empty:
      periods = runs.apply(lambda x: GetPeriod(x, det))
      hadvisE[periods > 2] = 0.9949*hadvisE[periods > 2]
      hadvisE[periods <= 2] = 0.9844*hadvisE[periods <= 2]
  
  df = pd.concat([runs, det, isRHC, hadvisE], axis=1)
  hadE = df.apply(lambda x: \
                 GetSpline(x['run'], x['det'], x['isRHC'], "had")(x['hadvisE']), axis = 1)
  return hadE.where(ntracks, -5.)

def PandAna.var.analysis_vars.kNumuMuE

(

tables

)

Definition at line 185 of file analysis_vars.py.

References PandAna.var.numuE_utils.GetSpline(), PandAna.var.analysis_vars.kDetID, PandAna.var.analysis_vars.kNumuMuE, and PandAna.var.analysis_vars.kRHC.

def kNumuMuE(tables):
  det = kDetID(tables)
  isRHC = kRHC(tables)
  
  hdr_df = tables['rec.hdr']
  runs = hdr_df.assign(run=hdr_df.index.get_level_values('run'))['run']
  ntracks = (tables['rec.trk.kalman']['ntracks'] != 0)
  
  if (det == detector.kFD).agg(np.all) and not det.empty:
    ismc = tables['rec.hdr']['ismc']
    
    trks = tables['rec.trk.kalman.tracks']
    trklen = trks['len'][(trks['rec.trk.kalman.tracks_idx'] == 0)]/100
    if not ismc.agg(np.all) and not ismc.empty:
      trklen = trklen*0.9957
    
    df = pd.concat([runs, det, isRHC, trklen], axis=1).dropna()
    muE = df.apply(lambda x: \
                   GetSpline(x[0], x[1], x[2], "muon")(x[3]), axis = 1)
    return muE.where(ntracks, -5.)
  
  else:
    trklenact = tables['rec.energy.numu']['ndtrklenact']/100.
    trklencat = tables['rec.energy.numu']['ndtrklencat']/100.
    trkcalactE = tables['rec.energy.numu']['ndtrkcalactE']
    trkcaltranE = tables['rec.energy.numu']['ndtrkcaltranE']

    df = pd.concat([runs, det, isRHC, trklenact, trklencat], axis=1)
    muE = df.apply(lambda x: \
                   GetSpline(x['run'], x['det'], x['isRHC'], "act")(x['ndtrklenact']) + \
                   GetSpline(x['run'], x['det'], x['isRHC'], "cat")(x['ndtrklencat']), axis = 1)
    muE[(trkcalactE == 0.) & (trkcaltranE == 0.)] = -5.
    return muE.where(ntracks, -5.)

def PandAna.var.analysis_vars.kNusEnergy

(

tables

)

Definition at line 260 of file analysis_vars.py.

References PandAna.var.analysis_vars.kCaloE, PandAna.var.analysis_vars.kDetID, PandAna.var.analysis_vars.kNusEnergy, PandAna.var.analysis_vars.kRHC, and PandAna.var.analysis_vars.NusScale().

def kNusEnergy(tables):
  det = kDetID(tables)
  isRHC = kRHC(tables)
  cale = kCaloE(tables)
  df = pd.concat([det, isRHC, cale], axis=1)
  return df.apply(lambda x: \
                 NusScale(x['isRHC'], x['det'])*x['calE'], axis = 1)

def PandAna.var.analysis_vars.NusScale	(		isRHC,
			det
	)

Nus Vars.

Definition at line 253 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNusEnergy().

def NusScale(isRHC, det):
  if (isRHC and det == detector.kFD): return 1.18
  if (isRHC and det == detector.kND): return 1.15
  if (not isRHC and det == detector.kFD): return 1.2
  if (not isRHC and det == detector.kND): return 1.11
  else: return -5.

Variable Documentation

PandAna.var.analysis_vars.kCaloE = Var(lambda tables: tables['rec.slc']['calE'])

Definition at line 65 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNusEnergy().

PandAna.var.analysis_vars.kCCE = kNumuE

Definition at line 245 of file analysis_vars.py.

PandAna.var.analysis_vars.kClosestSlcMinDist = Var(lambda tables: tables['rec.slc']['closestslicemindist'])

Definition at line 270 of file analysis_vars.py.

PandAna.var.analysis_vars.kClosestSlcMinTop = Var(lambda tables: tables['rec.slc']['closestsliceminfromtop'])

Definition at line 271 of file analysis_vars.py.

PandAna.var.analysis_vars.kClosestSlcTime = Var(lambda tables: tables['rec.slc']['closestslicetime'])

Definition at line 269 of file analysis_vars.py.

PandAna.var.analysis_vars.kCosNumi = Var(kCosNumi)

Definition at line 183 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kCosNumi().

PandAna.var.analysis_vars.kCVNcos20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['cosmicid'])

Definition at line 31 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNe = Var(lambda tables: tables['rec.sel.cvn2017']['nueid'])

General Vars.

Definition at line 16 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNe20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['nueid'])

Definition at line 28 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNm = Var(lambda tables: tables['rec.sel.cvn2017']['numuid'])

Definition at line 17 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNm20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['numuid'])

Definition at line 29 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNnc = Var(lambda tables: tables['rec.sel.cvn2017']['ncid'])

Definition at line 18 of file analysis_vars.py.

PandAna.var.analysis_vars.kCVNnc20 = Var(lambda tables: tables['rec.sel.cvnloosepreselptp']['ncid'])

Definition at line 30 of file analysis_vars.py.

PandAna.var.analysis_vars.kDetID = Var(lambda tables: tables['rec.hdr']['det'])

Definition at line 23 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kCosNumi(), PandAna.var.analysis_vars.kNumuHadE(), PandAna.var.analysis_vars.kNumuMuE(), and PandAna.var.analysis_vars.kNusEnergy().

PandAna.var.analysis_vars.kDistAllBack = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngback'])

Definition at line 50 of file analysis_vars.py.

PandAna.var.analysis_vars.kDistAllBottom = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngbottom'])

Definition at line 47 of file analysis_vars.py.

PandAna.var.analysis_vars.kDistAllEast = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngeast'])

Definition at line 49 of file analysis_vars.py.

PandAna.var.analysis_vars.kDistAllFront = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngfront'])

Definition at line 51 of file analysis_vars.py.

PandAna.var.analysis_vars.kDistAllTop = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngtop'])

Definition at line 46 of file analysis_vars.py.

PandAna.var.analysis_vars.kDistAllWest = Var(lambda tables: tables['rec.sel.nuecosrej']['distallpngwest'])

Definition at line 48 of file analysis_vars.py.

PandAna.var.analysis_vars.kEMEnergy = Var(kEMEnergy)

Definition at line 114 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kEMEnergy(), PandAna.var.analysis_vars.kHadEnergy(), and PandAna.var.analysis_vars.kNueEnergy().

PandAna.var.analysis_vars.kHadEnergy = Var(kHadEnergy)

Definition at line 131 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kHadEnergy(), and PandAna.var.analysis_vars.kNueEnergy().

PandAna.var.analysis_vars.kHitsPerPlane = Var(lambda tables: tables['rec.sel.nuecosrej']['hitsperplane'])

Definition at line 53 of file analysis_vars.py.

PandAna.var.analysis_vars.kLongestProng = Var(kLongestProng)

Definition at line 44 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kEMEnergy(), and PandAna.var.analysis_vars.kLongestProng().

PandAna.var.analysis_vars.kMaxY = Var(kMaxY)

Definition at line 61 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kMaxY().

PandAna.var.analysis_vars.kMeantime = Var(lambda tables: tables['rec.slc']['meantime'])

Definition at line 25 of file analysis_vars.py.

PandAna.var.analysis_vars.kNHit = Var(lambda tables: tables['rec.slc']['nhit'])

Definition at line 20 of file analysis_vars.py.

PandAna.var.analysis_vars.kNueEnergy = Var(kNueEnergy)

Definition at line 160 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNueEnergy(), nue_pid_effs(), and nue_pid_effs_miniprod().

PandAna.var.analysis_vars.kNumuE = kNumuMuE+kNumuHadE

Definition at line 244 of file analysis_vars.py.

PandAna.var.analysis_vars.kNumuHadE = Var(kNumuHadE)

Definition at line 242 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNumuHadE().

PandAna.var.analysis_vars.kNumuMuE = Var(kNumuMuE)

Definition at line 218 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNumuMuE().

PandAna.var.analysis_vars.kNusEnergy = Var(kNusEnergy)

Definition at line 267 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kNusEnergy().

PandAna.var.analysis_vars.kPtP = Var(lambda tables: tables['rec.sel.nuecosrej']['partptp'])

Definition at line 55 of file analysis_vars.py.

PandAna.var.analysis_vars.kRHC = Var(lambda tables: tables['rec.spill']['isRHC'])

Definition at line 22 of file analysis_vars.py.

Referenced by PandAna.var.analysis_vars.kEMEnergy(), PandAna.var.analysis_vars.kNueEnergy(), PandAna.var.analysis_vars.kNumuHadE(), PandAna.var.analysis_vars.kNumuMuE(), and PandAna.var.analysis_vars.kNusEnergy().

PandAna.var.analysis_vars.kSparsenessAsymm = Var(lambda tables: tables['rec.sel.nuecosrej']['sparsenessasymm'])

Definition at line 63 of file analysis_vars.py.