3d-interface/analysis/replay/main.js

"use strict";

// REQUIRES AND INITS
let width =  Math.floor(1134/10);
let height = Math.floor(768 /10);

let XMLHttpRequest = require('xmlhttprequest').XMLHttpRequest;
let THREE = require('three');
let L3D = require('../../static/js/l3d.min.js');
let fs = require('fs');

let scene = new THREE.Scene();
let geometry = new THREE.Geometry();

let peach = new THREE.Object3D();
let bobomb = new THREE.Object3D();
let whomp = new THREE.Object3D();
let mountain = new THREE.Object3D();

var finished = false;
let frame = 0;
let raycaster = new THREE.Raycaster();

var total = 0;

let loader, progLoader;

let id = 56;


// INIT COLORS
var colors = [[0,0,0]];

for (let i = 0; i < 856; i++) {
    colors.push([
        Math.floor(255*Math.random()),
        Math.floor(255*Math.random()),
        Math.floor(255*Math.random())
    ]);
}

var camera; var finished = false;
var Recommendation = L3D.BaseRecommendation;
var forceFinished = false;
main();

// FUNCTIONS
function main() {
    let xhr = new XMLHttpRequest();
    xhr.open("GET", "http://localhost:4000/prototype/replay-info/" + id, true);

    xhr.onreadystatechange = function() {
        if (xhr.readyState == 4 && xhr.status == 200) {

            var data = JSON.parse(xhr.responseText);

            console.log('%sceneId = ' + data.sceneInfo.sceneId);
            console.log('%recoStyle = ' + data.sceneInfo.recommendationStyle);
            console.log('%expId = ' + id);
            console.log();
            console.log('M = [');

            camera = new L3D.ReplayCamera(50, width / height, 0.01, 100000, [], data, () => finished = true);
            var path = initElements(camera, data.sceneInfo);

            loader = new L3D.ProgressiveLoader(
                path, new THREE.Object3D(), null
            );

            progLoader = new L3D.ProgressiveLoader(
                path, scene, camera, null, null, true
            );

            loader.load(function(){
                process.stderr.write('Loading complete.\n');
                progLoader.load(function() {
                    process.stderr.write("Loading complete\n");
                    forceFinished = true;
                });
                init(data);
            });
        }
    };
    xhr.send();
}

function testDistance(old, newP) {

    return (
        L3D.Tools.norm2(L3D.Tools.diff(old.position, newP.position)) +
        L3D.Tools.norm2(L3D.Tools.diff(old.target, newP.target)) > 0.1
    );
}


function initElements(camera, sceneInfo) {
    switch (sceneInfo.sceneId) {
        case 1:
            camera.resetElements = L3D.resetPeachElements();
            camera.cameras = L3D.createPeachRecommendations(width, height);
            camera.speed = 0.001;
            return '/static/data/castle/princess peaches castle (outside).obj';
        case 2:
            camera.resetElements = L3D.resetBobombElements();
            camera.cameras = L3D.createBobombRecommendations(width, height);
            camera.speed = 0.005;
            return '/static/data/bobomb/bobomb battlefeild.obj';
        case 3:
            camera.resetElements = L3D.resetMountainElements();
            camera.cameras = L3D.createMountainRecommendations(width, height);
            camera.speed = 0.005;
            return '/static/data/mountain/coocoolmountain.obj';
        case 4:
            camera.resetElements = L3D.resetWhompElements();
            camera.cameras = L3D.createWhompRecommendations(width, height);
            camera.speed = 0.002;
            return '/static/data/whomp/Whomps Fortress.obj';
        default:
            console.err('This sceneId doesn\'t exist');
            process.exit(-1);


    }
}

function init(data) {
    scene.add(camera);

    camera.reset();
    camera.speed = 0.001;
    camera.start();
    setTimeout(loop, 0);
}

function equalFaces(face1, face2) { return face1.a === face2.a && face1.b === face2.b && face1.c === face2.c; };

function printVector(vec) { console.log(`(${vec.x},${vec.y},${vec.z})`); }

let old;

function loop() {

    old = {
        position: new THREE.Vector3().copy(camera.position),
        target: new THREE.Vector3().copy(camera.target)
    };

    for (let i = 0; i < 10; i++)
        finished = camera.update(20);

    total++;

    if (!testDistance(old, camera) && !finished && !forceFinished) {
        process.nextTick(loop);
        return;
    }


    camera.look();
    camera.updateMatrixWorld(true);

    // printVector(camera.position);

    let buf = [];
    // let buf2 = [];

    let score = 0;

    var totalPixel = 0;
    for (let i = 0; i < width; i++) {

        buf[i] = [];

        let x = (i / width) * 2 - 1;

        // process.stderr.write('\b\r' + Math.floor(100*(i / width)) + '%\n');

        for (let j = 0; j < height; j++) {

            let y = (j / height) * 2 - 1;

            raycaster.setFromCamera({x:x, y:y}, camera);

            let intersectsProg = raycaster.intersectObject(progLoader.obj, true);
            let intersects = raycaster.intersectObject(loader.obj, true);

            if (intersectsProg.length === 0 && intersects.length === 0) {
                // Not good
            } else if (intersectsProg.length !== intersects.length) {
                // Not good
                totalPixel++;
            } else if (equalFaces(intersectsProg[0].face, intersects[0].face)) {
                score++;
                totalPixel++;
            } else {
                totalPixel++;
            }

            if (intersectsProg.length > 0) {
                buf[i][j] = intersectsProg[0].faceIndex;
            }


        }

    }

    score /= totalPixel;

    // for (let i = 0; i < 255; i++) {
    //     colorsOccurence[i+1] += colorsOccurence[i];
    // }

    var buffer = [];
    buffer.push('P3\n');
    buffer.push(width+'\n');
    buffer.push(height+'\n');
    buffer.push(255+'\n');

    for (let i = 0; i < height; i++) {
        for (let j = 0; j < width; j++) {
            var grey = buf[j][height - i - 1];
            if (colors[grey] === undefined)
                buffer.push('0 0 0\n');
            else {
                buffer.push(colors[grey][0] + ' ' + colors[grey][1] + ' ' + colors[grey][2] + '\n');
            }
        }
    }

    let frameName = "" + frame;
    var pad = "00000";
    frameName = pad.substring(0, pad.length - frameName.length) + frameName;
    fs.writeFileSync(`img/${frameName}.ppm`, buffer.join(''));


    // Write image buffer to disk
    console.log(score);
    process.stderr.write('Frame : ' + frame++ + ', score = ' + score + '\n');

    if (!finished && !forceFinished)
        setTimeout(loop,50);
}