RootTrajectorySummaryWriter.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file RootTrajectorySummaryWriter.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019-2021 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Definitions/TrackParametrization.hpp"
#include "Acts/Utilities/Logger.hpp"
#include "ActsExamples/EventData/Index.hpp"
#include "ActsExamples/EventData/SimParticle.hpp"
#include "ActsExamples/EventData/Trajectories.hpp"
#include "ActsExamples/Framework/DataHandle.hpp"
#include "ActsExamples/Framework/ProcessCode.hpp"
#include "ActsExamples/Framework/WriterT.hpp"

#include <cstdint>
#include <mutex>
#include <string>
#include <vector>

#include <TMatrixD.h>

class TFile;
class TTree;
namespace ActsFatras {
class Barcode;
}  // namespace ActsFatras

namespace ActsExamples {
struct AlgorithmContext;

class RootTrajectorySummaryWriter final
    : public WriterT<TrajectoriesContainer> {
 public:
  using HitParticlesMap = IndexMultimap<ActsFatras::Barcode>;

  struct Config {
    std::string inputTrajectories;
    std::string inputParticles;
    std::string inputMeasurementParticlesMap;
    std::string filePath = "tracksummary.root";
    std::string treeName = "tracksummary";
    std::string fileMode = "RECREATE";
    bool writeCovMat = false;
  };

  RootTrajectorySummaryWriter(const Config& config, Acts::Logging::Level level);
  ~RootTrajectorySummaryWriter() override;

  ProcessCode finalize() override;

  const Config& config() const { return m_cfg; }

 protected:
  ProcessCode writeT(const AlgorithmContext& ctx,
                     const TrajectoriesContainer& trajectories) override;

 private:
  Config m_cfg;  

  ReadDataHandle<SimParticleContainer> m_inputParticles{this, "InputParticles"};
  ReadDataHandle<HitParticlesMap> m_inputMeasurementParticlesMap{
      this, "InputMeasurementParticlesMaps"};

  std::mutex m_writeMutex;  
  TFile* m_outputFile{nullptr};  
  TTree* m_outputTree{nullptr};  
  uint32_t m_eventNr{0};         
  std::vector<uint32_t>
      m_multiTrajNr;  
  std::vector<unsigned int>
      m_subTrajNr;  

  std::vector<unsigned int> m_nStates;        
  std::vector<unsigned int> m_nMeasurements;  
  std::vector<unsigned int> m_nOutliers;      
  std::vector<unsigned int> m_nHoles;         
  std::vector<unsigned int> m_nSharedHits;    
  std::vector<float> m_chi2Sum;               
  std::vector<unsigned int>
      m_NDF;  
  std::vector<std::vector<double>>
      m_measurementChi2;  
  std::vector<std::vector<double>>
      m_outlierChi2;  
  std::vector<std::vector<double>>
      m_measurementVolume;  
  std::vector<std::vector<double>>
      m_measurementLayer;  
  std::vector<std::vector<double>>
      m_outlierVolume;  
  std::vector<std::vector<double>>
      m_outlierLayer;  

  // The majority truth particle info
  std::vector<unsigned int>
      m_nMajorityHits;  
  std::vector<uint64_t>
      m_majorityParticleId;      
  std::vector<int> m_t_charge;   
  std::vector<float> m_t_time;   
  std::vector<float> m_t_vx;     
  std::vector<float> m_t_vy;     
  std::vector<float> m_t_vz;     
  std::vector<float> m_t_px;     
  std::vector<float> m_t_py;     
  std::vector<float> m_t_pz;     
  std::vector<float> m_t_theta;  
  std::vector<float> m_t_phi;    
  std::vector<float> m_t_p;      
  std::vector<float> m_t_pT;     
  std::vector<float> m_t_eta;    
  std::vector<float>
      m_t_d0;  
  std::vector<float>
      m_t_z0;  

  std::vector<bool> m_hasFittedParams;  
  // The fitted parameters
  std::vector<float> m_eLOC0_fit;   
  std::vector<float> m_eLOC1_fit;   
  std::vector<float> m_ePHI_fit;    
  std::vector<float> m_eTHETA_fit;  
  std::vector<float> m_eQOP_fit;    
  std::vector<float> m_eT_fit;      
  // The error of fitted parameters
  std::vector<float> m_err_eLOC0_fit;  
  std::vector<float>
      m_err_eLOC1_fit;  
  std::vector<float> m_err_ePHI_fit;  
  std::vector<float>
      m_err_eTHETA_fit;               
  std::vector<float> m_err_eQOP_fit;  
  std::vector<float> m_err_eT_fit;    
  // The residual of fitted parameters
  std::vector<float> m_res_eLOC0_fit;  
  std::vector<float>
      m_res_eLOC1_fit;  
  std::vector<float> m_res_ePHI_fit;  
  std::vector<float>
      m_res_eTHETA_fit;               
  std::vector<float> m_res_eQOP_fit;  
  std::vector<float> m_res_eT_fit;    
  // The pull of fitted parameters
  std::vector<float>
      m_pull_eLOC0_fit;  
  std::vector<float>
      m_pull_eLOC1_fit;  
  std::vector<float> m_pull_ePHI_fit;  
  std::vector<float>
      m_pull_eTHETA_fit;  
  std::vector<float> m_pull_eQOP_fit;  
  std::vector<float> m_pull_eT_fit;    

  // entries of the full covariance matrix. One block for every row of the
  // matrix
  std::vector<float> m_cov_eLOC0_eLOC0;
  std::vector<float> m_cov_eLOC0_eLOC1;
  std::vector<float> m_cov_eLOC0_ePHI;
  std::vector<float> m_cov_eLOC0_eTHETA;
  std::vector<float> m_cov_eLOC0_eQOP;
  std::vector<float> m_cov_eLOC0_eT;

  std::vector<float> m_cov_eLOC1_eLOC0;
  std::vector<float> m_cov_eLOC1_eLOC1;
  std::vector<float> m_cov_eLOC1_ePHI;
  std::vector<float> m_cov_eLOC1_eTHETA;
  std::vector<float> m_cov_eLOC1_eQOP;
  std::vector<float> m_cov_eLOC1_eT;

  std::vector<float> m_cov_ePHI_eLOC0;
  std::vector<float> m_cov_ePHI_eLOC1;
  std::vector<float> m_cov_ePHI_ePHI;
  std::vector<float> m_cov_ePHI_eTHETA;
  std::vector<float> m_cov_ePHI_eQOP;
  std::vector<float> m_cov_ePHI_eT;

  std::vector<float> m_cov_eTHETA_eLOC0;
  std::vector<float> m_cov_eTHETA_eLOC1;
  std::vector<float> m_cov_eTHETA_ePHI;
  std::vector<float> m_cov_eTHETA_eTHETA;
  std::vector<float> m_cov_eTHETA_eQOP;
  std::vector<float> m_cov_eTHETA_eT;

  std::vector<float> m_cov_eQOP_eLOC0;
  std::vector<float> m_cov_eQOP_eLOC1;
  std::vector<float> m_cov_eQOP_ePHI;
  std::vector<float> m_cov_eQOP_eTHETA;
  std::vector<float> m_cov_eQOP_eQOP;
  std::vector<float> m_cov_eQOP_eT;

  std::vector<float> m_cov_eT_eLOC0;
  std::vector<float> m_cov_eT_eLOC1;
  std::vector<float> m_cov_eT_ePHI;
  std::vector<float> m_cov_eT_eTHETA;
  std::vector<float> m_cov_eT_eQOP;
  std::vector<float> m_cov_eT_eT;
};

}  // namespace ActsExamples