FP420Test.h

Go to the documentation of this file.

#ifndef FP420Test_H
#define FP420Test_H

// system include files
#include <vector>
#include <iostream>
#include <memory>
#include <string>
//
// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/MakerMacros.h"

// necessary objects:
#include "FWCore/Framework/interface/ESHandle.h"

//
//
#include "SimG4Core/Notification/interface/Observer.h"
#include "SimG4Core/Notification/interface/BeginOfJob.h"
#include "SimG4Core/Notification/interface/BeginOfRun.h"
#include "SimG4Core/Notification/interface/EndOfRun.h"
#include "SimG4Core/Notification/interface/BeginOfEvent.h"
#include "SimG4Core/Notification/interface/EndOfEvent.h"
#include "SimG4Core/Notification/interface/BeginOfTrack.h"
#include "SimG4Core/Notification/interface/EndOfTrack.h"
#include "SimG4Core/Watcher/interface/SimWatcher.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "SimG4CMS/FP420/interface/FP420G4Hit.h"

#include "G4VTouchable.hh"
#include <map>
#include <cmath>
#include <CLHEP/Random/Randomize.h>

// ----------------------------------------------------------------
// Includes needed for Root ntupling
//
#include "TROOT.h"
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "TProfile.h"
#include "TNtuple.h"
#include "TRandom.h"
#include "TLorentzVector.h"
#include "TUnixSystem.h"
#include "TSystem.h"
#include "TMath.h"
#include "TF1.h"

#include <TObjArray.h>
#include <TObjString.h>
#include <TNamed.h>

class Fp420AnalysisHistManager : public TNamed {
public:
  Fp420AnalysisHistManager(const TString& managername);
  ~Fp420AnalysisHistManager() override;

  TH1F* GetHisto(Int_t Number);
  TH1F* GetHisto(const TObjString& histname);

  TH2F* GetHisto2(Int_t Number);
  TH2F* GetHisto2(const TObjString& histname);

  void WriteToFile(const TString& fOutputFile, const TString& fRecreateFile);

private:
  void BookHistos();
  void StoreWeights();
  void HistInit(const char* name, const char* title, Int_t nbinsx, Axis_t xlow, Axis_t xup);
  void HistInit(
      const char* name, const char* title, Int_t nbinsx, Axis_t xlow, Axis_t xup, Int_t nbinsy, Axis_t ylow, Axis_t yup);

  const char* fTypeTitle;
  TObjArray* fHistArray;
  TObjArray* fHistNamesArray;
};

class FP420NumberingScheme;

class BeginOfJob;
class BeginOfRun;
class EndOfRun;
class BeginOfEvent;
class EndOfEvent;
class BeginOfTrack;
class EndOfTrack;
class G4Step;

/*
class ObserveBeginOfRun : private Observer<const BeginOfRun *> 
{
public:
    ObserveBeginOfRun();
private:
    void update(const BeginOfRun * run);
};

class ObserveEndOfRun : private Observer<const EndOfRun *> 
{
public:
    ObserveEndOfRun();
private:
    void update(const EndOfRun * run);
};

class ObserveBeginOfEvent : private Observer<const BeginOfEvent *> 
{
public:
    ObserveBeginOfEvent();
private:
    void update(const BeginOfEvent * evt);
};

class ObserveEndOfEvent : private Observer<const EndOfEvent *> 
{
public:
    ObserveEndOfEvent();
private:
    void update(const EndOfEvent * evt);
    //    std::vector<FP420G4Hit> theStripHits;
};

class ObserveBeginOfTrack : private Observer<const BeginOfTrack *> 
{
public:
    ObserveBeginOfTrack();
private:
    void update(const BeginOfTrack * trk);
};

class ObserveEndOfTrack : private Observer<const EndOfTrack *> 
{
public:
    ObserveEndOfTrack();
private:
    void update(const EndOfTrack * trk);
};

class ObserveStep : private Observer<const G4Step *> 
{
public:
    ObserveStep();
private:
    void update(const G4Step * step);
};
*/
//class FP420Test: public SimProducer,
class FP420Test : public SimWatcher,
                  public Observer<const BeginOfJob*>,
                  public Observer<const BeginOfRun*>,
                  public Observer<const EndOfRun*>,
                  public Observer<const BeginOfEvent*>,
                  public Observer<const BeginOfTrack*>,
                  public Observer<const G4Step*>,
                  public Observer<const EndOfTrack*>,
                  public Observer<const EndOfEvent*> {
public:
  FP420Test(const edm::ParameterSet& p);
  ~FP420Test() override;
  //MyActions();
  //MyActions();
private:
  // observer classes
  void update(const BeginOfJob* run) override;
  void update(const BeginOfRun* run) override;
  void update(const EndOfRun* run) override;
  void update(const BeginOfEvent* evt) override;
  void update(const BeginOfTrack* trk) override;
  void update(const G4Step* step) override;
  void update(const EndOfTrack* trk) override;
  void update(const EndOfEvent* evt) override;

private:
  //UHB_Analysis* UserNtuples;

  int iev;
  int itrk;
  G4double entot0, tracklength0;

private:
  // Utilities to get detector levels during a step

  int detLevels(const G4VTouchable*) const;
  G4String detName(const G4VTouchable*, int, int) const;
  void detectorLevel(const G4VTouchable*, int&, int*, G4String*) const;

  double rinCalo, zinCalo;
  int lastTrackID;
  int verbosity;

  // SumEnerDeposit - all deposited energy on all steps ;  SumStepl - length in steel !!!
  G4double SumEnerDeposit, SumStepl, SumStepc;
  // numofpart - # particles produced along primary track
  int numofpart;
  // last point of the primary track
  G4ThreeVector lastpo;

  // z:
  double z1, z2, z3, z4, zD2, zD3;
  int sn0, dn0, pn0, rn0;
  int rn00;
  double ZSiDet, z420;
  double ZGapLDet, ZBoundDet, ZSiStep, ZSiPlane, zinibeg;
  double zBlade, gapBlade;

private:
  Float_t fp420eventarray[1];
  TNtuple* fp420eventntuple;
  TFile fp420OutputFile;
  int whichevent;

  Fp420AnalysisHistManager* TheHistManager;  //Histogram Manager of the analysis
  std::string fDataLabel;                    // Data type label
  std::string fOutputFile;                   // The output file name
  std::string fRecreateFile;                 // Recreate the file flag, default="RECREATE"

  //  //  //  //  //  //  TObjString fHistType;
  //   TString fDataLabel;             // Data type label
  //   TString fOutputFile;            // The output file name
  //   TString fRecreateFile;          // Recreate the file flag, default="RECREATE"
};

#endif