ProcessingGcodeViewer.pde

/*
Author: Noah Levy

This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program 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 General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see http://www.gnu.org/licenses.
*/
import java.io.BufferedReader;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.util.ArrayList;


import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.filechooser.FileFilter;
import javax.swing.filechooser.FileNameExtensionFilter;
import javax.vecmath.Point3f;
import javax.media.opengl.*;
import javax.swing.*;

import peasy.PeasyCam;

import controlP5.*;
import processing.opengl.*;


	private boolean dualExtrusionColoring = false ;

	PeasyCam cam; //The camera object, PeasyCam extends the default processing camera and enables user interaction
	ControlP5 controlP5; //ControlP5 object, ControlP5 is a library used for drawing GUI items
	PMatrix3D currCameraMatrix; //By having 2 camera matrix I'm able to switch a 3D pannable area and a fixed gui in relation to the user
	PGraphicsOpenGL g3; //The graphics object, necessary for openGL manipulations
        ControlGroup panButts; //The group of controlP5 buttons related to panning
        private boolean is2D = false;
        private boolean isDrawable = false; //True if a file is loaded; false if not
        private boolean isPlatformed = false;
        private boolean isSpeedColored = true;
	private String gCode; //The path of the gcode File
	private ArrayList<LineSegment> objCommands; //An ArrayList of linesegments composing the model
	private int curScale = 20; 
        private int curLayer = 0;
        

	////////////ALPHA VALUES//////////////

	private final int TRANSPARENT = 20;
	private final int SOLID = 100;
	private final int SUPERSOLID = 255;

	//////////////////////////////////////

	////////////COLOR VALUES/////////////

	
	private final int RED = color(255,200,200);
	private final int BLUE = color(0, 255, 255);
	private final int PURPLE = color(242, 0, 255);
	private final int YELLOW = color(237, 255, 0);
	private final int OTHER_YELLOW = color(234, 212, 7);
	private final int GREEN = color(33, 255, 0);
	private final int WHITE = color(255, 255, 255);

	//////////////////////////////////////

	///////////SPEED VALUES///////////////

	private float LOW_SPEED = 700;
	private float MEDIUM_SPEED = 1400;
	private float HIGH_SPEED = 1900;

	//////////////////////////////////////

	//////////SLIDER VALUES/////////////

	private int minSlider = 1;
	private int maxSlider;
	private int defaultValue;

	////////////////////////////////////

	/////////Canvas Size///////////////

	private int xSize = 5*screen.width/6;
	private int ySize = 5*screen.height/6;

	////////////////////////////////////
        private int camOffset = 70;
        private int textBoxOffset = 200;
        
        /*
        private static String argument =null;


        public static void main(String args[]) {
        PApplet.main(new String[] {"ProcessingGcodeViewer" });
        if(args.length >= 1)
        {
        argument = args[0];
        }
        }
        */
        
        
	public void setup() {
              
               //gCode = ("RectangularServoHorn2.gcode");
               //gCode = ("C:/Users/noah/Downloads/RoboArm/pig.gcode");
		//gCode = ("C:/Users/noah/Dropbox/Rep26Stuff/Example Files/Cupcake/Merged.gcode");
                size(xSize,ySize, OPENGL); //OpenGL is the renderer; untested with p3d
		frameRate(25);
                hint(ENABLE_NATIVE_FONTS); //I've found this hint prevents some weird ControlP5 font not in box issues.
		background(0); //Make background black

		g3 = (PGraphicsOpenGL)g;
                hint(DISABLE_OPENGL_2X_SMOOTH); //AntiAliasing really hurts performance and I've found it to be distracting when on
                noSmooth();
                /*
                GL gl = g3.beginGL();  // always use the GL object returned by beginGL
               gl.glHint(gl.GL_CLIP_VOLUME_CLIPPING_HINT_EXT, gl.GL_FASTEST); //This line does not work with discrete graphcis
                g3.endGL();
                */
                setupCamera();

		controlP5 = new ControlP5(this);
              
              /*
                Textfield lowField = controlP5.addTextfield("lowSpeed",textBoxOffset,30,40,20);
                Textfield mediumField = controlP5.addTextfield("mediumSpeed",textBoxOffset + 60,30,40,20);
                Textfield highField = controlP5.addTextfield("highSpeed",textBoxOffset + 120,30,40,20);
                lowField.setText(Float.toString(LOW_SPEED));
                mediumField.setText(Float.toString(MEDIUM_SPEED));
                highField.setText(Float.toString(HIGH_SPEED));
               */
                
                CheckBox cb =  controlP5.addCheckBox("2DBox", width - 200, 38); //The object that contains the gui checkoxes
                cb.addItem("2D View",0);
                cb.addItem("Enable DualExtrusion Coloring",0);
                cb.addItem("Show Platform",0);
                //cb.addItem("Consecutive Coloring",0);
               // cb.addItem("Full Screen",0);
                controlP5.setAutoDraw(false); //THIS IS VERY IMPORTANT, this allows me to designate specifically when the GUI gets drawn, without this and my gui() method the buttons would exist in the 3D plane with everything else
      		controlP5.addButton("Choose File...",10f,(width - 110),30,80,20); //Opens JFileChooser
                
                make3D(); //By default the model should be displayed in 3D
                if(gCode != null)
                {
                generateObject();
                }
	}
	public void controlEvent(ControlEvent theEvent) 
        {
          if(theEvent.isGroup()) 
          {
            if(theEvent.group().name() == "2DBox")
            {
             int i = 0;
             int choice2D = (int)theEvent.group().arrayValue()[0];
            println("2D view is" + choice2D);
            if(choice2D == 1)
            {
            make2D();
            }
            if(choice2D == 0)
            {
            make3D();
            }
           int dualChoice = (int)theEvent.group().arrayValue()[1];
           
           if(dualChoice == 1)
           {
             dualExtrusionColoring = true; 
  
           }
          if(dualChoice == 0)
           {
            dualExtrusionColoring = false; 
           }
          int platformChoice = (int)theEvent.group().arrayValue()[2];
            if(platformChoice == 1)
           {
             isPlatformed = true; 
  
           }
          if(platformChoice == 0)
           {
            isPlatformed = false; 
           }
            }
          }
          else if(theEvent.controller().name() == "Choose File...")
          {
          selectFile();
          }
          else if(theEvent.controller().name() == "lowSpeed")
          {
            LOW_SPEED = Float.parseFloat(theEvent.controller().stringValue());
          }
          else if(theEvent.controller().name() == "mediumSpeed")
          {
            MEDIUM_SPEED = Float.parseFloat(theEvent.controller().stringValue());
          }
          else if(theEvent.controller().name() == "highSpeed")
          {
            HIGH_SPEED = Float.parseFloat(theEvent.controller().stringValue());            
          }
          else
          {
            float pos[] = cam.getLookAt();
              if(theEvent.controller().name() == "Left")
              {
              cam.lookAt(pos[0] - 1,pos[1],pos[2],0);
              }
               else if(theEvent.controller().name() == "Up")
              {
              cam.lookAt(pos[0],pos[1] - 1,pos[2],0);
              }
               else if(theEvent.controller().name() == "Right")
              {
              cam.lookAt(pos[0] + 1,pos[1],pos[2],0);
              }
               else if(theEvent.controller().name() == "Down")
              {
              cam.lookAt(pos[0],pos[1] + 1,pos[2],0);
              }
          }
	}
        public void make2D()
        {
          is2D = true;
          cam.reset();
          cam.setActive(false);
          if(panButts == null)
          {
          panButts = panButtons();
          }
        }
        public void make3D()
        {
          is2D = false;
          cam.rotateX(-.37); //Make it obvious it is 3d to start
	  cam.rotateY(.1);
          cam.setActive(true);
          
          controlP5.remove("Pan Buttons");
          panButts = null;
        }
        public void generateObject()
	{
  		scale(1, -1, 1); //orient cooridnate plane right-handed props to whosawwhatsis for discovering this

		GcodeViewParse gcvp = new GcodeViewParse();
		objCommands = (gcvp.toObj(readFiletoArrayList(gCode)));
                float bounds[] = gcvp.getExtremes();
                System.out.println("Object bounds = " + Arrays.toString(bounds));
                float cent[] = findCenter(bounds);
                System.out.println("Center = " + Arrays.toString(cent));
                //cam.lookAt(-1*bounds[0], -1*bounds[1], 0, camOffset);
                println("objComBumands :" + objCommands.size());
		maxSlider = objCommands.get(objCommands.size() - 1).getLayer() - 1; // Maximum slider value is highest layer
		defaultValue = maxSlider;
                controlP5.remove("Layer Slider");
		controlP5.addSlider("Layer Slider",minSlider,maxSlider,defaultValue,20,100,10,300).setNumberOfTickMarks(maxSlider);
		//controlP5.addControlWindow("ControlWindow", 50, 50, 20, 20);
		
                curLayer = (int)Math.round(controlP5.controller("Layer Slider").value());
	      isDrawable = true;
         }
         private float[] findCenter(float[] bounds)
         {
          float cent[] = {(bounds[0] + bounds[3]),(bounds[1] + bounds[4]),(bounds[2] + bounds[5])};
          return cent;
         }
         private void setupCamera()
         {
          setupCamera(0,0); 
         }
         private void setupCamera(float centerX, float centerY)
         {
           float fov = PI/3.0;
                float cameraZ = (height/2.0) / tan(fov/2.0);
                perspective(fov, float(width)/float(height), 0.1, cameraZ*10.0); //Calling perspective allows me to set 0.1 as the frustum's zNear which prevents a bunch of clipping issues.
                cam = new PeasyCam(this, centerX,  centerY, 0, camOffset); // parent, x, y, z, initial distance
         
		cam.setMinimumDistance(2);
		cam.setMaximumDistance(200);
                cam.setResetOnDoubleClick(false);
           
         }
         public ControlGroup panButtons()
         {
            ControlGroup panButts = controlP5.addGroup("Pan Buttons",20,height - 100);
            panButts.hideBar();
            //DragHandler panHandle = cam.getPanDragHandler();
            controlP5.addBang("Up",30,4,20,20).setGroup(panButts);
            controlP5.addBang("Left",0,34,20,20).setGroup(panButts);
            controlP5.addBang("Right",60,34,20,20).setGroup(panButts);
            controlP5.addBang("Down",30,64,20,20).setGroup(panButts);
            return panButts;
         }
	public void draw() {
                lights();
              //directionalLight(255, 255, 255, -1, 0, 0);
//spotLight(51, 102, 126, 80, 20, 40, -1, 0, 0, PI/2, 2);

		//ambientLight(255,255,255);
                background(0);
                
		hint(ENABLE_DEPTH_TEST);
		pushMatrix();
		noSmooth();
               if(isPlatformed)
                {
                  fill(6,13,137);
                  beginShape();
                  vertex(-50,-50,0);
                  vertex(-50,50,0);
                  vertex(50,50,0);
                  vertex(50,-50,0);
                  endShape();
                  noFill();
                }
                if(isDrawable)
                {
		scale(1, -1, 1); //orient cooridnate plane right-handed props to whosawwhatsis for discovering this

		float[] points = new float[6];
               
		int maxLayer = (int)Math.round(controlP5.controller("Layer Slider").value());
                
		int curTransparency = 0;
                int curColor = 0;
               beginShape(LINES);
		for(LineSegment ls : objCommands)
		{
			if(ls.getLayer() < maxLayer)
			{
				curTransparency = SOLID;
			}
			if(ls.getLayer() == maxLayer)
			{
				curTransparency = SUPERSOLID;
			}
			if(ls.getLayer() > maxLayer)
			{
				curTransparency = TRANSPARENT;
			}
			if(!ls.getExtruding())
			{
				stroke(WHITE,TRANSPARENT);
			}
			if(!dualExtrusionColoring)
			{
				if(ls.getExtruding())
				{
                                    if(isSpeedColored)
                                    {
					if(ls.getSpeed() > LOW_SPEED && ls.getSpeed() < MEDIUM_SPEED)
					{
						stroke(PURPLE, curTransparency);
					}
					if(ls.getSpeed() > MEDIUM_SPEED && ls.getSpeed() < HIGH_SPEED)
					{
						stroke(BLUE, curTransparency);
					}
					else if(ls.getSpeed() >= HIGH_SPEED)
					{
						stroke(OTHER_YELLOW, curTransparency);
					}
					else //Very low speed....
					{
						stroke(GREEN, curTransparency);
					}
                                    }
                                    if(!isSpeedColored)
                                    {
                                        if(curColor == 0)
                                        {
                                         stroke(GREEN, SUPERSOLID);
                                        } 
                                        if(curColor == 1)
                                        {
                                        stroke(RED, SUPERSOLID); 
                                        } 
                                        if(curColor == 2)
                                        {
                                         stroke(BLUE, SUPERSOLID);
                                        } 
                                        if(curColor == 3)
                                        {
                                         stroke(YELLOW, SUPERSOLID);
                                        }
                                         curColor++; 
                                        if(curColor == 4)
                                        {
                                         curColor = 0; 
                                        }
                                    }
				}
			}
			if(dualExtrusionColoring)
			{
				if(ls.getExtruding())
				{
					if(ls.getToolhead() == 0)
					{
						stroke(BLUE, curTransparency);
					}
					if(ls.getToolhead() == 1)
					{
						stroke(GREEN, curTransparency);
					}
				}
			}
                      
                        if(!is2D || (ls.getLayer() == maxLayer))
                        {
			points = ls.getPoints();
                        
			//vertex(points[0],points[1],points[2],points[3], points[4], points[5]);
                          vertex(points[0],points[1],points[2]);
                          vertex(points[3],points[4],points[5]);
        		}
                    }                  endShape();
                    if((curLayer != maxLayer) && is2D)
                    {
                      cam.setDistance(cam.getDistance() + (maxLayer - curLayer)*.3,0);
                    }
                     curLayer = maxLayer;
                }
		popMatrix();
		// makes the gui stay on top of elements
		// drawn before.
		hint(DISABLE_DEPTH_TEST);
                
		gui();
	}

	private void gui() {
		noSmooth();
		//currCameraMatrix = new PMatrix3D(g3.camera);
		//camera();
                cam.beginHUD();
		controlP5.draw();
                cam.endHUD();
		//g3.camera = currCameraMatrix;
	}
        void selectFile() {
          
          try
          {
            
            SwingUtilities.invokeLater(new Runnable() {
            public void run()
            {
               JFileChooser fc = new JFileChooser(".");
                FileFilter gcodeFilter = new FileNameExtensionFilter("Gcode file", "gcode", "ngc");		
               fc.setDialogTitle("Choose a file...");
               fc.setFileFilter(gcodeFilter);
          
               int returned = fc.showOpenDialog(frame);
                if (returned == JFileChooser.APPROVE_OPTION) 
                {
                  isDrawable = false;
                 File file = fc.getSelectedFile();
                  gCode = (String)file.getPath();
                 println(gCode);
                generateObject();
                  
                }
            }
        });
          }
          catch(Exception e)
          {
            e.printStackTrace();
          }
        
      }
	public void mouseMoved() {
		if(mouseX < 35 || (mouseY < 50 && mouseX > (width - 130)) || is2D)
		{
			cam.setActive(false);
		}
		else
		{
			cam.setActive(true);
		}
	}

	public ArrayList<String> readFiletoArrayList(String s) {
		ArrayList<String> vect;
		String lines[] = loadStrings(s);
                vect = new ArrayList<String>(Arrays.asList(lines));
		return vect;
	}
public void mousePressed() {
 redraw();
}