SurfaceCollector.hpp

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

// This file is part of the Acts project.
//
// Copyright (C) 2016-2018 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 <sstream>
#include "Acts/Surfaces/Surface.hpp"

namespace Acts {

struct SurfaceSelector {
  bool selectSensitive = true;
  bool selectMaterial = false;
  bool selectPassive = false;

  SurfaceSelector(bool sSensitive = true, bool sMaterial = false,
                  bool sPassive = false)
      : selectSensitive(sSensitive),
        selectMaterial(sMaterial),
        selectPassive(sPassive) {}

  bool operator()(const Acts::Surface& surface) const {
    if (selectSensitive && surface.associatedDetectorElement() != nullptr) {
      return true;
    }
    if (selectMaterial && surface.surfaceMaterial() != nullptr) {
      return true;
    }
    if (selectPassive) {
      return true;
    }
    return false;
  }
};

struct SurfaceHit {
  const Surface* surface = nullptr;
  Vector3D position;
  Vector3D direction;
};

template <typename Selector = SurfaceSelector>
struct SurfaceCollector {
  Selector selector;

  struct this_result {
    std::vector<SurfaceHit> collected;
  };

  using result_type = this_result;

  template <typename propagator_state_t, typename stepper_t>
  void operator()(propagator_state_t& state, const stepper_t& stepper,
                  result_type& result) const {
    // The current surface has been assigned by the navigator
    if (state.navigation.currentSurface &&
        selector(*state.navigation.currentSurface)) {
      // Create for recording
      SurfaceHit surface_hit;
      surface_hit.surface = state.navigation.currentSurface;
      surface_hit.position = stepper.position(state.stepping);
      surface_hit.direction = stepper.direction(state.stepping);
      // Save if in the result
      result.collected.push_back(surface_hit);
      // Screen output
      debugLog(state, [&] {
        std::stringstream dstream;
        dstream << "Collect surface  "
                << state.navigation.currentSurface->geoID();
        return dstream.str();
      });
    }
  }

  template <typename propagator_state_t, typename stepper_t>
  void operator()(propagator_state_t& /*state*/,
                  const stepper_t& /*unused*/) const {}

 private:
  template <typename propagator_state_t>
  void debugLog(propagator_state_t& state,
                const std::function<std::string()>& logAction) const {
    if (state.options.debug) {
      std::stringstream dstream;
      dstream << "   " << std::setw(state.options.debugPfxWidth);
      dstream << "surface collector"
              << " | ";
      dstream << std::setw(state.options.debugMsgWidth) << logAction() << '\n';
      state.options.debugString += dstream.str();
    }
  }
};

}  // namespace Acts