#include "pxr/pxr.h"
#include "pxr/base/tf/refPtr.h"
#include "pxr/base/tf/weakPtr.h"
#include "pxr/base/tf/diagnosticLite.h"
#include "pxr/base/tf/preprocessorUtilsLite.h"
#include "pxr/base/tf/pySafePython.h"
#include "pxr/base/tf/pyInterpreter.h"
#include "pxr/base/tf/pyLock.h"
#include "pxr/base/tf/api.h"
#include <functional>
#include <typeinfo>
#include <string>
#include <vector>
#include <hboost/python/dict.hpp>
#include <hboost/python/extract.hpp>
#include <hboost/python/handle.hpp>
#include <hboost/python/object.hpp>
#include <hboost/python/type_id.hpp>

Include dependency graph for pyUtils.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	TF_PY_REPR_PREFIX

Functions
TF_API bool	TfPyIsInitialized ()
	Returns true if python is initialized. More...

TF_API void	TfPyThrowIndexError (const char *msg)

void	TfPyThrowIndexError (std::string const &msg)

TF_API void	TfPyThrowRuntimeError (const char *msg)

void	TfPyThrowRuntimeError (std::string const &msg)

TF_API void	TfPyThrowStopIteration (const char *msg)

void	TfPyThrowStopIteration (std::string const &msg)

TF_API void	TfPyThrowKeyError (const char *msg)

void	TfPyThrowKeyError (std::string const &msg)

TF_API void	TfPyThrowValueError (const char *msg)

void	TfPyThrowValueError (std::string const &msg)

TF_API void	TfPyThrowTypeError (const char *msg)

void	TfPyThrowTypeError (std::string const &msg)

TF_API bool	TfPyIsNone (hboost::python::object const &obj)
	Return true iff obj is None. More...

TF_API bool	TfPyIsNone (hboost::python::handle<> const &obj)
	Return true iff obj is None. More...

TF_API void	Tf_PyObjectError (bool printError)

template<typename T >
hboost::python::object	TfPyObject (T const &t, bool complainOnFailure=true)

hboost::python::object	TfPyObject (PyObject *t, bool complainOnFailure=true)

TF_API std::string	TfPyObjectRepr (hboost::python::object const &t)

template<typename T >
std::string	TfPyRepr (T const &t)

template<typename T >
std::string	TfPyRepr (const std::vector< T > &v)
	Return repr(t) for a vector as a python list. More...

TF_API hboost::python::object	TfPyEvaluate (std::string const &expr, hboost::python::dict const &extraGlobals=hboost::python::dict())

TF_API int64_t	TfPyNormalizeIndex (int64_t index, uint64_t size, bool throwError=false)

TF_API std::string	TfPyGetClassName (hboost::python::object const &obj)
	Return the name of the class of obj. More...

TF_API hboost::python::object	TfPyGetClassObject (std::type_info const &type)

template<typename T >
hboost::python::object	TfPyGetClassObject ()

TF_API void	Tf_PyWrapOnceImpl (hboost::python::type_info const &, std::function< void()> const &, bool *)

template<typename T >
void	TfPyWrapOnce (std::function< void()> const &wrapFunc)

TF_API void	Tf_PyLoadScriptModule (std::string const &name)

template<class Map >
hboost::python::dict	TfPyCopyMapToDictionary (Map const &map)
	Creates a python dictionary from a std::map. More...

template<class Seq >
hboost::python::list	TfPyCopySequenceToList (Seq const &seq)

template<class Seq >
hboost::python::object	TfPyCopySequenceToSet (Seq const &seq)

template<class Seq >
hboost::python::tuple	TfPyCopySequenceToTuple (Seq const &seq)

TF_API hboost::python::object	TfPyCopyBufferToByteArray (const char *buffer, size_t size)

TF_API std::vector< std::string >	TfPyGetTraceback ()

TF_API void	TfPyGetStackFrames (std::vector< uintptr_t > *frames)

TF_API void	TfPyDumpTraceback ()
	Print the current python traceback to stdout. More...

TF_API bool	TfPySetenv (const std::string &name, const std::string &value)

TF_API bool	TfPyUnsetenv (const std::string &name)

TF_API bool	Tf_PyEvaluateWithErrorCheck (const std::string &expr, hboost::python::object *obj)

template<typename T >
bool	TfPyEvaluateAndExtract (const std::string &expr, T *t)

TF_API void	TfPyPrintError ()

Detailed Description

Miscellaneous Utilities for dealing with script.

Definition in file pyUtils.h.

Macro Definition Documentation

#define TF_PY_REPR_PREFIX

A macro which expands to the proper repr prefix for a library. This is the "canonical" name of the module that the system uses to identify it followed by a '.'. This can be used in the implementation of repr

Definition at line 59 of file pyUtils.h.

Function Documentation

TF_API bool Tf_PyEvaluateWithErrorCheck	(	const std::string &	expr,
		hboost::python::object *	obj
	)

TF_API void Tf_PyLoadScriptModule ( std::string const & name )

Load the python module moduleName. This is used by some low-level infrastructure code to load python wrapper modules corresponding to C++ shared libraries when they are needed. It should generally not need to be called from normal user code.

TF_API void Tf_PyObjectError ( bool printError )

TF_API void Tf_PyWrapOnceImpl	(	hboost::python::type_info const &	,
		std::function< void()> const &	,
		bool *
	)

TF_API hboost::python::object TfPyCopyBufferToByteArray	(	const char *	buffer,
		size_t	size
	)

Create a python bytearray from an input buffer and size.

If a size of zero is passed in this function will return a valid python bytearray of size zero.

An invalid object handle is returned on failure.

template<class Map >

hboost::python::dict TfPyCopyMapToDictionary ( Map const & map )

Creates a python dictionary from a std::map.

Definition at line 275 of file pyUtils.h.

template<class Seq >

hboost::python::list TfPyCopySequenceToList ( Seq const & seq )

Definition at line 284 of file pyUtils.h.

template<class Seq >

hboost::python::object TfPyCopySequenceToSet ( Seq const & seq )

Create a python set from an iterable sequence.

If Seq::value_type is not hashable, TypeError is raised via throwing hboost::python::error_already_set.

Definition at line 298 of file pyUtils.h.

template<class Seq >

hboost::python::tuple TfPyCopySequenceToTuple ( Seq const & seq )

Definition at line 314 of file pyUtils.h.

TF_API void TfPyDumpTraceback ( )

Print the current python traceback to stdout.

TF_API hboost::python::object TfPyEvaluate	(	std::string const &	expr,
		hboost::python::dict const &	extraGlobals = `hboost::python::dict()`
	)

Evaluate python expression expr with all the known script modules imported under their standard names. Additional globals may be provided in the extraGlobals dictionary.

template<typename T >

bool TfPyEvaluateAndExtract	(	const std::string &	expr,
		T *	t
	)

Safely evaluates expr and extracts the return object of type T. If successful, returns true and sets *t to the return value, otherwise returns false.

Definition at line 403 of file pyUtils.h.

TF_API std::string TfPyGetClassName ( hboost::python::object const & obj )

Return the name of the class of obj.

TF_API hboost::python::object TfPyGetClassObject ( std::type_info const & type )

Return the python class object for type if type has been wrapped. Otherwise return None.

template<typename T >

hboost::python::object TfPyGetClassObject ( )

Return the python class object for T if T has been wrapped. Otherwise return None.

Definition at line 232 of file pyUtils.h.

TF_API void TfPyGetStackFrames ( std::vector< uintptr_t > * frames )

Populates the vector passed in with pointers to strings containing the python interpreter stack frames. Note that TfPyGetStackFrames allocates these strings on the heap and its the caller's responsibility to free them.

TF_API std::vector<std::string> TfPyGetTraceback ( )

Return a vector of strings containing the current python traceback.

The vector contains the same strings that python's traceback.format_stack() returns.

TF_API bool TfPyIsInitialized ( )

Returns true if python is initialized.

TF_API bool TfPyIsNone ( hboost::python::object const & obj )

Return true iff obj is None.

TF_API bool TfPyIsNone ( hboost::python::handle<> const & obj )

Return true iff obj is None.

TF_API int64_t TfPyNormalizeIndex	(	int64_t	index,
		uint64_t	size,
		bool	throwError = `false`
	)

Return a positive index in the range [0,size). If throwError is true, this will throw an index error if the resulting index is out of range.

template<typename T >

hboost::python::object TfPyObject	(	T const &	t,
		bool	complainOnFailure = `true`
	)

Return a python object for the given C++ object, loading the appropriate wrapper code if necessary. Spams users if complainOnFailure is true and conversion fails.

Definition at line 144 of file pyUtils.h.

hboost::python::object TfPyObject	(	PyObject *	t,
		bool	complainOnFailure = `true`
	)

inline

Definition at line 165 of file pyUtils.h.

TF_API std::string TfPyObjectRepr ( hboost::python::object const & t )

Return repr(t).

Calls PyObject_Repr on the given python object.

TF_API void TfPyPrintError ( )

Print a standard traceback to sys.stderr and clear the error indicator. If the error is a KeyboardInterrupt then this does nothing. Call this function only when the error indicator is set.

template<typename T >

std::string TfPyRepr ( T const & t )

Return repr(t).

Converts t to its equivalent python object and then calls PyObject_Repr on that.

Definition at line 180 of file pyUtils.h.

template<typename T >

std::string TfPyRepr ( const std::vector< T > & v )

Return repr(t) for a vector as a python list.

Definition at line 189 of file pyUtils.h.

TF_API bool TfPySetenv	(	const std::string &	name,
		const std::string &	value
	)

Set an environment variable in os.environ.

This function is equivalent to

def PySetenv(name, value):
    try:
        import os
        os.environ[name] = value
        return True
    except:
        return False

Calling this function without first initializing Python is an error and returns false.

Note that this function will import the os module, causing os.environ to be poputated. All modifications to the environment after os has been imported must be made with this function or TfSetenv if it important that they appear in os.environ.

TF_API void TfPyThrowIndexError ( const char * msg )

Raises a Python IndexError with the given error msg and throws a hboost::python::error_already_set exception. Callers must hold the GIL before calling this function.

void TfPyThrowIndexError ( std::string const & msg )

inline