src/CONVERTOR/VISU_ConvertorUtils.cxx

   1 //  VISU OBJECT : interactive object for VISU entities implementation
   2 //
   3 //  Copyright (C) 2003  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
   4 //  CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
   5 //
   6 //  This library is free software; you can redistribute it and/or
   7 //  modify it under the terms of the GNU Lesser General Public
   8 //  License as published by the Free Software Foundation; either
   9 //  version 2.1 of the License.
  10 //
  11 //  This library is distributed in the hope that it will be useful,
  12 //  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 //  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14 //  Lesser General Public License for more details.
  15 //
  16 //  You should have received a copy of the GNU Lesser General Public
  17 //  License along with this library; if not, write to the Free Software
  18 //  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  19 //
  20 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  21 //
  22 //
  23 //  File   : VISU_Convertor_impl.cxx
  24 //  Author : Alexey PETROV
  25 //  Module : VISU
  26
  27 #include "VISU_ConvertorUtils.hxx"
  28
  29 #include <vtkCellType.h>
  30
  31 #include <vtkUnstructuredGridWriter.h>
  32 #include <vtkUnstructuredGrid.h>
  33
  34 #include <vtkPolyDataWriter.h>
  35 #include <vtkPolyData.h>
  36
  37 #include <vtkInformationVector.h>
  38 #include <vtkInformation.h>
  39 #include <vtkExecutive.h>
  40
  41 #include <vtkTimerLog.h>
  42 #include <vtkPointData.h>
  43 #include <vtkCellData.h>
  44 #include <vtkDataSet.h>
  45
  46 #include <vtkIntArray.h>
  47 #include <algorithm>
  48
  49 #ifdef _DEBUG_
  50 static int MYDEBUG = 0;
  51 #else
  52 static int MYDEBUG = 0;
  53 #endif
  54
  55 namespace VISU
  56 {
  57   //---------------------------------------------------------------
  58   vtkIdType
  59   VISUGeom2NbNodes(EGeometry theGeom)
  60   {
  61     switch(theGeom){
  62 #ifndef VISU_ENABLE_QUADRATIC
  63     case VISU::eSEG3:
  64       return 2;
  65     case VISU::eTRIA6:
  66       return 3;
  67     case VISU::eQUAD8:
  68       return 4;
  69     case VISU::eTETRA10:
  70       return 4;
  71     case VISU::eHEXA20:
  72       return 8;
  73     case VISU::ePENTA15:
  74       return 6;
  75     case VISU::ePYRA13:
  76       return 5;
  77 #endif
  78     case VISU::ePOLYGONE:
  79     case VISU::ePOLYEDRE:
  80       return -1;
  81     default:
  82       return theGeom % 100;
  83     }
  84   }
  85
  86
  87   //---------------------------------------------------------------
  88   vtkIdType
  89   VISUGeom2VTK(EGeometry theGeom)
  90   {
  91     switch(theGeom){
  92     case VISU::ePOINT1:
  93       return VTK_VERTEX;
  94     case VISU::eSEG2:
  95       return VTK_LINE;
  96     case VISU::eTRIA3:
  97       return VTK_TRIANGLE;
  98     case VISU::eQUAD4:
  99       return VTK_QUAD;
 100     case VISU::eTETRA4:
 101       return VTK_TETRA;
 102     case VISU::eHEXA8:
 103       return VTK_HEXAHEDRON;
 104     case VISU::ePENTA6:
 105       return VTK_WEDGE;
 106     case VISU::ePYRA5:
 107       return VTK_PYRAMID;
 108
 109     case VISU::ePOLYGONE:
 110       return VTK_POLYGON;
 111     case VISU::ePOLYEDRE:
 112       return VTK_CONVEX_POINT_SET;
 113
 114 #ifndef VISU_ENABLE_QUADRATIC
 115     case VISU::eSEG3:
 116       return VTK_LINE;
 117     case VISU::eTRIA6:
 118       return VTK_TRIANGLE;
 119     case VISU::eQUAD8:
 120       return VTK_QUAD;
 121     case VISU::eTETRA10:
 122       return VTK_TETRA;
 123     case VISU::eHEXA20:
 124       return VTK_HEXAHEDRON;
 125     case VISU::ePENTA15:
 126       return VTK_WEDGE;
 127     case VISU::ePYRA13:
 128       return VTK_PYRAMID;
 129
 130 #else
 131
 132     case VISU::eSEG3:
 133 #if defined(VTK_QUADRATIC_EDGE) && defined(VISU_USE_VTK_QUADRATIC)
 134       return VTK_QUADRATIC_EDGE;
 135 #else
 136       return VTK_POLY_LINE;
 137 #endif
 138
 139     case VISU::eTRIA6:
 140 #if defined(VTK_QUADRATIC_TRIANGLE) && defined(VISU_USE_VTK_QUADRATIC)
 141       return VTK_QUADRATIC_TRIANGLE;
 142 #else
 143       return VTK_POLYGON;
 144 #endif
 145
 146     case VISU::eQUAD8:
 147 #if defined(VTK_QUADRATIC_QUAD) && defined(VISU_USE_VTK_QUADRATIC)
 148       return VTK_QUADRATIC_QUAD;
 149 #else
 150       return VTK_POLYGON;
 151 #endif
 152
 153     case VISU::eTETRA10:
 154 #if defined(VTK_QUADRATIC_TETRA) && defined(VISU_USE_VTK_QUADRATIC)
 155       return VTK_QUADRATIC_TETRA;
 156 #else
 157       return VTK_CONVEX_POINT_SET;
 158 #endif
 159
 160     case VISU::eHEXA20:
 161 #if defined(VTK_QUADRATIC_HEXAHEDRON) && defined(VISU_USE_VTK_QUADRATIC)
 162       return VTK_QUADRATIC_HEXAHEDRON;
 163 #else
 164       return VTK_CONVEX_POINT_SET;
 165 #endif
 166
 167     case VISU::ePENTA15:
 168 #if defined(VTK_QUADRATIC_WEDGE) && defined(VISU_USE_VTK_QUADRATIC)
 169       return VTK_QUADRATIC_WEDGE;
 170 #else
 171       return VTK_CONVEX_POINT_SET;
 172 #endif
 173
 174     case VISU::ePYRA13:
 175 #if defined(VTK_QUADRATIC_PYRAMID) && defined(VISU_USE_VTK_QUADRATIC)
 176       return VTK_QUADRATIC_PYRAMID;
 177 #else
 178       return VTK_CONVEX_POINT_SET;
 179 #endif
 180
 181 #endif //VISU_ENABLE_QUADRATIC
 182
 183     default:
 184       return -1;
 185     }
 186   }
 187
 188
 189   //---------------------------------------------------------------
 190   void
 191   WriteToFile(vtkUnstructuredGrid* theDataSet,
 192               const std::string& theFileName)
 193   {
 194     vtkUnstructuredGridWriter* aWriter = vtkUnstructuredGridWriter::New();
 195     //aWriter->SetFileType(VTK_BINARY);
 196     aWriter->SetFileName(theFileName.c_str());
 197     aWriter->SetInput(theDataSet);
 198     aWriter->Write();
 199     aWriter->Delete();
 200   }
 201
 202
 203   //---------------------------------------------------------------
 204   void
 205   WriteToFile(vtkPolyData* theDataSet,
 206               const std::string& theFileName)
 207   {
 208     vtkPolyDataWriter* aWriter = vtkPolyDataWriter::New();
 209     //aWriter->SetFileType(VTK_BINARY);
 210     aWriter->SetFileName(theFileName.c_str());
 211     aWriter->SetInput(theDataSet);
 212     aWriter->Write();
 213     aWriter->Delete();
 214   }
 215
 216
 217   //---------------------------------------------------------------
 218   bool
 219   IsDataOnPoints(vtkDataSet* theDataSet)
 220   {
 221     theDataSet->Update();
 222     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetPointData();
 223     return aDataSetAttributes->GetArray("VISU_FIELD") != NULL;
 224   }
 225
 226
 227   //---------------------------------------------------------------
 228   bool
 229   IsDataOnCells(vtkDataSet* theDataSet)
 230   {
 231     theDataSet->Update();
 232     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetCellData();
 233     return aDataSetAttributes->GetArray("VISU_FIELD") != NULL;
 234   }
 235
 236
 237   //---------------------------------------------------------------
 238   vtkIdType
 239   GetVTKID(vtkDataArray *theIDDataArray, vtkIdType theID, int theEntity)
 240   {
 241     if(vtkIntArray *anIntArray = dynamic_cast<vtkIntArray*>(theIDDataArray)){
 242       int aNbTuples = anIntArray->GetNumberOfTuples();
 243       int* aPointer = anIntArray->GetPointer(0);
 244       for(int aTupleId = 0; aTupleId < aNbTuples; aTupleId++){
 245         if(*aPointer == theID && *(aPointer + 1) == theEntity){
 246           return aTupleId;
 247         }
 248         aPointer += 2;
 249       }
 250     }
 251     return -1;
 252   }
 253
 254
 255   //---------------------------------------------------------------
 256   vtkIdType
 257   GetObjectID(vtkDataArray *theIDDataArray, vtkIdType theID)
 258   {
 259     if(vtkIntArray *anIntArray = dynamic_cast<vtkIntArray*>(theIDDataArray)){
 260       int aNbComp = anIntArray->GetNumberOfComponents();
 261       int* aPointer = anIntArray->GetPointer(theID*aNbComp);
 262       return *aPointer;
 263     }
 264     return -1;
 265   }
 266
 267
 268   //---------------------------------------------------------------
 269   vtkIdType
 270   GetElemVTKID(vtkDataSet *theDataSet, vtkIdType theID, int theEntity)
 271   {
 272     theDataSet->Update();
 273     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetCellData();
 274     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_CELLS_MAPPER")){
 275       if(theEntity < 0){
 276         {
 277           vtkIdType anID = GetVTKID(aDataArray, theID, VISU::CELL_ENTITY);
 278           if(anID != -1)
 279             return anID;
 280         }
 281         {
 282           vtkIdType anID = GetVTKID(aDataArray, theID, VISU::FACE_ENTITY);
 283           if(anID != -1)
 284             return anID;
 285         }
 286         {
 287           vtkIdType anID = GetVTKID(aDataArray, theID, VISU::EDGE_ENTITY);
 288           if(anID != -1)
 289             return anID;
 290         }
 291         {
 292           vtkIdType anID = GetVTKID(aDataArray, theID, VISU::NODE_ENTITY);
 293           if(anID != -1)
 294             return anID;
 295         }
 296       }else
 297         return GetVTKID(aDataArray, theID, theEntity);
 298     }
 299     return -1;
 300   }
 301
 302
 303   //---------------------------------------------------------------
 304   vtkIdType
 305   GetElemObjID(vtkDataSet *theDataSet, vtkIdType theID)
 306   {
 307     theDataSet->Update();
 308     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetCellData();
 309     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_CELLS_MAPPER"))
 310       return GetObjectID(aDataArray, theID);
 311
 312     return -1;
 313   }
 314
 315
 316   //---------------------------------------------------------------
 317   vtkCell*
 318   GetElemCell(vtkDataSet *theDataSet, vtkIdType  theObjID)
 319   {
 320     vtkIdType aVTKID = GetElemVTKID(theDataSet, theObjID);
 321     return theDataSet->GetCell(aVTKID);
 322   }
 323
 324
 325   //---------------------------------------------------------------
 326   vtkIdType
 327   GetNodeVTKID(vtkDataSet *theDataSet, vtkIdType theID)
 328   {
 329     theDataSet->Update();
 330     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetPointData();
 331     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_POINTS_MAPPER"))
 332       return GetVTKID(aDataArray, theID, VISU::NODE_ENTITY);
 333
 334     return -1;
 335   }
 336
 337
 338   //---------------------------------------------------------------
 339   vtkIdType
 340   GetNodeObjID(vtkDataSet *theDataSet, vtkIdType theID)
 341   {
 342     theDataSet->Update();
 343     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetPointData();
 344     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_POINTS_MAPPER"))
 345       return GetObjectID(aDataArray, theID);
 346
 347     return -1;
 348   }
 349
 350
 351   //---------------------------------------------------------------
 352   vtkFloatingPointType*
 353   GetNodeCoord(vtkDataSet *theDataSet, vtkIdType theObjID)
 354   {
 355     vtkIdType aVTKID = GetNodeVTKID(theDataSet, theObjID);
 356     return theDataSet->GetPoint(aVTKID);
 357   }
 358
 359
 360   //---------------------------------------------------------------
 361   TGaussPointID
 362   GetObjID(vtkDataSet *theDataSet, vtkIdType theID)
 363   {
 364     theDataSet->Update();
 365     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetCellData();
 366     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_CELLS_MAPPER")){
 367       if(vtkIntArray *anIntArray = dynamic_cast<vtkIntArray*>(aDataArray)){
 368         vtkIdType anID = 2 * theID;
 369         TCellID aCellID = anIntArray->GetValue(anID);
 370         TLocalPntID aLocalPntID = anIntArray->GetValue(anID + 1);
 371         return TGaussPointID(aCellID, aLocalPntID);
 372       }
 373     }
 374     return TGaussPointID();
 375   }
 376
 377
 378   //---------------------------------------------------------------
 379   TInputCellID
 380   GetInputCellID(vtkDataSet *theDataSet, vtkIdType theObjID)
 381   {
 382     theDataSet->Update();
 383     vtkDataSetAttributes *aDataSetAttributes = theDataSet->GetCellData();
 384     if(vtkDataArray *aDataArray = aDataSetAttributes->GetArray("VISU_INPUTS_MAPPER")){
 385       if(vtkIntArray *anIntArray = dynamic_cast<vtkIntArray*>(aDataArray)){
 386         vtkIdType aVTKID = GetElemVTKID(theDataSet, theObjID);
 387         vtkIdType aTupleID = 2 * aVTKID;
 388         TCellID aCellID = anIntArray->GetValue(aTupleID);
 389         TInputID anInputID = anIntArray->GetValue(aTupleID + 1);
 390         return TInputCellID(anInputID, aCellID);
 391       }
 392     }
 393     return TInputCellID();
 394   }
 395
 396
 397   //---------------------------------------------------------------
 398   vtkDataSet*
 399   GetInput(vtkInformationVector **theInputVector,
 400            vtkIdType theInputId)
 401   {
 402     if(vtkInformation* anInformation = theInputVector[0]->GetInformationObject(theInputId))
 403       return vtkDataSet::SafeDownCast(anInformation->Get(vtkDataObject::DATA_OBJECT()));
 404     return NULL;
 405   }
 406
 407
 408   //---------------------------------------------------------------
 409   vtkDataSet*
 410   GetOutput(vtkInformationVector *theOutputVector)
 411   {
 412     if(vtkInformation* anInformation = theOutputVector->GetInformationObject(0))
 413       return vtkDataSet::SafeDownCast(anInformation->Get(vtkDataObject::DATA_OBJECT()));
 414     return NULL;
 415   }
 416
 417
 418   //---------------------------------------------------------------
 419   TTimerLog
 420   ::TTimerLog(int theIsDebug,
 421               const std::string& theName):
 422     myIsDebug(MYDEBUG + theIsDebug),
 423     myTimerLog(vtkTimerLog::New()),
 424     myPrefixPrinter(myIsDebug == 1),
 425     myName(theName)
 426   {
 427     myCPUTime = myTimerLog->GetCPUTime();
 428     BEGMSG(myIsDebug > 1,"{\n");
 429   }
 430
 431   //---------------------------------------------------------------
 432   TTimerLog
 433   ::~TTimerLog()
 434   {
 435     myCPUTime = myTimerLog->GetCPUTime() - myCPUTime;
 436
 437     if(myIsDebug > 1){
 438       BEGMSG(myIsDebug,"} = "<<myCPUTime<<" secs ("<<myName<<")\n");
 439     }else{
 440       BEGMSG(myIsDebug,myName<<" takes "<<myCPUTime<<" secs\n");
 441     }
 442
 443     myTimerLog->Delete();
 444     myTimerLog = NULL;
 445   }
 446
 447
 448   //---------------------------------------------------------------
 449 }