ui/uiMeasurementQt/include/uiMeasurementQt/editor/SLandmarks.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2017-2018.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#pragma once

#include "uiMeasurementQt/config.hpp"

#include <fwData/Landmarks.hpp>

#include <fwDataTools/PickingInfo.hpp>

#include <fwGui/editor/IEditor.hpp>

#include <QCheckBox>
#include <QComboBox>
#include <QObject>
#include <QPointer>
#include <QPushButton>
#include <QSlider>
#include <QTreeWidget>

namespace uiMeasurementQt
{
namespace editor
{

class UIMEASUREMENTQT_CLASS_API SLandmarks : public QObject,
                                             public ::fwGui::editor::IEditor
{

Q_OBJECT

public:

    fwCoreServiceClassDefinitionsMacro( (SLandmarks)(::fwGui::editor::IEditor) );

    UIMEASUREMENTQT_API SLandmarks() noexcept;

    UIMEASUREMENTQT_API virtual ~SLandmarks() noexcept;

    UIMEASUREMENTQT_API virtual KeyConnectionsMap getAutoConnections() const override;

protected:

    typedef ::fwRuntime::ConfigurationElement::sptr Configuration;

    virtual void starting() override;

    virtual void stopping() override;

    virtual void updating() override;

    virtual void configuring() override;

private:

    void onColorButton();

    void onGroupNameEdited(QTreeWidgetItem* item, int column);

    void onSelectionChanged(QTreeWidgetItem* current, QTreeWidgetItem* previous);

    void onSizeChanged(int newSize);

    void onOpacityChanged(int newOpacity);

    void onVisibilityChanged(int visibility);

    void onShapeChanged(const QString& shape);

    void onAddNewGroup();

    void onRemoveSelection();

    void addPickedPoint(::fwDataTools::PickingInfo pickingInfo);

    void addPoint(std::string groupName);

    void modifyPoint(std::string groupName, size_t index);

    void selectPoint(std::string groupName, size_t index);

    void deselectPoint(std::string groupName, size_t index);

    void addGroup(std::string name);

    void removeGroup(std::string name);

    void removePoint(std::string groupName, size_t index);

    void renameGroup(std::string oldName, std::string newName);

    void modifyGroup(std::string name);

    bool currentSelection(std::string& selection) const;

    QTreeWidgetItem* getGroupItem(const std::string& groupName) const;

    std::string generateNewGroupName() const;

    std::array<float, 4> generateNewColor();

    static QColor convertToQColor(const ::fwData::Landmarks::ColorType& color);

    static void setColorButtonIcon(QPushButton* button, const QColor& color);

    QPointer<QTreeWidget> m_treeWidget;

    QPointer<QWidget> m_groupEditorWidget;

    QPointer<QSlider> m_sizeSlider;

    QPointer<QSlider> m_opacitySlider;

    QPointer<QCheckBox> m_visibilityCheckbox;

    QPointer<QComboBox> m_shapeSelector;

    QPointer<QPushButton> m_newGroupButton;

    QPointer<QPushButton> m_removeButton;

    bool m_advancedMode;

    float m_defaultLandmarkSize;

    float m_defaultLandmarkOpacity;

};
} // namespace editor
} // uiMeasurementQt