QtBaseViewerPicking.cc 22.2 KB
Newer Older
1
/*===========================================================================*\
Jan Möbius's avatar
Jan Möbius committed
2 3
*                                                                            *
*                              OpenFlipper                                   *
Jan Möbius's avatar
Jan Möbius committed
4
*      Copyright (C) 2001-2011 by Computer Graphics Group, RWTH Aachen       *
Jan Möbius's avatar
Jan Möbius committed
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
*                           www.openflipper.org                              *
*                                                                            *
*--------------------------------------------------------------------------- *
*  This file is part of OpenFlipper.                                         *
*                                                                            *
*  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 with the               *
*  following exceptions:                                                     *
*                                                                            *
*  If other files instantiate templates or use macros                        *
*  or inline functions from this file, or you compile this file and          *
*  link it with other files to produce an executable, this file does         *
*  not by itself cause the resulting executable to be covered by the         *
*  GNU Lesser General Public License. This exception does not however        *
*  invalidate any other reasons why the executable file might be             *
*  covered by the GNU Lesser General Public License.                         *
*                                                                            *
*  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 LesserGeneral Public           *
*  License along with OpenFlipper. If not,                                   *
*  see <http://www.gnu.org/licenses/>.                                       *
*                                                                            *
33 34 35
\*===========================================================================*/

/*===========================================================================*\
Jan Möbius's avatar
Jan Möbius committed
36 37 38 39 40
*                                                                            *
*   $Revision$                                                       *
*   $LastChangedBy$                                                *
*   $Date$                     *
*                                                                            *
41
\*===========================================================================*/
Jan Möbius's avatar
 
Jan Möbius committed
42 43 44 45 46 47




//=============================================================================
//
Jan Möbius's avatar
 
Jan Möbius committed
48
//  CLASS glViewer - IMPLEMENTATION
Jan Möbius's avatar
 
Jan Möbius committed
49 50 51 52 53 54 55 56 57 58
//
//=============================================================================


//== INCLUDES =================================================================

#include "QtBaseViewer.hh"
#include "QtGLGraphicsScene.hh"
#include "QtGLGraphicsView.hh"

59
#if QT_VERSION < 0x050000
60
  #include <QGLFramebufferObject>
61 62 63 64 65
#else // QT_VERSION > 0x050000
  #undef QT_NO_OPENGL
  #include <QOpenGLFramebufferObject>
  #define QT_NO_OPENGL
#endif //QT_VERSION < 0x050000
Jan Möbius's avatar
Jan Möbius committed
66

Jan Möbius's avatar
 
Jan Möbius committed
67 68 69 70 71 72 73 74 75 76
//== NAMESPACES ===============================================================

//== IMPLEMENTATION ==========================================================

static const unsigned int  SELECTION_BUFFER_SIZE = 10000;
static const unsigned int  NAME_STACK_SIZE       = 2;

//== IMPLEMENTATION ==========================================================


Jan Möbius's avatar
 
Jan Möbius committed
77
bool glViewer::pick( ACG::SceneGraph::PickTarget _pickTarget,
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
78 79 80 81 82 83 84
                     const QPoint&               _mousePos,
                     unsigned int&               _nodeIdx,
                     unsigned int&               _targetIdx,
                     ACG::Vec3d*                 _hitPointPtr )
{
  if (sceneGraphRoot_)
  {
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
85 86 87
    // unsigned int node, target;
    // QTime time;
    // time.start ();
Jan Möbius's avatar
Jan Möbius committed
88 89 90 91 92
    int rv = pickFromCache (_pickTarget, _mousePos, _nodeIdx, _targetIdx, _hitPointPtr);

    // cache will return -1 if a update is needed or caching is not supported
    if (rv < 0)
      rv = pickColor (_pickTarget, _mousePos, _nodeIdx, _targetIdx, _hitPointPtr);
Dirk Wilden's avatar
Dirk Wilden committed
93

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
94 95 96 97
    // printf ("ColorPicking took %d msec\n",time.restart ());
    // rv = -1;
    // node = _nodeIdx;
    // target = _targetIdx;
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
98 99
    if (rv < 0)
      rv = pickGL (_pickTarget, _mousePos, _nodeIdx, _targetIdx, _hitPointPtr);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
100
    // printf ("GLPicking took %d msec\n",time.restart ());
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
101

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
102 103
    // if (rv > 0 && (node != _nodeIdx || target != _targetIdx))
    //   printf ("***** Picking difference Color %d/%d GL %d/%d\n",node, target, _nodeIdx, _targetIdx);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
104 105 106 107 108 109 110 111 112 113
    if (rv > 0)
      return rv;
  }
  return false;
}


//-----------------------------------------------------------------------------

int glViewer::pickColor( ACG::SceneGraph::PickTarget _pickTarget,
Jan Möbius's avatar
 
Jan Möbius committed
114 115 116 117 118
                         const QPoint&               _mousePos,
                         unsigned int&               _nodeIdx,
                         unsigned int&               _targetIdx,
                         ACG::Vec3d*                 _hitPointPtr )
{
119 120 121 122 123
  GLint         w = glWidth(),
                h = glHeight(),
                l = scenePos().x(),
                b = scene()->height () - scenePos().y() - h,
                x = _mousePos.x(),
Jan Möbius's avatar
Jan Möbius committed
124 125 126
                y = scene()->height () - _mousePos.y(),
                pW = 1,
                pH = 1;
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
127
  GLubyte       pixels[9][4];
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
128 129
  GLfloat       depths[9];
  int           hit = -1;
Jan Möbius's avatar
Jan Möbius committed
130 131

  // traversing order (center, top, bottom, ...)
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
132 133
  unsigned char order[9] = { 4, 7, 1, 3, 5, 0, 2, 6, 8 };

Jan Möbius's avatar
Jan Möbius committed
134 135 136 137 138 139 140 141 142 143 144
  if (pickCacheSupported_)
  {
    // delete pick cache if the size changed
    if (pickCache_ && pickCache_->size () != QSize (glWidth (), glHeight ()))
    {
      delete pickCache_;
      pickCache_ = NULL;
    }
    // create a new pick cache frambuffer object
    if (!pickCache_)
    {
145
      pickCache_ = new QFrameBufferObject (glWidth (), glHeight (), QFrameBufferObject::Depth);
Jan Möbius's avatar
Jan Möbius committed
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168
      if (!pickCache_->isValid ())
      {
        pickCacheSupported_ = false;
        delete pickCache_;
        pickCache_ = NULL;
      }
    }
    if (pickCache_)
    {
      // the viewport for the framebuffer object
      l = 0;
      b = 0;
      x = _mousePos.x() - scenePos().x();
      y = glHeight() - (_mousePos.y() - scenePos().y());

      // we can only pick inside of our window
      if (x < 0 || y < 0 || x >= (int)glWidth() || y >= (int)glHeight())
        return 0;

      pickCache_->bind ();
    }
  }

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
169 170 171
  const ACG::GLMatrixd&  modelview  = properties_.glState().modelview();
  const ACG::GLMatrixd&  projection = properties_.glState().projection();

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
172 173 174
  ACG::Vec4f clear_color = properties_.glState().clear_color();
  properties_.glState().set_clear_color (ACG::Vec4f (0.0, 0.0, 0.0, 0.0));

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
175 176 177
  // prepare GL state
  makeCurrent();

178
  glViewport (l, b, w, h);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
179 180
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
181

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
182 183 184
  glMultMatrixd(projection.get_raw_data());
  glMatrixMode(GL_MODELVIEW);
  glLoadMatrixd(modelview.get_raw_data());
185 186 187
  ACG::GLState::disable(GL_LIGHTING);
  ACG::GLState::disable(GL_BLEND);
  ACG::GLState::enable(GL_DEPTH_TEST);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
188 189 190 191
  glClear(GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT);
  properties_.glState().pick_init (true);

  // do the picking
Jan Möbius's avatar
Jan Möbius committed
192
  ACG::SceneGraph::PickAction action(properties_.glState(), _pickTarget, properties_.drawMode());
193
  ACG::SceneGraph::traverse_multipass(sceneGraphRoot_, action,properties_.glState() );
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
194 195 196 197 198 199

  // restore GL state
  glMatrixMode( GL_PROJECTION );
  glLoadMatrixd(projection.get_raw_data());
  glMatrixMode( GL_MODELVIEW );
  glLoadMatrixd(modelview.get_raw_data());
200
  ACG::GLState::enable(GL_LIGHTING);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
201

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
202 203
  properties_.glState().set_clear_color (clear_color);

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
204
  if (properties_.glState().pick_error ())
Jan Möbius's avatar
Jan Möbius committed
205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253
  {
    if (pickCache_ && pickCache_->isBound ())
      pickCache_->release ();
    return -1;
  }

  glPixelStorei(GL_PACK_ALIGNMENT, 1);
  glPixelStorei(GL_UNPACK_ALIGNMENT, 1);

  // we can only read inside our viewport
  if (x + 1 < w)
    pW++;

  if (y + 1 < h)
    pH++;

  if (x > 0)
  {
    x--;
    pW++;
  }
  if (y > 0)
  {
    y--;
    pH++;
  }

  if (pH != 3 || pW != 3)
  {
    // initialize unused values with 0
    for (int i = 0; i < 9; i++)
    {
      pixels[i][0] = 0;
      pixels[i][1] = 0;
      pixels[i][2] = 0;
      pixels[i][3] = 0;
      depths[i] = 0.0;
    }
  }

  // read from framebuffer
  glReadPixels (x, y, pW, pH, GL_RGBA, GL_UNSIGNED_BYTE, pixels);
  glReadPixels (x, y, pW, pH, GL_DEPTH_COMPONENT, GL_FLOAT, depths);

  // unbind pick cache
  if (pickCache_ && pickCache_->isBound ())
  {
    pickCache_->release ();
    updatePickCache_ = false;
254
    pickCacheTarget_ = _pickTarget;
Jan Möbius's avatar
Jan Möbius committed
255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280
  }

  // get first found pixel
  for (int i = 0; i < 9; i++)
  {
    if (hit < 0 && (pixels[order[i]][2] != 0 || pixels[order[i]][1] != 0 || pixels[order[i]][0] != 0 || pixels[order[i]][3] != 0))
    {
      hit = order[i];
      break;
    }
  }

  if (hit < 0)
    return 0;


  ACG::Vec4uc rgba;
  rgba[0] = pixels[hit][0];
  rgba[1] = pixels[hit][1];
  rgba[2] = pixels[hit][2];
  rgba[3] = pixels[hit][3];

  std::vector<unsigned int> rv = properties_.glState().pick_color_to_stack (rgba);

  // something wrong with the color stack ?
  if (rv.size () < 2)
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
281 282
    return -1;

Jan Möbius's avatar
Jan Möbius committed
283 284 285
  _nodeIdx   = rv[1];
  _targetIdx = rv[0];

286 287 288 289 290 291 292 293 294 295 296
//   // Debug Code to visualize picking cache ( DO NOT REMOVE!!!! Jan )
//   QImage murks(glWidth (),glHeight (),QImage::Format_ARGB32);
//   murks = pickCache_->toImage();
//   for ( int i = 0 ; i < glWidth() ; ++i ) 
//     for ( int j = 0 ; j < glHeight() ; ++j )  {
//       QColor bla (murks.pixel(i,j));
//       bla.setAlpha(255);
//       murks.setPixel(i,j,bla.rgba());
//     }
//   murks.save("murks.png");

Jan Möbius's avatar
Jan Möbius committed
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
  if (_hitPointPtr)
  {
    *_hitPointPtr = properties_.glState().unproject (
      ACG::Vec3d(_mousePos.x(), scene()->height () - _mousePos.y(),depths[hit]));
  }

  return 1;
}

//-----------------------------------------------------------------------------

int glViewer::pickFromCache( ACG::SceneGraph::PickTarget _pickTarget,
                             const QPoint&               _mousePos,
                             unsigned int&               _nodeIdx,
                             unsigned int&               _targetIdx,
                             ACG::Vec3d*                 _hitPointPtr )
{
  // do we need an update?
315 316
  if (!pickCacheSupported_ || updatePickCache_ || !pickCache_ ||
      pickCacheTarget_ != _pickTarget)
Jan Möbius's avatar
Jan Möbius committed
317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336
    return -1;

  GLint         x = _mousePos.x() - scenePos().x(),
                y = glHeight() - (_mousePos.y() - scenePos().y()),
                pW = 1,
                pH = 1;
  GLubyte       pixels[9][4];
  GLfloat       depths[9];
  int           hit = -1;

  // traversing order (center, top, bottom, ...)
  unsigned char order[9] = { 4, 7, 1, 3, 5, 0, 2, 6, 8 };

  // can't pick outside
  if (x < 0 || y < 0 || x >= (int)glWidth() || y >= (int)glHeight())
    return 0;

  // bind cache framebuffer object
  pickCache_->bind ();

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
337 338 339
  glPixelStorei(GL_PACK_ALIGNMENT, 1);
  glPixelStorei(GL_UNPACK_ALIGNMENT, 1);

Jan Möbius's avatar
Jan Möbius committed
340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374 375 376
  // we can only read inside our viewport
  if (x + 1 < (int)glWidth ())
    pW++;

  if (y + 1 < (int)glHeight ())
    pH++;

  if (x > 0)
  {
    x--;
    pW++;
  }
  if (y > 0)
  {
    y--;
    pH++;
  }

  if (pH != 3 || pW != 3)
  {
    // initialize unused values with 0
    for (int i = 0; i < 9; i++)
    {
      pixels[i][0] = 0;
      pixels[i][1] = 0;
      pixels[i][2] = 0;
      pixels[i][3] = 0;
      depths[i] = 0.0;
    }
  }

  // read from framebuffer
  glReadPixels (x, y, pW, pH, GL_RGBA, GL_UNSIGNED_BYTE, pixels);
  glReadPixels (x, y, pW, pH, GL_DEPTH_COMPONENT, GL_FLOAT, depths);

  // unbind
  pickCache_->release ();
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
377

Jan Möbius's avatar
Jan Möbius committed
378
  // get first found pixel
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
379
  for (int i = 0; i < 9; i++)
Jan Möbius's avatar
 
Jan Möbius committed
380
  {
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
381
    if (hit < 0 && (pixels[order[i]][2] != 0 || pixels[order[i]][1] != 0 || pixels[order[i]][0] != 0 || pixels[order[i]][3] != 0))
Jan Möbius's avatar
 
Jan Möbius committed
382
    {
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
383 384 385 386 387 388 389
      hit = order[i];
      break;
    }
  }

  if (hit < 0)
    return 0;
Jan Möbius's avatar
 
Jan Möbius committed
390 391


Jan Möbius's avatar
Dennis:  
Jan Möbius committed
392 393 394 395 396
  ACG::Vec4uc rgba;
  rgba[0] = pixels[hit][0];
  rgba[1] = pixels[hit][1];
  rgba[2] = pixels[hit][2];
  rgba[3] = pixels[hit][3];
Jan Möbius's avatar
 
Jan Möbius committed
397

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
398
  std::vector<unsigned int> rv = properties_.glState().pick_color_to_stack (rgba);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
399

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
400 401 402 403 404 405 406 407 408
  // something wrong with the color stack ?
  if (rv.size () < 2)
    return -1;

  _nodeIdx   = rv[1];
  _targetIdx = rv[0];

  if (_hitPointPtr)
  {
Jan Möbius's avatar
Jan Möbius committed
409 410
    *_hitPointPtr = properties_.glState().unproject(
      ACG::Vec3d(_mousePos.x(), scene()->height () - _mousePos.y(),depths[hit]));
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
  }

  return 1;
}

//-----------------------------------------------------------------------------

bool glViewer::pickGL( ACG::SceneGraph::PickTarget _pickTarget,
                       const QPoint&               _mousePos,
                       unsigned int&               _nodeIdx,
                       unsigned int&               _targetIdx,
                       ACG::Vec3d*                 _hitPointPtr )
{
  GLint         w = glWidth(),
                h = glHeight(),
                l = scenePos().x(),
                b = scene()->height () - scenePos().y() - h,
                x = _mousePos.x(),
                y = scene()->height () - _mousePos.y();
  GLint         viewport[4] = {l,b,w,h};
  GLuint        selectionBuffer[ SELECTION_BUFFER_SIZE ],
                nameBuffer[ NAME_STACK_SIZE ];

  const ACG::GLMatrixd&  modelview  = properties_.glState().modelview();
  const ACG::GLMatrixd&  projection = properties_.glState().projection();

  // prepare GL state
  makeCurrent();

  glSelectBuffer( SELECTION_BUFFER_SIZE, selectionBuffer );
  glRenderMode(GL_SELECT);
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
  gluPickMatrix((GLdouble) x, (GLdouble) y, 3, 3, viewport);
  glMultMatrixd(projection.get_raw_data());
  glMatrixMode(GL_MODELVIEW);
  glLoadMatrixd(modelview.get_raw_data());
448 449
  ACG::GLState::disable(GL_LIGHTING);
  ACG::GLState::disable(GL_BLEND);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
450 451 452 453
  glClear(GL_DEPTH_BUFFER_BIT);
  properties_.glState().pick_init (false);

  // do the picking
Jan Möbius's avatar
Jan Möbius committed
454
  ACG::SceneGraph::PickAction action(properties_.glState(), _pickTarget, properties_.drawMode());
455
  ACG::SceneGraph::traverse_multipass(sceneGraphRoot_, action, properties_.glState());
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
456 457 458 459 460 461 462
  int hits = glRenderMode(GL_RENDER);

  // restore GL state
  glMatrixMode( GL_PROJECTION );
  glLoadMatrixd(projection.get_raw_data());
  glMatrixMode( GL_MODELVIEW );
  glLoadMatrixd(modelview.get_raw_data());
463 464
  ACG::GLState::enable(GL_LIGHTING);
  ACG::GLState::enable(GL_BLEND);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484

  // process hit record
  if ( hits > 0 )
  {
    GLuint *ptr = selectionBuffer,
    num_names,
    z,
    min_z=~(0u),
    max_z=0;

    for (int i=0; i<hits; ++i)
    {
      num_names = *ptr++;
      if ( num_names != NAME_STACK_SIZE )
      {
        std::cerr << "glViewer::pick() : namestack error\n\n";
        return false;
      }

      if ( (z = *ptr++) < min_z )
Jan Möbius's avatar
 
Jan Möbius committed
485
      {
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
486 487 488 489
        min_z = z;
        max_z = *ptr++;
        for (unsigned int j=0; j<num_names; ++j)
          nameBuffer[j] = *ptr++;
Jan Möbius's avatar
 
Jan Möbius committed
490
      }
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
491 492 493 494 495
      else ptr += 1+num_names;
    }

    _nodeIdx   = nameBuffer[0];
    _targetIdx = nameBuffer[1];
Jan Möbius's avatar
 
Jan Möbius committed
496

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
497 498 499 500 501 502 503
    if (_hitPointPtr)
    {
      GLuint zscale=~(0u);
      GLdouble min_zz = ((GLdouble)min_z) / ((GLdouble)zscale);
      GLdouble max_zz = ((GLdouble)max_z) / ((GLdouble)zscale);
      GLdouble zz     = 0.5F * (min_zz + max_zz);
      *_hitPointPtr = properties_.glState().unproject(ACG::Vec3d(x,y,zz));
Jan Möbius's avatar
 
Jan Möbius committed
504
    }
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
505 506

    return true;
Jan Möbius's avatar
 
Jan Möbius committed
507
  }
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
508 509
  else if (hits < 0)
    std::cerr << "glViewer::pick() : selection buffer overflow\n\n";
Jan Möbius's avatar
 
Jan Möbius committed
510 511 512 513 514 515

  return false;
}

//-----------------------------------------------------------------------------

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
516 517
bool glViewer::pick_region( ACG::SceneGraph::PickTarget                _pickTarget,
                            const QRegion&                             _region,
518 519 520
                            QList<QPair<unsigned int, unsigned int> >& _list,
                            QVector<float>*                            _depths,
                            QVector<ACG::Vec3d>*                       _points)
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
521
{
522
  QRect    rect = _region.boundingRect();
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
523 524 525
  GLint    w = glWidth(),
           h = glHeight(),
           l = scenePos().x(),
526 527 528
           b = scene()->height () - scenePos().y() - h,
           x = rect.x(),
           y = scene()->height () - rect.bottom();
529 530 531

  GLubyte* buffer = 0;
  GLfloat* depths = 0;
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
532

533 534 535 536 537 538 539 540 541 542 543
  if (pickCacheSupported_)
  {
    // delete pick cache if the size changed
    if (pickCache_ && pickCache_->size () != QSize (glWidth (), glHeight ()))
    {
      delete pickCache_;
      pickCache_ = NULL;
    }
    // create a new pick cache frambuffer object
    if (!pickCache_)
    {
544
      pickCache_ = new QFrameBufferObject (glWidth (), glHeight (), QFrameBufferObject::Depth);
545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566
      if (!pickCache_->isValid ())
      {
        pickCacheSupported_ = false;
        delete pickCache_;
        pickCache_ = NULL;
      }
    }
    if (pickCache_)
    {
      // the viewport for the framebuffer object
      l = 0;
      b = 0;
      x = rect.x() - scenePos().x();
      y = glHeight() - (rect.bottom() - scenePos().y());

      // we can only pick inside of our window
      if (x < 0 || y < 0 || x >= (int)glWidth() || y >= (int)glHeight())
        return 0;

      pickCache_->bind ();
    }
  }
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
567 568 569 570

  const ACG::GLMatrixd&  modelview  = properties_.glState().modelview();
  const ACG::GLMatrixd&  projection = properties_.glState().projection();

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
571 572 573
  ACG::Vec4f clear_color = properties_.glState().clear_color();
  properties_.glState().set_clear_color (ACG::Vec4f (0.0, 0.0, 0.0, 0.0));

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
574 575 576 577 578 579 580 581 582 583
  // prepare GL state
  makeCurrent();

  glViewport (l, b, w, h);
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();

  glMultMatrixd(projection.get_raw_data());
  glMatrixMode(GL_MODELVIEW);
  glLoadMatrixd(modelview.get_raw_data());
584 585 586
  ACG::GLState::disable(GL_LIGHTING);
  ACG::GLState::disable(GL_BLEND);
  ACG::GLState::enable(GL_DEPTH_TEST);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
587 588 589 590
  glClear(GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT);
  properties_.glState().pick_init (true);

  // do the picking
Jan Möbius's avatar
Jan Möbius committed
591
  ACG::SceneGraph::PickAction action(properties_.glState(), _pickTarget, properties_.drawMode());
592
  ACG::SceneGraph::traverse_multipass(sceneGraphRoot_, action,properties_.glState());
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
593 594 595 596 597 598

  // restore GL state
  glMatrixMode( GL_PROJECTION );
  glLoadMatrixd(projection.get_raw_data());
  glMatrixMode( GL_MODELVIEW );
  glLoadMatrixd(modelview.get_raw_data());
599 600
  ACG::GLState::enable(GL_LIGHTING);
  ACG::GLState::enable(GL_BLEND);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
601

602
  properties_.glState().set_clear_color(clear_color);
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
603

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
604
  if (properties_.glState().pick_error ())
605 606 607
  {
    if (pickCache_ && pickCache_->isBound ())
      pickCache_->release ();
Jan Möbius's avatar
Jan Möbius committed
608
    return false;
609
  }
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
610

Jan Möbius's avatar
Dennis:  
Jan Möbius committed
611
  buffer = new GLubyte[4 * rect.width() * rect.height()];
Jan Möbius's avatar
Dennis:  
Jan Möbius committed
612

Henrik Zimmer's avatar
Henrik Zimmer committed
613 614 615 616 617 618 619 620


  glPixelStorei(GL_PACK_ALIGNMENT, 1);
  glPixelStorei(GL_UNPACK_ALIGNMENT, 1);

  glReadPixels (x, y, rect.width(),
                rect.height(), GL_RGBA, GL_UNSIGNED_BYTE, buffer);

621 622 623
  if (_depths || _points ) {
    depths = new GLfloat[ rect.width() * rect.height() ];
    glReadPixels (x, y, rect.width(), rect.height(), GL_DEPTH_COMPONENT, GL_FLOAT, depths);
Henrik Zimmer's avatar
Henrik Zimmer committed
624

625 626
    /*  Debug code, writing out the depth image
    QImage depthmapimage(rect.width(), rect.height(), QImage::Format_Indexed8);
Henrik Zimmer's avatar
Henrik Zimmer committed
627

628 629 630
    // color map
    for ( int i = 0 ; i <= 255 ; i++ )
      depthmapimage.setColor( i, qRgb( i, i, i ) );
Henrik Zimmer's avatar
Henrik Zimmer committed
631

632 633
    for ( int i = 0 ; i < rect.width() ; i++ )
      for ( int j = 0 ; j < rect.height() ; j++ )
Henrik Zimmer's avatar
Henrik Zimmer committed
634
      {
635
        depthmapimage.setPixel(i,rect.height()-j-1, (unsigned int)(depths[j*rect.width()+i]*255));
Henrik Zimmer's avatar
Henrik Zimmer committed
636 637
      }

638 639
    depthmapimage.save("test.png");
    */
Henrik Zimmer's avatar
Henrik Zimmer committed
640 641
  }

642
  // Iterate over the bounding rectangle of the region
Henrik Zimmer's avatar
Henrik Zimmer committed
643 644 645 646
  for (int y = 0; y < rect.height (); y++)
    for (int x = 0; x < rect.width (); x++)
    {

647 648
      // Check if the current point is in the polygon of the region
      if (_region.contains (QPoint (rect.x() + x, rect.y() + y)))
Henrik Zimmer's avatar
Henrik Zimmer committed
649 650
      {

651 652
        // Calculate position inside the buffer
        const int bPos = (((rect.height () - (y + 1)) * rect.width ()) + x) * 4;
Henrik Zimmer's avatar
Henrik Zimmer committed
653

654
        // Get the picking color from the buffer at the current position
Henrik Zimmer's avatar
Henrik Zimmer committed
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673
        if (buffer[bPos + 2] != 0 || buffer[bPos + 1] != 0 || buffer[bPos] != 0 || buffer[bPos + 3] != 0)
        {
          ACG::Vec4uc rgba;
          rgba[0] = buffer[bPos];
          rgba[1] = buffer[bPos + 1];
          rgba[2] = buffer[bPos + 2];
          rgba[3] = buffer[bPos + 3];

          std::vector<unsigned int> rv = properties_.glState().pick_color_to_stack (rgba);
          if (rv.size () < 2)
            continue;

          QPair<unsigned int, unsigned int> curr(rv[1], rv[0]);
          
          // added a new (targetidx/nodeidx) pair
          if( !_list.contains(curr))
          {
            _list << curr;

674
            if ( _depths  || _points ) {
Henrik Zimmer's avatar
Henrik Zimmer committed
675

676 677 678 679 680 681 682 683 684 685
              const double curr_depth(depths[(rect.height()-(y+1))*rect.width() + x]);

              // If depths should be returned, we extract it here
              if (_depths)
                (*_depths) << curr_depth;

              // unproject depth to real (3D) depth value
              if ( _points )
                (*_points) << properties_.glState().unproject(ACG::Vec3d(x+rect.x(),h-(y+rect.y()),curr_depth));
            }
Henrik Zimmer's avatar
Henrik Zimmer committed
686 687 688 689 690
          }
        }
      }
    }

691
  delete[] buffer;
692 693 694

  if ( _depths  || _points )
    delete[] depths;
Henrik Zimmer's avatar
Henrik Zimmer committed
695 696 697 698 699 700 701 702 703 704 705 706 707

  // unbind pick cache
  if (pickCache_ && pickCache_->isBound ())
  {
    pickCache_->release ();
    updatePickCache_ = false;
    pickCacheTarget_ = _pickTarget;
  }

  return true;
}


Jan Möbius's avatar
Dennis:  
Jan Möbius committed
708 709
//-----------------------------------------------------------------------------

Jan Möbius's avatar
 
Jan Möbius committed
710
bool
Jan Möbius's avatar
 
Jan Möbius committed
711
glViewer::
Jan Möbius's avatar
 
Jan Möbius committed
712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727
fast_pick( const QPoint&  _mousePos,
           ACG::Vec3d&    _hitPoint )
{
  // get x,y,z values of pixel
  GLint     x(_mousePos.x()), y(glHeight() - _mousePos.y());
  GLfloat   z;


  makeCurrent();
  glPixelStorei(GL_PACK_ALIGNMENT, 1);
  glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
  glReadPixels(x, y, 1, 1, GL_DEPTH_COMPONENT, GL_FLOAT, &z);


  if (z < 0.99999)
  {
728
    _hitPoint = properties_.glState().unproject( ACG::Vec3d(x, y, z) );
Jan Möbius's avatar
 
Jan Möbius committed
729 730 731 732 733 734 735 736 737
    return true;
  }
  else return false;
}


//=============================================================================

//=============================================================================