ClustererTask.h

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.
 
 
#ifndef ALICEO2_ITS_CLUSTERERTASK
#define ALICEO2_ITS_CLUSTERERTASK
 
#include "ITSMFTReconstruction/ChipMappingITS.h"
#include "ITSMFTReconstruction/PixelReader.h"
#include "ITSMFTReconstruction/RawPixelReader.h"
#include "ITSMFTReconstruction/DigitPixelReader.h"
#include "ITSMFTReconstruction/Clusterer.h"
#include "DataFormatsITSMFT/CompCluster.h"
#include "DataFormatsITSMFT/ROFRecord.h"
#include "SimulationDataFormat/MCTruthContainer.h"
#include "SimulationDataFormat/MCCompLabel.h"
#include <memory>
#include <limits>
 
namespace o2
{
class MCCompLabel;
namespace dataformats
{
template <typename T>
class MCTruthContainer;
}
 
namespace its
{
 
class ClustererTask
{
  using Clusterer = o2::itsmft::Clusterer;
  using CompCluster = o2::itsmft::CompCluster;
  using CompClusterExt = o2::itsmft::CompClusterExt;
  using MCTruth = o2::dataformats::MCTruthContainer<o2::MCCompLabel>;
 
 public:
  ClustererTask(bool useMC = true, bool raw = false);
  ~ClustererTask();
 
  void Init();
  Clusterer& getClusterer() { return mClusterer; }
  void run(const std::string inpName, const std::string outName);
  o2::itsmft::PixelReader* getReader() const { return (o2::itsmft::PixelReader*)mReader; }
 
  void writeTree(std::string basename, int i);
  void setMaxROframe(int max) { maxROframe = max; }
  int getMaxROframe() const { return maxROframe; }
 
 private:
  int maxROframe = std::numeric_limits<int>::max();                                   
  bool mRawDataMode = false;                                                          
  bool mUseMCTruth = true;                                                            
  o2::itsmft::PixelReader* mReader = nullptr;                                         
  std::unique_ptr<o2::itsmft::DigitPixelReader> mReaderMC;                            
  std::unique_ptr<o2::itsmft::RawPixelReader<o2::itsmft::ChipMappingITS>> mReaderRaw; 
 
  Clusterer mClusterer;                                
 
  std::vector<CompClusterExt> mCompClus;               
 
  std::vector<o2::itsmft::ROFRecord> mROFRecVec;               
 
  MCTruth mClsLabels;               
 
  std::vector<unsigned char> mPatterns;
 
  ClassDefNV(ClustererTask, 2);
};
} // namespace its
} // namespace o2
 
#endif /* ALICEO2_ITS_CLUSTERERTASK */