pyobject_util.h

Go to the documentation of this file.

/** @file
 *  @author Bram de Greve (bram@cocamware.com)
 *  @author Tom De Muer (tom@cocamware.com)
 *
 *  *** BEGIN LICENSE INFORMATION ***
 *  
 *  The contents of this file are subject to the Common Public Attribution License 
 *  Version 1.0 (the "License"); you may not use this file except in compliance with 
 *  the License. You may obtain a copy of the License at 
 *  http://lass.sourceforge.net/cpal-license. The License is based on the 
 *  Mozilla Public License Version 1.1 but Sections 14 and 15 have been added to cover 
 *  use of software over a computer network and provide for limited attribution for 
 *  the Original Developer. In addition, Exhibit A has been modified to be consistent 
 *  with Exhibit B.
 *  
 *  Software distributed under the License is distributed on an "AS IS" basis, WITHOUT 
 *  WARRANTY OF ANY KIND, either express or implied. See the License for the specific 
 *  language governing rights and limitations under the License.
 *  
 *  The Original Code is LASS - Library of Assembled Shared Sources.
 *  
 *  The Initial Developer of the Original Code is Bram de Greve and Tom De Muer.
 *  The Original Developer is the Initial Developer.
 *  
 *  All portions of the code written by the Initial Developer are:
 *  Copyright (C) 2004-2011 the Initial Developer.
 *  All Rights Reserved.
 *  
 *  Contributor(s):
 *
 *  Alternatively, the contents of this file may be used under the terms of the 
 *  GNU General Public License Version 2 or later (the GPL), in which case the 
 *  provisions of GPL are applicable instead of those above.  If you wish to allow use
 *  of your version of this file only under the terms of the GPL and not to allow 
 *  others to use your version of this file under the CPAL, indicate your decision by 
 *  deleting the provisions above and replace them with the notice and other 
 *  provisions required by the GPL License. If you do not delete the provisions above,
 *  a recipient may use your version of this file under either the CPAL or the GPL.
 *  
 *  *** END LICENSE INFORMATION ***
 */
 
 
 
#ifndef LASS_GUARDIAN_OF_INCLUSION_UTIL_PYOBJECT_UTIL_H
#define LASS_GUARDIAN_OF_INCLUSION_UTIL_PYOBJECT_UTIL_H
 
#include "python_common.h"
#include "gil.h"
 
namespace lass
{
    namespace python
    {
        namespace impl
        {
            
            inline bool isOfType(PyObject* iObject, PyTypeObject* iTypeObject)
            {
                return (iObject->ob_type == iTypeObject);
            }
 
            template<typename ForwardIterator>
            TPyObjPtr pyBuildTuple(ForwardIterator iB, ForwardIterator iE )
            {
                LockGIL LASS_UNUSED(lock);
                const Py_ssize_t size = static_cast<Py_ssize_t>(std::distance(iB, iE));
                LASS_ASSERT(size >= 0);
                TPyObjPtr r(PyTuple_New(size));
                if (r)
                {
                    for (Py_ssize_t i = 0; iB != iE; ++iB, ++i)
                    {
                        PyObject* item = pyBuildSimpleObject(*iB);
                        if (!item)
                        {
                            return TPyObjPtr();
                        }
                        PyTuple_SET_ITEM(r.get(), i, item);
                    }
                }
                return r;
            }
 
            template<typename ForwardIterator>
            TPyObjPtr pyBuildList(ForwardIterator iB, ForwardIterator iE )
            {
                LockGIL LASS_UNUSED(lock);
                const Py_ssize_t size = static_cast<Py_ssize_t>(std::distance(iB, iE));
                LASS_ASSERT(size >= 0);
                TPyObjPtr r(PyList_New(size));
                if (r)
                {
                    for (Py_ssize_t i = 0; iB != iE; ++iB, ++i)
                    {
                        PyObject* item = pyBuildSimpleObject(*iB);
                        if (!item)
                        {
                            return TPyObjPtr();
                        }
                        PyList_SET_ITEM(r.get(), i, item);
                    }
                }
                return r;
            }
 
            template<typename InputIterator>
            TPyObjPtr pyBuildMap(InputIterator iB, InputIterator iE )
            {
                LockGIL LASS_UNUSED(lock);
                TPyObjPtr r(PyDict_New());
                if (r)
                {
                    for (;iB!=iE;++iB)
                    {
                        TPyObjPtr key(pyBuildSimpleObject(iB->first));
                        if (!key)
                        {
                            return TPyObjPtr();
                        }
                        TPyObjPtr value(pyBuildSimpleObject(iB->second));
                        if (!value)
                        {
                            return TPyObjPtr();
                        }
                        if (PyDict_SetItem(r.get(), key.get(), value.get()) != 0)
                        {
                            return TPyObjPtr();
                        }
                    }
                }
                return r;
            }
        }
    }
}
 
#endif