git.sur5r.net Git - fstl/blob - src/canvas.cpp

   1 #include <QMouseEvent>
   2 #include <QDebug>
   3
   4 #include <cmath>
   5
   6 #include "canvas.h"
   7 #include "backdrop.h"
   8 #include "glmesh.h"
   9 #include "mesh.h"
  10
  11 Canvas::Canvas(const QGLFormat& format, QWidget *parent)
  12     : QGLWidget(format, parent), mesh(NULL),
  13       scale(1), tilt(90), yaw(0)
  14 {
  15     // Nothing to do here
  16 }
  17
  18 Canvas::~Canvas()
  19 {
  20     delete mesh;
  21 }
  22
  23 void Canvas::load_mesh(Mesh* m)
  24 {
  25     mesh = new GLMesh(m);
  26     center = QVector3D(m->xmin() + m->xmax(),
  27                        m->ymin() + m->ymax(),
  28                        m->zmin() + m->zmax()) / 2;
  29     scale = 2 / sqrt(
  30                 pow(m->xmax() - m->xmin(), 2) +
  31                 pow(m->ymax() - m->ymin(), 2) +
  32                 pow(m->zmax() - m->zmin(), 2));
  33     update();
  34
  35     delete m;
  36 }
  37
  38 void Canvas::set_status(const QString &s)
  39 {
  40     status = s;
  41     update();
  42 }
  43
  44 void Canvas::initializeGL()
  45 {
  46     mesh_shader.addShaderFromSourceFile(QGLShader::Vertex, ":/gl/mesh.vert");
  47     mesh_shader.addShaderFromSourceFile(QGLShader::Fragment, ":/gl/mesh.frag");
  48     mesh_shader.link();
  49
  50     backdrop = new Backdrop();
  51 }
  52
  53 void Canvas::paintEvent(QPaintEvent *event)
  54 {
  55     Q_UNUSED(event);
  56
  57     glClearColor(0.0, 0.0, 0.0, 0.0);
  58     glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
  59     glEnable(GL_DEPTH_TEST);
  60
  61     backdrop->draw();
  62     if (mesh)  draw_mesh();
  63
  64     QPainter painter(this);
  65     painter.setRenderHint(QPainter::Antialiasing);
  66     if (!status.isNull())
  67         painter.drawText(10, height() - 10, status);
  68 }
  69
  70
  71 void Canvas::draw_mesh()
  72 {
  73     mesh_shader.bind();
  74
  75     // Load the transform and view matrices into the shader
  76     glUniformMatrix4fv(
  77                 mesh_shader.uniformLocation("transform_matrix"),
  78                 1, GL_FALSE, transform_matrix().data());
  79     glUniformMatrix4fv(
  80                 mesh_shader.uniformLocation("view_matrix"),
  81                 1, GL_FALSE, view_matrix().data());
  82
  83     // Find and enable the attribute location for vertex position
  84     const GLuint vp = mesh_shader.attributeLocation("vertex_position");
  85     glEnableVertexAttribArray(vp);
  86
  87     // Then draw the mesh with that vertex position
  88     mesh->draw(vp);
  89
  90     // Clean up state machine
  91     glDisableVertexAttribArray(vp);
  92     mesh_shader.release();
  93 }
  94
  95 QMatrix4x4 Canvas::transform_matrix() const
  96 {
  97     QMatrix4x4 m;
  98     m.rotate(tilt, QVector3D(1, 0, 0));
  99     m.rotate(yaw,  QVector3D(0, 0, 1));
 100     m.scale(scale);
 101     m.translate(-center);
 102     return m;
 103 }
 104
 105 QMatrix4x4 Canvas::view_matrix() const
 106 {
 107     QMatrix4x4 m;
 108     if (width() > height())
 109     {
 110         m.scale(height() / float(width()), 1, 0.5);
 111     }
 112     else
 113     {
 114         m.scale(1, width() / float(height()), 0.5);
 115     }
 116     return m;
 117 }
 118
 119 void Canvas::mousePressEvent(QMouseEvent* event)
 120 {
 121     if (event->button() == Qt::LeftButton)
 122     {
 123         mouse_pos = event->pos();
 124         setCursor(Qt::ClosedHandCursor);
 125     }
 126 }
 127
 128 void Canvas::mouseReleaseEvent(QMouseEvent* event)
 129 {
 130     if (event->button() == Qt::LeftButton)
 131     {
 132         unsetCursor();
 133     }
 134 }
 135
 136 void Canvas::mouseMoveEvent(QMouseEvent* event)
 137 {
 138     if (event->buttons() & Qt::LeftButton)
 139     {
 140         auto p = event->pos();
 141         auto d = p - mouse_pos;
 142         yaw = fmod(yaw + d.x(), 360);
 143         tilt = fmax(0, fmin(180, tilt - d.y()));
 144         mouse_pos = p;
 145         update();
 146     }
 147 }