/*
 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code 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
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package bugs;

import com.sun.javafx.perf.PerformanceTracker;
import javafx.animation.AnimationTimer;
import javafx.animation.KeyFrame;
import javafx.animation.Timeline;
import javafx.animation.TranslateTransition;
import javafx.application.Application;
import javafx.collections.ObservableFloatArray;
import javafx.event.ActionEvent;
import javafx.event.EventHandler;
import javafx.scene.*;
import javafx.scene.input.KeyEvent;
import javafx.scene.paint.Color;
import javafx.scene.paint.PhongMaterial;
import javafx.scene.shape.*;
import javafx.scene.transform.Rotate;
import javafx.stage.Stage;
import javafx.util.Duration;

public class SimpleIssueViewerApp extends Application {

    Group root;
    PointLight pointLight;
    MeshView meshView;
    TriangleMesh triMesh;
    PhongMaterial material;

    int divX = 70;
    int divY = 70;
    boolean smooth = true;
    int smoothGroups[];
    float rotateAngle = 0.0f;
    float funcValue = 0.0f;
    float resolution = 0.1f;
    boolean texture = false;
    boolean textureSwitch = false;

    final static float meshScale = 20;
    final static float minX = -10;
    final static float minY = -10;
    final static float maxX = 10;
    final static float maxY = 10;
    
    boolean enableChangeOne = false;
    boolean enableChangeAll = false;
    boolean oddity = false;

    double getSinDivX(double x, double y) {
        double r = Math.sqrt(x*x + y*y);
        return funcValue * (r == 0 ? 1 : Math.sin(r) / r);
    }

    void computeMesh(int subDivX, int subDivY, float scale) {
         final int pointSize = 3;
        int numDivX = subDivX + 1;
        float[] points = triMesh.getPoints().toArray(null);

        // Initial points and texCoords
        for (int y = 0; y <= subDivY; y++) {
            float dy = (float) y / subDivY;
            double fy = (1 - dy) * minY + dy * maxY;
//            System.err.println("dy = " + dy + ", fy = " + fy);
            for (int x = 0; x <= subDivX; x++) {
                float dx = (float) x / subDivX;
                double fx = (1 - dx) * minX + dx * maxX;
//                System.err.println("dx = " + dx + ", fx = " + fx);
                int index = y * numDivX * pointSize + (x * pointSize);
//                System.err.println("point index = " + index);
                points[index] = (float) fx * scale;
                points[index + 1] = (float) fy * scale;
                points[index + 2] = (float) getSinDivX(fx, fy) * scale;
//                System.err.println("points[" + (index + 2) + " = " + points[index + 2]);
            }
        }

        triMesh.getPoints().set(0, points, 0, points.length);
    }

    static TriangleMesh buildTriangleMesh(int subDivX, int subDivY,
            float scale) {

//        System.err.println("subDivX = " + subDivX + ", subDivY = " + subDivY);

        final int pointSize = 3;
        final int texCoordSize = 2;
        final int faceSize = 6; // 3 point indices and 3 texCoord indices per triangle
        int numDivX = subDivX + 1;
        int numVerts = (subDivY + 1) * numDivX;
        float points[] = new float[numVerts * pointSize];
        float texCoords[] = new float[numVerts * texCoordSize];
        int faceCount = subDivX * subDivY * 2;
        int faces[] = new int[ faceCount * faceSize]; 

        // Create points and texCoords
        for (int y = 0; y <= subDivY; y++) {
            float dy = (float) y / subDivY;
            double fy = (1 - dy) * minY + dy * maxY;
//            System.err.println("dy = " + dy + ", fy = " + fy);
            for (int x = 0; x <= subDivX; x++) {
                float dx = (float) x / subDivX;
                double fx = (1 - dx) * minX + dx * maxX;
//                System.err.println("dx = " + dx + ", fx = " + fx);
                int index = y * numDivX * pointSize + (x * pointSize);
//                System.err.println("point index = " + index);
                points[index] = (float) fx * scale;
                points[index + 1] = (float) fy * scale;
                points[index + 2] = 0.0f;
                index = y * numDivX * texCoordSize + (x * texCoordSize);
//                System.err.println("texCoord index = " + index);
                texCoords[index] = dx;
                texCoords[index + 1] = dy;
            }
        }

        // Create faces
        for (int y = 0; y < subDivY; y++) {
            for (int x = 0; x < subDivX; x++) {
                int p00 = y * numDivX + x;
                int p01 = p00 + 1;
                int p10 = p00 + numDivX;
                int p11 = p10 + 1;
                int tc00 = y * numDivX + x;
                int tc01 = tc00 + 1;
                int tc10 = tc00 + numDivX;
                int tc11 = tc10 + 1;

                int index = (y * subDivX * faceSize + (x * faceSize)) * 2;
//                System.err.println("face  0 index = " + index);
                faces[index + 0] = p00;
                faces[index + 1] = tc00;
                faces[index + 2] = p10;
                faces[index + 3] = tc10;
                faces[index + 4] = p11;
                faces[index + 5] = tc11;
                index += faceSize;
//                System.err.println("face  1 index = " + index);
                faces[index + 0] = p11;
                faces[index + 1] = tc11;
                faces[index + 2] = p01;
                faces[index + 3] = tc01;
                faces[index + 4] = p00;
                faces[index + 5] = tc00;
            }
        }
//        for(int i = 0; i < points.length; i++) {
//            System.err.println("points[" + i + "] = " + points[i]);
//        }
//        for(int i = 0; i < texCoords.length; i++) {
//            System.err.println("texCoords[" + i + "] = " + texCoords[i]);
//        }
//        for(int i = 0; i < faces.length; i++) {
//            System.err.println("faces[" + i + "] = " + faces[i]);
//        }
        TriangleMesh triangleMesh = new TriangleMesh();
        triangleMesh.getPoints().setAll(points);
        triangleMesh.getTexCoords().setAll(texCoords);
        triangleMesh.getFaces().setAll(faces);

        return triangleMesh;
    }

    private Scene buildScene(int width, int height, boolean depthBuffer) {

        triMesh = buildTriangleMesh(divX, divY, meshScale);
        smoothGroups = new int[divX * divY * 2];
        material = new PhongMaterial();
        material.setDiffuseColor(Color.LIGHTGRAY);
        material.setSpecularColor(Color.WHITE);
        material.setSpecularPower(64);
        meshView = new MeshView(triMesh);
        meshView.setMaterial(material);

        //Set Wireframe mode
        meshView.setDrawMode(DrawMode.FILL);
        meshView.setCullFace(CullFace.BACK);

        final Group grp1 = new Group(meshView);
        grp1.setRotate(0);
        grp1.setRotationAxis(Rotate.X_AXIS);
        Group grp2 = new Group(grp1);
        grp2.setRotate(-60);
        grp2.setRotationAxis(Rotate.X_AXIS);
        Group grp3 = new Group(grp2);
        grp3.setTranslateX(400);
        grp3.setTranslateY(400);
        grp3.setTranslateZ(10);

        pointLight = new PointLight(Color.ANTIQUEWHITE);
        pointLight.setTranslateX(300);
        pointLight.setTranslateY(-50);
        pointLight.setTranslateZ(-1000);

        root = new Group(grp3, pointLight);
        final Scene scene = new Scene(root, width, height, depthBuffer);
        scene.setOnKeyTyped(new EventHandler<KeyEvent>() {
            @Override
            public void handle(KeyEvent e) {
                switch (e.getCharacter()) {
                    case "[":
                        funcValue -= resolution;
                        if (funcValue < -20.0f) {
                            funcValue = -20.0f;
                        }
                        computeMesh(divX, divY, meshScale);
                        break;
                    case "]":
                        funcValue += resolution;
                        if (funcValue > 20.0f) {
                            funcValue = 20.0f;
                        }
                        computeMesh(divX, divY, meshScale);
                        break;
                    case "p":
                        funcValue = 0;
                        computeMesh(divX, divY, meshScale);
                        break;
                    case "l":
                        System.err.print("l ");
                        boolean wireframe = meshView.getDrawMode() == DrawMode.LINE;
                        meshView.setDrawMode(wireframe ? DrawMode.FILL : DrawMode.LINE);
                        break;
                    case "m":
                        enableChangeOne = !enableChangeOne;
                        enableChangeAll = false;
                        break;
                    case "n":
                        enableChangeAll = !enableChangeAll;
                        enableChangeOne = false;
                        break;
                }
            }
        });
        
        TranslateTransition translateTransition = new TranslateTransition(Duration.seconds(2), meshView);
        translateTransition.setCycleCount(Timeline.INDEFINITE);
        translateTransition.setAutoReverse(true);
        translateTransition.setByX(100.0);
        translateTransition.play();
        
        // MEASURE FPS
        Timeline fpsTimeline = new Timeline(new KeyFrame(Duration.seconds(2), new EventHandler<ActionEvent>() {
            @Override public void handle(ActionEvent t) {
                System.out.println("fps = " + PerformanceTracker.getSceneTracker(scene).getInstantFPS());
                
            }
        }));
        fpsTimeline.setCycleCount(Timeline.INDEFINITE);
        fpsTimeline.play();
        
        // CHANGE ONE POINT OF MESH IF CHANGE IS ENABLED
        AnimationTimer animationTimer = new AnimationTimer() {
            @Override
            public void handle(long l) {
                oddity = !oddity;
                if (enableChangeOne) {
                    ObservableFloatArray points = triMesh.getPoints();
                    points.set(0, points.get(0) + ((oddity) ? 5.0f : -5.0f));
                }
                if (enableChangeAll) {
                    ObservableFloatArray points = triMesh.getPoints();
                    int size = points.size();
                    for (int i = 0; i < size; i++) {
                        points.set(i, points.get(i) + ((oddity) ? 5.0f : -5.0f));
                    }
                }
            }
        };
        animationTimer.start();
        
        return scene;
    }

    private PerspectiveCamera addCamera(Scene scene) {
        PerspectiveCamera perspectiveCamera = new PerspectiveCamera();
        scene.setCamera(perspectiveCamera);
        return perspectiveCamera;
    }

    @Override
    public void start(Stage primaryStage) {
        Scene scene = buildScene(800, 800, true);
        scene.setFill(Color.rgb(10, 10, 40));
        addCamera(scene);
        primaryStage.setTitle("SimpleIssueViewerApp");
        primaryStage.setScene(scene);
        primaryStage.show();
        
        
    }

    public static void main(String[] args) {
        launch(args);
    }
}