added support for 'set'

Author: FadyEssam

include/boost/python/set.hpp

@@ -0,0 +1,95 @@
+#ifndef SET_BOOST_PYTHON_HH
+#define SET_BOOST_PYTHON_HH
+
+#include <boost/python.hpp>
+
+namespace boost {
+	namespace python {
+
+		namespace detail
+		{
+			struct BOOST_PYTHON_DECL set_base : object
+			{
+				void add(object_cref); // add object to set
+
+				object pop(); // remove and return item at top
+
+				void discard(object_cref x); // discard value from set
+
+				long __len__(); // length of set
+
+				void clear(); // empties set
+
+			protected:
+				set_base(); 
+				explicit set_base(object_cref sequence); // new set initialized from sequence's items
+
+				BOOST_PYTHON_FORWARD_OBJECT_CONSTRUCTORS(set_base, object)
+			private:
+				static detail::new_non_null_reference call(object const&);
+			};
+
+
+		}
+
+
+
+
+
+
+		class set : public detail::set_base
+		{
+			typedef detail::set_base base;
+		public:
+			set() {} 
+
+			template <class T>
+			explicit set(T const& sequence)
+				: base(object(sequence))
+			{
+			}
+
+			template <class T>
+			void add(T const& x)
+			{
+				base::add(object(x));
+			}
+
+			object pop() { return base::pop(); }
+
+			template <class T>
+			void discard(T const& value)
+			{
+				base::discard(object(value));
+			}
+
+			void clear() {base::clear(); }
+
+			long __len__() { return base::__len__(); }
+
+
+
+		public: 
+			BOOST_PYTHON_FORWARD_OBJECT_CONSTRUCTORS(set, base)
+		};
+
+		
+		namespace converter
+		{
+			template <>
+			struct object_manager_traits<set>
+				: pytype_object_manager_traits<&PySet_Type, set>
+			{
+			};
+		}
+
+
+	}
+
+}
+
+
+
+
+
+#endif // !SET_BOOST_PYTHON_HH

src/set.cpp

@@ -0,0 +1,81 @@
+#include "set.hpp"
+#include <boost/python/ssize_t.hpp>
+#include <boost/python.hpp>
+
+namespace boost {
+	namespace python {
+		namespace detail {
+
+
+			detail::new_non_null_reference set_base::call(object_cref arg_)
+			{
+				return (detail::new_non_null_reference)
+					(expect_non_null)(
+						PySet_New(arg_.ptr())
+						);
+			}
+
+			set_base::set_base()
+				: object(detail::new_reference(PySet_New(NULL)))
+			{}
+
+			set_base::set_base(object_cref sequence)
+				: object(set_base::call(sequence))
+			{}
+
+			void set_base::add(object_cref x)
+			{
+				if (PyAnySet_CheckExact(this->ptr()))
+				{
+					if (PySet_Add(this->ptr(), x.ptr()) == -1)
+						throw_error_already_set();
+				}
+				else
+				{
+					this->attr("add")(x);
+				}
+			}
+
+
+			void set_base::discard(object_cref x)
+			{
+				if (PyAnySet_CheckExact(this->ptr()))
+				{
+					if (PySet_Discard(this->ptr(), x.ptr()) == -1)
+						throw_error_already_set();
+				}
+				else
+				{
+					this->attr("discrad")(x);
+				}
+			}
+
+			object set_base::pop()
+			{
+				return this->attr("pop")();
+			}
+
+			void set_base::clear()
+			{
+				this->attr("clear")();
+			}
+
+			long set_base::__len__()
+			{
+				return extract<long>(object(PySet_Size(this->ptr())))();
+			}
+
+
+			static struct register_set_pytype_ptr
+			{
+				register_set_pytype_ptr()
+				{
+					const_cast<converter::registration &>(
+						converter::registry::lookup(boost::python::type_id<boost::python::set>())
+						).m_class_object = &PySet_Type;
+				}
+			}register_set_pytype_ptr_;
+
+		}
+	}
+} // namespace boost::python