target_selector.hpp

#pragma once
 
#include <optional>
 
#include "vipra/vipra_behaviors/targets/target.hpp"
 
namespace VIPRA::Behaviors {

class TargetSelector {
  DEFAULT_CONSTRUCTIBLE(TargetSelector)
  COPYABLE(TargetSelector)
  MOVEABLE(TargetSelector)
 
 public:
  explicit TargetSelector(TargetFunc&& func) : _select(func) {}
 
  void get_targets(Simpack pack, const VIPRA::idxVec& peds, std::vector<Target>& targets)
  {
    if ( ! _select ) {
      std::for_each(peds.begin(), peds.end(), [&](auto idx) {
        targets[idx] = Target{TargetType::PEDESTRIAN, idx};
      });
      return;
    }
 
    std::for_each(peds.begin(), peds.end(), [&](auto idx) {
      targets[idx] = _select.value()(pack, Self{Target{TargetType::PEDESTRIAN, idx}});
    });
  }
 
 private:
  std::optional<TargetFunc> _select;
};
}  // namespace VIPRA::Behaviors