test/VisuPrs/ScalarMap/B3.py

   1 # Copyright (C) 2010-2015  CEA/DEN, EDF R&D
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20 # This case corresponds to: /visu/ScalarMap/B3 case
  21 # Test ScalarMap interface methods.
  22
  23 from paravistest import datadir
  24 import presentations
  25 import pvsimple
  26
  27 # 1. Import MED file
  28
  29 print 'Importing "fra.med"...',
  30 file_path = datadir + "fra.med"
  31 pvsimple.OpenDataFile(file_path)
  32 med_reader = pvsimple.GetActiveSource()
  33
  34 if med_reader is None:
  35     print "FAILED"
  36 else:
  37     print "OK"
  38
  39 # 2. Create ScalarMap
  40 field_name = 'TAUX_DE_VIDE'
  41 entity =  presentations.EntityType.NODE
  42 myMeshName='LE VOLUME'
  43
  44 scalarmap = presentations.ScalarMapOnField(med_reader, entity, field_name, 1)
  45 if scalarmap is None:
  46     print "FAILED"
  47
  48 # 3. Scalar mode
  49 lookup_table = scalarmap.LookupTable
  50 print "Vector mode .....", lookup_table.VectorMode
  51 print "Vector component .....", lookup_table.VectorComponent
  52
  53 lookup_table.VectorMode = 'Component'
  54 lookup_table.VectorComponent = 0
  55
  56 scalarmap.LookupTable = lookup_table
  57
  58 print "Vector mode .....", scalarmap.LookupTable.VectorMode
  59 print "Vector component .....", scalarmap.LookupTable.VectorComponent
  60
  61 # 4. Scaling mode
  62 scalarmap.LookupTable.UseLogScale = 1
  63 print "Use logarithmic scaling ....", scalarmap.LookupTable.UseLogScale
  64
  65 scalarmap.LookupTable.UseLogScale = 0
  66 print "Use logarithmic scaling ....", scalarmap.LookupTable.UseLogScale
  67
  68 # 5. Scalar range
  69 print "Set scalar range min=12 < max=120 ...",
  70 rmin = 12
  71 rmax = 120
  72 scalarmap.LookupTable.RGBPoints[0] = rmin
  73 scalarmap.LookupTable.RGBPoints[4] = rmax
  74 print "min = ", scalarmap.LookupTable.RGBPoints[0],"  : max = ",scalarmap.LookupTable.RGBPoints[4]
  75
  76 print "Set scalar range min=max=12 ...",
  77 rmin = 120
  78 rmax = rmin
  79 scalarmap.LookupTable.RGBPoints[0] = rmin
  80 scalarmap.LookupTable.RGBPoints[4] = rmax
  81 print "min = ", scalarmap.LookupTable.RGBPoints[0],"  : max = ",scalarmap.LookupTable.RGBPoints[4]
  82
  83 print "Set scalar range min=120 > max=15 ...",
  84 rmin = 120
  85 rmax = 15
  86 scalarmap.LookupTable.RGBPoints[0] = rmin
  87 scalarmap.LookupTable.RGBPoints[4] = rmax
  88 print "min = ", scalarmap.LookupTable.RGBPoints[0],"  : max = ",scalarmap.LookupTable.RGBPoints[4]
  89
  90 # 6. Bar orientation
  91 bar = presentations.get_bar()
  92
  93 print "Set bar orientation = 'Horizontal'"
  94 bar.Orientation = 'Horizontal'
  95 print "Bar orientation ....", bar.Orientation
  96
  97 print "Set bar orientation = 'Vertical'"
  98 bar.Orientation = 'Vertical'
  99 print "Bar orientation ....", bar.Orientation
 100
 101 # 7. Position of scalar bar
 102 print "Default position ....", bar.Position
 103
 104 print "Set left down corner position"
 105 bar.Position = [0, 0]
 106 print "Position =", bar.Position
 107
 108 print "Set position outside of the screen"
 109 bar.Position = [-1, -1]
 110 print "Position =", bar.Position
 111
 112 # 8. Size of scalar bar
 113 print "Default Height=", bar.Position2[1],"  :  Width=", bar.Position2[0]
 114
 115 print "Set positive Height and Width"
 116 h = 0.4
 117 w = 0.2
 118 bar.Position2 = [w, h]
 119 print "Size =", bar.Position2
 120
 121 print "Set negative Height and Width"
 122 h = -0.4
 123 w = -0.2
 124 bar.Position2 = [w, h]
 125 print "Size =", bar.Position2
 126
 127 # 9. Number of colors
 128 print "Default number of colors = ", scalarmap.LookupTable.NumberOfTableValues
 129
 130 scalarmap.LookupTable.Discretize = 1
 131
 132 #print "Set negative number of colors"
 133 #nb_colors = -128
 134 #scalarmap.LookupTable.NumberOfTableValues = nb_colors
 135 #print "Number of colors =", scalarmap.LookupTable.NumberOfTableValues
 136
 137 print "Set zero number of colors"
 138 nb_colors = 0
 139 scalarmap.LookupTable.NumberOfTableValues = nb_colors
 140 print "Number of colors =", scalarmap.LookupTable.NumberOfTableValues
 141
 142 print "Set positive number of colors"
 143 nb_colors = 256
 144 scalarmap.LookupTable.NumberOfTableValues = nb_colors
 145 print "Number of colors =", scalarmap.LookupTable.NumberOfTableValues
 146
 147 # 10. Number of labels
 148 print "Default number of labels = ", bar.NumberOfLabels
 149
 150 print "Set negative number of labels"
 151 nb_labels = -10
 152 bar.NumberOfLabels = nb_labels
 153 print "Number of labels=", bar.NumberOfLabels
 154
 155 print "Set zero number of labels"
 156 nb_labels = 0
 157 bar.NumberOfLabels = nb_labels
 158 print "Number of labels=", bar.NumberOfLabels
 159
 160 print "Set positive number of labels"
 161 nb_labels = 10
 162 bar.NumberOfLabels = nb_labels
 163 print "Number of labels=", bar.NumberOfLabels
 164
 165 # 11. Scalar bar title
 166 print 'Default Title ="', bar.Title, '"'
 167
 168 print "Set not null title"
 169 title = "Scalar Bar Title"
 170 bar.Title = title
 171 print '   Title ="', bar.Title, '"'
 172
 173 print "Set title from spaces"
 174 title='    '
 175 bar.Title = title
 176 print '   Title ="', bar.Title, '"'