Salome HOME
add the template command
[tools/sat.git] / data / templates / PythonComponent / src / Model / Polyline.py
1 from Model import *
2 from PyQt4.QtGui import *
3 from PyQt4.QtCore import *
4
5 class Polyline( Model ):
6
7    def __init__( self, name, points, controller ):
8        """Constructor"""
9
10        Model.__init__( self, controller )
11        self._name = name
12        self._points = points
13        self.addTreeWidgetItem( self.getName(), controller )
14        self.addGraphicScene( controller )
15        pass
16
17    def getPoints( self ):
18        return self._points
19
20    def setPoints( self, points ):
21        self._points = points
22        pass
23
24    def editPoint( self, pointRange, newPoint ) :
25        self._points[ pointRange ] = newPoint
26        pass
27
28    def addTreeWidgetItem( self, name, controller ):
29        from PolyTreeWidgetItem import PolyTreeWidgetItem
30        from TreeWidgetItem import TreeWidgetItem
31
32        myTreeWidgetItem = PolyTreeWidgetItem( name, controller, ["Show", "Rename", "Delete"] )
33        # Adding list of points
34        for point in self.getPoints() :
35           x = point[0]
36           y = point[1]
37           newTreeWidgetItem = TreeWidgetItem( str(x) + ":" + str(y), controller, ["Edit"] )
38           myTreeWidgetItem.addChild( newTreeWidgetItem )
39           pass
40        myTreeWidgetItem.setModel( self )
41        self.getViews().append( myTreeWidgetItem )
42        return myTreeWidgetItem
43
44    def addGraphicScene( self, controller ) :
45        from PolyGraphicsScene import PolyGraphicsScene
46
47        myGraphicsScene = PolyGraphicsScene( controller )
48        myGraphicsScene.setModel( self )
49        self.getViews().append( myGraphicsScene )
50        return myGraphicsScene
51
52    def save( self ):
53        pass
54
55 pass