src/VISU_SWIG/VISU_Example_05.py

   1 #  -*- coding: iso-8859-1 -*-
   2 #  Copyright (C) 2007-2008  CEA/DEN, EDF R&D, OPEN CASCADE
   3 #
   4 #  Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   5 #  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   6 #
   7 #  This library is free software; you can redistribute it and/or
   8 #  modify it under the terms of the GNU Lesser General Public
   9 #  License as published by the Free Software Foundation; either
  10 #  version 2.1 of the License.
  11 #
  12 #  This library is distributed in the hope that it will be useful,
  13 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15 #  Lesser General Public License for more details.
  16 #
  17 #  You should have received a copy of the GNU Lesser General Public
  18 #  License along with this library; if not, write to the Free Software
  19 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  20 #
  21 #  See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  22 #
  23 # Create a table and show it in Plot2d viewer
  24 # This script is equivalent to script VISU_SWIG/visu_big_table.py
  25 #
  26 import salome
  27 import math
  28 import SALOMEDS
  29 import VISU
  30 #from visu_gui import *
  31
  32 # >>> Getting study builder ==================================================
  33 myStudy = salome.myStudy
  34 myBuilder = myStudy.NewBuilder()
  35
  36 # >>> Getting (loading) VISU component =======================================
  37 myVisu = salome.lcc.FindOrLoadComponent("FactoryServer", "VISU")
  38 myComponent = myStudy.FindComponent("VISU")
  39 myVisu.SetCurrentStudy(myStudy)
  40 if not myComponent:
  41    myComponent = myBuilder.NewComponent("VISU")
  42    aName = myBuilder.FindOrCreateAttribute(myComponent, "AttributeName")
  43    aName.SetValue( salome.sg.getComponentUserName("VISU") )
  44
  45    A2 = myBuilder.FindOrCreateAttribute(myComponent, "AttributePixMap");
  46    aPixmap = A2._narrow(SALOMEDS.AttributePixMap);
  47    aPixmap.SetPixMap( "ICON_OBJBROWSER_Visu" );
  48
  49    myBuilder.DefineComponentInstance(myComponent,myVisu)
  50
  51 # >>> Creating object with Table of real[ 200 * 20 ] =========================
  52 myTRealObject = myBuilder.NewObject(myComponent)
  53 AName = myBuilder.FindOrCreateAttribute(myTRealObject, "AttributeName")
  54 AName.SetValue("Table Of Real")
  55 ARealTable = myBuilder.FindOrCreateAttribute(myTRealObject, "AttributeTableOfReal")
  56 myHorNb = 10
  57 myVerNb = 200
  58
  59 k={}
  60 for j in range(0,myHorNb):
  61    k[j] = j*10+1
  62 ARealTable.AddRow(k.values())
  63 ARealTable.SetRowTitle(1, "Frequency")
  64 ARealTable.SetRowUnit(1, "Hz")
  65
  66 for i in range(1,myVerNb+1):
  67    for j in range(0,myHorNb):
  68       if j % 2 == 1:
  69          k[j] = math.log10(j*30*math.pi/180) * 20 + i * 15 + j*5
  70       else:
  71          k[j] = math.sin(j*30*math.pi/180) * 20 + i * 15 + j*5
  72    ARealTable.AddRow(k.values())
  73    ARealTable.SetRowTitle(i+1, "Power " + str(i))
  74    ARealTable.SetRowUnit(i+1, "Wt")
  75
  76 ARealTable.SetTitle("Very useful data")
  77
  78 # >>> Create Visu table ======================================================
  79 myVisuTableReal = myVisu.CreateTable( myTRealObject.GetID() )
  80
  81 # >>> Create container and insert curves
  82 myContainer = myVisu.CreateContainer()
  83
  84 # >>> Create curves ==========================================================
  85 for i in range(1,myVerNb+1):
  86    myCurve = myVisu.CreateCurve( myVisuTableReal, 1, i+1 )
  87    myContainer.AddCurve(myCurve)
  88
  89 # >>> Updating Object Browser ================================================
  90 salome.sg.updateObjBrowser(1)
  91
  92 # >>> Display curves in Plot2d viewer ========================================
  93 myViewManager = myVisu.GetViewManager();
  94 myView = myViewManager.CreateXYPlot();
  95 myView.SetTitle("The viewer for Curves from the Table")
  96 myView.Display(myContainer)
  97
  98 # ============================================================================