src/VISU_SWIG/VISU_Example_05.py

   1 #  -*- coding: iso-8859-1 -*-
   2 #  Copyright (C) 2007-2010  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
  24 # Create a table and show it in Plot2d viewer
  25 # This script is equivalent to script VISU_SWIG/visu_big_table.py
  26 #
  27 import salome
  28 import math
  29 import SALOMEDS
  30 import VISU
  31 #from visu_gui import *
  32
  33 # >>> Getting study builder ==================================================
  34 myStudy = salome.myStudy
  35 myBuilder = myStudy.NewBuilder()
  36
  37 # >>> Getting (loading) VISU component =======================================
  38 myVisu = salome.lcc.FindOrLoadComponent("FactoryServer", "VISU")
  39 myComponent = myStudy.FindComponent("VISU")
  40 myVisu.SetCurrentStudy(myStudy)
  41 if not myComponent:
  42    myComponent = myBuilder.NewComponent("VISU")
  43    aName = myBuilder.FindOrCreateAttribute(myComponent, "AttributeName")
  44    aName.SetValue( salome.sg.getComponentUserName("VISU") )
  45
  46    A2 = myBuilder.FindOrCreateAttribute(myComponent, "AttributePixMap");
  47    aPixmap = A2._narrow(SALOMEDS.AttributePixMap);
  48    aPixmap.SetPixMap( "ICON_OBJBROWSER_Visu" );
  49
  50    myBuilder.DefineComponentInstance(myComponent,myVisu)
  51
  52 # >>> Creating object with Table of real[ 200 * 20 ] =========================
  53 myTRealObject = myBuilder.NewObject(myComponent)
  54 AName = myBuilder.FindOrCreateAttribute(myTRealObject, "AttributeName")
  55 AName.SetValue("Table Of Real")
  56 ARealTable = myBuilder.FindOrCreateAttribute(myTRealObject, "AttributeTableOfReal")
  57 myHorNb = 10
  58 myVerNb = 200
  59
  60 k={}
  61 for j in range(0,myHorNb):
  62    k[j] = j*10+1
  63 ARealTable.AddRow(k.values())
  64 ARealTable.SetRowTitle(1, "Frequency")
  65 ARealTable.SetRowUnit(1, "Hz")
  66
  67 for i in range(1,myVerNb+1):
  68    for j in range(0,myHorNb):
  69       if j % 2 == 1:
  70          k[j] = math.log10(j*30*math.pi/180) * 20 + i * 15 + j*5
  71       else:
  72          k[j] = math.sin(j*30*math.pi/180) * 20 + i * 15 + j*5
  73    ARealTable.AddRow(k.values())
  74    ARealTable.SetRowTitle(i+1, "Power " + str(i))
  75    ARealTable.SetRowUnit(i+1, "Wt")
  76
  77 ARealTable.SetTitle("Very useful data")
  78
  79 # >>> Create Visu table ======================================================
  80 myVisuTableReal = myVisu.CreateTable( myTRealObject.GetID() )
  81
  82 # >>> Create container and insert curves
  83 myContainer = myVisu.CreateContainer()
  84
  85 # >>> Create curves ==========================================================
  86 for i in range(1,myVerNb+1):
  87    myCurve = myVisu.CreateCurve( myVisuTableReal, 1, i+1 )
  88    myContainer.AddCurve(myCurve)
  89
  90 # >>> Updating Object Browser ================================================
  91 salome.sg.updateObjBrowser(1)
  92
  93 # >>> Display curves in Plot2d viewer ========================================
  94 myViewManager = myVisu.GetViewManager();
  95 myView = myViewManager.CreateXYPlot();
  96 myView.SetTitle("The viewer for Curves from the Table")
  97 myView.Display(myContainer)
  98
  99 # ============================================================================