bd40d1301181a138e2df01ef132ccd1026b271d8
[anneal.git] / src / Main.java
1 import java.awt.*;
2 import java.awt.event.*;
3 import javax.swing.*;
4 import javax.media.opengl.*;
5 import javax.media.opengl.glu.*;
6
7 public class Main implements GLEventListener {
8
9     private Geom geom = new Geom();
10
11     /** magnification factor */
12     private static final float MAG = 1;
13
14     public Main(StlFile stlf) {
15         for(int i=0; i<stlf.coordArray.length; i+=3) {
16             Geom.P p0 = geom.newP(stlf.coordArray[i+0].x * MAG, stlf.coordArray[i+0].y * MAG, stlf.coordArray[i+0].z * MAG);
17             Geom.P p1 = geom.newP(stlf.coordArray[i+1].x * MAG, stlf.coordArray[i+1].y * MAG, stlf.coordArray[i+1].z * MAG);
18             Geom.P p2 = geom.newP(stlf.coordArray[i+2].x * MAG, stlf.coordArray[i+2].y * MAG, stlf.coordArray[i+2].z * MAG);
19             Geom.V n  = geom.new V(stlf.normArray[i/3].x * MAG, stlf.normArray[i/3].y  * MAG, stlf.normArray[i/3].z * MAG);
20             Geom.T t  = geom.newT(geom.newE(p0, p1), geom.newE(p1, p2), geom.newE(p2, p0), n);
21         }
22     }
23
24     /**
25      * Take care of initialization here.
26      */
27     public void init(GLAutoDrawable gld) {
28         GL gl = gld.getGL();
29         GLU glu = new GLU();
30         gl.glClearColor(0.0f, 0.0f, 0.0f, 1.0f);
31         gl.glViewport(0, 0, 500, 300);
32         gl.glMatrixMode(GL.GL_PROJECTION);
33         gl.glLoadIdentity();
34         //glu.gluOrtho2D(0.0, 500.0, 0.0, 300.0);
35         display(gld);
36     }
37
38     public void reshape(GLAutoDrawable drawable, int x, int y, int width, int height) { }
39     public void displayChanged(GLAutoDrawable drawable, boolean modeChanged, boolean deviceChanged) { }
40     public void display(GLAutoDrawable drawable) {
41         float red = 0.0f;
42         float green = 0.0f;
43         float blue = 0.0f;
44         GL gl = drawable.getGL();
45         gl.glClear(GL.GL_COLOR_BUFFER_BIT);
46         gl.glPointSize(5.0f);
47
48         for(Geom.T t : geom) {
49             red -= .09f;
50             green -= .12f;
51             blue -= .15f;
52             if (red < 0.15) red = 1.0f;
53             if (green < 0.15) green = 1.0f;
54             if (blue < 0.15) blue = 1.0f;
55             gl.glColor3f(red, green, blue);
56             gl.glBegin(GL.GL_TRIANGLES);
57             t.glVertices(gl);
58             gl.glEnd();
59
60             Geom.P centroid = t.centroid();
61             gl.glBegin(GL.GL_LINES);
62             centroid.glVertex(gl);
63             centroid.plus(t.norm().times(t.diameter())).glVertex(gl);
64             gl.glEnd();
65             
66
67         }
68     }
69
70     public static void main(String[] s) throws Exception {
71         StlFile stlf = new StlFile();
72         stlf.load("teapot.stl");
73         Main main = new Main(stlf);
74         Frame f = new Frame();
75         GLCapabilities glcaps = new GLCapabilities();
76         GLCanvas glcanvas = new GLCanvas();
77         glcanvas.addGLEventListener(main);
78         f.add(glcanvas, BorderLayout.CENTER);
79         f.show();
80         f.setSize(500, 300);
81     }
82
83 }