LED-Cube/CubeControl/Led3D.java

/*
* Led3D.java
*
*
* Copyright 2011 Thomas Buck <xythobuz@me.com>
* Copyright 2011 Max Nuding <max.nuding@gmail.com>
* Copyright 2011 Felix Bäder <baeder.felix@gmail.com>
*
* This file is part of LED-Cube.
*
* LED-Cube 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.
*
* LED-Cube 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 LED-Cube. If not, see <http://www.gnu.org/licenses/>.
*/

import com.sun.j3d.utils.universe.*;
import com.sun.j3d.utils.geometry.*;
import javax.media.j3d.*;
import javax.vecmath.*;
import com.sun.j3d.utils.behaviors.mouse.*;
import java.awt.Color;
import com.sun.j3d.utils.image.TextureLoader;

/**
 * This class is responsible for displaying the 3D View of our Cube.
 * @author Thomas Buck
 * @author Max Nuding
 * @author Felix Bäder
 * @version 1.0
 */

public class Led3D {
	private Canvas3D canvas = null;
	private SimpleUniverse universe = null;
	private BranchGroup group = null;
	private Transform3D trans3D = null;
	private BranchGroup inBetween = null;
	private TransformGroup transGroup = null;

	private Sphere[][][] leds = new Sphere[8][8][8];
	private static ColoringAttributes redColor = new ColoringAttributes(new Color3f(1.0f, 0.0f, 0.0f), ColoringAttributes.FASTEST);
	private static ColoringAttributes whiteColor = new ColoringAttributes(new Color3f(0.2f, 0.2f, 0.2f), ColoringAttributes.FASTEST);
	private static Material whiteMat = new Material(new Color3f(0.2f, 0.2f, 0.2f), new Color3f(0.0f, 0.0f, 0.0f), new Color3f(0.2f, 0.2f, 0.2f), new Color3f(0.2f, 0.2f, 0.2f), 64.0f);
	private static Material redMat = new Material(new Color3f(1.0f, 0.0f, 0.0f), new Color3f(1.0f, 0.0f, 0.0f), new Color3f(1.0f, 0.0f, 0.0f), new Color3f(1.0f, 0.0f, 0.0f), 64.0f);

  /**
   * @param canv The Canvas3D we render our cube in
   */
  public Led3D(Canvas3D canv) {

    canvas = canv;
    group = new BranchGroup();
    // Position viewer, so we are looking at object
    trans3D = new Transform3D();
	Matrix4d mat = new Matrix4d();
	mat.setRow(0, 0.744, 0.0237, -0.66756, -0.34);
	mat.setRow(1, 0.136, -0.9837, 0.117, 3.24);
	mat.setRow(2, -0.6536, -0.1785, -0.735, -8.32);
	mat.setRow(3, 0.0, 0.0, 0.0, 1.0);
	trans3D.set(mat);

	transGroup = new TransformGroup(trans3D);
	transGroup.setCapability(TransformGroup.ALLOW_TRANSFORM_READ);
	transGroup.setCapability(TransformGroup.ALLOW_TRANSFORM_WRITE);
    ViewingPlatform viewingPlatform = new ViewingPlatform();
    Viewer viewer = new Viewer(canvas);
    universe = new SimpleUniverse(viewingPlatform, viewer);

    BoundingBox boundBox = new BoundingBox(new Point3d(-5.0, -5.0, -5.0), new Point3d(13.0, 13.0, 13.0));
    // roration with left mouse button
    MouseRotate behaviour = new MouseRotate(transGroup);
	behaviour.setSchedulingBounds(boundBox);
    transGroup.addChild(behaviour);

    // zoom with middle mouse button
    MouseZoom beh2 = new MouseZoom(transGroup);
	beh2.setSchedulingBounds(boundBox);
	transGroup.addChild(beh2);

    // move with right mouse button
    MouseTranslate beh3 = new MouseTranslate(transGroup);
    beh3.setSchedulingBounds(boundBox);
	transGroup.addChild(beh3);

	group.addChild(createBackground());

    // Add all our led sphares to the universe
    for (int x = 0; x < 8; x++) {
      for (int y = 0; y < 8; y++) {
        for (int z = 0; z < 8; z++) {
          Appearance a = new Appearance();
          a.setMaterial(whiteMat);
		  a.setColoringAttributes(whiteColor);
		  
		  leds[x][y][z] = new Sphere(0.08f, Sphere.ENABLE_APPEARANCE_MODIFY, a);

          TransformGroup tg = new TransformGroup();
          Transform3D transform = new Transform3D();
          Vector3f vector = new Vector3f(x, y, z);
          transform.setTranslation(vector);
          tg.setTransform(transform);
          tg.addChild(leds[x][y][z]);
          transGroup.addChild(tg);

		  drawLedFeetVertical((double)x, y + 0.5, (double)z, 0.9f, 0.01f);
		  if (x < 7)
			  drawLedFeetHorizontal(x + 0.5, (double)y, (double)z, 0.9f, 0.01f, 0);
        }
      }
	  // 8 times, use x as y
	  drawLedFeetHorizontal(0.5, (double)x, 3.5, 7.0f, 0.02f, 90);
	  drawLedFeetHorizontal(6.5, (double)x, 3.5, 7.0f, 0.02f, 90);
	  drawLedFeetHorizontal(3.5, (double)x, 3.5, 7.0f, 0.02f, 90);
    }

    // Add an ambient light
    Color3f light2Color = new Color3f(1.0f, 1.0f, 1.0f);
    AmbientLight light2 = new AmbientLight(light2Color);
    BoundingSphere bounds = new BoundingSphere(new Point3d(0.0, 0.0, 0.0), 100.0);
    light2.setInfluencingBounds(bounds);
    light2.setEnable(true);
    transGroup.addChild(light2);
	group.addChild(transGroup);
	universe.addBranchGraph(group); // Add group to universe
  }

	private void drawLedFeetVertical(double x, double y, double z, float length, float rad) {
		// draw Feet going down
		Appearance feetApp = new Appearance();
		feetApp.setMaterial(whiteMat);
		feetApp.setColoringAttributes(whiteColor);
		Cylinder c = new Cylinder(rad, length, feetApp);

		TransformGroup tg = new TransformGroup();
		Transform3D transform = new Transform3D();
		Vector3d vector = new Vector3d(x, y, z);
		transform.setTranslation(vector);
		tg.setTransform(transform);
		tg.addChild(c);

		transGroup.addChild(tg);
	}

	private void drawLedFeetHorizontal(double x, double y, double z, float length, float rad, int deg) {
		// draw Feet going down
		Appearance feetApp = new Appearance();
		feetApp.setMaterial(whiteMat);
		feetApp.setColoringAttributes(whiteColor);
		Cylinder c = new Cylinder(rad, length, feetApp);

		TransformGroup tg = new TransformGroup();
		Transform3D transform = new Transform3D();
		Vector3d vector = new Vector3d(x, y, z);
		transform.rotZ(Math.toRadians(90));
		if (deg != 0)
			transform.rotX(Math.toRadians(deg));
		transform.setTranslation(vector);
		tg.setTransform(transform);
		tg.addChild(c);

		transGroup.addChild(tg);
	}

	/**
	 * Rotate the cube back to its initial position.
	 */
	public void resetView() {
		Matrix4d mat = new Matrix4d();
		mat.setRow(0, 0.744, 0.0237, -0.66756, -0.34);
		mat.setRow(1, 0.136, -0.9837, 0.117, 3.24);
		mat.setRow(2, -0.6536, -0.1785, -0.735, -8.32);
		mat.setRow(3, 0.0, 0.0, 0.0, 1.0);
		trans3D.set(mat);
		transGroup.setTransform(trans3D);
	}

	/**
	 * Prints the translation matrix that is changed by moving/rotating the 3D Cube with your mouse.
	 */
  	public void printTranslationData() {
		Matrix4d mat = new Matrix4d();
		Transform3D t = new Transform3D();
		transGroup.getTransform(t);
		t.get(mat);
		String s = mat.toString();
		System.out.println(s.replaceAll(", ", "\t"));
	}

	/**
	 * Sets the data that is displayed by the LEDs
	 * @param data 64 byte array with the data (8 bits/LEDs per byte)
	 */
	public void setData(short[] data) {
		for (int y = 0; y < 8; y++) {
			for (int z = 0; z < 8; z++) {
				for (int x = 0; x < 8; x++) {
					Appearance a = new Appearance();
					if ((data[y + (z * 8)] & (1 << x)) != 0) {
						// Activate led
						a.setColoringAttributes(redColor);
						a.setMaterial(redMat);
					} else {
						// Deactivate led
						a.setColoringAttributes(whiteColor);
						a.setMaterial(whiteMat);
					}
					leds[x][y][z].setAppearance(a);
				}
			}
		}
	}

	// create nice background
	private Background createBackground() {
		ImageComponent2D image = new TextureLoader(getClass().getResource("bg.png"), null).getImage();
		Background bg = new Background(image);
		bg.setApplicationBounds(new BoundingSphere(new Point3d(0.0, 0.0, 0.0), 100.0));
		return bg;
	}
}