GGSScoringManager.cpp

Go to the documentation of this file.

/*
 * GGSScoringManager.cpp
 *
 *  Created on: 02 Aug 2013
 *      Author: Nicola Mori
 */

#include "montecarlo/scoring/manager/GGSScoringManager.h"
#include "montecarlo/scoring/manager/GGSMultiSensitiveDetector.h"
#include "montecarlo/scoring/manager/GGSScoringManagerMessenger.h"
#include "utils/GGSSmartLog.h"

#include "G4LogicalVolumeStore.hh"
#include "G4SDManager.hh"

#include <iostream>

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

GGSScoringManager &GGSScoringManager::GetInstance() {
  static GGSScoringManager instance;
  return instance;
}

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

GGSScoringManager::GGSScoringManager() : _messenger(new GGSScoringManagerMessenger) {}

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

GGSScoringManager::~GGSScoringManager() { delete _messenger; }

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

void GGSScoringManager::RegisterSDClass(const G4String &className, G4VSensitiveDetector *(*sdBuilder)(G4String)) {
  static const std::string routineName("GGSScoringManager::RegisterSDClass");
  std::pair<SDBuildersMap::iterator, bool> insertResult =
      _buildersMap.insert(std::pair<G4String, G4VSensitiveDetector *(*)(G4String)>(className, sdBuilder));
  if (!insertResult.second) {
    COUT(WARNING) << "Sensitive detector class " << className << " is already present." << ENDL;
  }
}

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

void GGSScoringManager::AddSDToLogVol(const G4String &sdClassName, const G4String &logVolNameAndParams) {
  static const std::string routineName("GGSScoringManager::AddSDToLogVol");

  G4VSensitiveDetector *sd = nullptr;

  // Attach the sensitive detector to all the logical volumes
  G4String logVolName = logVolNameAndParams.substr(0, logVolNameAndParams.find_first_of(' '));
  auto separatorPos = logVolNameAndParams.find_first_of(' ');
  G4String params;
  if (separatorPos != std::string::npos) {
    params = logVolNameAndParams.substr(logVolNameAndParams.find_first_of(' ') + 1);
  }
  for (G4LogicalVolumeStore::const_iterator logVol = G4LogicalVolumeStore::GetInstance()->begin();
       logVol != G4LogicalVolumeStore::GetInstance()->end(); logVol++) {
    if ((*logVol)->GetName() == logVolName) {
      // Retrieve the sd builder
      if (!sd) {
        SDBuildersMap::iterator builder = _buildersMap.find(sdClassName);
        if (builder == _buildersMap.end()) {
          throw std::runtime_error(routineName + ": builder for sensitive detector class " + sdClassName +
                                   " is not present.");
        }
        // Build the sd
        G4String sdName = sdClassName;
        if (params != "") {
          sdName.append('.').append(params);
        }
        sd = builder->second(logVolName + "." + sdName);
      }
      // Add sd to logvol
      AddSDToLogVol(sd, *logVol);
    }
  }

  if (!sd) {
    throw std::runtime_error(routineName + ": logical volume " + logVolName + " not present. Sensitive detector " +
                             sdClassName + " not added.");
  }
}

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

void GGSScoringManager::AddSDToLogVol(G4VSensitiveDetector *sd, G4LogicalVolume *logVol) {

  // Add sd to logical volume
  if (logVol->GetSensitiveDetector() == NULL) {
    // Case 1: logical has not a sd. Add the newly built sd to it
    logVol->SetSensitiveDetector(sd);
  } else {
    GGSMultiSensitiveDetector *multiSD = dynamic_cast<GGSMultiSensitiveDetector *>(logVol->GetSensitiveDetector());
    if (multiSD) {
      // Case 2: the current sd is a multisd. Add the newly built sd to it
      multiSD->AddSensitiveDetector(sd);
    } else {
      // Case 3: the current sd is a standard sd. Replace it with a multisd and add it and the newly built sd to the
      // multisd
      multiSD = new GGSMultiSensitiveDetector(G4String(logVol->GetName()).append(".GGSMultiSD"));
      multiSD->AddSensitiveDetector(logVol->GetSensitiveDetector());
      multiSD->AddSensitiveDetector(sd);
      G4SDManager::GetSDMpointer()->AddNewDetector(multiSD);
      logVol->SetSensitiveDetector(multiSD);
    }
  }

  // Add the sd to the sdmanager if it has not already been added
  GGSSmartLog::MuteOutput();
  if (G4SDManager::GetSDMpointer()->GetCollectionID(sd->GetName()) == -1) {
    G4SDManager::GetSDMpointer()->AddNewDetector(sd);
  }
  GGSSmartLog::UnmuteOutput();
}

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