From 8d9bf59fb1a06d2e2aa9143bae80a5bbb688c051 Mon Sep 17 00:00:00 2001
From: Hugo Parente Lima <hugo.pl@gmail.com>
Date: Tue, 26 Apr 2011 17:58:45 -0300
Subject: [PATCH] Fix bugs 776 and 777 Bug 776 - "Operator
 "QPolygon::operator<<(QPoint)" missing" Bug 777 - "Operator
 "QPolygon::operator<<(QVector<QPoint>)" missing"

---
 PySide/QtGui/typesystem_gui_common.xml | 18 ++++++++++++++----
 tests/QtGui/qpolygonf_test.py          |  6 ++++++
 2 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/PySide/QtGui/typesystem_gui_common.xml b/PySide/QtGui/typesystem_gui_common.xml
index 50da933..8882367 100644
--- a/PySide/QtGui/typesystem_gui_common.xml
+++ b/PySide/QtGui/typesystem_gui_common.xml
@@ -651,10 +651,20 @@
     <modify-function signature="QPolygon(int, const int *)" remove="all"/>
     <!-- ### A QVector parameter, for no defined type, will generate wrong code. -->
     <modify-function signature="operator+=(QVector)" remove="all"/>
-    <!-- ### See bug 776 -->
-    <modify-function signature="operator&lt;&lt;(QPoint)" remove="all"/>
-    <!-- ### See bug 777 -->
-    <modify-function signature="operator&lt;&lt;(QVector&lt;QPoint&gt;)" remove="all"/>
+    <modify-function signature="operator&lt;&lt;(QPoint)">
+        <inject-code>
+            // %FUNCTION_NAME()
+            *%CPPSELF &lt;&lt; %1;
+            %PYARG_0 = %CONVERTTOPYTHON[QPolygon*](%CPPSELF);
+        </inject-code>
+    </modify-function>
+    <modify-function signature="operator&lt;&lt;(QVector&lt;QPoint&gt;)">
+        <inject-code>
+            // %FUNCTION_NAME()
+            *%CPPSELF &lt;&lt; %1;
+            %PYARG_0 = %CONVERTTOPYTHON[QPolygon*](%CPPSELF);
+        </inject-code>
+    </modify-function>
     <!-- ### -->
   </value-type>
   <value-type name="QPolygonF">
diff --git a/tests/QtGui/qpolygonf_test.py b/tests/QtGui/qpolygonf_test.py
index 83e643c..f842eb4 100644
--- a/tests/QtGui/qpolygonf_test.py
+++ b/tests/QtGui/qpolygonf_test.py
@@ -18,5 +18,11 @@ class QPolygonFNotIterableTest(unittest.TestCase):
             self.assertEqual(int(point.y()), i)
             i += 1;
 
+    def testPolygonShiftOperators(self):
+        p = QPolygon()
+        self.assertEqual(len(p), 0)
+        p << QPoint(10, 20) << QPoint(20, 30) << [QPoint(20, 30), QPoint(40, 50)]
+        self.assertEqual(len(p), 4)
+
 if __name__ == '__main__':
     unittest.main()