pymhrw.h

/* This file is part of the Tomographer project, which is distributed under the
 * terms of the MIT license.
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2016 ETH Zurich, Institute for Theoretical Physics, Philippe Faist
 * Copyright (c) 2017 Caltech, Institute for Quantum Information and Matter, Philippe Faist
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#ifndef TOMOGRAPHERPY_MHRW_H
#define TOMOGRAPHERPY_MHRW_H

#include <string>

#include <tomographerpy/common.h>

#include <tomographer/mhrw.h>
//#include <tomographer/mhrwtasks.h>


namespace tpy {

typedef Tomographer::MHRWParams<py::object, IterCountIntType>  MHRWParams;



template<typename MHWalkerParams>
struct TOMOGRAPHER_EXPORT PyMHWalkerParamsToDict
{
  static inline py::dict makeDict(const MHWalkerParams & ) { return {}; }
  static inline MHWalkerParams fromPyObj(py::object ) { return {}; }
};
template<typename StepRealType>
struct TOMOGRAPHER_EXPORT PyMHWalkerParamsToDict<Tomographer::MHWalkerParamsStepSize<StepRealType> >
{
  static inline py::dict makeDict(const Tomographer::MHWalkerParamsStepSize<StepRealType> & p) {
    return py::dict(py::arg("step_size") = p.step_size);
  }
  static Tomographer::MHWalkerParamsStepSize<StepRealType> fromPyObj(py::object d)
  {
    if ( d.is_none() ) {
      // None: let the underlying mhwalker decide what to do with this
      return 0;
    }
    if ( py::hasattr(d, "__getitem__") ) {
      // dict or dict-like, go. If key doesn't exist, send in zero and let the underlying
      // mhwalker handle it
      return d.attr("get")("step_size", 0).cast<StepRealType>();
    }
    // try to iterpret the object itself as a float
    return d.cast<StepRealType>();
  }
};

template<typename MHWalkerParams>
inline py::dict pyMHWalkerParamsToDictInvoke(const MHWalkerParams & p)
{
  return PyMHWalkerParamsToDict<MHWalkerParams>::makeDict(p);
}
template<typename MHWalkerParams>
inline MHWalkerParams pyMHWalkerParamsFromPyObj(py::object o)
{
  return PyMHWalkerParamsToDict<MHWalkerParams>::fromPyObj(o);
}




} // namespace Py



#endif