PulseHeight.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_PULSEHEIGHT_H
#define O2_TRD_PULSEHEIGHT_H
 
#include "DataFormatsTRD/TrackTRD.h"
#include "DataFormatsTRD/Tracklet64.h"
#include "DataFormatsTRD/Digit.h"
#include "DataFormatsTRD/TriggerRecord.h"
#include "DataFormatsTRD/TrackTriggerRecord.h"
#include "DataFormatsTRD/PHData.h"
#include "TRDCalibration/CalibrationParams.h"
#include "TFile.h"
#include "TTree.h"
 
namespace o2
{
 
namespace globaltracking
{
class RecoContainer;
}
 
namespace trd
{
 
class PulseHeight
{
 
 public:
  PulseHeight() = default;
 
  void init();
 
  void setInput(const o2::globaltracking::RecoContainer& input, gsl::span<const Digit>* digits);
 
  void reset();
 
  void setApplyShift(bool f) { mApplyShift = f; }
 
  void process();
 
  void findDigitsForTracklet(const Tracklet64& trklt, const TriggerRecord& trig, int type);
 
  const std::vector<PHData>& getPHData() { return mPHValues; }
  const std::vector<PHDataHD>& getPHDataHD() { return mPHValuesHD; }
 
  void createOutputFile();
  void closeOutputFile();
 
 private:
  // input arrays which should not be modified since they are provided externally
  gsl::span<const TrackTRD> mTracksInITSTPCTRD;                      
  gsl::span<const TrackTRD> mTracksInTPCTRD;                         
  gsl::span<const Tracklet64> mTrackletsRaw;                         
  const gsl::span<const Digit>* mDigits = nullptr;                   
  gsl::span<const TriggerRecord> mTriggerRecords;                    
  gsl::span<const TrackTriggerRecord> mTrackTriggerRecordsTPCTRD;    
  gsl::span<const TrackTriggerRecord> mTrackTriggerRecordsITSTPCTRD; 
 
  // output
  std::vector<PHData> mPHValues, *mPHValuesPtr{&mPHValues}; 
  std::vector<PHDataHD> mPHValuesHD, *mPHValuesHDPtr{&mPHValuesHD}; 
  std::vector<int> mDistances, *mDistancesPtr{&mDistances}; 
  std::unique_ptr<TFile> mOutFile{nullptr};                 
  std::unique_ptr<TTree> mOutTree{nullptr};                 
 
  // settings
  const TRDCalibParams& mParams{TRDCalibParams::Instance()}; 
  bool mWriteOutput{false};                                  
  bool mApplyShift{true};
 
  ClassDefNV(PulseHeight, 1);
};
 
} // namespace trd
} // namespace o2
 
#endif // O2_TRD_PULSEHEIGHT_H