BCalTagEfficiency.h

Go to the documentation of this file.

#ifndef BCalTagEfficiency_h
#define BCalTagEfficiency_h 1

#include "marlin/Processor.h"
#include "lcio.h"
#include <string>


// includes for the BcEnergyDensity

#include "BcEnergyDensity.h"

// root includes 

#include <TTree.h>
#include <TFile.h>
#include <TH2D.h>
#include <TLorentzVector.h>
#include <TMath.h>
#include <TObjArray.h>
#include <TClonesArray.h>


using namespace lcio ;
using namespace marlin ;


/**  Marlin processor to calculate BCAL tagging efficiency
 * 
 *  If compiled with MARLIN_USE_AIDA 
 *  it creates a histogram (cloud) of the MCParticle energies.
 * 
 *  <h4>Input - Prerequisites</h4>
 *  Needs the collection of MCParticles.
 *
 *  <h4>Output</h4> 
 *  A histogram.
 * 
 * @param CollectionName Name of the MCParticle collection
 * 
 * @author F. Gaede, DESY
 * @version $Id$ 
 */

class BCalTagEfficiency : public Processor {
 
 public:
  
  virtual Processor*  newProcessor() { return new BCalTagEfficiency ; }
  
  
  BCalTagEfficiency() ;
  
  /** Called at the begin of the job before anything is read.
   * Use to initialize the processor, e.g. book histograms.
   */
  virtual void init() ;
  
  /** Called for every run.
   */
  virtual void processRunHeader( LCRunHeader* run ) ;
  
  /** Called for every event - the working horse.
   */
  virtual void processEvent( LCEvent * evt ) ; 
  
  
  virtual void check( LCEvent * evt ) ; 
  
  
  /** Called after data processing for clean up.
   */
  virtual void end() ;
  
  
 protected:

  /** Input collection name.
   */
  std::string _MCParticleName ;
  std::string _BCALInputTruthLinkName;
  std::string _BCALEffiMCTruthLinkName;
  std::string _BCALClusterName;
  std::string _BCALParticleName ;


  int _nRun ;
  int _nEvt ;


  // root variables to create tree 

  TFile* rootfile;
  TTree* tree;
  std::string rootFileName;

  // per particle
  int mcp;
  enum {MCP = 100};
  int pdg[MCP];
  float energy[MCP], ePrime[MCP], pxPrime[MCP], pxIP[MCP], pyIP[MCP], pzIP[MCP];
  float phiIP[MCP], theIP[MCP];
  float lposx[MCP], lposy[MCP], lposz[MCP]; 
  float gposx[MCP], gposy[MCP], gposz[MCP]; 
  float radius[MCP], phi[MCP], ebkg[MCP], ebkg_err[MCP]; 
  float efficiency[MCP], rand[MCP];
  int tag[MCP];
  float scaleP[MCP];

  // parameters of detector
  std::string backgroundfilename;
  std::string SGVmapfilename;
  float densityScaling;
  float bField;
  float eBeam;
  float zbcal;
  float thresholdMin, thresholdMax;
  float xingangle;

  //
  bool DBDsample;
  bool newMap;
  bool writeTree;
  bool detectAll;
  bool smearEnergy;
  bool writeSGVMap;
  bool useInputClusters;
  
  // background handler
  BcEnergyDensity *bc_en;
  
  //helpers
  double alpha;
  double gamma;
  double betagamma; 
  double PI;
  
  
  // binning for SGV map
  int nbinx, nbiny;
  float xmin, xmax, ymin, ymax;
  TH2D* SGVmapP;
  TH2D* SGVmapN;
} ;

#endif