VolumeMeshSelectionPlugin.hh

/*===========================================================================*\
 *                                                                           *
 *                              OpenFlipper                                  *
 *           Copyright (c) 2001-2015, RWTH-Aachen University                 *
 *           Department of Computer Graphics and Multimedia                  *
 *                          All rights reserved.                             *
 *                            www.openflipper.org                            *
 *                                                                           *
 *---------------------------------------------------------------------------*
 * This file is part of OpenFlipper.                                         *
 *---------------------------------------------------------------------------*
 *                                                                           *
 * Redistribution and use in source and binary forms, with or without        *
 * modification, are permitted provided that the following conditions        *
 * are met:                                                                  *
 *                                                                           *
 * 1. Redistributions of source code must retain the above copyright notice, *
 *    this list of conditions and the following disclaimer.                  *
 *                                                                           *
 * 2. Redistributions in binary form must reproduce the above copyright      *
 *    notice, this list of conditions and the following disclaimer in the    *
 *    documentation and/or other materials provided with the distribution.   *
 *                                                                           *
 * 3. Neither the name of the copyright holder nor the names of its          *
 *    contributors may be used to endorse or promote products derived from   *
 *    this software without specific prior written permission.               *
 *                                                                           *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS       *
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED *
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A           *
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER *
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,  *
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,       *
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR        *
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF    *
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING      *
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS        *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.              *
 *                                                                           *
\*===========================================================================*/
 
#ifndef VOLUMEMESHSELECTIONPLUGIN_HH
#define VOLUMEMESHSELECTIONPLUGIN_HH
 
#include <QObject>
 
#include <OpenFlipper/BasePlugin/BaseInterface.hh>
#include <OpenFlipper/BasePlugin/KeyInterface.hh>
#include <OpenFlipper/BasePlugin/LoggingInterface.hh>
#include <OpenFlipper/BasePlugin/ToolboxInterface.hh>
#include <OpenFlipper/BasePlugin/ToolbarInterface.hh>
#include <OpenFlipper/BasePlugin/LoadSaveInterface.hh>
#include <OpenFlipper/BasePlugin/BackupInterface.hh>
#include <OpenFlipper/BasePlugin/INIInterface.hh>
#include <OpenFlipper/BasePlugin/ScriptInterface.hh>
#include <OpenFlipper/BasePlugin/SelectionInterface.hh>
#include <OpenFlipper/common/Types.hh>
#include <OpenFlipper/BasePlugin/PluginFunctions.hh>
#include <OpenFlipper/INIFile/INIFile.hh>
 
#include <ObjectTypes/PolyhedralMesh/PolyhedralMesh.hh>
#include <ObjectTypes/HexahedralMesh/HexahedralMesh.hh>
#ifdef ENABLE_TETRAHEDRALMESH_SUPPORT
#include <ObjectTypes/TetrahedralMesh/TetrahedralMesh.hh>
#endif
 
class VolMeshParamWidget;
 
class VolumeMeshSelectionPlugin : public QObject, BaseInterface, KeyInterface,
            INIInterface, BackupInterface, LoggingInterface, ScriptInterface, SelectionInterface
{
    Q_OBJECT
    Q_INTERFACES(BaseInterface)
    Q_INTERFACES(KeyInterface)
    Q_INTERFACES(INIInterface)
    Q_INTERFACES(BackupInterface)
    Q_INTERFACES(LoggingInterface)
    Q_INTERFACES(ScriptInterface)    
    Q_INTERFACES(SelectionInterface)
 
  Q_PLUGIN_METADATA(IID "org.OpenFlipper.Plugins.Plugin-SelectionVolumeMesh")
 
public:
    
    VolumeMeshSelectionPlugin();
 
    ~VolumeMeshSelectionPlugin();
    
    friend class SelectVolumeAction;
 
    void   set_max_angle(const double _a);
    double get_max_angle();
 
signals:
 
    // BaseInterface
    void updateView() override;
    void updatedObject(int, const UpdateType&) override;
    void nodeVisibilityChanged(int _identifier) override;
    void setSlotDescription(QString     _slotName,   QString     _slotDescription,
                            QStringList _parameters, QStringList _descriptions) override;
 
    // LoggingInterface
    void log(Logtype _type, QString _message) override;
    void log(QString _message) override;
 
    // ScriptInterface
    void scriptInfo(QString _functionName);
 
    // SelectionInterface
    void addSelectionEnvironment(QString _modeName, QString _description, QString _icon, QString& _handleName) override;
    void registerType(QString _handleName, DataType _type) override;
    void addPrimitiveType(QString _handleName, QString _name, QString _icon, SelectionInterface::PrimitiveType& _typeHandle) override;
    void addSelectionOperations(QString _handleName, QStringList _operationsList, QString _category, SelectionInterface::PrimitiveType _type = 0u) override;
    void addSelectionParameters(QString _handleName, QWidget* _widget, QString _category, SelectionInterface::PrimitiveType _type = 0u) override;
 
    void showToggleSelectionMode(QString _handleName, bool _show, SelectionInterface::PrimitiveType _associatedTypes) override;
 
    void showVolumeLassoSelectionMode(QString _handleName, bool _show, SelectionInterface::PrimitiveType _associatedTypes) override;
    void showFloodFillSelectionMode(QString _handleName, bool _show, SelectionInterface::PrimitiveType _associatedTypes) override;
    void addCustomSelectionMode(QString _handleName, QString _modeName, QString _description, QString _icon,
           SelectionInterface::PrimitiveType _associatedTypes, QString& _customIdentifier) override;
 
    void getActiveDataTypes(SelectionInterface::TypeList& _types) override;
    void getActivePrimitiveType(SelectionInterface::PrimitiveType& _type) override;
    void targetObjectsOnly(bool& _targetsOnly) override;
    
    void registerKeyShortcut(int _key, Qt::KeyboardModifiers _modifiers = Qt::NoModifier) override;
 
private slots:
 
    // INIInterface
    void loadIniFile(INIFile& _ini, int _id) override;
    void saveIniFile(INIFile& _ini, int _id) override;
 
    // BaseInterface
    void initializePlugin() override;
    void pluginsInitialized() override;
    void noguiSupported() override {};
 
    // SelectionInterface
    void slotSelectionOperation(QString _operation) override;
    void slotToggleSelection(QMouseEvent* _event, SelectionInterface::PrimitiveType _currentType, bool _deselect) override;
 
    void slotVolumeLassoSelection(QMouseEvent* _event, SelectionInterface::PrimitiveType _currentType, bool _deselect) override;
    void slotFloodFillSelection(QMouseEvent* _event, SelectionInterface::PrimitiveType _currentType, bool _deselect) override;
    void slotCustomSelection(QMouseEvent *_event, SelectionInterface::PrimitiveType _currentType, QString _customIdentifier, bool _deselect) override;
 
    void slotLoadSelection(const INIFile& _file) override;
    void slotSaveSelection(INIFile& _file) override;
    
    void slotKeyShortcutEvent(int _key, Qt::KeyboardModifiers _modifiers) override;
 
public slots:
 
    // SelectionInterface
    void loadSelection(int _objId, const QString& _filename) override;
 
public:
 
    // BaseInterface
    QString name() override {
        return (QString(tr("VolumeMesh Selection")));
    };
    
    QString description() override {
        return (QString(tr("Allows to select VolumeMeshes")));
    };
 
    //===========================================================================
    //===========================================================================
private:
    
    void updateSlotDescriptions();
    
public slots:
 
    QString version() override {
        return QString("1.0");
    };
 
    //===========================================================================
    //===========================================================================
 
    bool vertexTypeActive();
 
    bool edgeTypeActive();
 
    bool faceTypeActive();
 
    bool cellTypeActive();
 
    //===========================================================================
    
    void selectAllVertices(int _objectId);
 
    void deselectAllVertices(int _objectId);
 
    void invertVertexSelection(int _objectId);
 
    void selectVertices(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getVertexSelection(int _objectId);
 
    void deleteSelectedVertices(int _objectId, bool _preserveManifoldness = true);
 
    //===========================================================================
 
    void selectAllEdges(int _objectId);
 
    void deselectAllEdges(int _objectId);
 
    void invertEdgeSelection(int _objectId);
 
    void selectEdges(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getEdgeSelection(int _objectId);
 
    void deleteSelectedEdges(int _objectId, bool _preserveManifoldness = true);
 
    //===========================================================================
 
    void selectAllHalfEdges(int _objectId);
 
    void deselectAllHalfEdges(int _objectId);
 
    void invertHalfEdgeSelection(int _objectId);
 
    void selectHalfEdges(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getHalfEdgeSelection(int _objectId);
 
    //===========================================================================
 
    void selectAllFaces(int _objectId);
 
    void deselectAllFaces(int _objectId);
 
    void invertFaceSelection(int _objectId);
 
    void selectFaces(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getFaceSelection(int _objectId);
 
    void deleteSelectedFaces(int _objectId, bool _preserveManifoldness = true);
 
    //===========================================================================
 
    void selectAllHalfFaces(int _objectId);
 
    void deselectAllHalfFaces(int _objectId);
 
    void invertHalfFaceSelection(int _objectId);
 
    void selectHalfFaces(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getHalfFaceSelection(int _objectId);
 
    //===========================================================================
 
    void selectAllCells(int _objectId);
 
    void deselectAllCells(int _objectId);
 
    void invertCellSelection(int _objectId);
 
    void selectCells(int _objectId, const IdList& _ids, bool _deselect = false);
 
    IdList getCellSelection(int _objectId);
 
    void deleteSelectedCells(int _objectId, bool _preserveManifoldness = true);
 
    //===========================================================================
    //===========================================================================
 
private:
    template<class MeshT>
    bool volumeSelection(MeshT* _mesh, ACG::GLState& _state, QRegion *_region,
                         PrimitiveType _primitiveTypes, bool _deselection);
 
    template<class MeshT>
    void floodFillSelection(MeshT* _mesh, uint _fh, double _maxAngle,
                            PrimitiveType _primitiveTypes, bool _deselection);
 
    // Get orthogonal orientation of neighboring cell
    unsigned char getOrthogonalOrientationOfNeighborCell(const OpenVolumeMesh::CellHandle& _ch1,
            const OpenVolumeMesh::CellHandle& _ch2, unsigned char _firstOrthDirection, const HexahedralMesh* _mesh) const;
 
    // Get face shared by two adjacent cells
    typedef std::pair<OpenVolumeMesh::HalfFaceHandle,OpenVolumeMesh::HalfFaceHandle> HFPair;
    HFPair getCommonFace(const OpenVolumeMesh::CellHandle& _ch1,
            const OpenVolumeMesh::CellHandle& _ch2, const HexahedralMesh* _mesh) const;
 
 
    OpenVolumeMesh::StatusAttrib* getStatus(int _objectId);
    OpenVolumeMesh::StatusAttrib* getStatus(BaseObjectData* object);
 
    QString environmentHandle_;
 
    QString columnSelectionHandle_;
    QString sheetSelectionHandle_;
 
    PrimitiveType vertexType_;
    PrimitiveType edgeType_;
    PrimitiveType faceType_;
    PrimitiveType cellType_;
    
    PrimitiveType allSupportedTypes_;
    PrimitiveType floodFillSupportedTypes_;
 
    VolMeshParamWidget* parameterWidget_;
 
    double max_angle_;
    
    QVector<QPoint> volumeLassoPoints_;
 
    // Store last picked cell for sheet selection
    OpenVolumeMesh::CellHandle lastPickedCell_;
    unsigned char lastPickedOrientation_;
    std::map<OpenVolumeMesh::CellHandle, unsigned char> orientationMap_;
 
};
 
class SelectVolumeAction
{
public:
    SelectVolumeAction(QRegion&_region, VolumeMeshSelectionPlugin* _plugin,
            unsigned int _type, bool _deselection, ACG::GLState& _state)
        : state_(_state)
        , region_(_region)
        , plugin_(_plugin)
        , type_(_type)
        , deselection_(_deselection)
    {
    }
 
    void enter(BaseNode* /*_node*/) {}
    void leave(BaseNode* /*_node*/) {}
 
    bool operator()(BaseNode* _node);
 
private:
    ACG::GLState&               state_;
    QRegion&                    region_;
    VolumeMeshSelectionPlugin*  plugin_;
    unsigned int                type_;
    bool                        deselection_;
};
//=============================================================================
#if defined(INCLUDE_TEMPLATES) && !defined(VOLUMEMESHSELECTIONPLUGINT_CC)
#define VOLUMEMESHSELECTIONPLUGINT_TEMPLATES
#include "VolumeMeshSelectionPluginT_impl.hh"
#endif
 
//=============================================================================
#endif // VOLUMEMESHSELECTIONPLUGIN_HH defined
//=============================================================================