TBNoiseWrite.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TBREC_TBNOISEWRITE_H
#define TBREC_TBNOISEWRITE_H
 
// 
// class TBNoiseWrite 
// Write ROOT Tree with noise in randomly triggered events
//
#include "AthenaBaseComps/AthAlgorithm.h"
 
class TFile;
class TTree;
class CaloCell_ID;
class IToolSvc;
class LArDigitContainer;
 
class TBNoiseWrite: public AthAlgorithm {
 public:    
  
  TBNoiseWrite(const std::string& name, ISvcLocator* pSvcLocator);
  virtual ~TBNoiseWrite();
  virtual StatusCode initialize() override;
  virtual StatusCode execute(const EventContext& ctx) override;
  virtual StatusCode finalize() override;
  
 private: 
  void clear();
  StatusCode getXcryoYtable(float &x, float &y, float &eBeam);
  
  int m_nEvent;                     // counter
  int m_nEventRandomTrigger;        // counter
  bool m_first;                     // First event flag
 
  // Run header
  struct NoiseHeader{
    int m_nRun;               
    float m_beamMom;          
    float m_xCryo;            
    float m_yTable;           
  };
  NoiseHeader m_header;
 
  // Variables to be in the TTree
  //--------------------------
  // Cell parameters
  std::vector<unsigned int>* m_cell_id;
  std::vector<float>* m_cell_energy;
  
  // Names and pointers
  std::string m_digitContainerName;
  std::string m_caloCellContainerName;    // Cell container name
  std::string m_headerTreeName;           // Header Tree name
  std::string m_noiseTreeName;            // Noise Tree name
  std::string m_rootfile_name;            // name of the ROOT file with TBTree
  TFile* m_rootfile;                      // and its pointer
  TTree* m_header_tree;                   // Header Tree pointer
  TTree* m_tree;                          // Noise Tree pointer

  std::string m_txtFileWithXY;
};
 
#endif