ML2SDWithVoxels.cc

이 파일의 문서화 페이지로 가기

//
// ********************************************************************
// * 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.          *
// ********************************************************************
//
// The code was written by :
//  ^Claudio Andenna  claudio.andenna@ispesl.it, claudio.andenna@iss.infn.it
//      *Barbara Caccia barbara.caccia@iss.it
//      with the support of Pablo Cirrone (LNS, INFN Catania Italy)
//  with the contribute of Alessandro Occhigrossi*
//
// ^INAIL DIPIA - ex ISPESL and INFN Roma, gruppo collegato Sanità, Italy
// *Istituto Superiore di Sanità and INFN Roma, gruppo collegato Sanità, Italy
//  Viale Regina Elena 299, 00161 Roma (Italy)
//  tel (39) 06 49902246
//  fax (39) 06 49387075
//
// more information:
// http://g4advancedexamples.lngs.infn.it/Examples/medical-linac
//
//*******************************************************//

#include "ML2SDWithVoxels.hh"
#include "ML2ExpVoxels.hh"
#include "G4SystemOfUnits.hh"

// *************************************************************************************

CML2SDWithVoxels::CML2SDWithVoxels(G4String name, G4int voxSave,
                                   G4int seed, G4String ROGOutFile,
                                   G4bool bROG, G4ThreeVector ctr, G4ThreeVector hSiz,
                                   G4int NumVX, G4int NumVY, G4int NumVZ)
: G4VSensitiveDetector(name),voxelsSum(0), voxelsSingle(0)
{
  saving_in_ROG_Voxels_every_events=voxSave;
  bSaveROG=bROG;
  bActive=true;
  nParticle=0;
  nParticleValatile=0;
  nTotalEvents=0;
  nSingleTotalEvents=0;
  density=1.;
  voxelVolume=0.;
  voxelMass=0.;
  nRecycling=1;

  G4String seedName;
  char a[10];
  sprintf(a,"%d", seed);
  seedName=(G4String)a;

  fullOutFileData=ROGOutFile+"_"+seedName+".txt";
  fullOutFileDataSingle="";
  if(bSaveROG)
  {
    centre=ctr;
    halfSize=hSiz;
    NumberOfVoxelsAlongX=NumVX;
    NumberOfVoxelsAlongY=NumVY;
    NumberOfVoxelsAlongZ=NumVZ;
    halfXVoxelDimensionX=halfSize.getX()/NumberOfVoxelsAlongX;
    halfXVoxelDimensionY=halfSize.getY()/NumberOfVoxelsAlongY;
    halfXVoxelDimensionZ=halfSize.getZ()/NumberOfVoxelsAlongZ;

    voxelVolume=halfXVoxelDimensionX*halfXVoxelDimensionY*halfXVoxelDimensionZ*8.;

// voxels to store and save the sum of the geometry configurations
    voxelsSum=new Svoxel**[NumberOfVoxelsAlongX];
    for (int ix=0; ix< NumberOfVoxelsAlongX; ix++)
    {
      voxelsSum[ix]=new Svoxel*[NumberOfVoxelsAlongY];
      for (int iy=0; iy< NumberOfVoxelsAlongY; iy++)
      {
        voxelsSum[ix][iy]=new Svoxel[NumberOfVoxelsAlongZ];
        for (int iz=0; iz< NumberOfVoxelsAlongZ; iz++)
        {
          voxelsSum[ix][iy][iz].volumeId=-1;
          voxelsSum[ix][iy][iz].depEnergy=0.;
          voxelsSum[ix][iy][iz].depEnergy2=0.;
          voxelsSum[ix][iy][iz].depEnergyNorm=0.;
          voxelsSum[ix][iy][iz].depEnergyNormError=0.;
          voxelsSum[ix][iy][iz].expDose=0.;
          voxelsSum[ix][iy][iz].halfSize.set(halfXVoxelDimensionX, halfXVoxelDimensionY, halfXVoxelDimensionZ);
          voxelsSum[ix][iy][iz].pos.set(2.*(ix)*halfXVoxelDimensionX  -halfSize.getX()+halfXVoxelDimensionX + centre.getX(), 
      2.*(iy)*halfXVoxelDimensionY  -halfSize.getY()+halfXVoxelDimensionY + centre.getY(), 
      2.*(iz)*halfXVoxelDimensionZ  -halfSize.getZ()+halfXVoxelDimensionZ + centre.getZ());
          voxelsSum[ix][iy][iz].nEvents=0;
        }
      }
    }


// voxels to store and save the single geometry configuration
    voxelsSingle=new Svoxel**[NumberOfVoxelsAlongX];
    for (int ix=0; ix< NumberOfVoxelsAlongX; ix++)
    {
      voxelsSingle[ix]=new Svoxel*[NumberOfVoxelsAlongY];
      for (int iy=0; iy< NumberOfVoxelsAlongY; iy++)
      {
        voxelsSingle[ix][iy]=new Svoxel[NumberOfVoxelsAlongZ];
        for (int iz=0; iz< NumberOfVoxelsAlongZ; iz++)
        {
          voxelsSingle[ix][iy][iz].volumeId=-1;
          voxelsSingle[ix][iy][iz].depEnergy=0.;
          voxelsSingle[ix][iy][iz].depEnergy2=0.;
          voxelsSingle[ix][iy][iz].depEnergyNorm=0.;
          voxelsSingle[ix][iy][iz].depEnergyNormError=0.;
          voxelsSingle[ix][iy][iz].expDose=0.;
          voxelsSingle[ix][iy][iz].halfSize.set(halfXVoxelDimensionX, halfXVoxelDimensionY, halfXVoxelDimensionZ);
          voxelsSingle[ix][iy][iz].pos.set(2.*(ix)*halfXVoxelDimensionX  -halfSize.getX()+halfXVoxelDimensionX + centre.getX(), 
      2.*(iy)*halfXVoxelDimensionY  -halfSize.getY()+halfXVoxelDimensionY + centre.getY(), 
      2.*(iz)*halfXVoxelDimensionZ  -halfSize.getZ()+halfXVoxelDimensionZ + centre.getZ());
          voxelsSingle[ix][iy][iz].nEvents=0;
        }
      }
    }
  }
}

CML2SDWithVoxels::~CML2SDWithVoxels()
{
  if(bSaveROG)
  {
    delete [] voxelsSum;
    delete [] voxelsSingle;
  }
}
void CML2SDWithVoxels::resetVoxelsSingle()
{
  for (int ix=0; ix< NumberOfVoxelsAlongX; ix++)
  {
    for (int iy=0; iy< NumberOfVoxelsAlongY; iy++)
    {
      for (int iz=0; iz< NumberOfVoxelsAlongZ; iz++)
      {
        voxelsSingle[ix][iy][iz].volumeId=-1;
        voxelsSingle[ix][iy][iz].depEnergy=0.;
        voxelsSingle[ix][iy][iz].depEnergy2=0.;
        voxelsSingle[ix][iy][iz].depEnergyNorm=0.;
        voxelsSingle[ix][iy][iz].depEnergyNormError=0.;
        voxelsSingle[ix][iy][iz].expDose=0.;
        voxelsSingle[ix][iy][iz].halfSize.set(halfXVoxelDimensionX, halfXVoxelDimensionY, halfXVoxelDimensionZ);
        voxelsSingle[ix][iy][iz].pos.set(2.*(ix)*halfXVoxelDimensionX  -halfSize.getX()+halfXVoxelDimensionX + centre.getX(), 
    2.*(iy)*halfXVoxelDimensionY  -halfSize.getY()+halfXVoxelDimensionY + centre.getY(), 
    2.*(iz)*halfXVoxelDimensionZ  -halfSize.getZ()+halfXVoxelDimensionZ + centre.getZ());
        voxelsSingle[ix][iy][iz].nEvents=0;
      }
    }
  }
  nSingleTotalEvents=0;
}
G4bool CML2SDWithVoxels::ProcessHits(G4Step *aStep, G4TouchableHistory *ROHist)
{
  
  if (bActive)
  {
    G4double energyDep = aStep->GetTotalEnergyDeposit();
    if (bSaveROG && energyDep>0.) 
    {
      G4int ix, iy, iz;
      G4String volumeName;

      ix=ROHist->GetReplicaNumber(2);
      iy=ROHist->GetReplicaNumber(0);
      iz=ROHist->GetReplicaNumber(1);

      density=aStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume()->GetMaterial()->GetDensity();

      voxelMass=voxelVolume*density;
      energyDep/=voxelMass*nRecycling;
      voxelsSum[ix][iy][iz].volumeId=getIdFromVolumeName(aStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume()->GetMaterial()->GetName());
      voxelsSum[ix][iy][iz].depEnergy+=energyDep;
      voxelsSum[ix][iy][iz].depEnergy2+=energyDep*energyDep;
      voxelsSum[ix][iy][iz].nEvents++;

      voxelsSingle[ix][iy][iz].volumeId=getIdFromVolumeName(aStep->GetPreStepPoint()->GetPhysicalVolume()->GetLogicalVolume()->GetMaterial()->GetName());
      voxelsSingle[ix][iy][iz].depEnergy+=energyDep;
      voxelsSingle[ix][iy][iz].depEnergy2+=energyDep*energyDep;
      voxelsSingle[ix][iy][iz].nEvents++;
      nTotalEvents++;
      nSingleTotalEvents++;
      if (nTotalEvents%saving_in_ROG_Voxels_every_events==0 && nTotalEvents>0)
      {
        save();
      }
    }
  }
  return true;
}
G4int CML2SDWithVoxels::getIdFromVolumeName(G4String name)
{
  for (int i=0; i<(int)volumeNameIdLink.size(); i++)
  {
    if (volumeNameIdLink[i].volumeName==name)
    {
      return volumeNameIdLink[i].volumeId;
      break;
    }
  }
  return -1;
}
void CML2SDWithVoxels::save()
{
std::cout<< "n. of events collected in the whole ROG phantom for all geometries: "<< nTotalEvents<< G4endl;
std::cout<< "n. of events collected in the whole ROG phantom for the current geometry: "<< nSingleTotalEvents<< G4endl;
  if (nTotalEvents>0)
  {saveData(fullOutFileData, voxelsSum);}
  if (nSingleTotalEvents>0)
  {saveData(fullOutFileDataSingle, voxelsSingle);}
}
void CML2SDWithVoxels::saveData(G4String Filename, Svoxel ***voxels)
{
  std::ofstream out;
  out.open(Filename, std::ios::out);
  out << "Sensitive Detector-Voxels. Total number of events, [mm]->centreX centreY centreZ HalfSizeX HalfSizeY HalfSizeZ minX maxX, minY maxY, minZ maxZ, Dx, Dy, Dz, nX, nY, nZ: \n";
  out <<nTotalEvents<<'\t';
  out <<centre.getX()/mm << '\t' << centre.getY()/mm<< '\t'<< centre.getZ()/mm<<'\t';
  out <<halfSize.getX()/mm << '\t' << halfSize.getY()/mm <<'\t'<< halfSize.getZ()/mm<<'\t';
  out <<(centre.getX()-halfSize.getX())/mm<<'\t'<<(centre.getX()+halfSize.getX())/mm<<'\t';
  out <<(centre.getY()-halfSize.getY())/mm<<'\t'<<(centre.getY()+halfSize.getY())/mm<<'\t';
  out <<(centre.getZ()-halfSize.getZ())/mm<<'\t'<<(centre.getZ()+halfSize.getZ())/mm<<'\t';
  out <<halfXVoxelDimensionX/mm<<'\t'<<halfXVoxelDimensionY/mm<<'\t'<<halfXVoxelDimensionZ/mm<<'\t';
  out <<NumberOfVoxelsAlongX <<'\t'<<NumberOfVoxelsAlongY <<'\t'<<NumberOfVoxelsAlongZ <<'\n';
  out << "Number of physical volumes: "<< volumeNameIdLink.size() << '\n';
  for (int i=0; i<(int)volumeNameIdLink.size(); i++)
  {
    out << volumeNameIdLink[i].volumeName  <<'\t'<< volumeNameIdLink[i].volumeId << G4endl;
  }


  out << "Phys Volume x [mm], y [mm], z [mm], ix, iy, iz, Dose [Gy], Dose2 [Gy^2], nEvents" << G4endl;

  for (int ix=0; ix< NumberOfVoxelsAlongX; ix++)
  {
    for (int iy=0; iy< NumberOfVoxelsAlongY; iy++)
    {
      for (int iz=0; iz< NumberOfVoxelsAlongZ; iz++)
      {
        if (voxels[ix][iy][iz].nEvents>0)
        {
          out << voxels[ix][iy][iz].volumeId << '\t';
          out << voxels[ix][iy][iz].pos.getX()/mm << '\t';
          out << voxels[ix][iy][iz].pos.getY()/mm << '\t';
          out << voxels[ix][iy][iz].pos.getZ()/mm << '\t';
          out << ix << '\t';
          out << iy << '\t';
          out << iz << '\t';
          out << voxels[ix][iy][iz].depEnergy/(joule/kg) << '\t';
          out << voxels[ix][iy][iz].depEnergy2/((joule/kg)*(joule/kg)) << '\t';
          out << voxels[ix][iy][iz].nEvents << G4endl;
        }
      }
    }
  }
  out.close();

}
void CML2SDWithVoxels::setFullOutFileDataSingle(G4String val)
{
  unsigned int ind = fullOutFileData.find(".txt");
  G4String onlyName=fullOutFileData.substr( 0, ind);
  if (val=="")
  {
    static unsigned int indGeom=0;
    char cT[5];
    sprintf(cT,"%d",indGeom);
    fullOutFileDataSingle=onlyName+"Single_"+G4String(cT)+".txt";
    indGeom++;
  }
  else
  {
    fullOutFileDataSingle=onlyName+val+".txt";
  }
}