KrClusterFinder.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 O2_TRD_KRCLUSTERFINDER_H
#define O2_TRD_KRCLUSTERFINDER_H
 
#include "DataFormatsTRD/Digit.h"
#include "DataFormatsTRD/TriggerRecord.h"
#include "DataFormatsTRD/KrCluster.h"
#include "DataFormatsTRD/KrClusterTriggerRecord.h"
 
#include "Rtypes.h"
#include "TF1.h"
#include "Fit/Fitter.h"
#include "Fit/FitResult.h"
 
#include <memory>
#include <gsl/span>
 
namespace o2
{
 
namespace trd
{
 
class KrClusterFinder
{
 
 public:
  KrClusterFinder() = default;
  KrClusterFinder(const KrClusterFinder&) = delete;
  ~KrClusterFinder() = default;
 
  struct LandauChi2Functor {
    double operator()(const double* par) const;
    std::vector<float> x; 
    std::vector<float> y; 
    int xLowerBound;
    int xUpperBound;
  };
 
  void init();
 
  // Reset output containers
  void reset();
 
  void setInput(const gsl::span<const Digit>& digitsIn, const gsl::span<const TriggerRecord>& trigRecIn);
 
  void findClusters();
 
  double getRms(const std::vector<uint64_t>& adcIndices, int itTrunc, double nRmsTrunc, int minAdc, double& rmsTime, uint32_t& sumAdc) const;
 
  const std::vector<KrCluster>& getKrClusters() const { return mKrClusters; }
  const std::vector<KrClusterTriggerRecord>& getKrTrigRecs() const { return mTrigRecs; }
 
 private:
  // input
  gsl::span<const Digit> mDigits;                 
  gsl::span<const TriggerRecord> mTriggerRecords; 
  // output
  std::vector<KrCluster> mKrClusters{};            
  std::vector<KrClusterTriggerRecord> mTrigRecs{}; 
  // helpers
  LandauChi2Functor mLandauChi2Functor;                                         
  std::shared_ptr<ROOT::Fit::FitResult> mFitResult{new ROOT::Fit::FitResult()}; 
  ROOT::Fit::Fitter mFitter{mFitResult};                                        
  std::array<double, 3> mInitialFitParams{1., 1., 1.};                          
  std::unique_ptr<TF1> mFuncLandauFit;                                          
  // settings
  const int mBaselineAdc{10};        
  const int mMinAdcForMax{70};       
  const int mMinAdcClContrib{40};    
  const int mMinAdcForSecondMax{50}; 
  const int mMinAdcClEoverT{60};     
 
  ClassDefNV(KrClusterFinder, 1);
};
 
} // namespace trd
} // namespace o2
 
#endif // O2_TRD_KRCLUSTERFINDER_H