VTKViewer_ExtractUnstructuredGrid.h

Go to the documentation of this file.

// Copyright (C) 2007-2016  CEA/DEN, EDF R&D, OPEN CASCADE
//
// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
//
// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
//

#ifndef VTKVIEWER_EXTRACTUNSTRUCTUREDGRID_H
#define VTKVIEWER_EXTRACTUNSTRUCTUREDGRID_H

#include "VTKViewer.h"

#include <vtkUnstructuredGridAlgorithm.h>

#include <set>
#include <map>
#include <vector>

#ifdef WIN32
#pragma warning ( disable:4251 )
#endif

class VTKVIEWER_EXPORT VTKViewer_ExtractUnstructuredGrid : public vtkUnstructuredGridAlgorithm
{
public:
  vtkTypeMacro( VTKViewer_ExtractUnstructuredGrid, vtkUnstructuredGridAlgorithm );

  static VTKViewer_ExtractUnstructuredGrid *New();

  enum EExtraction{ eCells, ePoints};
  void SetModeOfExtraction(EExtraction theExtractionMode){
    if ( myExtractionMode != theExtractionMode ) { myExtractionMode = theExtractionMode; Modified(); }
  }
  EExtraction GetModeOfExtraction(){ return myExtractionMode;}

  enum EChanging{ ePassAll, eAdding, eRemoving};
  void SetModeOfChanging(EChanging theChangeMode){
    if ( myChangeMode != theChangeMode ) { myChangeMode = theChangeMode; Modified(); }
  }
  EChanging GetModeOfChanging(){ return myChangeMode; }

  void RegisterCell(vtkIdType theCellId);
  int IsCellsRegistered() { return !myCellIds.empty(); }
  void ClearRegisteredCells() {
    if ( !myCellIds.empty() ) { myCellIds.clear(); Modified(); }
  }

  void RegisterCellsWithType(vtkIdType theCellType);
  int IsCellsWithTypeRegistered() { return !myCellTypes.empty(); }
  void ClearRegisteredCellsWithType() {
    if ( !myCellTypes.empty() ) { myCellTypes.clear(); Modified(); }
  }

  int IsChanging() { return IsCellsRegistered() || IsCellsWithTypeRegistered(); }

  void SetStoreMapping(int theStoreMapping);
  int GetStoreMapping(){ return myStoreMapping; }

  void BuildOut2InMap();
  vtkIdType GetInputId(int theOutId) const;
  //vtkIdType GetOutputId(int theInId) const;

  typedef std::vector<vtkIdType>        TVectorId;
  typedef std::map<vtkIdType,vtkIdType> TMapId;
  typedef std::set<vtkIdType>           TSetId;

protected:
  VTKViewer_ExtractUnstructuredGrid();
  ~VTKViewer_ExtractUnstructuredGrid();

  virtual int RequestData(vtkInformation *, vtkInformationVector **, vtkInformationVector *);

  EExtraction myExtractionMode;
  
  EChanging myChangeMode;
  TSetId myCellIds;
  TSetId myCellTypes;

  bool myStoreMapping, myPassAll;
  TVectorId myOut2InId;
  //TMapId myIn2OutId;

private:
  VTKViewer_ExtractUnstructuredGrid(const VTKViewer_ExtractUnstructuredGrid&);
  void operator=(const VTKViewer_ExtractUnstructuredGrid&);
};

#ifdef WIN32
#pragma warning ( default:4251 )
#endif

#endif