155 lines
5.4 KiB
C++
155 lines
5.4 KiB
C++
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c) 2007-2011, Industrial Light & Magic, a division of Lucas
|
|
// Digital Ltd. LLC
|
|
//
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following disclaimer
|
|
// in the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Industrial Light & Magic nor the names of
|
|
// its contributors may be used to endorse or promote products derived
|
|
// from this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////
|
|
|
|
#ifndef _PyImathShear_h_
|
|
#define _PyImathShear_h_
|
|
|
|
#include <Python.h>
|
|
#include <boost/python.hpp>
|
|
#include <ImathShear.h>
|
|
|
|
|
|
namespace PyImath {
|
|
|
|
template <class T> boost::python::class_<IMATH_NAMESPACE::Shear6<T> > register_Shear();
|
|
|
|
//
|
|
|
|
// Other code in the Zeno code base assumes the existance of a class with the
|
|
// same name as the Imath class, and with static functions wrap() and
|
|
// convert() to produce a PyImath object from an Imath object and vice-versa,
|
|
// respectively. The class Boost generates from the Imath class does not
|
|
// have these properties, so we define a companion class here.
|
|
// The template argument, T, is the element type (e.g.,float, double).
|
|
|
|
template <class T>
|
|
class S6 {
|
|
public:
|
|
static PyObject * wrap (const IMATH_NAMESPACE::Shear6<T> &s);
|
|
static int convert (PyObject *p, IMATH_NAMESPACE::Shear6<T> *s);
|
|
};
|
|
|
|
template <class T>
|
|
PyObject *
|
|
S6<T>::wrap (const IMATH_NAMESPACE::Shear6<T> &s)
|
|
{
|
|
typename boost::python::return_by_value::apply < IMATH_NAMESPACE::Shear6<T> >::type converter;
|
|
PyObject *p = converter (s);
|
|
return p;
|
|
}
|
|
|
|
template <class T>
|
|
int
|
|
S6<T>::convert (PyObject *p, IMATH_NAMESPACE::Shear6<T> *s)
|
|
{
|
|
boost::python::extract <IMATH_NAMESPACE::Shear6f> extractorShear6f (p);
|
|
if (extractorShear6f.check())
|
|
{
|
|
IMATH_NAMESPACE::Shear6f s6f = extractorShear6f();
|
|
float xy, xz, yz, yx, zx, zy;
|
|
s6f.getValue (xy, xz, yz, yx, zx, zy);
|
|
s->setValue(T(xy), T(xz), T(yz), T(yx), T(zx), T(zy));
|
|
return 1;
|
|
}
|
|
|
|
boost::python::extract <IMATH_NAMESPACE::Shear6d> extractorShear6d (p);
|
|
if (extractorShear6d.check())
|
|
{
|
|
IMATH_NAMESPACE::Shear6d s6d = extractorShear6d();
|
|
double xy, xz, yz, yx, zx, zy;
|
|
s6d.getValue (xy, xz, yz, yx, zx, zy);
|
|
s->setValue(T(xy), T(xz), T(yz), T(yx), T(zx), T(zy));
|
|
return 1;
|
|
}
|
|
|
|
boost::python::extract <boost::python::tuple> extractorTuple (p);
|
|
if (extractorTuple.check())
|
|
{
|
|
boost::python::tuple t = extractorTuple();
|
|
if (t.attr ("__len__") () == 3)
|
|
{
|
|
double xy = boost::python::extract <double> (t[0]);
|
|
double xz = boost::python::extract <double> (t[1]);
|
|
double yz = boost::python::extract <double> (t[2]);
|
|
s->setValue (T(xy), T(xz), T(yz), T(0), T(0), T(0));
|
|
return 1;
|
|
}
|
|
|
|
else if (t.attr ("__len__") () == 6)
|
|
{
|
|
double xy = boost::python::extract <double> (t[0]);
|
|
double xz = boost::python::extract <double> (t[1]);
|
|
double yz = boost::python::extract <double> (t[2]);
|
|
double yx = boost::python::extract <double> (t[3]);
|
|
double zx = boost::python::extract <double> (t[4]);
|
|
double zy = boost::python::extract <double> (t[5]);
|
|
s->setValue (T(xy), T(xz), T(yz), T(yx), T(zx), T(zy));
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
boost::python::extract <IMATH_NAMESPACE::V3i> extractorV3i (p);
|
|
if (extractorV3i.check())
|
|
{
|
|
IMATH_NAMESPACE::V3i v3i = extractorV3i();
|
|
s->setValue (T(v3i[0]), T(v3i[1]), T(v3i[2]), T(0), T(0), T(0));
|
|
return 1;
|
|
}
|
|
|
|
boost::python::extract <IMATH_NAMESPACE::V3f> extractorV3f (p);
|
|
if (extractorV3f.check())
|
|
{
|
|
IMATH_NAMESPACE::V3f v3f = extractorV3f();
|
|
s->setValue (T(v3f[0]), T(v3f[1]), T(v3f[2]), T(0), T(0), T(0));
|
|
return 1;
|
|
}
|
|
|
|
boost::python::extract <IMATH_NAMESPACE::V3d> extractorV3d (p);
|
|
if (extractorV3d.check())
|
|
{
|
|
IMATH_NAMESPACE::V3d v3d = extractorV3d();
|
|
s->setValue (T(v3d[0]), T(v3d[1]), T(v3d[2]), T(0), T(0), T(0));
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
typedef S6<float> Shear6f;
|
|
typedef S6<double> Shear6d;
|
|
|
|
}
|
|
|
|
#endif
|