qMRMLSegmentEditorWidget.h

Go to the documentation of this file.

/*==============================================================================

  Program: 3D Slicer

  Copyright (c) Laboratory for Percutaneous Surgery (PerkLab)
  Queen's University, Kingston, ON, Canada. All Rights Reserved.

  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.

  This file was originally developed by Csaba Pinter, PerkLab, Queen's University
  and was supported through the Applied Cancer Research Unit program of Cancer Care
  Ontario with funds provided by the Ontario Ministry of Health and Long-Term Care

==============================================================================*/

#ifndef __qMRMLSegmentEditorWidget_h
#define __qMRMLSegmentEditorWidget_h

// Segmentations includes
#include "qSlicerSegmentationsModuleWidgetsExport.h"

// MRMLWidgets includes
#include "qMRMLWidget.h"

// Qt includes
#include <QVariant>

// CTK includes
#include <ctkVTKObject.h>

// STD includes
#include <cstdlib>

class vtkMRMLNode;
class vtkMRMLSegmentationNode;
class vtkMRMLSegmentEditorNode;
class vtkMRMLVolumeNode;
class vtkObject;
class QItemSelection;
class QAbstractButton;
class qMRMLSegmentEditorWidgetPrivate;
class qSlicerSegmentEditorAbstractEffect;
class qSlicerAbstractModuleWidget;

class Q_SLICER_MODULE_SEGMENTATIONS_WIDGETS_EXPORT qMRMLSegmentEditorWidget : public qMRMLWidget
{
  Q_OBJECT
  QVTK_OBJECT
  Q_PROPERTY(bool segmentationNodeSelectorVisible READ segmentationNodeSelectorVisible WRITE setSegmentationNodeSelectorVisible)
  Q_PROPERTY(bool masterVolumeNodeSelectorVisible READ masterVolumeNodeSelectorVisible WRITE setMasterVolumeNodeSelectorVisible)
  Q_PROPERTY(bool autoShowMasterVolumeNode READ autoShowMasterVolumeNode WRITE setAutoShowMasterVolumeNode)
  Q_PROPERTY(bool switchToSegmentationsButtonVisible READ switchToSegmentationsButtonVisible WRITE setSwitchToSegmentationsButtonVisible)
  Q_PROPERTY(bool undoEnabled READ undoEnabled WRITE setUndoEnabled)
  Q_PROPERTY(int maximumNumberOfUndoStates READ maximumNumberOfUndoStates WRITE setMaximumNumberOfUndoStates)
  Q_PROPERTY(bool readOnly READ readOnly WRITE setReadOnly)
  Q_PROPERTY(Qt::ToolButtonStyle effectButtonStyle READ effectButtonStyle WRITE setEffectButtonStyle)
  Q_PROPERTY(bool unorderedEffectsVisible READ unorderedEffectsVisible WRITE setUnorderedEffectsVisible)

public:
  typedef qMRMLWidget Superclass;
  explicit qMRMLSegmentEditorWidget(QWidget* parent = 0);
  virtual ~qMRMLSegmentEditorWidget();

  Q_INVOKABLE vtkMRMLSegmentEditorNode* mrmlSegmentEditorNode()const;

  Q_INVOKABLE vtkMRMLNode* segmentationNode()const;
  Q_INVOKABLE QString segmentationNodeID()const;
  Q_INVOKABLE vtkMRMLNode* masterVolumeNode()const;
  Q_INVOKABLE QString masterVolumeNodeID()const;

  Q_INVOKABLE QString currentSegmentID()const;

  Q_INVOKABLE qSlicerSegmentEditorAbstractEffect* activeEffect()const;
  Q_INVOKABLE void setActiveEffect(qSlicerSegmentEditorAbstractEffect* effect);

  Q_INVOKABLE qSlicerSegmentEditorAbstractEffect* effectByName(QString name);

  Q_INVOKABLE QStringList availableEffectNames();

  Q_INVOKABLE void setEffectNameOrder(const QStringList& effectNames);

  Q_INVOKABLE QStringList effectNameOrder() const;

  void setUnorderedEffectsVisible(bool visible);

  bool unorderedEffectsVisible() const;

  Q_INVOKABLE int effectCount();

  Q_INVOKABLE qSlicerSegmentEditorAbstractEffect* effectByIndex(int index);

  Q_INVOKABLE void setupViewObservations();

  Q_INVOKABLE void removeViewObservations();

  bool segmentationNodeSelectorVisible() const;
  bool masterVolumeNodeSelectorVisible() const;
  bool autoShowMasterVolumeNode() const;
  bool switchToSegmentationsButtonVisible() const;
  bool undoEnabled() const;
  int maximumNumberOfUndoStates() const;
  bool readOnly() const;

  Qt::ToolButtonStyle effectButtonStyle() const;

  Q_INVOKABLE void segmentationNodeSelectorAddAttribute(const QString& nodeType,
    const QString& attributeName,
    const QVariant& attributeValue = QVariant());
  Q_INVOKABLE void segmentationNodeSelectorRemoveAttribute(const QString& nodeType,
    const QString& attributeName);

  Q_INVOKABLE void masterVolumeNodeSelectorAddAttribute(const QString& nodeType,
    const QString& attributeName,
    const QVariant& attributeValue = QVariant());
  Q_INVOKABLE void masterVolumeNodeSelectorRemoveAttribute(const QString& nodeType,
    const QString& attributeName);

public slots:
  virtual void setMRMLScene(vtkMRMLScene* newScene);

  void setMRMLSegmentEditorNode(vtkMRMLSegmentEditorNode* newSegmentEditorNode);

  virtual void updateWidgetFromMRML();

  void setSegmentationNode(vtkMRMLNode* node);
  void setSegmentationNodeID(const QString& nodeID);
  void setMasterVolumeNode(vtkMRMLNode* node);
  void setMasterVolumeNodeID(const QString& nodeID);

  void setCurrentSegmentID(const QString segmentID);

  void setActiveEffectByName(QString effectName);

  void saveStateForUndo();

  void updateVolume(void* volumePtr, bool& success);

  void setSegmentationNodeSelectorVisible(bool);
  void setMasterVolumeNodeSelectorVisible(bool);
  void setAutoShowMasterVolumeNode(bool);
  void setSwitchToSegmentationsButtonVisible(bool);
  void setUndoEnabled(bool);
  void setMaximumNumberOfUndoStates(int);
  void setReadOnly(bool aReadOnly);
  void toggleMasterVolumeIntensityMaskEnabled();

  void undo();

  void redo();

  void installKeyboardShortcuts(QWidget* parent = NULL);

  void uninstallKeyboardShortcuts();

  bool turnOffLightboxes();

  void hideLabelLayer();

  void setEffectButtonStyle(Qt::ToolButtonStyle toolButtonStyle);

  void updateEffectLayouts();

  void showMasterVolumeInSliceViewers(bool forceShowInBackground = false, bool fitSlice = false);

signals:
  void currentSegmentIDChanged(const QString&);

  void masterVolumeNodeChanged(vtkMRMLVolumeNode*);

  void segmentationNodeChanged(vtkMRMLSegmentationNode*);

protected slots:
  void onSegmentationNodeChanged(vtkMRMLNode* node);
  void onMasterVolumeNodeChanged(vtkMRMLNode* node);
  void onSegmentSelectionChanged(const QItemSelection &selected, const QItemSelection &deselected);

  void onInteractionNodeModified();

  void onEffectButtonClicked(QAbstractButton* button);

  void onSelectEffectShortcut();

  void onSelectSegmentShortcut();

  void onAddSegment();
  void onRemoveSegment();
  void onSwitchToSegmentations();
  void onCreateSurfaceToggled(bool on);
  void onSegmentAddedRemoved();
  void onMasterVolumeImageDataModified();
  void onLayoutChanged(int layoutIndex);
  void onSegmentationDisplayModified();

  void onMaskModeChanged(int);

  void onMasterVolumeIntensityMaskChecked(bool checked);
  void onMasterVolumeIntensityMaskRangeChanged(double low, double high);

  void onOverwriteModeChanged(int);

  void onMRMLSceneEndCloseEvent();

  void initializeParameterSetNode();

  void onSegmentationHistoryChanged();

  void anchorClicked(const QUrl &url);

  void onSetSurfaceSmoothingClicked();
  void onImportExportActionClicked();

  void updateMaskingSection();

protected:
  static void processEvents(vtkObject* caller, unsigned long eid, void* clientData, void* callData);

  void updateWidgetFromSegmentationNode();
  void updateWidgetFromMasterVolumeNode();
  void updateEffectsSectionFromMRML();

  bool setMasterRepresentationToBinaryLabelmap();

  qSlicerAbstractModuleWidget* switchToSegmentationsModule();

protected:
  QScopedPointer<qMRMLSegmentEditorWidgetPrivate> d_ptr;

private:
  Q_DECLARE_PRIVATE(qMRMLSegmentEditorWidget);
  Q_DISABLE_COPY(qMRMLSegmentEditorWidget);
};

#endif