PxdPackerUnpackerTestModule Class Reference

Inheritance diagram for PxdPackerUnpackerTestModule:

Collaboration diagram for PxdPackerUnpackerTestModule:

Public Member Functions
	__init__ (self, rawhits_collection='PXDRawHits', digits_collection="PXDDigits")
	sortDigits (self, unsortedPyStoreArray)
	sortRawHits (self, unsortedPyStoreArray)
	event (self)

Public Attributes
	rawhits_collection = rawhits_collection
	the PXDRawHits
	digits_collection = digits_collection
	the PXDDigits

Detailed Description

Module which checks if a collection of PXDDigits and
a collection of PXDRawHits from the packing/unpacking procedure are equal.
The PXDUnpacker does not create PXDDigits but PXDRawHits and therefore these two lists
must be compared.

Definition at line 30 of file pxd_packer_unpacker.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			rawhits_collection = 'PXDRawHits',
			digits_collection = "PXDDigits" )

constructor

Definition at line 39 of file pxd_packer_unpacker.py.

    def __init__(self, rawhits_collection='PXDRawHits', digits_collection="PXDDigits"):
        """constructor"""
        # call constructor of base class, required if you implement __init__ yourself!
        super().__init__()
        # and do whatever else is necessary like declaring member variables
        ## the PXDRawHits
        self.rawhits_collection = rawhits_collection
        ## the PXDDigits
        self.digits_collection = digits_collection
 

Member Function Documentation

event()

event

(

self

)

 load the PXD Digits of the simulation and the packed/unpacked ones
and compare them

Definition at line 81 of file pxd_packer_unpacker.py.

    def event(self):
        """ load the PXD Digits of the simulation and the packed/unpacked ones
        and compare them"""
 
        # load the digits and the collection which results from the packer and unpacker
        # processed by packer and unpacker
        pxdRawHitsPackedUnpacked_unsorted = Belle2.PyStoreArray(self.rawhits_collection)
        # direct from simulation
        pxdDigits_unsorted = Belle2.PyStoreArray(self.digits_collection)
 
        # sort the digits, because the order gets
        # lost during the packing/unpacking process
        pxdDigits = self.sortDigits(pxdDigits_unsorted)
        pxdRawHitsPackedUnpacked = self.sortRawHits(pxdRawHitsPackedUnpacked_unsorted)
 
        if not len(pxdDigits) == len(pxdRawHitsPackedUnpacked):
            b2.B2FATAL("PXDDigits and PXDRawHits count not equal after packing and unpacking")
 
        print(f"Comparing {len(pxdDigits)} pxd digits ")
 
        # check all quantities between the direct and the packed/unpacked pxd digits
        for i in range(len(pxdDigits)):
            digit = pxdDigits[i]
            rawHitPackedUnpacked = pxdRawHitsPackedUnpacked[i]
 
            # compare all available quantities
            # cannot compare frame number, because it is not available on PXDDigits
            assert rawHitPackedUnpacked.getSensorID().getID() == digit.getSensorID().getID()
            assert rawHitPackedUnpacked.getRow() == digit.getVCellID()
            assert rawHitPackedUnpacked.getColumn() == digit.getUCellID()
            # There are some rare cases (~ every 10th event), where the PXD Digits have a charge
            # larger than 255 which will be clipped by the packer to 8bit (at most 255)
            # therefore, limit the maximal charge of the digit here in the comparison
            assert numpy.isclose(min(255.0, digit.getCharge()), rawHitPackedUnpacked.getCharge())
 
 
# to run the framework the used modules need to be registered

sortDigits()

sortDigits	(		self,
			unsortedPyStoreArray )

 use a some digit information to sort the PXDDigits list
    Returns a python-list containing the PXDDigts

Definition at line 49 of file pxd_packer_unpacker.py.

    def sortDigits(self, unsortedPyStoreArray):
        """ use a some digit information to sort the PXDDigits list
            Returns a python-list containing the PXDDigts
        """
 
        # first convert to a python-list to be abple to sort
        py_list = [x for x in unsortedPyStoreArray]
 
        # sort via a hierarchy of sort keys
        return sorted(
            py_list,
            key=lambda x: (
                x.getSensorID(),
                x.getVCellID(),
                x.getUCellID()))
 

sortRawHits()

sortRawHits	(		self,
			unsortedPyStoreArray )

 use a some digit information to sort the PXDRawHits list
    Returns a python-list containing the PXDRawHits

Definition at line 65 of file pxd_packer_unpacker.py.

    def sortRawHits(self, unsortedPyStoreArray):
        """ use a some digit information to sort the PXDRawHits list
            Returns a python-list containing the PXDRawHits
        """
 
        # first convert to a python-list to be able to sort
        py_list = [x for x in unsortedPyStoreArray]
 
        # sort via a hierarchy of sort keys
        return sorted(
            py_list,
            key=lambda x: (
                x.getSensorID(),
                x.getRow(),
                x.getColumn()))
 

Member Data Documentation

digits_collection

digits_collection = digits_collection

the PXDDigits

Definition at line 47 of file pxd_packer_unpacker.py.

rawhits_collection

rawhits_collection = rawhits_collection

the PXDRawHits

Definition at line 45 of file pxd_packer_unpacker.py.

---

The documentation for this class was generated from the following file:

• pxd/tests/pxd_packer_unpacker.py