qsrc/terraindisplay.cpp

   1 #include "terraindisplay.h"
   2
   3 #include <QtGui/QMessageBox>
   4 #include <QtGui/QMouseEvent>
   5 //#include <QtScript/QScriptEngine>
   6 //#include <QtCore/QDebug>
   7
   8 #include <iostream>
   9
  10 #include <GL/gl.h>
  11 #include <math.h>
  12 #include <stdlib.h>
  13 #include <time.h>
  14
  15 #define DEFAULT_SIZE 64
  16
  17 TerrainDisplay::TerrainDisplay(QWidget *parent) : QGLWidget(parent)
  18 {
  19     setSizePolicy(QSizePolicy(QSizePolicy::Expanding, QSizePolicy::Expanding));
  20
  21     m_mesh = terrain_create(DEFAULT_SIZE);
  22
  23     setZoom(10.0);
  24
  25     m_mouseSensitivity = 0.5;
  26
  27     setXRotation(45.0);
  28     setYRotation(45.0);
  29     emit zRotationChanged(0.0); // to fill in the UI
  30
  31     makeCurrent();
  32
  33     setSize(DEFAULT_SIZE);
  34     setSeed(time(0));
  35     setMagnitude(DEFAULT_SIZE);
  36     setRoughness(0.65);
  37     zoomFit();
  38 }
  39
  40 TerrainDisplay::~TerrainDisplay()
  41 {
  42     makeCurrent();
  43     terrain_destroy(m_mesh);
  44 }
  45
  46 QSize TerrainDisplay::minimumSizeHint() const
  47 {
  48     return QSize(64, 64);
  49 }
  50
  51 QSize TerrainDisplay::sizeHint() const
  52 {
  53     return QSize(320, 320);
  54 }
  55
  56 void TerrainDisplay::initializeGL()
  57 {
  58     glClearColor(0.0, 0.0, 0.0, 0.0);
  59
  60     glEnable(GL_DEPTH_TEST);
  61     glEnable(GL_NORMALIZE);
  62     glEnable(GL_CULL_FACE);
  63     glFrontFace(GL_CW);
  64
  65     glEnable(GL_DEPTH_TEST);
  66     glEnable(GL_CULL_FACE);
  67     glFrontFace(GL_CW);
  68
  69     glShadeModel(GL_SMOOTH);
  70
  71     glEnable(GL_LIGHTING);
  72     glEnable(GL_LIGHT0);
  73
  74 /*
  75     float position[] = { 10.0f, 10.0f, 10.0f, 1.0f };
  76     glLightfv(GL_LIGHT0, GL_POSITION, position);
  77
  78     float specular[] = { 1.0f, 1.0f, 1.0f, 1.0f };
  79     glLightfv(GL_LIGHT0, GL_SPECULAR, specular);
  80
  81     float ambient[] =  { 0.0f, 0.0f, 0.0f, 1.0f };
  82     glLightfv(GL_LIGHT0, GL_AMBIENT, ambient);
  83
  84     float diffuse[] =  { 1.0f, 1.0f, 1.0f, 1.0f };
  85     glLightfv(GL_LIGHT0, GL_DIFFUSE, diffuse);
  86
  87     glEnable(GL_COLOR_MATERIAL);
  88     glColorMaterial ( GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE ) ;
  89
  90     glMaterialf(GL_FRONT, GL_SHININESS, 128);
  91     glMaterialfv(GL_FRONT, GL_SPECULAR, specular);
  92 */
  93 }
  94
  95 void TerrainDisplay::resizeGL(int w, int h)
  96 {
  97     float ratio = (float)w/h;
  98     float scale = 10.0;
  99
 100     glViewport(0, 0, w, h);
 101
 102     glMatrixMode(GL_PROJECTION);
 103     glLoadIdentity();
 104     glOrtho(-scale*ratio, scale*ratio, -scale, scale, -1000.0, 1000.0);
 105     glMatrixMode(GL_MODELVIEW);
 106 }
 107
 108 void TerrainDisplay::paintGL()
 109 {
 110     glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
 111     glLoadIdentity();
 112     glTranslatef(0.0, 0.0, -10.0);
 113     glRotated(m_rot[0], 1.0, 0.0, 0.0);
 114     glRotated(m_rot[1], 0.0, 1.0, 0.0);
 115     glRotated(m_rot[2], 0.0, 0.0, 1.0);
 116
 117     glScalef(m_zoom, m_zoom, m_zoom);
 118     glColor3f(1.0, 1.0, 1.0);
 119     terrain_draw(m_mesh);
 120
 121     glPushMatrix();
 122
 123     glScalef(1.0/m_zoom, 1.0/m_zoom, 1.0/m_zoom);
 124
 125     glDisable(GL_LIGHTING);
 126     glDisable(GL_DEPTH_TEST);
 127
 128     glBegin(GL_LINES);
 129
 130     glColor3f(1.0, 0.0, 0.0); // x
 131     glVertex3f(0.0, 0.0, 0.0);
 132     glVertex3f(1.0, 0.0, 0.0);
 133
 134     glColor3f(0.0, 1.0, 0.0); // y
 135     glVertex3f(0.0, 0.0, 0.0);
 136     glVertex3f(0.0, 1.0, 0.0);
 137
 138     glColor3f(0.0, 0.0, 1.0); // z
 139     glVertex3f(0.0, 0.0, 0.0);
 140     glVertex3f(0.0, 0.0, 1.0);
 141
 142     glEnd();
 143
 144     glEnable(GL_DEPTH_TEST);
 145     glEnable(GL_LIGHTING);
 146
 147     glPopMatrix();
 148 }
 149
 150 void TerrainDisplay::mousePressEvent(QMouseEvent *event)
 151 {
 152     m_lastPos = event->pos();
 153 }
 154
 155 void TerrainDisplay::mouseMoveEvent(QMouseEvent *event)
 156 {
 157     int dx = event->x() - m_lastPos.x();
 158     int dy = event->y() - m_lastPos.y();
 159
 160     if (event->buttons() & Qt::LeftButton)
 161     {
 162         setXRotation(getXRotation() + m_mouseSensitivity * dy);
 163         setYRotation(getYRotation() + m_mouseSensitivity * dx);
 164     }
 165     else if (event->buttons() & Qt::RightButton)
 166     {
 167         setZoom(getZoom() + m_mouseSensitivity/100.0 * dy * getZoom());
 168     }
 169     else if (event->buttons() & Qt::MidButton)
 170     {
 171         makeCurrent();
 172         makeFractal();
 173         updateGL();
 174     }
 175
 176     m_lastPos = event->pos();
 177 }
 178
 179 void TerrainDisplay::zoomFit()
 180 {
 181     setZoom(10.0/getSize());
 182 }
 183
 184
 185 void TerrainDisplay::setZoom(double amount)
 186 {
 187     if (amount != m_zoom && amount > 0.0)
 188     {
 189         m_zoom = amount;
 190         emit zoomChanged(amount);
 191         updateGL();
 192     }
 193 }
 194
 195 void normalizeAngle(double *angle)
 196 {
 197     //int adj = *angle/360;
 198     //*angle -= adj*360;
 199 }
 200
 201 void TerrainDisplay::setXRotation(double angle)
 202 {
 203     normalizeAngle(&angle);
 204     if (angle != m_rot[0])
 205     {
 206         m_rot[0] = angle;
 207         emit xRotationChanged(angle);
 208         updateGL();
 209     }
 210 }
 211
 212 void TerrainDisplay::setYRotation(double angle)
 213 {
 214     normalizeAngle(&angle);
 215     if (angle != m_rot[1])
 216     {
 217         m_rot[1] = angle;
 218         emit yRotationChanged(angle);
 219         updateGL();
 220     }
 221 }
 222
 223 void TerrainDisplay::setZRotation(double angle)
 224 {
 225     normalizeAngle(&angle);
 226     if (angle != m_rot[2])
 227     {
 228         m_rot[2] = angle;
 229         emit zRotationChanged(angle);
 230         updateGL();
 231     }
 232 }
 233
 234 void TerrainDisplay::setRoughness(double rough)
 235 {
 236     if (rough != m_roughness)
 237     {
 238         m_roughness = rough;
 239         emit roughnessChanged(rough);
 240         makeFractal();
 241         updateGL();
 242     }
 243 }
 244
 245 void TerrainDisplay::setMagnitude(double mag)
 246 {
 247     if (mag != m_magnitude)
 248     {
 249         m_magnitude = mag;
 250         emit magnitudeChanged(mag);
 251         makeFractal();
 252         updateGL();
 253     }
 254 }
 255
 256 void TerrainDisplay::setSize(int size)
 257 {
 258     if (size != m_mesh->size)
 259     {
 260         TERRAIN_MODE m = m_mesh->drawmode;
 261         terrain_destroy(m_mesh);
 262         m_mesh = terrain_create(size);
 263         m_mesh->drawmode = m;
 264         makeFractal();
 265         emit sizeChanged(size);
 266         zoomFit();
 267         updateGL();
 268     }
 269 }
 270
 271 void TerrainDisplay::setDrawMode(int mode)
 272 {
 273     switch(mode)
 274     {
 275         case 0:
 276             m_mesh->drawmode = FLAT;
 277             m_mesh->dirty = 1;
 278             break;
 279         case 1:
 280             m_mesh->drawmode = NORMAL;
 281             m_mesh->dirty = 1;
 282             break;
 283         case 2:
 284             m_mesh->drawmode = NORMAL_DEBUG;
 285             m_mesh->dirty = 1;
 286             break;
 287         default:
 288             break;
 289     }
 290     updateGL();
 291 }
 292
 293 void TerrainDisplay::setSeed(int seed)
 294 {
 295     if ((unsigned int)seed != m_mesh->seed)
 296     {
 297         m_mesh->seed = seed;
 298         makeFractal();
 299         emit seedChanged(seed);
 300         updateGL();
 301     }
 302 }
 303
 304 void TerrainDisplay::makeFractal()
 305 {
 306     terrain_clear(m_mesh);
 307     terrain_fractal(m_mesh, m_magnitude);
 308     terrain_smooth(m_mesh, m_roughness);
 309     terrain_bake(m_mesh);
 310 }
 311