// 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");
    }
  }
}