vtkSciVizStatistics.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-FileCopyrightText: Copyright 2011 Sandia Corporation
// SPDX-License-Identifier: LicenseRef-BSD-3-Clause-Sandia-USGov
#ifndef vtkSciVizStatistics_h
#define vtkSciVizStatistics_h

#include "vtkPVVTKExtensionsFiltersStatisticsModule.h" //needed for exports
#include "vtkPassInputTypeAlgorithm.h"

class vtkCompositeDataSet;
class vtkDataObjectToTable;
class vtkFieldData;
class vtkGenerateStatistics;
class vtkInformationIntegerKey;
class vtkPartitionedDataSetCollection;
class vtkMultiProcessController;
class vtkSciVizStatisticsP;
class vtkStatisticsAlgorithm;

class VTKPVVTKEXTENSIONSFILTERSSTATISTICS_EXPORT vtkSciVizStatistics
  : public vtkPassInputTypeAlgorithm
{
public:
  vtkTypeMacro(vtkSciVizStatistics, vtkPassInputTypeAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;


  vtkGetMacro(AttributeMode, int);
  vtkSetMacro(AttributeMode, int);

  int GetNumberOfAttributeArrays();

  const char* GetAttributeArrayName(int n);

  int GetAttributeArrayStatus(const char* arrName);


  void EnableAttributeArray(const char* arrName);
  void ClearAttributeArrays();


  vtkSetClampMacro(TrainingFraction, double, 0.0, 1.0);
  vtkGetMacro(TrainingFraction, double);


  virtual void SetController(vtkMultiProcessController*);
  vtkGetObjectMacro(Controller, vtkMultiProcessController);

  enum Tasks
  {
    MODEL_INPUT,     
    CREATE_MODEL,    
    ASSESS_INPUT,    
    MODEL_AND_ASSESS 
  };


  vtkSetMacro(Task, int);
  vtkGetMacro(Task, int);


  vtkSetMacro(GlobalModel, vtkTypeBool);
  vtkGetMacro(GlobalModel, vtkTypeBool);
  vtkBooleanMacro(GlobalModel, vtkTypeBool);

  vtkInformationIntegerKey* MULTIPLE_MODELS();

protected:
  vtkSciVizStatistics();
  ~vtkSciVizStatistics() override;

  virtual bool PrepareInputArrays(vtkDataObject* inData, vtkGenerateStatistics* filter);

  virtual bool PrepareAlgorithm(vtkGenerateStatistics* filter) = 0;

  int FillInputPortInformation(int port, vtkInformation* info) override;
  int FillOutputPortInformation(int port, vtkInformation* info) override;

  int RequestData(
    vtkInformation* request, vtkInformationVector** input, vtkInformationVector* output) override;

  int AttributeMode;
  int Task;
  double TrainingFraction;
  vtkTypeBool GlobalModel;
  vtkSciVizStatisticsP* P;
  vtkMultiProcessController* Controller;

private:
  vtkSciVizStatistics(const vtkSciVizStatistics&) = delete;
  void operator=(const vtkSciVizStatistics&) = delete;
};

#endif // vtkSciVizStatistics_h