vcglib/apps/test/segmentation3d/simpleglwidget.cpp

#include <SimpleGLWidget.h>
#include <qlineedit.h> 
#include <qfiledialog.h> 
#include <qdir.h> 
#include <qmessagebox.h> 

extern Segmentator *s;

SimpleGLWidget::SimpleGLWidget( QWidget * parent, const char * name, const QGLWidget * shareWidget, WFlags f ):
QGLWidget(parent, name)
{

	///grabKeyboard();
	_showslides=false;
	blocked=false;
	wire=true;
	extForces=false;
	intForces=false;
	resultForces=false;
	continue_int=false;
	_numslide=0;
	Track.center=Point3f(0,0,0);
	Track.Reset();
	Track.radius= 100;
	zoom=1;
}

void SimpleGLWidget::SaveMatrix()
{
	glGetDoublev(GL_PROJECTION_MATRIX ,projection);
	glGetDoublev(GL_MODELVIEW_MATRIX ,modelMatrix);
	glGetIntegerv(GL_VIEWPORT,viewport);
}

void SimpleGLWidget::LoadMatrix()
{
	glMatrixMode(GL_PROJECTION);
	glLoadMatrixd(projection);
	glMatrixMode(GL_MODELVIEW);
	glLoadMatrixd(modelMatrix);
}


void SimpleGLWidget::drawSlide()
{	
glBegin(GL_QUADS);
for (int x=0;x<((s->V.dimX())-1);x++)
	for (int y=0;y<((s->V.dimY())-1);y++)
	{
		glNormal(Point3d(0,0,1));
		Point3<int> p0=Point3<int>(x,y,_numslide);
		double color=((double)s->V.getAt(p0))/256.f;
		glColor3d(color,color,color);
		glVertex(p0);

		Point3<int> p1=Point3<int>(x+1,y,_numslide);
		color=((double)s->V.getAt(p1))/256.f;
		glColor3d(color,color,color);
		glVertex(p1);

		Point3<int> p2=Point3<int>(x+1,y+1,_numslide);
		color=((double)s->V.getAt(p2))/256.f;
		glColor3d(color,color,color);
		glVertex(p2);

		Point3<int> p3=Point3<int>(x,y+1,_numslide);
		color=((double)s->V.getAt(p3))/256.f;
		glColor3d(color,color,color);
		glVertex(p3);
	}
glEnd();		
}

void drawForces(Segmentator::Part_VertexContainer *pv,int typeForce)
{
Segmentator::Part_VertexContainer::iterator vi;
glPushAttrib(GL_ALL_ATTRIB_BITS);
glLineWidth(0.3);
glDisable(GL_NORMALIZE);
glDisable(GL_LIGHTING);
glPolygonMode(GL_FRONT_AND_BACK,GL_LINE);

if (typeForce==0)
	glColor3d(1,0,0);
else
if (typeForce==1)
	glColor3d(0,0,1);
else
if (typeForce==2)
	glColor3d(0,1,0);

for (vi=pv->begin();vi<pv->end();vi++)
{
	glBegin(GL_LINE_LOOP);
	vcg::glVertex((*vi).P());
	if (typeForce==0)
		vcg::glVertex((*vi).P()+((*vi).IntForce()*4.f));
	else
	if (typeForce==1)
		vcg::glVertex((*vi).P()+((*vi).ExtForce()*4.f));
	else
	if (typeForce==2)
		vcg::glVertex((*vi).P()+(((*vi).ExtForce()+(*vi).IntForce())*4.f));
	glEnd();
}
glPopAttrib();
}

void SimpleGLWidget::Save()
{
	QString filename = QFileDialog::getSaveFileName("prova.ply",
													"Ply files (*.ply)",
													this,
													"save file dialog",
													"Choose a filename to save under" );
	if (filename!=NULL)
	{
		const char *path=filename.ascii();
		vcg::tri::io::ExporterPLY<Segmentator::MyTriMesh>::Save(s->m,path);	
	}
	
}

bool TimeSelfIntersection()
{
	static clock_t time=0;
	clock_t elapsedsecs=abs(time-clock());
	if (elapsedsecs>500)
	{
		time=clock();
		return true;
	}
	return false;
}

//void SimpleGLWidget::WriteInfo()
//{
//	
//	if (s!=0)
//	{
//		
//		glPushAttrib(0xffffffff);
//		
//		glEnable(GL_BLEND);
//		glBlendFunc(GL_SRC_ALPHA, GL_SRC_ALPHA);
//		glEnable(GL_LIGHTING);
//		glEnable(GL_NORMALIZE);
//		glEnable(GL_COLOR_MATERIAL);
//		glDisable(GL_CLIP_PLANE0);
//		glColor4d(0.7,0,0.7,0.6);
//
//		glDepthRange(0.0,0.1);
//
//		glBegin(GL_QUADS);
//			glVertex3d(-0.5,-0.5,0);
//			glVertex3d(-0.5,-0.3,0);
//			glVertex3d(0.5,-0.3,0);
//			glVertex3d(0.5,-0.5,0);
//		glEnd();
//		
//		renderText( (width() - 10) / 2, 15, "a" );
//
//		QFont f( "arial", 12 );
//		QFontMetrics fmc( f );
//		glColor3d(1,1,1);
//		
//		QString str="";
//		int level=0;
//		
//		glDisable( GL_LIGHTING );
//		glDisable( GL_TEXTURE_2D );
//
//		level++;
//		str.sprintf( "Triangles :  %i  Vertex: %i ",s->m.fn,s->m.vn);
//		renderText( 20, height() - level*20, str, f );
//	}
//}

void SimpleGLWidget::SmoothMesh()
{
	s->Smooth();
}

void SimpleGLWidget::glDraw(){
	
	glClearColor(0.2,0.2,0.2,1);
	glClear(GL_COLOR_BUFFER_BIT|GL_DEPTH_BUFFER_BIT);
	glMatrixMode(GL_PROJECTION);
	glLoadIdentity();
	gluPerspective(45,1,0.01,20);
	glMatrixMode(GL_MODELVIEW);
	glLoadIdentity();
	gluLookAt(0,0,1,0,0,0,0,10,0);

	if (s!=0){
		
		glPushMatrix();
		if (_showslides)
		{
			vcg::Point3f p=Point3f((float)s->BBox().Center().V(0),(float)s->BBox().Center().V(1),(float)s->BBox().Center().V(2));
			Track.radius=s->BBox().Diag();
			Track.GetView();
			Track.Apply();
			Track.Draw();
			glScalef(1/s->BBox().Diag(),1/s->BBox().Diag(),1/s->BBox().Diag());
			glScalef(GLfloat(zoom),GLfloat(zoom),GLfloat(zoom));
			glTranslate(-p);
			//save transformation matrixes
			SaveMatrix();
		}
		else
		{
			vcg::tri::UpdateBounding<Segmentator::MyTriMesh>::Box(s->m);
			vcg::Point3f p=s->m.bbox.Center();
			Track.radius=s->m.bbox.Diag();
			Track.GetView();
			Track.Apply();
			Track.Draw();
			glScalef(1/s->m.bbox.Diag(),1/s->m.bbox.Diag(),1/s->m.bbox.Diag());
			glScalef(GLfloat(zoom),GLfloat(zoom),GLfloat(zoom));
			glTranslate(-p);
		}
		
		glEnable(GL_NORMALIZE);
		glEnable(GL_LIGHTING);
		glEnable(GL_LIGHT0);
		glEnable(GL_COLOR_MATERIAL);
		if (_showslides)
			drawSlide();

		if (intForces)
			drawForces(&s->P_Vertex,0);
		if (extForces)
			drawForces(&s->P_Vertex,1);
		if (resultForces)
			drawForces(&s->P_Vertex,2);

		//draw faces
		glEnable(GL_NORMALIZE);
		glEnable(GL_LIGHTING);
		glEnable(GL_LIGHT0);
		glEnable(GL_COLOR_MATERIAL);

		Segmentator::MyTriMesh::FaceIterator fi;
		

		if (wire)
		{
			glDisable(GL_NORMALIZE);
			glDisable(GL_LIGHTING);
			glPolygonMode(GL_FRONT_AND_BACK,GL_LINE);
		}
		else
			glPolygonMode(GL_FRONT,GL_FILL);

		int i=0;
		glBegin(GL_TRIANGLES);
		for (fi=s->m.face.begin();fi<s->m.face.end();fi++)
		{
				glColor3d(0.4,0.8,0.8);
			if (fi->intersected)
				glColor3d(1.f,0,0);

			if (((blocked)&&(!fi->IsBlocked()))||(!blocked))
			{
				
				if (!wire)
					vcg::glNormal(fi->Normal());

				vcg::glVertex(fi->V(0)->P());
				vcg::glVertex(fi->V(1)->P());
				vcg::glVertex(fi->V(2)->P());
			}
		}
		glEnd();
	
		glPopMatrix();
		//WriteInfo();
		
	}
	QGLWidget::glDraw();
}

///open the directiry and initialize dataset
void SimpleGLWidget::OpenDirectory()
{
	QString filename = QFileDialog::getExistingDirectory(
              ".",
              this,
              "open file dialog"
              "Choose a Directory" );
	if (filename!=NULL)
	{
		filename+="/";
		const char *path=filename.ascii();
		char *p=(char*)path;
		s->LoadFromDir(p,"prova.txt");
	}
}

void SimpleGLWidget::resizeGL( int w, int h )
  {
	  // setup viewport, projection etc.:
	glMatrixMode (GL_PROJECTION);
	glLoadIdentity ();
	float ratio=(float)w/(float)h;
	gluPerspective(45,ratio,1,20);
	_W=w;
	_H=h;
	glViewport (0, 0, (GLsizei) w, (GLsizei) h);
	glMatrixMode (GL_MODELVIEW);
	repaint();

  }

void SimpleGLWidget::ClearMesh()
{
	s->m.Clear();	
	repaint();
}


void SimpleGLWidget::mousePressEvent ( QMouseEvent * e )
{
	if (e->button()==Qt::LeftButton )
		Track.MouseDown(e->x(),_H-e->y(),vcg::Trackball::BUTTON_LEFT);
	else
	//test mass spring model
	if ((e->button()==Qt::RightButton)&&(_showslides))
	{
		float winz;
		double x;
		double y;
		double z;
		//LoadMatrix();
		glReadPixels(e->x(),_H-e->y(),1,1,GL_DEPTH_COMPONENT,GL_FLOAT,&winz);
		gluUnProject(e->x(),_H-e->y(),winz,modelMatrix,projection,viewport,&x,&y,&z);
		QString color="";
		color.sprintf("%i",s->gray_init);
		//w->Color->text()=color;
		SetExtractionParameters();
		//s->SetInitialBarycenter(Point3f(x,y,_numslide));
		s->InitSegmentation(Point3f(x,y,_numslide));
		repaint();
	}
	//vcg::tri::UpdateBounding<Segmentator::MyTriMesh>::Box(s->m);
}

void SimpleGLWidget::wheelEvent(QWheelEvent *e)
{
	if (!_showslides)
	{
		zoom+=e->delta()/120.f;
		repaint();
	}
	else
	{
		int oldnum=_numslide;
		_numslide+=e->delta()/120.f;
		if ((_numslide<0)||(_numslide>=s->V.dimZ()))
				_numslide=oldnum;
	}

	repaint();
}

void SimpleGLWidget::mouseReleaseEvent(QMouseEvent * e )
  {
	  Track.MouseUp(e->x(),_H-e->y(),vcg::Trackball::BUTTON_LEFT);
	  repaint();
  }

void SimpleGLWidget::Step()
	{
		if ((s!=0)&&(continue_int))
			{
				s->AutoStep();
				repaint();
			}
	}

void SimpleGLWidget::SetExtractionParameters()
	{
		float mass=atof(w->M_particles->text());
		float k_elanst=atof(w->K_elanst->text());
		float dihedral=atof(w->D_angle->text());
		float timestep=atof(w->T_step->text());
		float edge=atof(w->E_size->text());
		float tolerance=atof(w->Tolerance->text());
		float dinstance=atof(w->S_dist->text());
		//int color =atoi(w->Color->text());
		///to modify

		s->SetSegmentParameters(s->gray_init,tolerance,mass,k_elanst,dihedral,timestep,edge,Point3f(1.f,1.f,dinstance));
	}

void SimpleGLWidget::mouseMoveEvent ( QMouseEvent * e )
	{
		Track.MouseMove(e->x(),_H-e->y());
		repaint();
	}


void SimpleGLWidget::initializeGL(){

	GLfloat f[4]={0.2,0.2,0.2,1.f};
	GLfloat p[4]={3,3,5,0};
	glLightfv(GL_LIGHT0, GL_AMBIENT,f);
	glLightfv(GL_LIGHT1, GL_POSITION,p);
	glLightfv(GL_LIGHT1, GL_DIFFUSE,f);
	glLightfv(GL_LIGHT1, GL_SPECULAR,f);

	glEnable(GL_LIGHT0);
	glEnable(GL_LIGHT1);
	glEnable(GL_LIGHTING);	
	glEnable(GL_DEPTH_TEST);
	glDepthFunc(GL_LESS);
	glPolygonMode(GL_FRONT,GL_FILL);
	glEnable(GL_BACK);
	glCullFace(GL_BACK);

	}