bundles/common/org.eclipse.apogy.common.topology.addons.primitives.ui.jme3/src/org/eclipse/apogy/common/topology/addons/primitives/ui/jme3/scene_objects/SpherePrimitiveJME3SceneObject.java

/*******************************************************************************
 * Copyright (c) 2018 Agence spatiale canadienne / Canadian Space Agency 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     Pierre Allard, 
 *     Regent L'Archeveque,
 *     Sebastien Gemme  - initial API and implementation
 *     
 * SPDX-License-Identifier: EPL-1.0
 *     
 *******************************************************************************/
package org.eclipse.apogy.common.topology.addons.primitives.ui.jme3.scene_objects;

import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;

import org.eclipse.apogy.common.topology.addons.primitives.ApogyCommonTopologyAddonsPrimitivesPackage;
import org.eclipse.apogy.common.topology.addons.primitives.SpherePrimitive;
import org.eclipse.apogy.common.topology.ui.jme3.JME3RenderEngineDelegate;
import org.eclipse.apogy.common.topology.ui.jme3.JME3Utilities;
import org.eclipse.apogy.common.topology.ui.jme3.scene_objects.DefaultJME3SceneObject;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.core.runtime.jobs.Job;
import org.eclipse.emf.common.notify.Adapter;
import org.eclipse.emf.common.notify.Notification;
import org.eclipse.emf.common.notify.Notifier;
import org.eclipse.swt.graphics.RGB;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.jme3.asset.AssetManager;
import com.jme3.material.Material;
import com.jme3.math.ColorRGBA;
import com.jme3.scene.Geometry;
import com.jme3.scene.Mesh;
import com.jme3.scene.shape.Sphere;

public class SpherePrimitiveJME3SceneObject extends DefaultJME3SceneObject<SpherePrimitive> {
	private static final Logger Logger = LoggerFactory.getLogger(SpherePrimitiveJME3SceneObject.class);

	public static int Z_SAMPLES = 100;
	public static int RADIAL_SAMPLES = 36;

	private Adapter sphereAdapter;

	private Geometry sphereGeometry = null;

	private AssetManager assetManager;

	public SpherePrimitiveJME3SceneObject(SpherePrimitive topologyNode,
			JME3RenderEngineDelegate jme3RenderEngineDelegate) {
		super(topologyNode, jme3RenderEngineDelegate);

		this.assetManager = this.jme3Application.getAssetManager();

		Job job = new Job("SpherePrimitiveJME3SceneObject : Initialize Geometry.") {
			@Override
			protected IStatus run(IProgressMonitor monitor) {
				SpherePrimitiveJME3SceneObject.this.jme3Application.enqueue(new Callable<Object>() {
					@Override
					public Object call() throws Exception {
						createGeometry();
						return null;
					}
				});
				return Status.OK_STATUS;
			}
		};
		job.schedule();

		topologyNode.eAdapters().add(getSphereAdapter());
	}

	@Override
	public void updateGeometry(float tpf) {
		if (this.sphereGeometry != null)
			getAttachmentNode().detachChild(this.sphereGeometry);

		// Create new geometry
		this.sphereGeometry = createSphereGeometry();
		getAttachmentNode().attachChild(this.sphereGeometry);
	}

	@Override
	public List<Geometry> getGeometries() {
		List<Geometry> geometries = new ArrayList<Geometry>();
		geometries.add(this.sphereGeometry);
		return geometries;
	}

	private void createGeometry() {
		this.sphereGeometry = createSphereGeometry();
		getAttachmentNode().attachChild(this.sphereGeometry);
	}

	private Geometry createSphereGeometry() {
		Mesh mesh = createMesh();
		Geometry geometry = new Geometry("Sphere", mesh);
		geometry.setMaterial(createMaterial());
		return geometry;
	}

	private Mesh createMesh() {
		Sphere sphere = new Sphere(Z_SAMPLES, RADIAL_SAMPLES, (float) getTopologyNode().getRadius());
		return sphere;
	}

	@Override
	public void setColor(RGB rgb) {
		this.jme3Application.enqueue(new Callable<Object>() {
			@Override
			public Object call() throws Exception {
				try {
					if (SpherePrimitiveJME3SceneObject.this.sphereGeometry != null) {
						Material mat = createMaterial();
						mat.setColor("Diffuse", JME3Utilities.convertToColorRGBA(rgb));
						mat.setColor("Ambient", JME3Utilities.convertToColorRGBA(rgb));
						mat.setColor("Specular", JME3Utilities.convertToColorRGBA(rgb));
						SpherePrimitiveJME3SceneObject.this.sphereGeometry.setMaterial(mat);
					}
				} catch (Throwable t) {
					Logger.error("Failed to set color to <" + rgb + ">.", t);
				}

				return null;
			}
		});

	}

	private Material createMaterial() {
		Material mat = new Material(this.assetManager, "Common/MatDefs/Light/Lighting.j3md");

		if (getColor() != null) {
			mat.setColor("Diffuse", JME3Utilities.convertToColorRGBA(getColor()));
			mat.setColor("Ambient", JME3Utilities.convertToColorRGBA(getColor()));
			mat.setColor("Specular", JME3Utilities.convertToColorRGBA(getColor()));
		} else {
			mat.setColor("Diffuse", ColorRGBA.White);
			mat.setColor("Ambient", ColorRGBA.White);
			mat.setColor("Specular", ColorRGBA.White);
		}

		mat.setFloat("Shininess", 64f);
		mat.setBoolean("UseMaterialColors", true);

		return mat;
	}

	private Adapter getSphereAdapter() {
		if (this.sphereAdapter == null) {
			this.sphereAdapter = new Adapter() {
				@Override
				public Notifier getTarget() {
					return null;
				}

				@Override
				public boolean isAdapterForType(Object type) {
					return false;
				}

				@Override
				public void notifyChanged(Notification notification) {
					if (notification.getFeatureID(
							SpherePrimitive.class) == ApogyCommonTopologyAddonsPrimitivesPackage.SPHERE_PRIMITIVE__RADIUS) {
						requestUpdate();
					}
				}

				@Override
				public void setTarget(Notifier newTarget) {
				}
			};
		}

		return this.sphereAdapter;
	}
}