GPUTPCClusterFilter.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "GPUCommonLogger.h"
#include "GPUTPCClusterFilter.h"
#include "DataFormatsTPC/ClusterNative.h"
 
using namespace o2::gpu;
 
GPUTPCClusterFilter::GPUTPCClusterFilter(const o2::tpc::ClusterNativeAccess& clusters, uint8_t filterType)
  : mFilterType(filterType)
{
  if (filterType == 1) {
    // Custom filter settings go here
 
  } else if (filterType == 2) {
    // PbPb23 filter
    mClusterStats = std::make_unique<std::vector<int>[]>(MaxStacks);
    static bool called = false;
    if (!called) {
      LOGP(info, "GPUTPCClusterFilter called for PbPb 2023 settings");
      called = true;
    }
 
    for (uint32_t iSector = 0; iSector < GPUCA_NSECTORS; iSector++) {
      for (uint32_t iRow = 0; iRow < GPUCA_ROW_COUNT; iRow++) {
        const uint32_t globalStack = getGlobalStack(iSector, iRow);
        mClusterStats[globalStack].resize(MaxTimeBin);
 
        for (uint32_t k = 0; k < clusters.nClusters[iSector][iRow]; k++) {
          const o2::tpc::ClusterNative& cl = clusters.clusters[iSector][iRow][k];
          const int clTime = static_cast<int>(cl.getTime());
          const float clQmax = cl.getQmax();
 
          if (clQmax < 12) {
            if (clTime >= static_cast<int>(mClusterStats[globalStack].size())) {
              mClusterStats[globalStack].resize(mClusterStats[globalStack].size() + 445);
            }
            ++mClusterStats[globalStack][clTime];
          }
        }
      }
    }
  }
}
 
bool GPUTPCClusterFilter::filter(uint32_t sector, uint32_t row, o2::tpc::ClusterNative& cl)
{
  // Return true to keep the cluster, false to drop it.
  // May change cluster properties by modifying the cl reference.
  // Note that this function might be called multiple times for the same cluster, in which case the final modified cl reference goes into the output clusters.
  if (mFilterType == 2) {
    const uint32_t globalStack = getGlobalStack(sector, row);
    const int clTime = static_cast<int>(cl.getTime());
    const float clQmax = cl.getQmax();
    if ((mClusterStats[globalStack][clTime] > 40 && clQmax < 12) || (mClusterStats[globalStack][clTime] > 200)) {
      return false;
    }
  }
 
  return true;
}