/* This file is part of Lemma, a geophysical modelling and inversion API. * More information is available at http://lemmasoftware.org */ /* 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/. */ /** * @file * @date 22/04/19 14:06:32 * @version $Id$ * @author Trevor Irons (ti) * @email Trevor.Irons@utah.edu * @copyright Copyright (c) 2019, University of Utah * @copyright Copyright (c) 2019, Lemma Software, LLC */ #include #include #include #include "FDEM1D" namespace py = pybind11; PYBIND11_MODULE(FDEM1D, m) { py::add_ostream_redirect(m, "ostream_redirect"); m.doc() = "Python binding of Lemma::FDEM1D, additional details can be found at https://lemmasoftware.org"; py::class_ > WireAntenna(m, "WireAntenna"); // lifecycle WireAntenna.def(py::init(&Lemma::WireAntenna::NewSP)) .def_static("DeSerialize", py::overload_cast(&Lemma::WireAntenna::DeSerialize), "Construct object from yaml representation") // print .def("Serialize", &Lemma::WireAntenna::Print, "YAML representation of the class") .def("__repr__", &Lemma::WireAntenna::Print) // modifiers .def("SetNumberOfPoints", &Lemma::WireAntenna::SetNumberOfPoints, "Sets the number of points comprising the antenna") .def("SetPoint", py::overload_cast(&Lemma::WireAntenna::SetPoint), "Sets a point in the antenna") .def("SetPoint", py::overload_cast(&Lemma::WireAntenna::SetPoint), "Sets a point in the antenna") .def("SetNumberOfTurns", &Lemma::WireAntenna::SetNumberOfTurns, "Sets the number of turns of the antenna") .def("SetNumberOfFrequencies", &Lemma::WireAntenna::SetNumberOfFrequencies, "Sets the number of frequencies of the transmitter") .def("SetFrequency", &Lemma::WireAntenna::SetFrequency, "Sets a single frequency of the transmitter") .def("SetCurrent", &Lemma::WireAntenna::SetCurrent, "Sets the current of the transmitter in amps") // accessors .def("GetCurrent", &Lemma::WireAntenna::GetCurrent, "Returns the current of the transmitter in amps") .def("GetPoints", &Lemma::WireAntenna::GetPoints, "Returns the points defining the transmitter") .def("GetNumberOfDipoles", &Lemma::WireAntenna::GetNumberOfDipoles, "Returns the number of dipoles defining the transmitter") .def("GetNumberOfFrequencies", &Lemma::WireAntenna::GetNumberOfFrequencies, "Returns the number of frequencies for the transmitter") .def("IsHorizontallyPlanar", &Lemma::WireAntenna::IsHorizontallyPlanar, "Returns true if the transmitter is flat") .def("GetName", &Lemma::WireAntenna::GetName, "Returns the class name of the object") // operations .def("ApproximateWithElectricDipoles", &Lemma::WireAntenna::ApproximateWithElectricDipoles, "Approximates loop with electric dipoles") ; py::class_ > PolygonalWireAntenna(m, "PolygonalWireAntenna", WireAntenna); // lifecycle PolygonalWireAntenna.def(py::init(&Lemma::PolygonalWireAntenna::NewSP)) .def_static("DeSerialize", py::overload_cast(&Lemma::PolygonalWireAntenna::DeSerialize), "Construct object from yaml representation") // print .def("__repr__", &Lemma::PolygonalWireAntenna::Print) .def("Serialize", &Lemma::PolygonalWireAntenna::Print, "YAML representation of the class") // accessors .def("GetName", &Lemma::PolygonalWireAntenna::GetName, "Returns the name of the class") // operations .def("ApproximateWithElectricDipoles", &Lemma::PolygonalWireAntenna::ApproximateWithElectricDipoles, "Approximates loop with series of electric dipoles around loop") ; py::class_ > DipoleSource(m, "DipoleSource"); // lifecycle DipoleSource.def(py::init(&Lemma::DipoleSource::NewSP)) .def_static("DeSerialize", py::overload_cast(&Lemma::DipoleSource::DeSerialize), "Construct object from yaml representation") // print .def("Serialize", &Lemma::DipoleSource::Print, "YAML representation of the class") .def("__repr__", &Lemma::DipoleSource::Print) // accessors .def("GetName", &Lemma::DipoleSource::GetName, "Returns the name of the class") .def("GetNumberOfFrequencies", &Lemma::DipoleSource::GetNumberOfFrequencies, "Returns the number of frequencies") .def("GetFrequencies", &Lemma::DipoleSource::GetFrequencies, "Returns an array of frequencies") .def("GetFrequency", &Lemma::DipoleSource::GetFrequency, "Returns the frequency of the argument index") .def("GetAngularFrequency", &Lemma::DipoleSource::GetAngularFrequency, "Returns the angular frequency of the argument index") .def("GetPhase", &Lemma::DipoleSource::GetPhase, "Returns the phase of the dipole") .def("GetMoment", &Lemma::DipoleSource::GetMoment, "Returns the moment of the dipole") .def("GetLocation", py::overload_cast< >(&Lemma::DipoleSource::GetLocation), "Returns the location of the dipole") .def("GetPolarisation", &Lemma::DipoleSource::GetPolarisation, "Returns the polarisation of the dipole") // modifiers .def("SetLocation", py::overload_cast (&Lemma::DipoleSource::SetLocation), "Sets the location of the dipole") .def("SetPolarisation", py::overload_cast (&Lemma::DipoleSource::SetPolarisation), "Sets the polarisation of the dipole") .def("SetType", &Lemma::DipoleSource::SetType, "Sets the type") .def("SetMoment", &Lemma::DipoleSource::SetMoment, "Sets the moment of the dipole") .def("SetPhase", &Lemma::DipoleSource::SetPhase, "Sets the phase of the dipole") .def("SetNumberOfFrequencies", &Lemma::DipoleSource::SetNumberOfFrequencies, "Sets the number of frequencies to calculate for the dipole") .def("SetFrequency", &Lemma::DipoleSource::SetFrequency, "Sets a single frequency, first argument is index, second argument is frequency") .def("SetFrequencies", &Lemma::DipoleSource::SetFrequencies, "Sets all frequencies, argument is numpy array of frequencies") ; } /* BOOST_PYTHON_MODULE(pyLemmaCore) { Py_Initialize(); np::initialize(); bp::class_ ("PolygonalWireAntenna", boost::python::no_init) // print .def(boost::python::self_ns::str(bp::self)) // Lifecycle .def("__init__", boost::python::make_constructor(&Lemma::PolygonalWireAntenna::NewSP)) //.def("Serialize", &Lemma::PolygonalWireAntenna::Serialize) //.def("DeSerialize", &Lemma::PolygonalWireAntenna::DeSerialize) // Accessors ; } */