SensitiveDetectorDebugHelper.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#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