Gabriele Cosmo
96 lines
2.4 KiB
C++
96 lines
2.4 KiB
C++
// This code implementation is the intellectual property of
|
|
// the RD44 GEANT4 collaboration.
|
|
//
|
|
// By copying, distributing or modifying the Program (or any work
|
|
// based on the Program) you indicate your acceptance of this statement,
|
|
// and all its terms.
|
|
//
|
|
// $Id: MyEventAction.cc,v 2.2 1998/07/13 17:29:43 urbi Exp $
|
|
// GEANT4 tag $Name: geant4-00 $
|
|
//
|
|
|
|
#include "MyEventAction.hh"
|
|
|
|
#include "MyCalorimeterHit.hh"
|
|
#include "MyCalorimeterHitsCollection.hh"
|
|
#include <rw/tvordvec.h>
|
|
|
|
#include "G4Event.hh"
|
|
#include "G4EventManager.hh"
|
|
#include "G4HCofThisEvent.hh"
|
|
#include "G4VHitsCollection.hh"
|
|
#include "G4TrajectoryContainer.hh"
|
|
#include "G4Trajectory.hh"
|
|
#include "G4VVisManager.hh"
|
|
#include "G4SDManager.hh"
|
|
#include "G4UImanager.hh"
|
|
#include "G4ios.hh"
|
|
|
|
#include "MyEventActionMessenger.cc"
|
|
|
|
MyEventAction::MyEventAction()
|
|
:drawFlag(false)
|
|
{
|
|
G4SDManager * SDman = G4SDManager::GetSDMpointer();
|
|
G4String colNam;
|
|
|
|
calorimeterCollID = SDman->GetCollectionID(colNam="CalCollection");
|
|
//////////bulkCollID = SDman->GetCollectionID(colNam="bulkCollection");
|
|
|
|
new MyEventActionMessenger(this);
|
|
}
|
|
|
|
MyEventAction::~MyEventAction()
|
|
{;}
|
|
|
|
void MyEventAction::BeginOfEventAction()
|
|
{
|
|
if(drawFlag)
|
|
{
|
|
G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
|
|
if(pVVisManager)
|
|
{
|
|
G4UImanager::GetUIpointer()->ApplyCommand("/vis~/Draw/current");
|
|
}
|
|
}
|
|
}
|
|
|
|
void MyEventAction::EndOfEventAction()
|
|
{
|
|
const G4Event* evt = fpEventManager->GetConstCurrentEvent();
|
|
|
|
G4cout << ">>> Event " << evt->GetEventID() << endl;
|
|
|
|
G4HCofThisEvent * HCE = evt->GetHCofThisEvent();
|
|
MyCalorimeterHitsCollection* CHC = NULL;
|
|
if(HCE)
|
|
CHC = (MyCalorimeterHitsCollection*)(HCE->GetHC(calorimeterCollID));
|
|
|
|
if(CHC)
|
|
{
|
|
RWTValOrderedVector<MyCalorimeterHit> theCollection
|
|
= CHC->GetVector();
|
|
int n_hit = theCollection.entries();
|
|
G4cout << " " << n_hit
|
|
<< " hits are stored in MyCalorimeterHitsCollection." << endl;
|
|
G4double totE = 0;
|
|
for(int i=0;i<n_hit;i++)
|
|
{ totE += theCollection[i].GetEdep(); }
|
|
G4cout << " Total energy deposition in calorimeter tubes : "
|
|
<< totE / GeV << " (GeV)" << endl;
|
|
}
|
|
|
|
if(drawFlag)
|
|
{
|
|
G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
|
|
if(pVVisManager)
|
|
{
|
|
if(CHC) CHC->DrawAllHits();
|
|
G4UImanager::GetUIpointer()->ApplyCommand("/vis~/show/view");
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|