EFTrackingTransient.h

Go to the documentation of this file.

/*
    Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/

 
#ifndef EFTRACKING_FPGA_INTEGRATION_EFTRACKING_TRANSIENT_H
#define EFTRACKING_FPGA_INTEGRATION_EFTRACKING_TRANSIENT_H
 
#include <vector>
 
#include "xAODMeasurementBase/UncalibratedMeasurementContainer.h"
 
namespace EFTrackingTransient
{
  // The struct of the StripCluster and PixelCluster are not aligned at the moment
  // They might be aligned in the future for efficient device memory usage
 
  constexpr uint32_t MAX_PIXEL_CLUSTERS = 326400;
  constexpr uint32_t MAX_STRIP_CLUSTERS = 307200;
  constexpr uint16_t CLUSTER_SEG_SIZE = 256;
  constexpr uint8_t  NUM_PIXEL_WORDS  = 10;
  constexpr uint8_t  NUM_STRIP_WORDS  = 9;
  constexpr uint8_t  NUM_PIXEL_FIELDS = 17;
  constexpr uint8_t  NUM_STRIP_FIELDS = 11;
  constexpr uint8_t  NUM_PIXEL_ROWS   = 22;
  constexpr uint8_t  NUM_STRIP_ROWS   = 14;
  constexpr uint32_t PIXEL_BLOCK_BUF_SIZE       = (NUM_PIXEL_WORDS*MAX_PIXEL_CLUSTERS/1024+1)*1024;        // align to 1024*sizeof(uint32_t)=4096
  constexpr uint32_t STRIP_BLOCK_BUF_SIZE       = (NUM_STRIP_WORDS*MAX_STRIP_CLUSTERS/1024+1)*1024;
  constexpr uint32_t PIXEL_CONTAINER_BUF_SIZE = ((NUM_PIXEL_ROWS*MAX_PIXEL_CLUSTERS+16)/1024+1)*1024;   // +16: for cluster number word
  constexpr uint32_t STRIP_CONTAINER_BUF_SIZE = ((NUM_STRIP_ROWS*MAX_STRIP_CLUSTERS+16)/1024+1)*1024;
 
 
 
  constexpr unsigned int MAX_NUM_CLUSTERS = 409600;
  constexpr unsigned int NUM_MAXCLUSTER_WORD_INTRACK = 13;
  constexpr unsigned int MAX_TRACK = 1000;
 
  constexpr unsigned long TRACK_CONTAINER_BUF_SIZE = (NUM_MAXCLUSTER_WORD_INTRACK*MAX_TRACK + 4096);
 
 
  // Optimize these, 
  constexpr unsigned int MAX_NUM_INPUTCLUSTERS = 409600;
  constexpr unsigned int NUM_MAXINPUT_PIXEL_ROW = 3;
  constexpr unsigned int NUM_MAXINPUT_STRIP_ROW = 2;
  constexpr unsigned long PIXEL_CONTAINER_INPUT_BUF_SIZE = (NUM_MAXINPUT_PIXEL_ROW*MAX_NUM_INPUTCLUSTERS + 4096);
  constexpr unsigned long STRIP_CONTAINER_INPUT_BUF_SIZE = (NUM_MAXINPUT_STRIP_ROW*MAX_NUM_INPUTCLUSTERS + 4096);

  struct StripCluster
  {
    float localPosition = 0.0f;
    float localCovariance = 0.0f;
    unsigned int idHash = 0;
    long unsigned int id = 0;
    float globalPosition[3] = {0.0f, 0.0f, 0.0f};
    unsigned long long rdoList[1000] = {0};
    int channelsInPhi = 0;
    int sizeOfRDOList = 0;
  };

  struct PixelCluster
  {
    long unsigned int id = 0;
    unsigned int idHash = 0;
    float localPosition[2] = {0.0f, 0.0f};
    float localCovariance[2] = {0.0f, 0.0f};
    float globalPosition[3] = {0.0f, 0.0f, 0.0f};
    unsigned long long rdoList[1000] = {0};
    int channelsInPhi = 0;
    int channelsInEta = 0;
    float widthInEta = 0.0f;
    int totList[1000] = {0};
    int totalToT = 0;
    float chargeList[1000] = {0.0f};
    float totalCharge = 0.0f;
    float energyLoss = 0.0f;
    char isSplit = 0;
    float splitProbability1 = 0.0f;
    float splitProbability2 = 0.0f;
    int lvl1a = 0;
    int sizeOfRDOList = 0;
    int sizeOfTotList = 0;
    int sizeOfChargeList = 0;
  };

  struct SpacePoint
  {
 
    unsigned int idHash[2] = {0};
    float globalPosition[3] = {0.0f, 0.0f, 0.0f};
    float radius = 0.0f;
    float cov_r = 0.0f;
    float cov_z = 0.0f;
    int measurementIndex[100] = {0};
    // Strip
    float topHalfStripLength = 0.0f;
    float bottomHalfStripLength = 0.0f;
    float topStripDirection[3] = {0.0f, 0.0f, 0.0f};
    float bottomStripDirection[3] = {0.0f, 0.0f, 0.0f};
    float stripCenterDistance[3] = {0.0f, 0.0f, 0.0f};
    float topStripCenter[3] = {0.0f, 0.0f, 0.0f};
  };

  struct Metadata
  {
    unsigned long numOfStripClusters = 0;
    unsigned long numOfPixelClusters = 0;
    unsigned long numOfStripSpacePoints = 0;
    unsigned int numOfPixelSpacePoints = 0;
    unsigned int scRdoIndex[500000] = {0};
    unsigned int pcRdoIndex[500000] = {0};
    unsigned int pcTotIndex[500000] = {0};
    unsigned int pcChargeIndex[500000] = {0};
    unsigned int pcRdoIndexSize = 0;
    unsigned int scRdoIndexSize = 0;
    unsigned int pcTotIndexSize = 0;
    unsigned int pcChargeIndexSize = 0;
  };

  struct StripClusterOutput
  {
 
    float *scLocalPosition;
    float *scLocalCovariance;
    unsigned int *scIdHash;
    long unsigned int *scId;
    float *scGlobalPosition;
    unsigned long long *scRdoList;
    int *scChannelsInPhi;
  };

  struct PixelClusterOutput
  {
 
    float *pcLocalPosition;
    float *pcLocalCovariance;
    unsigned int *pcIdHash;
    long unsigned int *pcId;
    float *pcGlobalPosition;
    unsigned long long *pcRdoList;
    int *pcChannelsInPhi;
    int *pcChannelsInEta;
    float *pcWidthInEta;
    int *pcTotList;
    int *pcTotalToT;
    float *pcChargeList;
    float *pcTotalCharge;
    float *pcEnergyLoss;
    char *pcIsSplit;
    float *pcSplitProbability1;
    float *pcSplitProbability2;
    int *pcLvl1a;
  };

 
  struct SpacePointOutput
  {
 
    unsigned int *spIdHash;
    float *spGlobalPosition;
    float *spRadius;
    float *spVarianceR;
    float *spVarianceZ;
    int *spMeasurementIndexes;
    unsigned int *spElementIdList;
    float *spTopHalfStripLength;
    float *spBottomHalfStripLength;
    float *spTopStripDirection;
    float *spBottomStripDirection;
    float *spStripCenterDistance;
    float *spTopStripCenter;
  };

  struct StripClusterAuxInput
  {
    std::vector<float> localPosition;
    std::vector<float> localCovariance;
    std::vector<unsigned int> idHash;
    std::vector<long unsigned int> id;
    std::vector<float> globalPosition;
    std::vector<unsigned long long> rdoList;
    std::vector<int> channelsInPhi;
  };

  struct PixelClusterAuxInput
  {
    std::vector<long unsigned int> id;
    std::vector<unsigned int> idHash;
    std::vector<float> localPosition;
    std::vector<float> localCovariance;
    std::vector<float> globalPosition;
    std::vector<unsigned long long> rdoList;
    std::vector<int> channelsInPhi;
    std::vector<int> channelsInEta;
    std::vector<float> widthInEta;
    std::vector<int> totList;
    std::vector<int> totalToT;
    std::vector<float> chargeList;
    std::vector<float> totalCharge;
    std::vector<float> energyLoss;
    std::vector<char> isSplit;
    std::vector<float> splitProbability1;
    std::vector<float> splitProbability2;
    std::vector<int> lvl1a;
    std::vector<int> sizeOfRDOList;
    std::vector<int> sizeOfTotList;
    std::vector<int> sizeOfChargeList;
  };

  struct SpacePointAuxInput
  {
    std::vector<unsigned int> elementIdList;
    std::vector<float> globalPosition;
    std::vector<float> radius;
    std::vector<float> varianceR;
    std::vector<float> varianceZ;
    std::vector<float> topHalfStripLength;
    std::vector<float> bottomHalfStripLength;
    std::vector<float> topStripDirection;
    std::vector<float> bottomStripDirection;
    std::vector<float> stripCenterDistance;
    std::vector<float> topStripCenter;
    std::vector<int> measurementIndexes;
  };
 
} // namespace EFTrackingDataTransient
 
#endif // EFTRACKING_FPGA_INTEGRATION_EFTRACKING_TRANSIENT_H