pyResultConversions.h

//
// Copyright 2016 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
#ifndef PXR_BASE_TF_PY_RESULT_CONVERSIONS_H
#define PXR_BASE_TF_PY_RESULT_CONVERSIONS_H
 
#include "pxr/pxr.h"
 
#include "pxr/base/tf/pyUtils.h"
 
#include "pxr/external/boost/python/tuple.hpp"
#include "pxr/external/boost/python/list.hpp"
#include "pxr/external/boost/python/dict.hpp"
 
#include <type_traits>
 
PXR_NAMESPACE_OPEN_SCOPE
 
template <typename T> struct Tf_PySequenceToListConverter;
template <typename T> struct Tf_PySequenceToSetConverter;
template <typename T> struct Tf_PyMapToDictionaryConverter;
template <typename T> struct Tf_PySequenceToTupleConverter;
template <typename First, typename Second> struct Tf_PyPairToTupleConverter;
 
struct TfPySequenceToList {
    template <typename T>
    struct apply {
        typedef Tf_PySequenceToListConverter<T> type;
    };
};
 
struct TfPySequenceToSet {
    template <typename T>
    struct apply {
        typedef Tf_PySequenceToSetConverter<T> type;
    };
};
 
struct TfPyMapToDictionary {
    template <typename T>
    struct apply {
        typedef Tf_PyMapToDictionaryConverter<T> type;
    };
};
 
struct TfPySequenceToTuple {
    template <typename T>
    struct apply {
        typedef Tf_PySequenceToTupleConverter<T> type;
    };
};
 
struct TfPyPairToTuple {
    template <typename T>
    struct apply {
        typedef Tf_PyPairToTupleConverter<typename T::first_type,
                                          typename T::second_type> type;
    };
};
 
template <typename T>
struct Tf_PySequenceToListConverter {
    typedef std::remove_reference_t<T> SeqType;
    bool convertible() const {
        return true;
    }
    PyObject *operator()(T seq) const {
        return pxr_boost::python::incref(TfPyCopySequenceToList(seq).ptr());
    }
    PyTypeObject *get_pytype() {
        return &PyList_Type;
    }
};
 
template <typename T>
struct Tf_PySequenceToSetConverter {
    typedef std::remove_reference_t<T> SeqType;
    bool convertible() const {
        return true;
    }
    PyObject *operator()(T seq) const {
        return pxr_boost::python::incref(TfPyCopySequenceToSet(seq).ptr());
    }
    PyTypeObject *get_pytype() {
        return &PySet_Type;
    }
};
 
template <typename T>
struct Tf_PyMapToDictionaryConverter {
    typedef std::remove_reference_t<T> SeqType;
    // TODO: convertible() should be made more robust by checking that the
    // value_type of the container is pair<const key_type, data_type> 
    bool convertible() const {
        return true;
    }
    PyObject *operator()(T seq) const {
        return pxr_boost::python::incref(TfPyCopyMapToDictionary(seq).ptr());
    }
    PyTypeObject *get_pytype() {
        return &PyDict_Type;
    }
};
 
template <typename T>
struct Tf_PySequenceToTupleConverter {
    typedef std::remove_reference_t<T> SeqType;
    bool convertible() const {
        return true;
    }
    PyObject *operator()(T seq) const {
        return pxr_boost::python::incref(TfPyCopySequenceToTuple(seq).ptr());
    }
    PyTypeObject *get_pytype() {
        return &PyTuple_Type;
    }
};
 
template <typename First, typename Second>
struct Tf_PyPairToTupleConverter {
    typedef std::pair<First, Second> PairType;
    bool convertible() const {
        return true;
    }
    PyObject *operator()(PairType const& a) const {
        pxr_boost::python::tuple result =
            pxr_boost::python::make_tuple(a.first, a.second);
        return pxr_boost::python::incref(result.ptr());
    }
    PyTypeObject *get_pytype() {
        return &PyTuple_Type;
    }
};
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif // TF_RESULT_CONVERSIONS_H