cgal/Polyhedron/demo/Polyhedron/Scene_implicit_function_ite...

648 lines
19 KiB
C++

#include "Scene_implicit_function_item.h"
#include <QColor>
#include <QApplication>
#include <map>
#include <CGAL/gl.h>
#include <CGAL/Simple_cartesian.h>
#include <QGLViewer/manipulatedFrame.h>
#include "Color_ramp.h"
#include <CGAL/Three/Viewer_interface.h>
#include <CGAL/double.h>
inline
bool is_nan(double d)
{
return !CGAL::Is_valid<double>()( d );
}
struct Scene_implicit_function_item_priv
{
Scene_implicit_function_item_priv(Implicit_function_interface* f, Scene_implicit_function_item* parent)
: function_(f)
, frame_(new ManipulatedFrame())
, need_update_(true)
, grid_size_(SCENE_IMPLICIT_GRID_SIZE)
, max_value_(0.)
, min_value_(0.)
, blue_color_ramp_()
, red_color_ramp_()
, textureId(-1)
{
item = parent;
texture = new Texture(grid_size_-1,grid_size_-1);
blue_color_ramp_.build_blue();
red_color_ramp_.build_red();
}
~Scene_implicit_function_item_priv()
{
delete frame_;
}
typedef qglviewer::Vec Point;
typedef std::pair <Point,double> Point_value;
typedef qglviewer::ManipulatedFrame ManipulatedFrame;
void compute_min_max();
void initialize_buffers(CGAL::Three::Viewer_interface *viewer) const;
void compute_vertices_and_texmap(void);
void compute_texture(int, int);
enum VAOs {
Plane = 0,
BBox,
Grid,
NbOfVaos
};
enum VBOs {
Quad_vertices = 0,
TexMap,
Cube_vertices,
Grid_vertices,
NbOfVbos
};
Implicit_function_interface* function_;
ManipulatedFrame* frame_;
mutable bool need_update_;
int grid_size_;
double max_value_;
double min_value_;
mutable Point_value implicit_grid_[SCENE_IMPLICIT_GRID_SIZE][SCENE_IMPLICIT_GRID_SIZE];
Color_ramp blue_color_ramp_;
Color_ramp red_color_ramp_;
std::vector<float> positions_cube;
std::vector<float> positions_grid;
std::vector<float> positions_tex_quad;
std::vector<float> texture_map;
Texture *texture;
GLuint vao;
GLuint buffer[4];
mutable QOpenGLShaderProgram *program;
mutable GLuint textureId;
Scene_implicit_function_item* item;
};
void Scene_implicit_function_item_priv::initialize_buffers(CGAL::Three::Viewer_interface *viewer = 0) const
{
if(GLuint(-1) == textureId) {
viewer->glGenTextures(1, &textureId);
}
//vao fot the cutting plane
{
program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITH_TEXTURE, viewer);
program->bind();
item->vaos[Plane]->bind();
item->buffers[Quad_vertices].bind();
item->buffers[Quad_vertices].allocate(positions_tex_quad.data(),
static_cast<int>(positions_tex_quad.size()*sizeof(float)));
program->enableAttributeArray("vertex");
program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
item->buffers[Quad_vertices].release();
item->buffers[TexMap].bind();
item->buffers[TexMap].allocate(texture_map.data(),
static_cast<int>(texture_map.size()*sizeof(float)));
program->enableAttributeArray("v_texCoord");
program->setAttributeBuffer("v_texCoord",GL_FLOAT,0,2);
item->buffers[TexMap].release();
program->setAttributeValue("normal", QVector3D(0.f,0.f,0.f));
program->release();
item->vaos[Plane]->release();
}
//vao fot the bbox
{
program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITHOUT_LIGHT, viewer);
program->bind();
item->vaos[BBox]->bind();
item->buffers[Cube_vertices].bind();
item->buffers[Cube_vertices].allocate(positions_cube.data(),
static_cast<int>(positions_cube.size()*sizeof(float)));
program->enableAttributeArray("vertex");
program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
item->buffers[Cube_vertices].release();
program->release();
item->vaos[BBox]->release();
}
//vao fot the grid
{
program = item->getShaderProgram(Scene_implicit_function_item::PROGRAM_WITHOUT_LIGHT, viewer);
program->bind();
item->vaos[Grid]->bind();
item->buffers[Grid_vertices].bind();
item->buffers[Grid_vertices].allocate(positions_grid.data(),
static_cast<int>(positions_grid.size()*sizeof(float)));
program->enableAttributeArray("vertex");
program->setAttributeBuffer("vertex",GL_FLOAT,0,3);
item->buffers[Grid_vertices].release();
program->release();
item->vaos[Grid]->release();
}
viewer->glPixelStorei(GL_UNPACK_ALIGNMENT, 1);
viewer->glBindTexture(GL_TEXTURE_2D, textureId);
viewer->glTexImage2D(GL_TEXTURE_2D,
0,
GL_RGB,
texture->getWidth(),
texture->getHeight(),
0,
GL_RGB,
GL_UNSIGNED_BYTE,
texture->getData());
viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S,GL_CLAMP_TO_EDGE );
viewer->glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T,GL_CLAMP_TO_EDGE );
item->are_buffers_filled = true;
}
void Scene_implicit_function_item_priv::compute_vertices_and_texmap(void)
{
QApplication::setOverrideCursor(Qt::WaitCursor);
positions_tex_quad.resize(0);
positions_cube.resize(0);
positions_grid.resize(0);
texture_map.resize(0);
const CGAL::Three::Scene_item::Bbox& b = item->bbox();
float x,y,z;
z = (b.zmax()+b.zmin())/2.0;
x = (b.xmax()+b.xmin())/2.0;
y = (b.ymax()+b.ymin())/2.0;
// The Quad
{
//A
positions_tex_quad.push_back(b.xmin()-x);
positions_tex_quad.push_back(b.ymin()-z);
positions_tex_quad.push_back(0);
//B
positions_tex_quad.push_back(b.xmin()-x);
positions_tex_quad.push_back(b.ymax()-y);
positions_tex_quad.push_back(0);
//C
positions_tex_quad.push_back(b.xmax()-x);
positions_tex_quad.push_back(b.ymax()-y);
positions_tex_quad.push_back(0);
//A
positions_tex_quad.push_back(b.xmin()-x);
positions_tex_quad.push_back(b.ymin()-y);
positions_tex_quad.push_back(0);
//C
positions_tex_quad.push_back(b.xmax()-x);
positions_tex_quad.push_back(b.ymax()-y);
positions_tex_quad.push_back(0);
//D
positions_tex_quad.push_back(b.xmax()-x);
positions_tex_quad.push_back(b.ymin()-y);
positions_tex_quad.push_back(0);
texture_map.push_back(0.0);
texture_map.push_back(0.0);
texture_map.push_back(0.0);
texture_map.push_back(1.0);
texture_map.push_back(1.0);
texture_map.push_back(1.0);
texture_map.push_back(0.0);
texture_map.push_back(0.0);
texture_map.push_back(1.0);
texture_map.push_back(1.0);
texture_map.push_back(1.0);
texture_map.push_back(0.0);
}
//The grid
{
double dx((b.xmax()-b.xmin())/10.0), dy((b.ymax()-b.ymin())/10.0);
for(int u = 0; u < 11; u++)
{
positions_grid.push_back(b.xmin()-x + dx* u);
positions_grid.push_back(b.ymin()-y);
positions_grid.push_back(0);
positions_grid.push_back(b.xmin()-x + dx* u);
positions_grid.push_back(b.ymax()-y);
positions_grid.push_back(0);
}
for(int v=0; v<11; v++)
{
positions_grid.push_back(b.xmin()-x);
positions_grid.push_back(b.ymin()-y + v * dy);
positions_grid.push_back(0);
positions_grid.push_back(b.xmax()-x);
positions_grid.push_back(b.ymin()-y + v * dy);
positions_grid.push_back(0);
}
}
//the Box
{
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmin());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmin());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymax());
positions_cube.push_back(b.zmax());
positions_cube.push_back(b.xmax());
positions_cube.push_back(b.ymin());
positions_cube.push_back(b.zmax());
}
//The texture
for( int i=0 ; i < texture->getWidth() ; i++ )
{
for( int j=0 ; j < texture->getHeight() ; j++)
{
compute_texture(i,j);
}
}
QApplication::restoreOverrideCursor();
}
Scene_implicit_function_item::
Scene_implicit_function_item(Implicit_function_interface* f)
:CGAL::Three::Scene_item(4,3)
{
d = new Scene_implicit_function_item_priv(f, this);
//Generates an integer which will be used as ID for each buffer
d->compute_min_max();
compute_function_grid();
double offset_x = (bbox().xmin() + bbox().xmax()) / 2;
double offset_y = (bbox().ymin() + bbox().ymax()) / 2;
double offset_z = (bbox().zmin() + bbox().zmax()) / 2;
d->frame_->setPosition(offset_x, offset_y, offset_z);
d->frame_->setOrientation(1., 0, 0, 0);
connect(d->frame_, SIGNAL(modified()), this, SLOT(plane_was_moved()));
plane_was_moved();
invalidateOpenGLBuffers();
}
Scene_implicit_function_item::~Scene_implicit_function_item()
{
delete d;
}
void
Scene_implicit_function_item::compute_bbox() const
{
_bbox = d->function_->bbox();
}
void
Scene_implicit_function_item::draw(CGAL::Three::Viewer_interface* viewer) const
{
if(!are_buffers_filled)
d->initialize_buffers(viewer);
if(d->frame_->isManipulated()) {
if(d->need_update_) {
compute_function_grid();
d->need_update_ = false;
}
}
vaos[Scene_implicit_function_item_priv::Plane]->bind();
viewer->glActiveTexture(GL_TEXTURE0);
viewer->glBindTexture(GL_TEXTURE_2D, d->textureId);
attribBuffers(viewer, PROGRAM_WITH_TEXTURE);
QMatrix4x4 f_mat;
GLdouble d_mat[16];
d->frame_->getMatrix(d_mat);
//Convert the GLdoubles matrices in GLfloats
for (int i=0; i<16; ++i){
f_mat.data()[i] = GLfloat(d_mat[i]);
}
d->program = getShaderProgram(PROGRAM_WITH_TEXTURE);
d->program->bind();
d->program->setUniformValue("f_matrix", f_mat);
d->program->setUniformValue("light_amb", QVector4D(1.f,1.f,1.f,1.f));
d->program->setUniformValue("light_diff", QVector4D(0.f,0.f,0.f,1.f));
d->program->setAttributeValue("color_facets", QVector3D(1.f,1.f,1.f));
viewer->glDrawArrays(GL_TRIANGLES, 0, static_cast<GLsizei>(d->positions_tex_quad.size()/3));
vaos[Scene_implicit_function_item_priv::Plane]->release();
d->program->release();
}
void
Scene_implicit_function_item::drawEdges(CGAL::Three::Viewer_interface* viewer) const
{
if(!are_buffers_filled)
d->initialize_buffers(viewer);
vaos[Scene_implicit_function_item_priv::BBox]->bind();
attribBuffers(viewer, PROGRAM_WITHOUT_LIGHT);
d->program = getShaderProgram(PROGRAM_WITHOUT_LIGHT);
d->program->bind();
d->program->setAttributeValue("colors", QVector3D(0.f,0.f,0.f));
viewer->glDrawArrays(GL_LINES, 0, static_cast<GLsizei>(d->positions_cube.size()/3));
vaos[Scene_implicit_function_item_priv::BBox]->release();
vaos[Scene_implicit_function_item_priv::Grid]->bind();
QMatrix4x4 f_mat;
GLdouble d_mat[16];
d->frame_->getMatrix(d_mat);
//Convert the GLdoubles matrices in GLfloats
for (int i=0; i<16; ++i){
f_mat.data()[i] = double(d_mat[i]);
}
d->program->setUniformValue("f_matrix", f_mat);
d->program->setAttributeValue("colors", QVector3D(0.6f, 0.6f, 0.6f));
viewer->glDrawArrays(GL_LINES, 0, static_cast<GLsizei>(d->positions_grid.size()/3));
vaos[Scene_implicit_function_item_priv::Grid]->release();
d->program->release();
}
QString
Scene_implicit_function_item::toolTip() const
{
return tr("<p>Function <b>%1</b>")
.arg(this->name());
}
bool
Scene_implicit_function_item::supportsRenderingMode(RenderingMode m) const
{
switch ( m )
{
case Splatting:
case Gouraud:
return false;
case Points:
case Wireframe:
case Flat:
case FlatPlusEdges:
return true;
default:
return false;
}
return false;
}
void Scene_implicit_function_item_priv::compute_texture(int i, int j)
{
double v = (implicit_grid_[i][j]).second;
if(is_nan(v)) {
texture->setData(i,j,51,51,51);
} else
// determines grey level
if ( v > 0 )
{
v = v/max_value_;
GLdouble r = red_color_ramp_.r(v), g = red_color_ramp_.g(v), b = red_color_ramp_.b(v);
texture->setData(i,j,255*r,255*g,255*b);
}
else
{
v = v/min_value_;
GLdouble r = blue_color_ramp_.r(v), g = blue_color_ramp_.g(v), b = blue_color_ramp_.b(v);
texture->setData(i,j,255*r,255*g,255*b);
}
}
void
Scene_implicit_function_item::
compute_function_grid() const
{
typedef CGAL::Simple_cartesian<double> K;
typedef K::Aff_transformation_3 Aff_transformation;
typedef K::Point_3 Point_3;
// Get transformation
const GLdouble* m = d->frame_->matrix();
// OpenGL matrices are row-major matrices
Aff_transformation t (m[0], m[4], m[8], m[12],
m[1], m[5], m[9], m[13],
m[2], m[6], m[10], m[14]);
double diag = (CGAL::sqrt((bbox().xmax()-bbox().xmin())*(bbox().xmax()-bbox().xmin()) + (bbox().ymax()-bbox().ymin())*(bbox().ymax()-bbox().ymin()) + (bbox().zmax()-bbox().zmin())*(bbox().zmax()-bbox().zmin()) )) * .6;
const double dx = diag;
const double dy = diag;
const double z (0);
int nb_quad = d->grid_size_ - 1;
for(int i=0 ; i<d->grid_size_ ; ++i)
{
double x = -diag/2. + double(i)/double(nb_quad) * dx;
for(int j=0 ; j<d->grid_size_ ; ++j)
{
double y = -diag/2. + double(j)/double(nb_quad) * dy;
Point_3 query = t( Point_3(x, y, z) );
double v = d->function_->operator()(query.x(), query.y(), query.z());
d->implicit_grid_[i][j] = Scene_implicit_function_item_priv::Point_value(Scene_implicit_function_item_priv::Point(query.x(),query.y(),query.z()),v);
}
}
// Update
const_cast<Scene_implicit_function_item*>(this)->invalidateOpenGLBuffers();
}
void
Scene_implicit_function_item_priv::
compute_min_max()
{
if(function_->get_min_max(min_value_, max_value_))
return;
double probes_nb = double(grid_size_) / 2;
// Probe bounding box
const CGAL::Three::Scene_item::Bbox& b = item->bbox();
for ( int i = 0 ; i <= probes_nb ; ++i )
{
double x = b.xmin() + double(i) * (b.xmax() - b.xmin()) / probes_nb;
for ( int j = 0 ; j <= probes_nb ; ++j )
{
double y = b.ymin() + double(j) * (b.ymax() - b.ymin()) / probes_nb;
for ( int k = 0 ; k <= probes_nb ; ++k )
{
double z = b.zmin() + double(k) * (b.zmax() - b.zmin()) / probes_nb;
double v = (*function_)(x,y,z);
if(is_nan(v)) continue;
max_value_ = (std::max)(v,max_value_);
min_value_ = (std::min)(v,min_value_);
}
}
}
}
void
Scene_implicit_function_item::invalidateOpenGLBuffers()
{
Scene_item::invalidateOpenGLBuffers();
compute_bbox();
d->compute_vertices_and_texmap();
are_buffers_filled = false;
}
void Scene_implicit_function_item::updateCutPlane()
{ // just handle deformation - paint like selection is handled in eventFilter()
if(d->need_update_) {
compute_function_grid();
d->compute_vertices_and_texmap();
d->need_update_= false;
}
}
Implicit_function_interface* Scene_implicit_function_item::function() const { return d->function_; }
Scene_implicit_function_item::ManipulatedFrame* Scene_implicit_function_item::manipulatedFrame() { return d->frame_; }
void Scene_implicit_function_item::plane_was_moved() { d->need_update_ = true; QTimer::singleShot(0, this, SLOT(updateCutPlane()));}