MouseAndKeyPlugin.cc

//=============================================================================
//
//                               OpenFlipper
//        Copyright (C) 2008 by Computer Graphics Group, RWTH Aachen
//                           www.openflipper.org
//
//-----------------------------------------------------------------------------
//
//                                License
//
//  OpenFlipper is free software: you can redistribute it and/or modify
//  it under the terms of the GNU Lesser General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
//
//  OpenFlipper is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU Lesser General Public License for more details.
//
//  You should have received a copy of the GNU Lesser General Public License
//  along with OpenFlipper.  If not, see <http://www.gnu.org/licenses/>.
//
//-----------------------------------------------------------------------------
//
//
//=============================================================================

#include "MouseAndKeyPlugin.hh"


#include <ObjectTypes/PolyMesh/PolyMesh.hh>
#include <ObjectTypes/TriangleMesh/TriangleMesh.hh>

#include "OpenFlipper/BasePlugin/PluginFunctions.hh"

/*
 * Constructor
 */
MouseAndKeyPlugin::MouseAndKeyPlugin() :
        contextMenuEntry_(0),
        tool_(0),
        pickButton_(0),
        activeObject_(-1),
        axis_x_(ACG::Vec3d(1.0, 0.0, 0.0)),
        axis_y_(ACG::Vec3d(0.0, 1.0, 0.0))
{

}

/*
 * Initialize plugin
 */
void MouseAndKeyPlugin::initializePlugin() {

        // Register keys
        emit registerKey(Qt::Key_J,     Qt::NoModifier, "Rotate object down");
        emit registerKey(Qt::Key_K,     Qt::NoModifier, "Rotate object up");
        emit registerKey(Qt::Key_H,     Qt::NoModifier, "Rotate object left");
        emit registerKey(Qt::Key_L,     Qt::NoModifier, "Rotate object right");

  tool_ = new QWidget();
  QSize size(300, 300);
  tool_->resize(size);

  // Create button that can be toggled
  // to (de)activate plugin's picking mode
  pickButton_ = new QPushButton(tr("Select object"));
  pickButton_->setCheckable(true);

  // Create label
  QLabel* label = new QLabel();
  label->setText("(De)activate pick mode");

  // Set label to be above the button
  QGridLayout* grid = new QGridLayout;
  grid->addWidget(label, 0, 0);
  grid->addWidget(pickButton_, 1, 0);

  tool_->setLayout(grid);

  // Connect button to slotButtonClicked()
  connect(pickButton_, SIGNAL(clicked()), this, SLOT(slotButtonClicked()));

  // Add the Toolbox
  emit addToolbox(tr("Mouse and Key"), tool_);

} // End initializePlugin

/*
 * Is called after all plugins have been initialized
 */
void MouseAndKeyPlugin::pluginsInitialized() {

        // Add pickmode
        emit addPickMode("MouseAndKeyPlugin");

        // Add context menu entry

        // Create submenu
        contextMenuEntry_ = new QMenu("Mouse and key plugin");

        QAction* lastAction;

        // Add action to recently created menu
        lastAction = contextMenuEntry_->addAction( "Hide object" );
        lastAction->setToolTip("Hide selected object");
        lastAction->setStatusTip( lastAction->toolTip() );

        // Finally insert created context submenu to OpenFlipper's context menu
        // We want our action to be visible for triangle and polygon meshes
        emit addContextMenuItem(contextMenuEntry_->menuAction() , DATA_TRIANGLE_MESH , CONTEXTOBJECTMENU );
        emit addContextMenuItem(contextMenuEntry_->menuAction() , DATA_POLY_MESH , CONTEXTOBJECTMENU );

        // Connect the created context menu entry to local function contextMenuItemSelected(QAction*)
        connect(contextMenuEntry_, SIGNAL(triggered(QAction*)), this, SLOT(contextMenuItemSelected(QAction*)));

} // End pluginsInitialized


/*
 * Is called when button in toolbox has been clicked
 */
void MouseAndKeyPlugin::slotButtonClicked() {

        if(pickButton_->isChecked()) {
                // Picking mode of our plugin shall be activated
                // set OpenFlipper's action mode to picking
                PluginFunctions::actionMode( Viewer::PickingMode );

                // Now activate our picking mode
                PluginFunctions::pickMode( "MouseAndKeyPlugin" );
        } else {
                // Picking mode shall be deactivated
                PluginFunctions::actionMode( Viewer::ExamineMode );
        }

} // End slotButtonClicked

/*
 * Is called when pick mode is changed in OpenFlipper
 */
void MouseAndKeyPlugin::slotPickModeChanged(const std::string& _mode) {

        // Set button checked if pick mode is our
        // plugin's pick mode
        pickButton_->setChecked(_mode == "MouseAndKeyPlugin");

} // End slotPickModeChanged

/*
 * Is called each time the mouse has moved or been clicked
 */
void MouseAndKeyPlugin::slotMouseEvent(QMouseEvent* _event) {

        if ( PluginFunctions::pickMode() == "MouseAndKeyPlugin" &&
                PluginFunctions::actionMode() == Viewer::PickingMode ) {

                // If double click has been performed
                if (_event->type() == QEvent::MouseButtonDblClick) {

                        size_t node_idx, target_idx;
                        OpenMesh::Vec3d hitPoint;

                        // Get picked object's identifier
                        if ( PluginFunctions::scenegraphPick(ACG::SceneGraph::PICK_ANYTHING,_event->pos(), node_idx, target_idx, &hitPoint) ) {

                                BaseObjectData* object;

                                // Get picked object
                                if ( PluginFunctions::getPickedObject(node_idx, object) ) {

                                        // Show small dialog window
                                        QDialog* dlg = new QDialog;

                                        QGridLayout* grid = new QGridLayout;
                                        QLabel* label = new QLabel;
                                        QString str = QString("You selected object ");
                                        str += QString::number(node_idx);
                                        label->setText(str);
                                        grid->addWidget(label, 0,0);

                                        dlg->setLayout(grid);

                                        dlg->show();

                                        // Set last selected object
                                        activeObject_ = node_idx;
                                }
                                else {
                                        emit log(LOGINFO, "Picking failed");
                                }
                        }

                        return;
                }

                // Continue traversing scene graph
                ACG::SceneGraph::MouseEventAction action(_event, PluginFunctions::viewerProperties().glState());
                PluginFunctions::traverse(action);
        }

} // End slotMouseEvent

/*
 * Is called when a key on the keyboard is pressed
 */
void MouseAndKeyPlugin::slotKeyEvent( QKeyEvent* _event ) {

        BaseObjectData* object;

        // Get last clicked object (selected in pick mode)
        if ( PluginFunctions::getPickedObject(activeObject_, object) ) {

                // Switch pressed keys
                switch (_event->key())
                {
                        case Qt::Key_J:

                                object->manipulatorNode()->loadIdentity();
                                object->manipulatorNode()->rotate(10.0, axis_x_);

                                break;

                        case Qt::Key_K :

                                object->manipulatorNode()->loadIdentity();
                                object->manipulatorNode()->rotate(-10.0, axis_x_);

                                break;

                        case Qt::Key_H :

                                object->manipulatorNode()->loadIdentity();
                                object->manipulatorNode()->rotate(10.0, axis_y_);

                                break;

                        case Qt::Key_L :

                                object->manipulatorNode()->loadIdentity();
                                object->manipulatorNode()->rotate(-10.0, axis_y_);

                                break;

                        default:
                        break;
                }

                // Perform rotation
                if ( object->dataType( DATA_TRIANGLE_MESH ) )
                        transformMesh(object->manipulatorNode()->matrix(), (*PluginFunctions::triMesh(object)));
                if ( object->dataType( DATA_POLY_MESH ) )
                        transformMesh(object->manipulatorNode()->matrix(), (*PluginFunctions::polyMesh(object)));

                // Tell core that object has been modified
                emit updatedObject(object->id(), UPDATE_GEOMETRY);

                // Update view
                emit updateView();
        } else {
                emit log(LOGINFO, "No object has been selected to rotate! Select object first.");
        }

} // End slotKeyEvent

/*
 * Transform a mesh with the given transformation matrix
 *
 * _mat : transformation matrix
 * _mesh : the mesh
 */
template<typename MeshT>
void MouseAndKeyPlugin::transformMesh(ACG::Matrix4x4d _mat, MeshT& _mesh) {

        typename MeshT::VertexIter v_it = _mesh.vertices_begin();
        typename MeshT::VertexIter v_end = _mesh.vertices_end();

        // Iterator over all vertices and transform them by _mat
        // Update normals
        for (; v_it != v_end; ++v_it) {
                _mesh.set_point(*v_it, (typename MeshT::Point) _mat.transform_point(
                                (OpenMesh::Vec3d)(_mesh.point(*v_it))));
                _mesh.set_normal(*v_it, (typename MeshT::Point) _mat.transform_vector(
                                (OpenMesh::Vec3d)(_mesh.normal(*v_it))));
        }

        typename MeshT::FaceIter f_it = _mesh.faces_begin();
        typename MeshT::FaceIter f_end = _mesh.faces_end();

        // Iterate over all faces and update face normals
        for (; f_it != f_end; ++f_it)
                _mesh.set_normal(*f_it, (typename MeshT::Point) _mat.transform_vector(
                                (OpenMesh::Vec3d)(_mesh.normal(*f_it))));

} // End transformMesh

/*
 * Is called when context menu entry has been clicked
 */
void MouseAndKeyPlugin::contextMenuItemSelected(QAction* _action) {

        // Get object id from QAction object
        QVariant contextObject = _action->data();
        int objectId = contextObject.toInt();

        if (objectId == -1) {

                log(LOGINFO, "Could not get associated object id.");
                return;
        }

        // Get node associated to object id
        ACG::SceneGraph::BaseNode* node = ACG::SceneGraph::find_node(
                        PluginFunctions::getSceneGraphRootNode(), objectId);

        // Return if node id was not valid
        if (!node) {

                log(LOGINFO, "Could not find associated object.");
                return;
        }

        BaseObjectData* obj;
        if (!PluginFunctions::getObject(objectId, obj))
                return;

        // Hide object
        obj->hide();

} // End contextMenuItemSelected