DeltaR Namespace Reference

Functions
def	bestMatch (object, matchCollection)
def	cleanObjectCollection (objects, masks, deltaRMin)
def	cleanObjectCollection2 (objects, masks, deltaRMin)
def	deltaPhi (p1, p2)
def	deltaR (args)
def	deltaR2 (e1, p1, e2, p2)
def	matchObjectCollection (objects, matchCollection, deltaR2Max)
def	matchObjectCollection2 (objects, matchCollection, deltaRMax=0.3)
def	matchObjectCollection3

Function Documentation

bestMatch()

def DeltaR.bestMatch	(		object,
			matchCollection
	)

Return the best match to object in matchCollection, which is the closest object in delta R

Definition at line 118 of file DeltaR.py.

References deltaR2(), and nano_mu_digi_cff.float.

Referenced by matchObjectCollection().

def bestMatch( object, matchCollection):
    '''Return the best match to object in matchCollection, which is the closest object in delta R'''
    deltaR2Min = float('+inf')
    bm = None
    for match in matchCollection:
        dR2 = deltaR2( object.eta(), object.phi(),
                       match.eta(), match.phi() )
        if dR2 < deltaR2Min:
            deltaR2Min = dR2
            bm = match
    return bm, deltaR2Min

cleanObjectCollection()

def DeltaR.cleanObjectCollection	(		objects,
			masks,
			deltaRMin
	)

Masks objects using a deltaR cut.

Definition at line 96 of file DeltaR.py.

References deltaR2().

def cleanObjectCollection( objects, masks, deltaRMin ):
    '''Masks objects using a deltaR cut.'''
    if len(objects)==0 or len(masks)==0:
        return objects, []
    deltaR2Min = deltaRMin*deltaRMin
    cleanObjects = []
    dirtyObjects = []
    for object in objects:
        ok = True 
        for mask in masks:
            dR2 = deltaR2( object.eta(), object.phi(),
                           mask.eta(), mask.phi() )
            if dR2 < deltaR2Min:
                ok = False
        if ok:
            cleanObjects.append( object )
        else:
            dirtyObjects.append( object )
    return cleanObjects, dirtyObjects

cleanObjectCollection2()

def DeltaR.cleanObjectCollection2	(		objects,
			masks,
			deltaRMin
	)

Masks objects using a deltaR cut, another algorithm (same results).

Definition at line 69 of file DeltaR.py.

References deltaR2().

def cleanObjectCollection2( objects, masks, deltaRMin ):
    '''Masks objects using a deltaR cut, another algorithm (same results).'''
    if len(objects)==0:
        return objects
    deltaR2Min = deltaRMin*deltaRMin
    cleanObjects = copy.copy( objects )
    for mask in masks:
        tooClose = []
        for idx, object in enumerate(cleanObjects):
            dR2 = deltaR2( object.eta(), object.phi(),
                           mask.eta(), mask.phi() )
            if dR2 < deltaR2Min:
                tooClose.append( idx )
        nRemoved = 0
        for idx in tooClose:
            # yes, everytime an object is removed, the list of objects is updated
            # so need to update the index accordingly.
            # example: to remove : ele 1 and 2
            #  first, ele 1 is removed
            #  -> ele 2 is now at index 1
            # one should again remove the element at index 1
            idx -= nRemoved
            del cleanObjects[idx]
            nRemoved += 1 
    return cleanObjects

deltaPhi()

def DeltaR.deltaPhi	(		p1,
			p2
	)

Computes delta phi, handling periodic limit conditions.

Definition at line 16 of file DeltaR.py.

Referenced by deltaR2().

def deltaPhi( p1, p2):
    '''Computes delta phi, handling periodic limit conditions.'''
    res = p1 - p2
    while res > math.pi:
        res -= 2*math.pi
    while res < -math.pi:
        res += 2*math.pi
    return res

deltaR()

def DeltaR.deltaR

(

args

)

Definition at line 12 of file DeltaR.py.

References deltaR2().

def deltaR( *args ):
    return math.sqrt( deltaR2(*args) )

deltaR2()

def DeltaR.deltaR2	(		e1,
			p1,
			e2,
			p2
	)

Definition at line 6 of file DeltaR.py.

References deltaPhi().

Referenced by bestMatch(), cleanObjectCollection(), cleanObjectCollection2(), and deltaR().

def deltaR2( e1, p1, e2, p2):
    de = e1 - e2
    dp = deltaPhi(p1, p2)
    return de*de + dp*dp

matchObjectCollection()

def DeltaR.matchObjectCollection	(		objects,
			matchCollection,
			deltaR2Max
	)

Definition at line 131 of file DeltaR.py.

References bestMatch(), and ComparisonHelper.zip().

def matchObjectCollection( objects, matchCollection, deltaR2Max):
    pairs = {}
    if len(objects)==0:
        return pairs
    if len(matchCollection)==0:
        return dict( zip(objects, [None]*len(objects)) )
    for object in objects:
        bm, dr2 = bestMatch( object, matchCollection )
        if dr2<deltaR2Max:
            pairs[object] = bm
        else:
            pairs[object] = None            
    return pairs

matchObjectCollection2()

def DeltaR.matchObjectCollection2	(		objects,
			matchCollection,
			deltaRMax = 0.3
	)

Univoque association of an element from matchCollection to an element of objects.
Reco and Gen objects get the "matched" attribute, true if they are part of a matched tuple.
By default, the matching is true only if delta R is smaller than 0.3.

Definition at line 146 of file DeltaR.py.

References ComparisonHelper.zip().

def matchObjectCollection2 ( objects, matchCollection, deltaRMax = 0.3 ):
    '''Univoque association of an element from matchCollection to an element of objects.
    Reco and Gen objects get the "matched" attribute, true if they are part of a matched tuple.
    By default, the matching is true only if delta R is smaller than 0.3.
    '''

    pairs = {}
    if len(objects)==0:
        return pairs
    if len(matchCollection)==0:
        return dict( zip(objects, [None]*len(objects)) )
    # build all possible combinations
    allPairs = sorted([(deltaR2 (object.eta(), object.phi(), match.eta(), match.phi()), (object, match)) for object in objects for match in matchCollection])

    # to flag already matched objects
    # FIXME this variable remains appended to the object, I do not like it
    for object in objects:
        object.matched = False
    for match in matchCollection:
        match.matched = False

    deltaR2Max = deltaRMax * deltaRMax
    for dR2, (object, match) in allPairs:
        if dR2 > deltaR2Max:
            break
        if dR2 < deltaR2Max and object.matched == False and match.matched == False:
            object.matched = True
            match.matched = True
            pairs[object] = match

    for object in objects:
        if object.matched == False:
            pairs[object] = None

    return pairs
    # by now, the matched attribute remains in the objects, for future usage
    # one could remove it with delattr (object, attrname)

matchObjectCollection3()

def DeltaR.matchObjectCollection3	(		objects,
			matchCollection,
			deltaRMax = 0.3,
			filter = lambda x,
			y
	)

Definition at line 25 of file DeltaR.py.

References ALCARECOTkAlBeamHalo_cff.filter, and ComparisonHelper.zip().

def matchObjectCollection3 ( objects, matchCollection, deltaRMax = 0.3, filter = lambda x,y : True ):
    '''Univoque association of an element from matchCollection to an element of objects. 
    Reco and Gen objects get the "matched" attribute, true if they are part of a matched tuple. 
    By default, the matching is true only if delta R is smaller than 0.3.
    '''
    #
    pairs = {}
    if len(objects)==0:
        return pairs
    if len(matchCollection)==0:
        return dict( zip(objects, [None]*len(objects)) )
    # build all possible combinations

    allPairs = sorted([(deltaR2 (object.eta(), object.phi(), match.eta(), match.phi()), (object, match)) for object in objects for match in matchCollection if list(filter(object,match)) ])
    #
    # to flag already matched objects
    # FIXME this variable remains appended to the object, I do not like it 

    for object in objects:
        object.matched = False
    for match in matchCollection:
        match.matched = False
    #

    deltaR2Max = deltaRMax * deltaRMax
    for dR2, (object, match) in allPairs:
        if dR2 > deltaR2Max:
            break
        if dR2 < deltaR2Max and object.matched == False and match.matched == False:
            object.matched = True
            match.matched = True
            pairs[object] = match
    #

    for object in objects:
        if object.matched == False:
            pairs[object] = None
    #

    return pairs
    # by now, the matched attribute remains in the objects, for future usage
    # one could remove it with delattr (object, attrname)