vtkTransferFunctionChartHistogram2D.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Kitware Inc.
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkTransferFunctionChartHistogram2D_h
#define vtkTransferFunctionChartHistogram2D_h

#include "vtkChartHistogram2D.h"

#include "vtkCommand.h"             // needed for vtkCommand::UserEvent
#include "vtkRemotingViewsModule.h" // needed for export macro
#include "vtkWeakPointer.h"         // needed for vtkWeakPointer

#include <vector> // needed for ivar

// Forward declarations
class vtkContext2D;
class vtkContextMouseEvent;
class vtkPVTransferFunction2D;
class vtkTransferFunctionBoxItem;

class VTKREMOTINGVIEWS_EXPORT vtkTransferFunctionChartHistogram2D : public vtkChartHistogram2D
{
public:
  static vtkTransferFunctionChartHistogram2D* New();
  vtkTypeMacro(vtkTransferFunctionChartHistogram2D, vtkChartHistogram2D);

  // Events fires by this class (and subclasses).
  // \li TransferFunctionModified is fired when the 2D transfer function is modified.
  enum
  {
    TransferFunctionModified = vtkCommand::UserEvent + 1000,
  };

  bool IsInitialized();


  vtkSmartPointer<vtkTransferFunctionBoxItem> AddNewBox();
  vtkSmartPointer<vtkTransferFunctionBoxItem> AddNewBox(
    const vtkRectd& r, double* color, double alpha, bool addToTF2D = true);
  void AddBox(vtkSmartPointer<vtkTransferFunctionBoxItem> box, bool addToTF2D = true);

  virtual void RemoveBox(vtkSmartPointer<vtkTransferFunctionBoxItem> box);

  bool MouseDoubleClickEvent(const vtkContextMouseEvent& mouse) override;

  bool KeyPressEvent(const vtkContextKeyEvent& key) override;

  void SetInputData(vtkImageData*, vtkIdType z = 0) override;


  virtual void SetTransferFunction2D(vtkPVTransferFunction2D* transfer2D);
  virtual vtkPVTransferFunction2D* GetTransferFunction2D();


  vtkSmartPointer<vtkTransferFunctionBoxItem> GetActiveBox() const;
  void SetActiveBox(vtkSmartPointer<vtkTransferFunctionBoxItem> box);


  void SetActiveBoxColorAlpha(double r, double g, double b, double a);
  void SetActiveBoxColorAlpha(double color[3], double alpha);

  bool Paint(vtkContext2D* painter) override;

protected:
  vtkTransferFunctionChartHistogram2D() = default;
  ~vtkTransferFunctionChartHistogram2D() override = default;

  void UpdateItemsBounds(double xMin, double xMax, double yMin, double yMax);

  virtual void GenerateTransfer2D();

  void OnTransferFunctionBoxItemModified(vtkObject* caller, unsigned long eid, void* callData);

  // Member variables;
  vtkWeakPointer<vtkPVTransferFunction2D> TransferFunction2D;
  vtkSmartPointer<vtkTransferFunctionBoxItem> ActiveBox;
  std::vector<int> BoxesToRemove;

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

#endif // vtkTransferFunctionChartHistogram2D_h