visu/visuVTKAdaptor/include/visuVTKAdaptor/SLandmarks.hpp

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

#ifndef __VISUVTKADAPTOR_SLANDMARKS_HPP__
#define __VISUVTKADAPTOR_SLANDMARKS_HPP__

#include "visuVTKAdaptor/config.hpp"

#include <fwData/Landmarks.hpp>

#include <fwRenderVTK/IAdaptor.hpp>
#include <fwRenderVTK/vtk/fwHandleRepresentation3D.hpp>

#include <fwThread/Timer.hpp>

#include <vtkHandleWidget.h>
#include <vtkSmartPointer.h>

class vtkCommand;
class vtkActor2D;

namespace visuVTKAdaptor
{

class VISUVTKADAPTOR_CLASS_API SLandmarks : public ::fwRenderVTK::IAdaptor
{

public:

    fwCoreServiceClassDefinitionsMacro( (SLandmarks)(::fwRenderVTK::IAdaptor) );
    fwCoreAllowSharedFromThis();

    typedef vtkSmartPointer< vtkHandleWidget > LandmarkWidgetType;

    VISUVTKADAPTOR_API SLandmarks() noexcept;

    VISUVTKADAPTOR_API virtual ~SLandmarks() noexcept;

    VISUVTKADAPTOR_API virtual void show(bool b = true);

    void deselect();

protected:

    VISUVTKADAPTOR_API void configuring() override;
    VISUVTKADAPTOR_API void starting() override;
    VISUVTKADAPTOR_API void updating() override;
    VISUVTKADAPTOR_API void stopping() override;

    VISUVTKADAPTOR_API virtual KeyConnectionsMap getAutoConnections() const override;

private:

    typedef std::vector< LandmarkWidgetType > LandmarksWidgetContainerType;

    typedef std::map< std::string, LandmarksWidgetContainerType > GroupWidgetsMapType;

    typedef std::map< LandmarkWidgetType, vtkActor2D* > WidgetLabelMapType;

    typedef std::map< LandmarkWidgetType, vtkCommand* > WidgetCommandMapType;

    typedef std::pair< std::string, size_t > PointPairType;

    void addPoint(std::string groupName);

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

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

    void modifyGroup(std::string groupName);

    void removeGroup(std::string groupName);

    void addGroup(std::string groupName);

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

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

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

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

    void changeColor(const vtkSmartPointer< ::fwRenderVTK::vtk::fwHandleRepresentation3D >& rep,
                     const std::array<double, 3>& color1, const std::array<double, 3>& color2);

    void clearLandmarks();

    vtkSmartPointer<vtkHandleWidget> newHandle(const ::fwData::Landmarks::sptr& landmarks,
                                               const std::string& groupName,
                                               size_t pointIndex);

    vtkCommand* m_noSelectionCommand;

    GroupWidgetsMapType m_handles;

    WidgetLabelMapType m_labels;

    WidgetCommandMapType m_commands;

    ::fwThread::Timer::sptr m_timer;

    PointPairType m_selectedPoint;

    size_t m_count;

    bool m_interaction;

};

} //namespace visuVTKAdaptor

#endif // __VISUVTKADAPTOR_SLANDMARKS_HPP__