// This file is part of the ACTS project.
//
// Copyright (C) 2016 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 https://mozilla.org/MPL/2.0/.

#include "ActsPlugins/Geant4/Geant4DetectorElement.hpp"
#include "ActsPlugins/Geant4/Geant4DetectorSurfaceFactory.hpp"
#include "ActsPlugins/Geant4/Geant4PhysicalVolumeSelectors.hpp"
#include "ActsPython/Utilities/Helpers.hpp"
#include "ActsPython/Utilities/Macros.hpp"

#include <memory>
#include <string>

#include <pybind11/pybind11.h>

namespace py = pybind11;
using namespace pybind11::literals;

PYBIND11_MODULE(ActsPluginsPythonBindingsGeant4, geant4) {
  using namespace Acts;
  using namespace ActsPlugins;

  // Detector Element
  py::class_<Geant4DetectorElement, std::shared_ptr<Geant4DetectorElement>,
             SurfacePlacementBase>(geant4, "Geant4DetectorElement")
      .def("surface", [](const Geant4DetectorElement& self) {
        return self.surface().getSharedPtr();
      });

  // Surface creation
  {
    using ISelector = IGeant4PhysicalVolumeSelector;
    auto is = py::class_<ISelector, std::shared_ptr<ISelector>>(
        geant4, "IVolumeSelector");

    using NameSelector = Geant4PhysicalVolumeSelectors::NameSelector;
    auto ns = py::class_<NameSelector, std::shared_ptr<NameSelector>>(
                  geant4, "VolumeNameSelector", is)
                  .def(py::init<const std::vector<std::string>&, bool>());

    using Factory = Geant4DetectorSurfaceFactory;
    auto o = py::class_<Factory::Options>(geant4, "SurfaceFactoryOptions")
                 .def(py::init<>());
    ACTS_PYTHON_STRUCT(o, scaleConversion, convertMaterial,
                       convertedMaterialThickness, sensitiveSurfaceSelector,
                       passiveSurfaceSelector);
  }
}