X-Git-Url: http://git.megacz.com/?p=anneal.git;a=blobdiff_plain;f=src%2Fedu%2Fberkeley%2Fqfat%2FMeshViewer.java;h=c5c6a3b9cd088fc6249f7df8e672527331da3d67;hp=7dc5b0a0cc31dba3d36f687661d199b4ba1f8511;hb=5e1dfbf83814f05b602d6463fca86470531fe4fe;hpb=2f48eabf8a07e99905e1eae0b64b5a2abecb01fe diff --git a/src/edu/berkeley/qfat/MeshViewer.java b/src/edu/berkeley/qfat/MeshViewer.java index 7dc5b0a..c5c6a3b 100644 --- a/src/edu/berkeley/qfat/MeshViewer.java +++ b/src/edu/berkeley/qfat/MeshViewer.java @@ -1,21 +1,32 @@ package edu.berkeley.qfat; +import java.io.*; import java.awt.*; import java.awt.event.*; import javax.swing.*; import javax.media.opengl.*; import javax.media.opengl.glu.*; +import com.sun.opengl.util.*; import java.util.*; import edu.berkeley.qfat.geom.*; import edu.berkeley.qfat.geom.Point; public class MeshViewer implements GLEventListener, MouseListener, MouseMotionListener, KeyListener, MouseWheelListener { - public Mesh tile = new Mesh(); - public Mesh goal = new Mesh(); - public Matrix[] translations; - public Mesh.Vert[] points; + public Mesh tile = new Mesh(false); + public Mesh goal = new Mesh(false); + public Matrix[] transforms; + public Mesh.Vertex[] points; + public double temp; + public boolean tileon = true; + public boolean tilemeshon = false; + public boolean goalon = true; + public boolean anneal = false; + public boolean neighbors = false; + public boolean neighborsWire = false; + public boolean neighborsWireOne = false; + public int breaks = 0; boolean alt = false; boolean shift = false; boolean control = false; @@ -30,8 +41,36 @@ public class MeshViewer implements GLEventListener, MouseListener, MouseMotionLi case KeyEvent.VK_CONTROL: control = true; break; case KeyEvent.VK_ALT: alt = true; break; case KeyEvent.VK_SHIFT: shift = true; break; + case KeyEvent.VK_SPACE: breaks++; break; + case KeyEvent.VK_UP: temp = temp * 2; break; + case KeyEvent.VK_ENTER: temp = 10; break; + case KeyEvent.VK_D: dump(); break; + case KeyEvent.VK_A: anneal = !anneal; break; + case KeyEvent.VK_N: neighbors = !neighbors; break; + case KeyEvent.VK_T: tileon = !tileon; break; + case KeyEvent.VK_G: goalon = !goalon; break; + case KeyEvent.VK_M: tilemeshon = !tilemeshon; break; } } + public synchronized void dump() { + try { + PrintWriter pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream("dump.stl"))); + pw.println("solid dump"); + for(Mesh.T t : tile) { + Vec normal = t.norm(); + pw.println("facet normal " + normal.x + " " + normal.y + " " + normal.z); + pw.println(" outer loop"); + for(Mesh.Vertex v : new Mesh.Vertex[] { t.v1(), t.v2(), t.v3() }) { + pw.println(" vertex " + v.p.x + " " + v.p.y + " " + v.p.z); + } + pw.println(" endloop"); + pw.println("endfacet"); + } + pw.println("endsolid dump"); + pw.flush(); + pw.close(); + } catch (Exception e) { throw new RuntimeException(e); } + } public void keyReleased(KeyEvent e) { switch(e.getKeyCode()) { case KeyEvent.VK_CONTROL: control = false; break; @@ -83,56 +122,118 @@ public class MeshViewer implements GLEventListener, MouseListener, MouseMotionLi gl.glMatrixMode(GL.GL_PROJECTION); gl.glLoadIdentity(); gl.glMatrixMode(GL.GL_MODELVIEW); + + float mat_specular[] = { 0.5f, 0.5f, 0.5f, 0.5f }; + float mat_shininess[] = { 50.0f }; + gl.glShadeModel(GL.GL_SMOOTH); + //gl.glMaterialfv(GL.GL_FRONT, GL.GL_DIFFUSE, mat_specular, 0); + //gl.glMaterialfv(GL.GL_FRONT, GL.GL_SPECULAR, mat_specular, 0); + //gl.glMaterialfv(GL.GL_FRONT, GL.GL_AMBIENT, new float[] { 0.3f, 0.3f, 0.3f, 0.3f }, 0); + //gl.glMaterialfv(GL.GL_FRONT, GL.GL_SHININESS, mat_shininess, 0); + gl.glLightfv(GL.GL_LIGHT0, GL.GL_POSITION, new float[] { 1.0f, 4.0f, -10.0f, 0.0f }, 0); + gl.glLightfv(GL.GL_LIGHT1, GL.GL_POSITION, new float[] { -10.0f, 10.0f, 10.0f, 0.0f }, 0); + gl.glLightfv(GL.GL_LIGHT2, GL.GL_POSITION, new float[] { 10.0f, -10.0f, 10.0f, 0.0f }, 0); + gl.glLightfv(GL.GL_LIGHT3, GL.GL_POSITION, new float[] { 10.0f, 10.0f, -10.0f, 0.0f }, 0); + gl.glLightfv(GL.GL_LIGHT4, GL.GL_POSITION, new float[] { -10.0f, 10.0f, -10.0f, 0.0f }, 0); + gl.glLightfv(GL.GL_LIGHT5, GL.GL_POSITION, new float[] { 10.0f, -10.0f, -10.0f, 0.0f }, 0); + gl.glEnable(GL.GL_LIGHTING); + gl.glEnable(GL.GL_LIGHT0); + /* + gl.glEnable(GL.GL_LIGHT1); + gl.glEnable(GL.GL_LIGHT2); + gl.glEnable(GL.GL_LIGHT3); + gl.glEnable(GL.GL_LIGHT4); + gl.glEnable(GL.GL_LIGHT5); + */ + gl.glColorMaterial(GL.GL_FRONT_AND_BACK, GL.GL_AMBIENT_AND_DIFFUSE); + gl.glEnable(GL.GL_COLOR_MATERIAL); + display(gld); } + public int temps; + public int accepts; + public int vertss; public void reshape(GLAutoDrawable drawable, int x, int y, int width, int height) { } public void displayChanged(GLAutoDrawable drawable, boolean modeChanged, boolean deviceChanged) { } public synchronized void display(GLAutoDrawable drawable) { - if (translations==null) return; + + if (transforms==null) return; + + + glcanvas.setSize(f.getWidth(), f.getHeight() - 100); + Graphics2D g = (Graphics2D)f.getGraphics(); + g.setColor(Color.black); + g.fillRect(0, f.getHeight()-100, f.getWidth(), f.getHeight()); + g.setColor(Color.red); + int top = f.getHeight()-70; + g.drawString("temperature: "+temps, 10, 30+top); + g.drawString("acceptance: "+accepts, 10, 50+top); + g.drawString("vertices: "+vertss, 10, 70+top); + g.fillRect(140, 25+top, temps, 10); + g.fillRect(140, 45+top, accepts, 10); + g.fillRect(140, 65+top, vertss, 10); + GL gl = drawable.getGL(); GLU glu = new GLU(); gl.glClear(GL.GL_COLOR_BUFFER_BIT | GL.GL_DEPTH_BUFFER_BIT); gl.glPointSize(5.0f); gl.glLoadIdentity(); - glu.gluPerspective(50-tz, ((float)drawable.getWidth())/drawable.getHeight(), 0.5, 10); - glu.gluLookAt(0, 0, -1, 0, 0, 0, 0, 1, 0); - gl.glTranslatef(tx/(float)20, ty/(float)20, 0); + glu.gluPerspective(50, ((float)drawable.getWidth())/drawable.getHeight(), 0.5, 10); + glu.gluLookAt(0, 0, (tz/10)-1, 0, 0, 0, 0, 1, 0); gl.glRotatef(anglex/3, 0, 1, 0); gl.glRotatef(angley/3, 1, 0, 0); gl.glBegin(GL.GL_TRIANGLES); - draw(gl, true, tile); + if (tileon) + draw(gl, true, safeTriangles); + if (tilemeshon) + draw(gl, false, safeTriangles); gl.glEnd(); + //draw(gl, false, tile); + gl.glBegin(GL.GL_TRIANGLES); gl.glColor4f((float)0.5, (float)0.5, (float)0.5, (float)0.8); - //draw(gl, false, goal); + if (goalon) + draw(gl, false, goal); gl.glEnd(); int i = 0; //gl.glDisable(GL.GL_DEPTH_TEST); gl.glColor4f(1,1,1,1); - for(Matrix m : translations) { + for(Matrix m : transforms) { //if (v1.z==0 && v1.y==0) continue; i++; - if (i != 1 /*&& i!=4*/) continue; + if (neighborsWireOne && i>1) continue; + //if (i>4) continue; + /* Point p = new Point(0, 0, 0).times(m); Vec v = new Vec(p.x, p.y, p.z); v = v.times((float)1.04); gl.glTranslatef(v.x, v.y, v.z); - draw(gl, false, tile); + */ + if (neighbors) draw(gl, true, safeTriangles, m); + else if (neighborsWire || neighborsWireOne) draw(gl, false, safeTriangles, m); + /* gl.glTranslatef(-v.x, -v.y, -v.z); + */ } //gl.glEnable(GL.GL_DEPTH_TEST); + gl.glEnable (GL.GL_LIGHTING); + } - private synchronized void draw(GL gl, boolean triangles, Mesh mesh) { + protected HashSet safeTriangles = new HashSet(); + + private void draw(GL gl, boolean triangles, Iterable tris) { draw(gl, triangles, tris, Matrix.ONE); } + private void draw(GL gl, boolean triangles, Iterable tris, Matrix m) { float red = 0.0f; float green = 0.0f; float blue = 0.0f; - for(Mesh.T t : mesh) { + synchronized(safeTriangles) { + for(Mesh.T t : tris) { if (red < 0.15) red = 1.0f; if (green < 0.15) green = 1.0f; if (blue < 0.15) blue = 1.0f; @@ -140,27 +241,45 @@ public class MeshViewer implements GLEventListener, MouseListener, MouseMotionLi green -= .12f; blue -= .15f; - if (triangles) switch(t.color) { + if (triangles) switch(t.color/*class*/) { case 0: gl.glColor4f((float)0.25, (float)0.25, (float)0.75, (float)0.3); break; case 1: gl.glColor4f((float)0.25, (float)0.75, (float)0.25, (float)0.3); break; case 2: gl.glColor4f((float)0.75, (float)0.25, (float)0.25, (float)0.3); break; case 3: gl.glColor4f((float)0.50, (float)0.50, (float)0.50, (float)0.3); break; + case 4: gl.glColor4f((float)0.25, (float)0.75, (float)0.75, (float)0.3); break; + case 5: gl.glColor4f((float)0.25, (float)0.75, (float)0.75, (float)0.3); break; + case 6: gl.glColor4f((float)0.75, (float)0.25, (float)0.75, (float)0.3); break; } + /* + if (t.e1().pair.t==null) gl.glColor4f((float)0.25, (float)0.25, (float)0.75, (float)0.3); + else if (t.e2().pair.t==null) gl.glColor4f((float)0.25, (float)0.25, (float)0.75, (float)0.3); + else if (t.e3().pair.t==null) gl.glColor4f((float)0.25, (float)0.25, (float)0.75, (float)0.3); + else gl.glColor4f((float)0.75, (float)0.25, (float)0.25, (float)0.3); + */ //gl.glBegin(GL.GL_LINES); if (triangles) { gl.glBegin(GL.GL_TRIANGLES); - t.glVertices(gl); + if (t.shouldBeDrawn()) { + m.times(t.norm()).glNormal(gl); + m.times(t.v1().goodp).glVertex(gl); + m.times(t.v2().goodp).glVertex(gl); + m.times(t.v3().goodp).glVertex(gl); + //t.glVertices(gl, m); + } gl.glEnd(); } else { + gl.glDisable(GL.GL_LIGHTING); gl.glBegin(GL.GL_LINES); - t.e1().p1.p.glVertex(gl); - t.e1().p2.p.glVertex(gl); - t.e2().p1.p.glVertex(gl); - t.e2().p2.p.glVertex(gl); - t.e3().p1.p.glVertex(gl); - t.e3().p2.p.glVertex(gl); + gl.glColor3f(1, 1, 1); + m.times(t.e1().p1.goodp).glVertex(gl); + m.times(t.e1().p2.goodp).glVertex(gl); + m.times(t.e2().p1.goodp).glVertex(gl); + m.times(t.e2().p2.goodp).glVertex(gl); + m.times(t.e3().p1.goodp).glVertex(gl); + m.times(t.e3().p2.goodp).glVertex(gl); gl.glEnd(); + gl.glEnable(GL.GL_LIGHTING); } Point centroid = t.centroid(); @@ -170,30 +289,36 @@ public class MeshViewer implements GLEventListener, MouseListener, MouseMotionLi centroid.glVertex(gl); centroid.plus(t.norm().times(t.diameter())).glVertex(gl); */ - + /* if (mesh==goal) - for(Mesh.Vert p : new Mesh.Vert[] { t.v1(), t.v2(), t.v3() }) { - p.p.glVertex(gl); - //p.plus(p.norm().times(p.score()*10)).glVertex(gl); - //p.partner().p.glVertex(gl); + for(Mesh.Vertex p : new Mesh.Vertex[] { t.v1(), t.v2(), t.v3() }) { + gl.glDisable(GL.GL_LIGHTING); + gl.glBegin(GL.GL_LINES); + gl.glColor3f(1, 1, 1); + p.p.glVertex(gl); + //p.p.plus(p.norm().times(p.score())).glVertex(gl); + if (p.nearest_in_other_mesh != null) p.nearest_in_other_mesh.p.glVertex(gl); //tile.nearest(p).centroid().glVertex(gl); + gl.glEnd(); + gl.glEnable(GL.GL_LIGHTING); } - + */ gl.glEnd(); - + } } } + + //private JTextArea ocanvas = new JTextArea(); + private JFrame f; private GLCanvas glcanvas; - public MeshViewer(Frame f) { - GLCapabilities glcaps = new GLCapabilities(); + public MeshViewer(JFrame f) { + this.f = f; + GLCapabilities glcaps = new GLCapabilities(); glcanvas = new GLCanvas(); glcanvas.addGLEventListener(this); f.add(glcanvas, BorderLayout.CENTER); - f.pack(); - f.show(); - f.setSize(900, 900); - f.doLayout(); + glcanvas.addMouseListener(this); glcanvas.addMouseMotionListener(this); glcanvas.addMouseWheelListener(this); @@ -202,4 +327,7 @@ public class MeshViewer implements GLEventListener, MouseListener, MouseMotionLi public void repaint() { glcanvas.repaint(); } + + + } \ No newline at end of file