TrackDump.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_TPC_TrackDump_H_
#define O2_TPC_TrackDump_H_
 
#include <Rtypes.h>
#include <cstdint>
#include <memory>
#include <string>
#include <gsl/span>
#include <string_view>
#include <vector>
 
#include "DataFormatsTPC/TrackTPC.h"
#include "SimulationDataFormat/MCCompLabel.h"
#include "CommonUtils/TreeStreamRedirector.h"
#include "DataFormatsTPC/Constants.h"
#include "CorrectionMapsHelper.h"
 
 
using namespace o2::tpc::constants;
 
namespace o2::tpc
{
 
class ClusterNativeAccess;
 
class TrackDump
{
 public:
  struct ClusterGlobal {
    float gx{};
    float gy{};
    uint16_t qMax; //< QMax of the cluster
    uint16_t qTot; //< Total charge of the cluster
    uint8_t sector = 0;
    uint8_t padrow = 0;
 
    ClassDefNV(ClusterGlobal, 1);
  };
 
  struct ClusterNativeAdd : public ClusterNative {
    ClusterNativeAdd() = default;
    ClusterNativeAdd(const ClusterNative& cl) : ClusterNative(cl){};
    ~ClusterNativeAdd() = default;
 
    // float z = 0.f;
 
    float tgl = 0.f;
    float snp = 0.f;
    uint8_t sector = 0;
    uint8_t padrow = 0;
 
    // uncorrected values
    float lx() const;
    float ly() const;
    float gx() const;
    float gy() const;
    float cpad() const;
 
    // corrected values
    float lxc(float vertexTime = 0) const;
    float lyc(float vertexTime = 0) const;
    float gxc(float vertexTime = 0) const;
    float gyc(float vertexTime = 0) const;
    float zc(float vertexTime = 0) const;
 
    static gpu::CorrectionMapsHelper sCorrHelper;
    static void loadCorrMaps(std::string_view corrMapFile, std::string_view corrMapFileRef = "");
    ClassDefNV(ClusterNativeAdd, 1);
  };
 
  struct TrackInfo : public TrackTPC {
    TrackInfo() = default;
    TrackInfo(const TrackTPC& track) : TrackTPC(track){};
    TrackInfo(const TrackInfo&) = default;
    ~TrackInfo() = default;
 
    std::vector<ClusterNativeAdd> clusters{};
 
    ClassDefNV(TrackInfo, 1);
  };
 
  enum class ClStorageType {
    InsideTrack = 0,
    SeparateBranch = 1,
    SeparateTree = 2,
    SeparateFile = 3,
  };
 
  enum class ClUnStorageType {
    DontStore = 0,
    SeparateTree = 1,
    SeparateFile = 2,
  };
 
  using ClExcludes = std::vector<int>[MAXSECTOR][MAXGLOBALPADROW];
 
  void filter(const gsl::span<const TrackTPC> tracks, ClusterNativeAccess const& clusterIndex, const gsl::span<const o2::tpc::TPCClRefElem> clRefs, const gsl::span<const MCCompLabel> mcLabels);
  void finalize();
  void fillClNativeAdd(ClusterNativeAccess const& clusterIndex, std::vector<ClusterNativeAdd>& clInfos, ClExcludes* excludes = nullptr);
 
  std::string outputFileName{"filtered-tracks-and-clusters.root"}; 
  bool writeTracks{true};                                          
  bool writeGlobal{false};                                         
  bool writeMC{false};                                             
  ClStorageType clusterStorageType{ClStorageType::InsideTrack};    
  ClUnStorageType noTrackClusterType{ClUnStorageType::DontStore};  
 
 private:
  std::unique_ptr<o2::utils::TreeStreamRedirector> mTreeDump;        
  std::unique_ptr<o2::utils::TreeStreamRedirector> mTreeDumpClOnly;  
  std::unique_ptr<o2::utils::TreeStreamRedirector> mTreeDumpUnassCl; 
};
} // namespace o2::tpc
#endif