//
// ********************************************************************
// * 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.          *
// ********************************************************************
//
///////////////////////////////////////////////////////////////////////////////
// File: CCalStackingAction.cc
// Description: Stacking action needed for the application
///////////////////////////////////////////////////////////////////////////////
#include "CCalStackingAction.hh"
#include "G4StackManager.hh"

#include "G4SDManager.hh"
#include "CCaloSD.hh"
#include "CCalSDList.hh"
#include "G4RunManager.hh"
#include "G4Navigator.hh"

//#define debug
//#define ddebug

CCalStackingAction::CCalStackingAction(): isInitialized(false) {}


CCalStackingAction::~CCalStackingAction(){}


void CCalStackingAction::PrepareNewEvent(){

  if(!isInitialized)   initialize();
  stage = firstStage;
  nurgent = 0;
  acceptSecondaries = 1;

}

void CCalStackingAction::initialize(){

  isInitialized = true;
 
  numberOfSD = CCalSDList::getInstance()->getNumberOfCaloSD();
#ifdef debug
  G4cout << "CCalStackingAction look for " << numberOfSD 
	 << " calorimeter-like SD" << G4endl;
#endif
  int i = 0;
  for (i=0; i<numberOfSD; i++) {
    G4String theName(CCalSDList::getInstance()->getCaloSDName(i));
    SDName[i] = theName;
#ifdef debug
    G4cout << "Found SD  name " << theName << G4endl;
#endif
    theCaloSD[i] = 0;
  }   

  G4SDManager* sd = G4SDManager::GetSDMpointerIfExist();
  if (sd != 0) {
    
    for (i=0; i<numberOfSD; i++){

      G4VSensitiveDetector* aSD = sd->FindSensitiveDetector(SDName[i]);
      if (aSD==0) {
#ifdef debug
	G4cout << "CCalStackingAction::initialize: No SD with name " << SDName[i]
	       << " in this Setup " << G4endl;
#endif
      } else {
	theCaloSD[i] = dynamic_cast<CCaloSD*>(aSD);
	theCaloSD[i]->SetPrimaryID(0);
      }	   
    }
#ifdef debug
    G4cout << "CCalStackingAction::initialize: Could not get SD Manager !" 
	   << G4endl;
#endif
  }
   
}

G4ClassificationOfNewTrack CCalStackingAction::ClassifyNewTrack(const G4Track* aTrack){

  G4ClassificationOfNewTrack classification=fKill;
  int parentID = aTrack->GetParentID();
#ifdef ddebug
  G4TrackStatus status = aTrack->GetTrackStatus();
  G4cout << "Classifying track " << aTrack->GetTrackID()
	 << " with status " << aTrack->GetTrackStatus() << G4endl;  
#endif
    
  if (parentID < 0) {
#ifdef debug     
    G4cout << "Killing track " << aTrack->GetTrackID() 
	   << " from previous event. Should not happen" << G4endl;
    G4cout << "returning classification= " << classification << G4endl;
#endif
    return classification= fKill;
  }
   
  if (aTrack->GetDefinition()->GetParticleName() == "gamma" && 
      aTrack->GetKineticEnergy() < 1.*eV) {
#ifdef debug
    G4cout << "Kills particle " << aTrack->GetDefinition()->GetParticleName() 
	   << " of energy " << aTrack->GetKineticEnergy()/MeV << " MeV" 
	   << G4endl;
#endif
    return classification= fKill;
  }
    
  if (stage<end) {
    /////////////////
    /// PRIMARIES ///
    /////////////////
    if (parentID == 0 ) {
      if ( nurgent == 0) {
	nurgent++;
	classification = fUrgent;
	setPrimaryID(aTrack->GetTrackID());
      }
      else  classification = fWaiting;   
    }

    ///////////////////
    /// SECONDARIES ///
    ///////////////////
       
    if (parentID > 0) {
      if (acceptSecondaries == 1) {
	if (trackStartsInCalo(const_cast<G4Track *>(aTrack))!=0 )
	  classification = fUrgent;
	else
	  classification = fWaiting; 
      } else {
	if(nurgent == 0){                     
	  nurgent++;
	  classification = fUrgent;
	  setPrimaryID(aTrack->GetTrackID());
	} else
	  classification = fWaiting;	
      }       
    }
       
       
  } else 
    classification = G4UserStackingAction::ClassifyNewTrack(aTrack);

#ifdef ddebug
  G4cout << " returning classification= " << classification
	 << " for track "<< aTrack->GetTrackID() << G4endl;
#endif
  return classification;

}


void CCalStackingAction::NewStage(){

#ifdef ddebug
  G4cout << "In NewStage with stage = " << stage << G4endl;
#endif
  if (stage <end) {
    nurgent = 0;		    
    setPrimaryID(0);
    acceptSecondaries = 0;
    stackManager->ReClassify();
    acceptSecondaries = 1;
    if (stackManager->GetNUrgentTrack() == 0) {
      stage = stageLevel(stage+1);
    }
	
  }
}

G4bool CCalStackingAction::trackStartsInCalo(const G4Track* ){

 /// This method should check that the secondary particle
 /// was produced inside the detector calorimeter and 
 /// really is part of the shower.
 /// If it has been produced before the calorimeter 
 /// for ex. Bremsstrahlung, it should be treated as a new
 /// particle producing a new shower.

 return true;
}

void CCalStackingAction::setPrimaryID(G4int id){
  
  for (int i=0; i<numberOfSD; i++){
    if(theCaloSD[i] != 0)theCaloSD[i]->SetPrimaryID(id);
  }

}