Geant4PhysicalVolumeSelectors.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Geant4PhysicalVolumeSelectors.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2022 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Utilities/Delegate.hpp"

#include <string>
#include <vector>

#include "G4VPhysicalVolume.hh"

namespace Acts {

class IGeant4PhysicalVolumeSelector {
 public:
  virtual ~IGeant4PhysicalVolumeSelector() = default;
  virtual bool select(const G4VPhysicalVolume& g4Phys) const = 0;
};

namespace Geant4PhysicalVolumeSelectors {

struct AllSelector : public IGeant4PhysicalVolumeSelector {
  bool select(const G4VPhysicalVolume& /*g4Phys*/) const final { return true; }
};

struct NameSelector : public IGeant4PhysicalVolumeSelector {
  std::vector<std::string> names = {};
  bool exact = false;

  NameSelector(const std::vector<std::string>& ns, bool e = false)
      : names(ns), exact(e) {}

  bool select(const G4VPhysicalVolume& g4PhysVol) const final {
    std::string volumeName = g4PhysVol.GetName();
    bool matched = false;
    for (const auto& name : names) {
      matched = exact ? (volumeName == name)
                      : volumeName.find(name) != std::string::npos;
      if (matched) {
        break;
      }
    }
    return matched;
  }
};

}  // namespace Geant4PhysicalVolumeSelectors
}  // namespace Acts