Exemplos de shift em Python

Exemplo n.º 1

Arquivo: pt_pk.py Projeto: gdwebb/UnderlyingEvent

def accept_track(event, track):
    eta_cut = abs(track.eta()) < 1.0
    dca_cut = abs(track.globalDca().mag()) < 1.0
    fit_cut = track.nHitsFit() > 25
    if isinstance(event, ROOT.StChargedPionMcEvent):
        pid_cut = track.geantId() in (11, 12, 14, 15)
    else:
        nsigpi = pid.shift(event.runId(), track.nSigmaPion())
        pid_cut = nsigpi < -2.1
    return eta_cut and dca_cut and fit_cut and pid_cut

Exemplo n.º 2

Arquivo: pt_pk.py Projeto: gdwebb/UnderlyingEvent

def accept_track(event, track):
    eta_cut = abs( track.eta() ) < 1.0
    dca_cut = abs( track.globalDca().mag() ) < 1.0
    fit_cut = track.nHitsFit() > 25
    if isinstance(event, ROOT.StChargedPionMcEvent):
        pid_cut = track.geantId() in (11,12,14,15)
    else:
        nsigpi = pid.shift(event.runId(), track.nSigmaPion())
        pid_cut = nsigpi < -2.1
    return eta_cut and dca_cut and fit_cut and pid_cut

Exemplo n.º 3

Arquivo: pt_e.py Projeto: gdwebb/UnderlyingEvent

def accept_track(event, track):
    eta_cut = abs(track.eta()) < 1.0
    dca_cut = abs(track.globalDca().mag()) < 1.0
    fit_cut = track.nHitsFit() > 25
    if isinstance(event, ROOT.StChargedPionMcEvent):
        pid_cut = track.geantId() in (2, 3)
    else:
        nsigpi = pid.shift(event.runId(), track.nSigmaPion())
        pt = track.Pt()
        if pt < 3.18:
            pid_cut = nsigpi > 2.60
        elif pt < 4.56:
            pid_cut = nsigpi > 2.40
        elif pt < 6.32:
            pid_cut = nsigpi > 2.40
        elif pt < 8.80:
            pid_cut = nsigpi > 2.40
        else:
            pid_cut = nsigpi > 2.10
    return eta_cut and dca_cut and fit_cut and pid_cut

Exemplo n.º 4

Arquivo: pt_e.py Projeto: gdwebb/UnderlyingEvent

def accept_track(event, track):
    eta_cut = abs( track.eta() ) < 1.0
    dca_cut = abs( track.globalDca().mag() ) < 1.0
    fit_cut = track.nHitsFit() > 25
    if isinstance(event, ROOT.StChargedPionMcEvent):
        pid_cut = track.geantId() in (2,3)
    else:
        nsigpi = pid.shift(event.runId(), track.nSigmaPion())
        pt = track.Pt()
        if pt < 3.18:
            pid_cut = nsigpi > 2.60
        elif pt < 4.56:
            pid_cut = nsigpi > 2.40
        elif pt < 6.32:
            pid_cut = nsigpi > 2.40
        elif pt < 8.80:
            pid_cut = nsigpi > 2.40
        else:
            pid_cut = nsigpi > 2.10
    return eta_cut and dca_cut and fit_cut and pid_cut

Exemplo n.º 5

Arquivo: dEdx_p.py Projeto: kocolosk/analysis

def accept_track(event, track):
    eta_cut = abs( track.eta() ) < 1.0
    dca_cut = abs( track.globalDca().mag() ) < 1.0
    fit_cut = track.nHitsFit() > 25
    if isinstance(event, ROOT.StChargedPionMcEvent):
        pid_cut = True
    else:
        nsigpi = pid.shift(event.runId(), track.nSigmaPion())
        pt = track.Pt()
        if pt < 3.18:
            pid_cut = -1.1 < nsigpi < 2.3
        elif pt < 4.56:
            pid_cut = -1.4 < nsigpi < 2.1
        elif pt < 6.32:
            pid_cut = -1.4 < nsigpi < 1.8
        elif pt < 8.80:
            pid_cut = -1.4 < nsigpi < 1.8
        else:
            pid_cut = -1.3 < nsigpi < 1.4
    return eta_cut and dca_cut and fit_cut and pid_cut

Exemplo n.º 6

Arquivo: pt_p_nSigmaPion.py Projeto: gdwebb/UnderlyingEvent

def analyze(event, jet_trigger_filter, **kw):
    for track in event.tracks():
        if event.charge_filter(track) and accept_track(event, track):
            nsigpi = pid.shift(event.runId(), track.nSigmaPion())
            yield (track.Pt(), track.P(), nsigpi + 6*track.charge())

Exemplo n.º 7

def analyze(event, jet_trigger_filter, **kw):
    for track in event.tracks():
        if event.charge_filter(track) and accept_track(event, track):
            nsigpi = pid.shift(event.runId(), track.nSigmaPion())
            yield (track.Pt(), track.P(), nsigpi + 6 * track.charge())