SensitiveDetectorDebugHelper.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 N**************************************************************************/
 
#pragma once
#ifndef VXD_SIMULATION_SENSITIVEDETECTORDEBUGHELPER_H
#define VXD_SIMULATION_SENSITIVEDETECTORDEBUGHELPER_H
 
#include <vxd/simulation/SensorTraversal.h>
#include <vxd/dataobjects/VxdID.h>
#include <vxd/dataobjects/VXDTrueHit.h>
#include <vxd/dataobjects/VXDSimHit.h>
 
#include <TFile.h>
#include <TTree.h>
 
namespace Belle2 {
  namespace VXD {
 
    class SensitiveDetectorDebugHelper {
    public:
      static SensitiveDetectorDebugHelper& getInstance();
      void startTraversal(VxdID sensorID, const SensorTraversal& traversal);
      void addTrueHit(const VXDTrueHit* truehit);
      void addSimHit(const VXDSimHit* simhit, int startPoint, int endPoint);
      void finishTraversal() {
        m_tree->Fill();
      }
    private:
      SensitiveDetectorDebugHelper();
      SensitiveDetectorDebugHelper(const SensitiveDetectorDebugHelper&) = delete;
      SensitiveDetectorDebugHelper& operator=(const SensitiveDetectorDebugHelper&) = delete;
      ~SensitiveDetectorDebugHelper() {
        m_tree->Write();
        m_file->Close();
      }
 
      TFile* m_file {0};
      TTree* m_tree {0};
 
      enum {
        MAX_STEPS = 5000,
        MAX_HITS = 500,
        MAX_EDEP = 10000,
 
        SIZE_STEP = 9,
        SIZE_TRUE = 20,
        SIZE_HITS = 13,
        SIZE_EDEP = 2
      };
 
      struct info {
        int   sensorID;
        int   pdg;
        int   contained;
        int   primary;
        double length;
 
        int   stepN;
        double stepInfo[SIZE_STEP* MAX_STEPS];
 
        int   trueN;
        double trueInfo[SIZE_TRUE];
 
        int edepN;
        double edepInfo[SIZE_EDEP* MAX_EDEP];
 
        int simhitN;
        double simhitInfo[SIZE_HITS* MAX_HITS];
      } m_info; 
    };
  }
} //Belle2 namespace
#endif // VXD_SIMULATION_DEBUGFILE_H