//
// ********************************************************************
// * 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 B1RunAction.cc
/// \brief Implementation of the B1RunAction class

#include "B1RunAction.hh"
#include "B1PrimaryGeneratorAction.hh"
#include "B1DetectorConstruction.hh"
// #include "B1Run.hh"

#include "G4RunManager.hh"
#include "G4Run.hh"
#include "G4AccumulableManager.hh"
#include "G4LogicalVolumeStore.hh"
#include "G4LogicalVolume.hh"
#include "G4UnitsTable.hh"
#include "G4SystemOfUnits.hh"
#include "g4analysis.hh"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

B1RunAction::B1RunAction()
: G4UserRunAction(),
  fEdep(0.),
  fEdep2(0.)
{ 
  // add new units for dose
  // 
  const G4double milligray = 1.e-3*gray;
  const G4double microgray = 1.e-6*gray;
  const G4double nanogray  = 1.e-9*gray;  
  const G4double picogray  = 1.e-12*gray;
   
  new G4UnitDefinition("milligray", "milliGy" , "Dose", milligray);
  new G4UnitDefinition("microgray", "microGy" , "Dose", microgray);
  new G4UnitDefinition("nanogray" , "nanoGy"  , "Dose", nanogray);
  new G4UnitDefinition("picogray" , "picoGy"  , "Dose", picogray); 

  // Register accumulable to the accumulable manager
  G4AccumulableManager* accumulableManager = G4AccumulableManager::Instance();
  accumulableManager->RegisterAccumulable(fEdep);
  accumulableManager->RegisterAccumulable(fEdep2); 

 
  //-----------START CREATE ROOT FILE-------------------------------//
  auto analysisManager = G4Analysis::ManagerInstance("root");
  G4cout << "Using " << analysisManager->GetType() << G4endl;

  // Default settings
  analysisManager->SetNtupleMerging(true);
     // Note: merging ntuples is available only with Root output
  analysisManager->SetVerboseLevel(1);
  analysisManager->SetFileName("B1");
  
  // Creating ntuple
    
 //ntupla id=0 Released energy in the two targets
  analysisManager->CreateNtuple("B1EventVariables", "Relesed_energy");
  analysisManager->CreateNtupleDColumn("ReleasedEnergy");
  analysisManager->CreateNtupleDColumn("ReleasedEnergy2");
  analysisManager->CreateNtupleDColumn("ReleasedEnergyReg");
  analysisManager->FinishNtuple();
  
  //ntupla id=1	Particles outgoing the targets
  analysisManager->CreateNtuple("B1Exitparticles", "Exit_targets");
  analysisManager->CreateNtupleDColumn("ExitParticleID");
  analysisManager->CreateNtupleDColumn("ExitParticleKinEn");
  analysisManager->CreateNtupleDColumn("ExitParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("ExitParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("ExitParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("ExitParticleVert.x");
  analysisManager->CreateNtupleDColumn("ExitParticleVert.y");
  analysisManager->CreateNtupleDColumn("ExitParticleVert.z");
  analysisManager->CreateNtupleDColumn("ExitParticleTotMom");
  analysisManager->CreateNtupleDColumn("ExitDirection.x");
  analysisManager->CreateNtupleDColumn("ExitDirection.y");
  analysisManager->CreateNtupleDColumn("ExitDirection.z");
  analysisManager->CreateNtupleDColumn("ExitParticleMom.x");
  analysisManager->CreateNtupleDColumn("ExitParticleMom.y");
  analysisManager->CreateNtupleDColumn("ExitParticleMom.z");
  analysisManager->CreateNtupleDColumn("ExitParticleAng.x");
  analysisManager->CreateNtupleDColumn("ExitParticleAng.y");
  analysisManager->CreateNtupleDColumn("ExitParticleParID");
  analysisManager->FinishNtuple();
  
  //ntupla id=2	Particles entering the targets
  analysisManager->CreateNtuple("B1Enteringparticles", "Entering_targets");
  analysisManager->CreateNtupleDColumn("EnteringParticleID");
  analysisManager->CreateNtupleDColumn("EnteringParticleKinEn");
  analysisManager->CreateNtupleDColumn("EnteringParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("EnteringParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("EnteringParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("EnteringParticleVert.x");
  analysisManager->CreateNtupleDColumn("EnteringParticleVert.y");
  analysisManager->CreateNtupleDColumn("EnteringParticleVert.z");
  analysisManager->CreateNtupleDColumn("EnteringParticleTotMom");
  analysisManager->CreateNtupleDColumn("EnteringDirection.x");
  analysisManager->CreateNtupleDColumn("EnteringDirection.y");
  analysisManager->CreateNtupleDColumn("EnteringDirection.z");
  analysisManager->CreateNtupleDColumn("EnteringParticleMom.x");
  analysisManager->CreateNtupleDColumn("EnteringParticleMom.y");
  analysisManager->CreateNtupleDColumn("EnteringParticleMom.z");
  analysisManager->CreateNtupleDColumn("EnteringParticleAng.x");
  analysisManager->CreateNtupleDColumn("EnteringParticleAng.y");
  analysisManager->CreateNtupleDColumn("EnteringParticleParID");
  analysisManager->FinishNtuple();
  
  //ntupla id=3	Particles entering the regenerator
  analysisManager->CreateNtuple("B1Enterinregengparticles", "Entering_regen");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleID");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleKinEn");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleVert.x");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleVert.y");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleVert.z");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleTotMom");
  analysisManager->CreateNtupleDColumn("EnteringRegenDirection.x");
  analysisManager->CreateNtupleDColumn("EnteringRegenDirection.y");
  analysisManager->CreateNtupleDColumn("EnteringRegenDirection.z");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMom.x");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMom.y");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleMom.z");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleAng.x");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleAng.y");
  analysisManager->CreateNtupleDColumn("EnteringRegenParticleParID");
  analysisManager->FinishNtuple();
  
   //ntupla id=4	Particles entering the regenerator
  analysisManager->CreateNtuple("B1EnterinPhotonregengparticles", "Entering_photon_regen");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleID");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleKinEn");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleVert.x");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleVert.y");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleVert.z");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleTotMom");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenDirection.x");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenDirection.y");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenDirection.z");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMom.x");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMom.y");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleMom.z");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleAng.x");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleAng.y");
  analysisManager->CreateNtupleDColumn("EnteringPhotonRegenParticleParID");
  analysisManager->FinishNtuple();
  
  //ntupla id=5	Particles outgoing the regenerator
  analysisManager->CreateNtuple("B1Exitregengparticles", "Exit_regen");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleID");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleKinEn");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleVert.x");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleVert.y");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleVert.z");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleTotMom");
  analysisManager->CreateNtupleDColumn("ExitRegenDirection.x");
  analysisManager->CreateNtupleDColumn("ExitRegenDirection.y");
  analysisManager->CreateNtupleDColumn("ExitRegenDirection.z");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMom.x");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMom.y");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleMom.z");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleAng.x");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleAng.y");
  analysisManager->CreateNtupleDColumn("ExitRegenParticleParID");
  analysisManager->FinishNtuple();
  
   //ntupla id=6	Particles hitting the scintillator
  analysisManager->CreateNtuple("B1Scintparticles", "Ent_scint");
  analysisManager->CreateNtupleDColumn("ScintParticleID");
  analysisManager->CreateNtupleDColumn("ScintParticleKinEn");
  analysisManager->CreateNtupleDColumn("ScintParticleMomDir.x");
  analysisManager->CreateNtupleDColumn("ScintParticleMomDir.y");
  analysisManager->CreateNtupleDColumn("ScintParticleMomDir.z");
  analysisManager->CreateNtupleDColumn("ScintParticleVert.x");
  analysisManager->CreateNtupleDColumn("ScintParticleVert.y");
  analysisManager->CreateNtupleDColumn("ScintParticleVert.z");
  analysisManager->CreateNtupleDColumn("ScintParticleTotMom");
  analysisManager->CreateNtupleDColumn("ScintDirection.x");
  analysisManager->CreateNtupleDColumn("ScintDirection.y");
  analysisManager->CreateNtupleDColumn("ScintDirection.z");
  analysisManager->CreateNtupleDColumn("ScintParticleMom.x");
  analysisManager->CreateNtupleDColumn("ScintParticleMom.y");
  analysisManager->CreateNtupleDColumn("ScintParticleMom.z");
  analysisManager->CreateNtupleDColumn("ScintParticleAng.x");
  analysisManager->CreateNtupleDColumn("ScintParticleAng.y");
  analysisManager->CreateNtupleDColumn("ScintParticleParID");
  analysisManager->FinishNtuple();
  //-----------END CREATE NMTUPLE-------------------------------//
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

B1RunAction::~B1RunAction()
{}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

void B1RunAction::BeginOfRunAction(const G4Run*)
{ 
  // inform the runManager to save random number seed
  G4RunManager::GetRunManager()->SetRandomNumberStore(false);

  // reset accumulables to their initial values
  G4AccumulableManager* accumulableManager = G4AccumulableManager::Instance();
  accumulableManager->Reset();
  
  
  // Get analysis manager
  auto analysisManager = G4AnalysisManager::Instance();

  // Open an output file 
  // The default file name is set in B5RunAction::B5RunAction(),
  // it can be overwritten in a macro
  analysisManager->OpenFile();

}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

void B1RunAction::EndOfRunAction(const G4Run* run)
{

     // save histograms & ntuple
  //
  auto analysisManager = G4AnalysisManager::Instance();
  analysisManager->Write();
  analysisManager->CloseFile();
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......

void B1RunAction::AddEdep(G4double edep)
{
  fEdep  += edep;
  fEdep2 += edep*edep;
  
   
}

void B1RunAction::AddEdep2(G4double edep)
{
  fEdep  += edep;
  
}

void B1RunAction::AddEdepReg(G4double edep)
{
  fEdep  += edep;
  
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo......