/[escript]/trunk/pyvisi/py_src/velocity.py
ViewVC logotype

Diff of /trunk/pyvisi/py_src/velocity.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1148 by jongui, Wed May 16 22:45:33 2007 UTC revision 1197 by jongui, Mon Jun 18 02:08:34 2007 UTC
# Line 1  Line 1 
1  """  """
2  @author: John NGUI  @var __author__: name of author
3    @var __copyright__: copyrights
4    @var __license__: licence agreement
5    @var __url__: url entry point on documentation
6    @var __version__: version
7    @var __date__: date of the version
8  """  """
9    
10    __author__="John Ngui, john.ngui@uq.edu.au"
11    __copyright__="""  Copyright (c) 2006 by ACcESS MNRF
12                        http://www.access.edu.au
13                    Primary Business: Queensland, Australia"""
14    __license__="""Licensed under the Open Software License version 3.0
15                 http://www.opensource.org/licenses/osl-3.0.php"""
16    __url__="http://www.iservo.edu.au/esys"
17    __version__="$Revision$"
18    __date__="$Date$"
19    
20    
21  import vtk  import vtk
22  from mapper import DataSetMapper  from mapper import DataSetMapper
23  from lookuptable import LookupTable  from lookuptable import LookupTable
# Line 59  class Velocity(DataSetMapper, Actor3D, A Line 75  class Velocity(DataSetMapper, Actor3D, A
75          @param outline: Places an outline around the domain surface          @param outline: Places an outline around the domain surface
76          """          """
77    
         self.__scene = scene  
78          self.__data_collector = data_collector          self.__data_collector = data_collector
79          self.__arrow = arrow          self.__arrow = arrow
80          self.__color_mode = color_mode          self.__color_mode = color_mode
# Line 91  class Velocity(DataSetMapper, Actor3D, A Line 106  class Velocity(DataSetMapper, Actor3D, A
106    
107              # Default line width is 1.              # Default line width is 1.
108              actor3D._setLineWidth(1)              actor3D._setLineWidth(1)
109              self.__scene._addActor3D(self.__viewport, actor3D._getActor3D())              scene._addActor3D(self.__viewport, actor3D._getActor3D())
110    
111          # ----- Velocity -----          # ----- Velocity -----
112    
# Line 125  class Velocity(DataSetMapper, Actor3D, A Line 140  class Velocity(DataSetMapper, Actor3D, A
140                  lookup_table._getLookupTable())                  lookup_table._getLookupTable())
141    
142          self._setupActor3D(self._getDataSetMapper())          self._setupActor3D(self._getDataSetMapper())
143          self.__scene._addActor3D(self.__viewport, self._getActor3D())          scene._addActor3D(self.__viewport, self._getActor3D())
144    
145      def _isModified(self):        def _isModified(self):  
146          """          """
# Line 137  class Velocity(DataSetMapper, Actor3D, A Line 152  class Velocity(DataSetMapper, Actor3D, A
152    
153          return self.__modified or self.__data_collector._isModified()          return self.__modified or self.__data_collector._isModified()
154    
155      def _render(self):      def _render(self, scene):
156          """          """
157          Render the velocity.          Render the velocity.
158    
159            @type scene: L{Scene <scene.Scene>} object
160            @param scene: Scene in which objects are to be rendered on
161          """          """
162    
163          if (self._isModified() == True):          if (self._isModified() == True):
# Line 151  class Velocity(DataSetMapper, Actor3D, A Line 169  class Velocity(DataSetMapper, Actor3D, A
169              # Color velocity by vector.              # Color velocity by vector.
170              if(self.__color_mode == ColorMode.VECTOR):                            if(self.__color_mode == ColorMode.VECTOR):              
171                  self._setColorModeByVector()                  self._setColorModeByVector()
172                  self._setRange(self.__data_collector._getVectorRange())  
173                  self._setScalarRange(self.__data_collector._getVectorRange())                  # self._isScalarRangeSet checks whether the scalar range has
174                    # been specified by the user. If it has, then the scalar range
175                    # read from the source will be ignored.
176                    if(not(self._isScalarRangeSet())):
177                        self._setRange(self.__data_collector._getVectorRange())
178                        self._setScalarRange(\
179                                self.__data_collector._getVectorRange())
180                    else:
181                        self._setRange(self._getDataSetMapperRange())
182                        
183              # Color velocity by scalar.              # Color velocity by scalar.
184              elif(self.__color_mode == ColorMode.SCALAR):                              elif(self.__color_mode == ColorMode.SCALAR):                
185                  self._setColorModeByScalar()                  self._setColorModeByScalar()
186                  self._setRange(self.__data_collector._getScalarRange())  
187                  self._setScalarRange(self.__data_collector._getScalarRange())                  # self._isScalarRangeSet checks whether the scalar range has
188                    # been specified by the user. If it has, then the scalar range
189                    # read from the source will be ignored.
190                    if(not(self._isScalarRangeSet())):
191                        self._setRange(self.__data_collector._getScalarRange())
192                        self._setScalarRange(\
193                                self.__data_collector._getScalarRange())
194                    else:
195                        self._setRange(self._getDataSetMapperRange())
196    
197              self.__modified = False              self.__modified = False
198    
# Line 213  class VelocityOnPlaneCut(DataSetMapper, Line 248  class VelocityOnPlaneCut(DataSetMapper,
248          @param outline: Places an outline around the domain surface          @param outline: Places an outline around the domain surface
249          """          """
250    
         self.__scene = scene  
251          self.__data_collector = data_collector          self.__data_collector = data_collector
252          self.__arrow = arrow          self.__arrow = arrow
253          self.__color_mode = color_mode          self.__color_mode = color_mode
# Line 249  class VelocityOnPlaneCut(DataSetMapper, Line 283  class VelocityOnPlaneCut(DataSetMapper,
283    
284              # Default line width is 1.              # Default line width is 1.
285              actor3D._setLineWidth(1)              actor3D._setLineWidth(1)
286              self.__scene._addActor3D(self.__viewport, actor3D._getActor3D())              scene._addActor3D(self.__viewport, actor3D._getActor3D())
287    
288          # ----- Velocity on a cut plane -----          # ----- Velocity on a cut plane -----
289    
# Line 288  class VelocityOnPlaneCut(DataSetMapper, Line 322  class VelocityOnPlaneCut(DataSetMapper,
322                  lookup_table._getLookupTable())                  lookup_table._getLookupTable())
323    
324          self._setupActor3D(self._getDataSetMapper())          self._setupActor3D(self._getDataSetMapper())
325          self.__scene._addActor3D(self.__viewport, self._getActor3D())          scene._addActor3D(self.__viewport, self._getActor3D())
326            
327      def _isModified(self):        def _isModified(self):  
328          """          """
# Line 301  class VelocityOnPlaneCut(DataSetMapper, Line 335  class VelocityOnPlaneCut(DataSetMapper,
335    
336          return self.__modified or self.__data_collector._isModified()          return self.__modified or self.__data_collector._isModified()
337    
338      def _render(self):      def _render(self, scene):
339          """          """
340          Render the velocity cut using a plane..          Render the velocity cut using a plane.
341    
342            @type scene: L{Scene <scene.Scene>} object
343            @param scene: Scene in which objects are to be rendered on
344          """          """
345    
346          if (self._isModified() == True):          if (self._isModified() == True):
# Line 312  class VelocityOnPlaneCut(DataSetMapper, Line 349  class VelocityOnPlaneCut(DataSetMapper,
349              # Color velocity by vector.              # Color velocity by vector.
350              if(self.__color_mode == ColorMode.VECTOR):                            if(self.__color_mode == ColorMode.VECTOR):              
351                  self._setColorModeByVector()                  self._setColorModeByVector()
352                  self._setRange(self.__data_collector._getVectorRange())  
353                  self._setScalarRange(self.__data_collector._getVectorRange())                  # self._isScalarRangeSet checks whether the scalar range has
354                    # been specified by the user. If it has, then the scalar range
355                    # read from the source will be ignored.
356                    if(not(self._isScalarRangeSet())):
357                        self._setRange(self.__data_collector._getVectorRange())
358                        self._setScalarRange(\
359                                self.__data_collector._getVectorRange())
360                    else:
361                        self._setRange(self._getDataSetMapperRange())
362    
363              # Color velocity by scalar.              # Color velocity by scalar.
364              elif(self.__color_mode == ColorMode.SCALAR):                              elif(self.__color_mode == ColorMode.SCALAR):                
365                  self._setColorModeByScalar()                  self._setColorModeByScalar()
366                  self._setRange(self.__data_collector._getScalarRange())  
367                  self._setScalarRange(self.__data_collector._getScalarRange())                  # self._isScalarRangeSet checks whether the scalar range has
368                    # been specified by the user. If it has, then the scalar range
369                    # read from the source will be ignored.
370                    if(not(self._isScalarRangeSet())):
371                        self._setRange(self.__data_collector._getScalarRange())
372                        self._setScalarRange(\
373                                self.__data_collector._getScalarRange())
374                    else:
375                        self._setRange(self._getDataSetMapperRange())
376    
377              self.__modified = False              self.__modified = False
378    
# Line 372  class VelocityOnPlaneClip(DataSetMapper, Line 426  class VelocityOnPlaneClip(DataSetMapper,
426          @param outline: Places an outline around the domain surface          @param outline: Places an outline around the domain surface
427          """          """
428    
         self.__scene = scene  
429          self.__data_collector = data_collector          self.__data_collector = data_collector
430          self.__arrow = arrow          self.__arrow = arrow
431          self.__color_mode = color_mode          self.__color_mode = color_mode
# Line 408  class VelocityOnPlaneClip(DataSetMapper, Line 461  class VelocityOnPlaneClip(DataSetMapper,
461    
462              # Default line width is 1.              # Default line width is 1.
463              actor3D._setLineWidth(1)              actor3D._setLineWidth(1)
464              self.__scene._addActor3D(self.__viewport, actor3D._getActor3D())              scene._addActor3D(self.__viewport, actor3D._getActor3D())
465    
466          # ----- Velocity on a clipped plane -----          # ----- Velocity on a clipped plane -----
467    
# Line 452  class VelocityOnPlaneClip(DataSetMapper, Line 505  class VelocityOnPlaneClip(DataSetMapper,
505                  lookup_table._getLookupTable())                  lookup_table._getLookupTable())
506    
507          self._setupActor3D(self._getDataSetMapper())          self._setupActor3D(self._getDataSetMapper())
508          self.__scene._addActor3D(self.__viewport, self._getActor3D())          scene._addActor3D(self.__viewport, self._getActor3D())
509            
510      def _isModified(self):        def _isModified(self):  
511          """          """
# Line 465  class VelocityOnPlaneClip(DataSetMapper, Line 518  class VelocityOnPlaneClip(DataSetMapper,
518    
519          return self.__modified or self.__data_collector._isModified()          return self.__modified or self.__data_collector._isModified()
520    
521      def _render(self):      def _render(self, scene):
522          """          """
523          Render the velocity clip using a plane..          Render the velocity clip using a plane..
524    
525            @type scene: L{Scene <scene.Scene>} object
526            @param scene: Scene in which objects are to be rendered on
527          """          """
528    
529          if (self._isModified() == True):          if (self._isModified() == True):
# Line 476  class VelocityOnPlaneClip(DataSetMapper, Line 532  class VelocityOnPlaneClip(DataSetMapper,
532              # Color velocity by vector.              # Color velocity by vector.
533              if(self.__color_mode == ColorMode.VECTOR):                            if(self.__color_mode == ColorMode.VECTOR):              
534                  self._setColorModeByVector()                  self._setColorModeByVector()
535                  self._setRange(self.__data_collector._getVectorRange())  
536                  self._setScalarRange(self.__data_collector._getVectorRange())                  # self._isScalarRangeSet checks whether the scalar range has
537                    # been specified by the user. If it has, then the scalar range
538                    # read from the source will be ignored.
539                    if(not(self._isScalarRangeSet())):
540                        self._setRange(self.__data_collector._getVectorRange())
541                        self._setScalarRange(\
542                                self.__data_collector._getVectorRange())
543                    else:
544                        self._setRange(self._getDataSetMapperRange())
545    
546              # Color velocity by scalar.              # Color velocity by scalar.
547              elif(self.__color_mode == ColorMode.SCALAR):                              elif(self.__color_mode == ColorMode.SCALAR):                
548                  self._setColorModeByScalar()                  self._setColorModeByScalar()
549                  self._setRange(self.__data_collector._getScalarRange())  
550                  self._setScalarRange(self.__data_collector._getScalarRange())                  # self._isScalarRangeSet checks whether the scalar range has
551                    # been specified by the user. If it has, then the scalar range
552                    # read from the source will be ignored.
553                    if(not(self._isScalarRangeSet())):
554                        self._setRange(self.__data_collector._getScalarRange())
555                        self._setScalarRange(\
556                                self.__data_collector._getScalarRange())
557                    else:
558                        self._setRange(self._getDataSetMapperRange())
559    
560              self.__modified = False              self.__modified = False
561    

Legend:
Removed from v.1148  
changed lines
  Added in v.1197

  ViewVC Help
Powered by ViewVC 1.1.26