GLCanvas.h

/*
Copyright (c) 2016, Jochen Kempfle
All rights reserved.


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.


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 GLCANVAS_H
#define GLCANVAS_H

#include <map>
#include <wx/glcanvas.h>
#include "GLImage.h"
#include "Skeleton.h"
#include "Vector3.h"
#include "JointConstraint.h"

#if defined(__WXMAC__) || defined(__WXCOCOA__)
    #include <OpenGL/gl.h>
    #include <OpenGL/glu.h>
#else
    #include <GL/gl.h>
    #include <GL/glu.h>
#endif


enum GLCanvasStyle
{
    STANDARD = 0,
    HIGHLIGHT_SELECTED_BONE = 1,
    DRAW_GRID = 2,
    DRAW_LOCAL_COORDINATE_SYSTEM = 4,
    DRAW_SPIN_ARROWS = 8,
    SELECTION_MODE = 16,
    DRAW_SENSORS = 32,
    SINGLE_SENSOR_MODE = 64,
    SINGLE_JOINT_MODE = 128,
    DRAW_AABB = 256,
    DRAW_LABEL = 512,
    DRAW_ROTATION_AXIS = 1024,
    DRAW_JOINT_CONSTRAINTS = 2048
    // TODO(JK#2#2017-06-30): add render skeleton normal and special mode (special = hidden or colored bones)
};


class GLCanvas : public wxGLCanvas
{
  public:
    GLCanvas(wxWindow* parent, wxWindowID id = wxID_ANY, const wxPoint &pos = wxDefaultPosition,
             const wxSize &size = wxDefaultSize, int style = 0, const wxString &name = _(""), int* attribList = NULL);

    virtual ~GLCanvas();

    void renderSkeleton();
    void drawSpinArrows(Vector3 pos, Vector3 dir, Vector3 up, Vector3 right) const;
    void renderSingleSensor() const;
    void renderSingleJoint() const;
    void InitGL();
    int getObjectIdAt(const wxPoint& pos);
    void setSkeleton(Skeleton* skeleton);
    const Skeleton* getSkeleton();
    void setSensorOrientation(const Quaternion &orientation) { _sensorOrientation = orientation; }
    void setConstraint(JointConstraint* constraint) { _constraint = constraint; }

    void setStyle(unsigned int flags) { _style = flags; }
    void setStyleFlag(unsigned int flags) { _style |= flags; }
    void unsetStyleFlag(unsigned int flags) { _style &= ~flags; }
    unsigned int getStyle() const { return _style; }

  private:
    bool init;
    wxGLContext* _GLRC;

    void OnPaint(wxPaintEvent &event);
    void OnSize(wxSizeEvent &event);
    void OnEraseBackground(wxEraseEvent &event);
    void OnEnterWindow(wxMouseEvent &event);
    void OnLeaveWindow(wxMouseEvent &event);
    void OnLeftDown(wxMouseEvent &event);
    void OnLeftUp(wxMouseEvent &event);
    void OnRightDown(wxMouseEvent &event);
    void OnRightUp(wxMouseEvent &event);
    void OnMouseMove(wxMouseEvent &event);
    void OnMouseWheel(wxMouseEvent &event);
    void OnMouseCaptureLost(wxMouseCaptureLostEvent& event);

    void drawUserInterface(wxDC &dc) const;

    void drawGrid() const;
    void drawJoint(const JointConstraint &constraint = JointConstraint(), float radius = 0.04f) const;
    void drawAABB(const AABB &box) const;

    wxPoint _prevMousePos;
    bool _lClicked;
    bool _rClicked;
    float _xRotation;
    float _yRotation;
    const float _cameraSpeed = 0.004f;
    Vector3 _cameraPosition;
    Vector3 _cameraFront;
    Vector3 _cameraUp;
    Vector3 _cameraRight;

    bool _showUI;

    wxColour _backgroundColor;
    float _lineWidth;

    unsigned int _traceLength;
    unsigned int _tracePos;
    std::map<int, std::vector<Vector3> > _boneIdsWithTracePoints;

    Skeleton* _skeleton;
    Quaternion _sensorOrientation;
    std::map<int, GLImage*> _labels;

    JointConstraint* _constraint;
    unsigned int _style;

    GLUquadric* _sphereQuad;

    const int _numButtons = 8;
    const int _buttonSize = 20;
    const int _buttonSizeHalf = _buttonSize/2;

    const GLubyte red[4] = {255, 0, 0, 255};
    const GLubyte green[4] = {0, 255, 0, 255};
    const GLubyte blue[4] = {0, 0, 255, 255};
    const GLubyte yellow[4] = {255, 255, 0, 255};
    const GLubyte white[4] = {255, 255, 255, 255};
    const GLubyte black[4] = {0, 0, 0, 255};

    const GLubyte pointColor[4] = {5, 15, 160, 255};
    const GLubyte boneStandardColor1[4] = {128, 128, 128, 255};
    const GLubyte boneStandardColor2[4] = {180, 180, 180, 255};
    const GLubyte boneHighlightedColor1[4] = {100, 200, 100, 255};
    const GLubyte boneHighlightedColor2[4] = {160, 255, 160, 255};

	DECLARE_EVENT_TABLE()
};

#endif // GLCANVAS_H