source: trunk/examples/extended/parameterisations/gflash/src/ExGflashEventAction.cc @ 807

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
// Created by Joanna Weng 26.11.2004
using namespace std;

#include "ExGflashEventAction.hh"
#include "ExGflashHit.hh"
#include "G4EventManager.hh"
#include "G4SDManager.hh"
#include "G4TrajectoryContainer.hh"
#include "G4Trajectory.hh"
#include "G4VVisManager.hh"
#include "G4UImanager.hh"
#include "G4TrajectoryContainer.hh"
#include "G4VVisManager.hh"
#include "G4Event.hh"
//std
#include <iostream>
#include <algorithm>
//Gflash


#include "G4Timer.hh"
extern G4Timer Timer;
extern G4Timer Timerintern;



ExGflashEventAction::ExGflashEventAction():
nevent(0),dtime(0.0),calorimeterCollectionId(-1)
{
}

ExGflashEventAction::~ExGflashEventAction()
{
        cout << "Internal Real Elapsed Time /event is: "<< dtime /nevent<< endl;
}


void ExGflashEventAction::BeginOfEventAction(const G4Event *evt){
        Timerintern.Start();
        cout<<" ------ Start ExGflashEventAction ----- "<<endl;
        nevent=evt->GetEventID();
        cout<<" Start generating event Nr "<<nevent<<endl<<endl;        
}

void ExGflashEventAction::EndOfEventAction(const G4Event *evt)
{  
        Timerintern.Stop();
        cout << endl;
        cout << "******************************************";
        cout << endl;
        cout << "Internal Real Elapsed Time is: "<< Timerintern.GetRealElapsed();
        cout << endl;
        cout << "Internal System Elapsed Time: " << Timerintern.GetSystemElapsed();
        cout << endl;
        cout << "Internal GetUserElapsed Time: " << Timerintern.GetUserElapsed();
        cout << endl;
        cout << "******************************************"<< endl;
        dtime+=Timerintern.GetRealElapsed();
        cout<<" ------ ExGflashEventAction::End of event nr. "<<nevent<<"  -----"<< endl;     

        G4SDManager * SDman = G4SDManager::GetSDMpointer();
        G4String colNam;
        calorimeterCollectionId=SDman->GetCollectionID(colNam="ExGflashCollection");
        if (calorimeterCollectionId<0) return;
        G4HCofThisEvent * HCE = evt->GetHCofThisEvent();
        ExGflashHitsCollection* THC = 0;
        G4double totE = 0;
        // Read out of the crysta ECAL
        THC=(ExGflashHitsCollection *)(HCE->GetHC(calorimeterCollectionId));
        if (THC) {
                /// Hits in sensitive Detector
                int n_hit = THC->entries();
                cout<<"  " << n_hit<< " hits are stored in ExGflashHitsCollection "<<endl;
                G4PrimaryVertex* pvertex=evt->GetPrimaryVertex();   
                ///Computing (x,y,z) of vertex of initial particles  
                G4ThreeVector vtx=pvertex->GetPosition();
                G4PrimaryParticle* pparticle=pvertex->GetPrimary();
                // direction of the Shower
                G4ThreeVector mom=pparticle->GetMomentum()/pparticle->GetMomentum().mag();
                
                //@@@ ExGflashEventAction: Magicnumber
                G4double energyincrystal[100];
                for (int i=0;i<100;i++) energyincrystal[i]=0.;
                
                //@@@ ExGflashEventAction: Magicnumber
                /// For all Hits in sensitive detector
                for (int i=0;i<n_hit;i++)
                {
                        G4double estep = (*THC)[i]->GetEdep()/GeV;
                        if (estep >0.0)
                        {
                                totE += (*THC)[i]->GetEdep()/GeV;
                                G4int num=(*THC)[i]->GetCrystalNum();
                                
                                energyincrystal[num]+=(*THC)[i]->GetEdep()/GeV;
                                //std::cout << num << std::endl;
                                //@@@ExGflashEventAction: was bringen die namespace statt using namespace std ?
                        //      std::cout << " Crystal Nummer " <<  (*THC)[i]->GetCrystalNum()  << std::endl;
                        //      std::cout <<  (*THC)[i]->GetCrystalNum() /10 << "  "<<(*THC)[i]->GetCrystalNum()%10 << std::endl;
                                
                                G4ThreeVector hitpos=(*THC)[i]->GetPos();                                       
                                G4ThreeVector l (hitpos.x(), hitpos.y(), hitpos.z());
                                // distance from shower start
                                l = l - vtx; 
                                // projection on shower axis = longitudinal profile
                                G4ThreeVector longitudinal  =  l;       
                                // shower profiles (Radial)
                                G4ThreeVector radial = vtx.cross(l);
                        }
                }
                G4double max=0;
                G4int index=0;
                //Find crystal with maximum energy
                for (int i=0;i<100;i++) 
                {
                        //std::cout << i <<"  " << energyincrystal[i] << std::endl;
                        if (max <energyincrystal[i])
                        {
                                max=energyincrystal[i];
                                index = i;
                        }
                }       
        //std::cout << index <<" NMAX  " << index << std::endl; 

        // 3x3 matrix of crystals around the crystal with the maximum energy deposit
        G4double e3x3 = energyincrystal[index]+energyincrystal[index+1]+energyincrystal[index-1]+
        energyincrystal[index-10]+energyincrystal[index-9]+energyincrystal[index-11]+
        energyincrystal[index+10]+energyincrystal[index+11]+energyincrystal[index+9];

        // 5x5 matrix of crystals around the crystal with the maximum energy deposit    
        G4double e5x5 = energyincrystal[index]+energyincrystal[index+1]+energyincrystal[index-1]+energyincrystal[index+2]+energyincrystal[index-2]+
        energyincrystal[index-10]+energyincrystal[index-9]+energyincrystal[index-11]+energyincrystal[index-8]+energyincrystal[index-12]+
        energyincrystal[index+10]+energyincrystal[index+11]+energyincrystal[index+9]+energyincrystal[index+12]+energyincrystal[index+8];
        
        std::cout << "   e1  " << energyincrystal[index]  << "   e3x3  " << e3x3<<  "   GeV  e5x5  "   <<e5x5  <<std::endl;     
        }
        
        G4cout << " Total energy deposited in the calorimeter: " << totE << " (GeV)" << G4endl;
        G4TrajectoryContainer * trajectoryContainer = evt->GetTrajectoryContainer();
        G4int n_trajectories = 0;
        if(trajectoryContainer){ n_trajectories = trajectoryContainer->entries(); }
        G4cout << "    " << n_trajectories  << " trajectories stored in this event." << endl;
        G4VVisManager* pVVisManager = G4VVisManager::GetConcreteInstance();
        
        if(pVVisManager)
        { 
                cout << "DRAWING " <<n_trajectories << endl;
                for(G4int i=0; i<n_trajectories; i++) { (*(evt->GetTrajectoryContainer()))[i]->DrawTrajectory(50); }
        }
        
}