vtkPythonExtractSelection.h 2.37 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*=========================================================================

  Program:   ParaView
  Module:    $RCSfile$

  Copyright (c) Kitware, Inc.
  All rights reserved.
  See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/
15 16 17 18 19 20
/**
 * @class   vtkPythonExtractSelection
 *
 * vtkPythonExtractSelection is a used to extra cells/points using numpy. This
 * enables creation of arbitrary queries to be used as the selection criteria.
*/
21

22 23
#ifndef vtkPythonExtractSelection_h
#define vtkPythonExtractSelection_h
24

25
#include "vtkExtractSelectionBase.h"
26
#include "vtkPVClientServerCoreCoreModule.h" //needed for exports
27

28
class vtkCompositeDataSet;
29

30
class VTKPVCLIENTSERVERCORECORE_EXPORT vtkPythonExtractSelection : public vtkExtractSelectionBase
31 32 33
{
public:
  static vtkPythonExtractSelection* New();
34
  vtkTypeMacro(vtkPythonExtractSelection, vtkExtractSelectionBase);
Cory Quammen's avatar
Cory Quammen committed
35
  void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;
36

37 38 39 40 41 42
  //@{
  /**
   * Method called by Python code to handle the extraction logic.
   * \c attributeType is vtkDataObject::AttributeTypes and not to be confused with
   * vtkSelectionNode::SelectionField
   */
43 44
  bool ExtractElements(int attributeType, vtkDataObject* input, vtkDataObject* output);
  bool ExtractElements(int attributeType, vtkCompositeDataSet* input, vtkCompositeDataSet* output);
45
  //@}
46 47 48

protected:
  vtkPythonExtractSelection();
49
  ~vtkPythonExtractSelection() override;
50

51 52
  int FillInputPortInformation(int port, vtkInformation* info) VTK_OVERRIDE;
  int RequestDataObject(vtkInformation* request, vtkInformationVector** inputVector,
Cory Quammen's avatar
Cory Quammen committed
53
    vtkInformationVector* outputVector) VTK_OVERRIDE;
54
  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) VTK_OVERRIDE;
55

56 57 58 59 60
  /**
   * Method used to initialize the output data object in request data.
   * The output data is initialized based on the state of
   * this->PreserveTopology.
   */
61
  void InitializeOutput(vtkDataObject* output, vtkDataObject* input);
62

63
private:
64 65
  vtkPythonExtractSelection(const vtkPythonExtractSelection&) = delete;
  void operator=(const vtkPythonExtractSelection&) = delete;
66 67 68
};

#endif