EcalTBH4Trigger.h

Go to the documentation of this file.

#ifndef HelpfulWatchers_EcalTBH4Trigger_h
#define HelpfulWatchers_EcalTBH4Trigger_h
// -*- C++ -*-
//
// Package:     HelpfulWatchers
// Class  :     EcalTBH4Trigger
//
//

// system include files
#include <iostream>

// user include files
#include "SimG4Core/Notification/interface/Observer.h"
#include "SimG4Core/Notification/interface/SimG4Exception.h"
#include "SimG4Core/Watcher/interface/SimWatcher.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "G4Step.hh"
#include "G4VProcess.hh"
#include "G4VTouchable.hh"

#include "CLHEP/Units/SystemOfUnits.h"

// forward declarations
class DDDWorld;
class BeginOfJob;
class BeginOfRun;
class BeginOfEvent;
class BeginOfTrack;

class EndOfRun;
class EndOfEvent;
class EndOfTrack;

#define OBSERVES(type) \
public                 \
  Observer<const type *>
#define UPDATEH4(type) \
  void update(const type *) {}
class EcalTBH4Trigger : public SimWatcher,
                        // OBSERVES(DDDWorld),
                        // OBSERVES(BeginOfJob),
                        // OBSERVES(BeginOfRun),
                        OBSERVES(BeginOfEvent),
                        // OBSERVES(BeginOfTrack),
                        OBSERVES(G4Step),
                        // OBSERVES(EndOfRun),
                        OBSERVES(EndOfEvent)
//,
// OBSERVES(EndOfTrack)
{
public:
  EcalTBH4Trigger(const edm::ParameterSet &pSet)
      : m_verbose(pSet.getUntrackedParameter<bool>("verbose", false)),
        nTriggeredEvents_(0),
        trigEvents_(pSet.getUntrackedParameter<int>("trigEvents", -1)) {}
  // virtual ~EcalTBH4Trigger();

  // ---------- const member functions ---------------------

  // ---------- static member functions --------------------

  // ---------- member functions ---------------------------
  //  UPDATEH4(DDDWorld)
  // UPDATEH4(BeginOfJob)
  // UPDATEH4(BeginOfRun)
  //  UPDATEH4(BeginOfEvent)
  void update(const BeginOfEvent *anEvent) override {
    //      std::cout <<"++ signal BeginOfEvent " ;
    //      m_enteringTBH4BeamLine=false;
    //      m_exitingTBH4BeamLine=false;
    //      m_passedTrigger=false;
    m_passedTrg1 = false;
    m_passedTrg3 = false;
    m_passedTrg4 = false;
    m_passedTrg5 = false;
    m_passedTrg6 = false;
  }

  //  UPDATEH4(BeginOfTrack)
  void update(const G4Step *iStep) override {
    if (trigEvents_ >= 0 && nTriggeredEvents_ >= trigEvents_)
      throw SimG4Exception("Number of needed trigger events reached in ECALTBH4");

    const G4StepPoint *pre = iStep->GetPreStepPoint();
    const G4StepPoint *post = iStep->GetPostStepPoint();
    if (m_verbose) {
      std::cout << "++ signal G4Step";
      const G4VTouchable *touch = iStep->GetPreStepPoint()->GetTouchable();
      // Get name and copy numbers
      if (touch->GetHistoryDepth() > 0) {
        for (int ii = 0; ii <= touch->GetHistoryDepth(); ii++) {
          std::cout << "EcalTBH4::Level " << ii << ": " << touch->GetVolume(ii)->GetName() << "["
                    << touch->GetReplicaNumber(ii) << "]";
        }
      }
      std::cout << std::endl;
      const G4Track *theTrack = iStep->GetTrack();
      const G4ThreeVector &pos = post->GetPosition();
      std::cout << "( " << pos.x() << "," << pos.y() << "," << pos.z() << ") ";
      std::cout << " released energy (MeV) " << iStep->GetTotalEnergyDeposit() / CLHEP::MeV;
      if (theTrack) {
        const G4ThreeVector mom = theTrack->GetMomentum();
        std::cout << " track length (cm) " << theTrack->GetTrackLength() / CLHEP::cm << " particle type "
                  << theTrack->GetDefinition()->GetParticleName() << " momentum "
                  << "( " << mom.x() << "," << mom.y() << "," << mom.z() << ") ";
        if (theTrack->GetCreatorProcess()) {
          std::cout << " created by " << theTrack->GetCreatorProcess()->GetProcessName();
        }
      }
      if (post->GetPhysicalVolume()) {
        std::cout << " " << pre->GetPhysicalVolume()->GetName() << "->" << post->GetPhysicalVolume()->GetName();
      }
      std::cout << std::endl;
    }

    if (post && post->GetPhysicalVolume()) {
      if (!m_passedTrg1 && post->GetPhysicalVolume()->GetName() == "TRG1")
        m_passedTrg1 = true;
      if (!m_passedTrg3 && post->GetPhysicalVolume()->GetName() == "TRG3")
        m_passedTrg3 = true;
      if (!m_passedTrg4 && post->GetPhysicalVolume()->GetName() == "TRG4")
        m_passedTrg4 = true;
      if (!m_passedTrg5 && post->GetPhysicalVolume()->GetName() == "TRG5")
        m_passedTrg5 = true;
      if (!m_passedTrg6 && post->GetPhysicalVolume()->GetName() == "TRG6")
        m_passedTrg6 = true;
      if (post->GetPhysicalVolume()->GetName() == "CMSSE")  // Exiting TBH4BeamLine
        if (!(m_passedTrg1 && m_passedTrg6))                // Trigger defined as Trg4 && Trg6
          throw SimG4Exception("Event is not triggered by ECALTBH4");
    }

    /*     if (!m_enteringTBH4BeamLine && ( post->GetPhysicalVolume()->GetName()
     * ==  */
  }
  // UPDATEH4(G4Step)
  // UPDATEH4(EndOfRun)
  // UPDATEH4(EndOfEvent)
  void update(const EndOfEvent *anEvent) override {
    //      std::cout <<"++ signal BeginOfEvent " ;
    //      m_enteringTBH4BeamLine=false;
    //      m_exitingTBH4BeamLine=false;
    //      m_passedTrigger=false;
    nTriggeredEvents_++;
  }
  // UPDATEH4(EndOfTrack)

private:
  // EcalTBH4Trigger(const EcalTBH4Trigger&); // stop default

  // const EcalTBH4Trigger& operator=(const EcalTBH4Trigger&); // stop default

  // ---------- member data --------------------------------
  bool m_verbose;
  // bool m_enteringTBH4BeamLine;
  // bool m_exitingTBH4BeamLine;
  bool m_passedTrg1;
  bool m_passedTrg3;
  bool m_passedTrg4;
  bool m_passedTrg5;
  bool m_passedTrg6;
  int nTriggeredEvents_;
  int trigEvents_;
  // bool m_passedTrigger;
};

#endif