source: js/viewer-f0/js/NeuronDrawer.js @ 143

function NeuronDrawer() {
    this._scene = undefined;
    this._camera = undefined;
    this._renderer = undefined;
    this._canvasWidth = 500;
    this._canvasHeight = 500;
    this._containerContext = undefined;
    this._controls = undefined;
    this._showAxis = false;
}

NeuronDrawer.prototype.showPartAxis = function () {
    this._showAxis = true;
}

NeuronDrawer.prototype._createRenderer = function () {
    this._renderer = new THREE.CanvasRenderer();//WebGLRenderer({antialias: true});
    this._renderer.setClearColor(0xffffff, 1);
    this._renderer.setSize(this._canvasWidth, this._canvasHeight);
    this._containerContext = $("#containerNeuron");
    this._containerContext.append(this._renderer.domElement);
}

NeuronDrawer.prototype._prepareCamera = function () {
    this._camera = new THREE.PerspectiveCamera(45, this._canvasWidth / this._canvasHeight, 1, 4000);
    this._camera.position.set(0, 0, 10);
    this._camera.lookAt(this._scene.position);
    this._scene.add(this._camera);
}

NeuronDrawer.prototype._addLight = function () {
    var directionalLight = new THREE.DirectionalLight(0xffffff);

    directionalLight.position = this._camera.position;
    directionalLight.intensity = 1;

    this._scene.add(directionalLight);
}

NeuronDrawer.prototype.initializeScene = function () {

    this._createRenderer();
    this._scene = new THREE.Scene();
    this._prepareCamera();
    this._addLight();
    this._controls = new THREE.TrackballControls(this._camera, this._renderer.domElement)
    this._debug();
}

NeuronDrawer.prototype.renderScene = function () {

    var self = this;
    requestAnimationFrame(
        function () {
            self.renderScene();
        });
    this._renderer.render(this._scene, this._camera);
    this._controls.update();

}

NeuronDrawer.prototype._debug = function()
{
    this._scene.add( new THREE.AxisHelper( 20 ) );
}

NeuronDrawer.prototype.drawNeuron = function(x, y){
    var scheme = [4,4,26,27,26,73,73,73,73,27,26,27,1,73,50,100,50,1,56,68,46,68,2,41,47,51,32,51,68];
    //pobieranie schematu
    //inne określenie położeń


    var obj = new THREE.Object3D();

    var material = new THREE.LineBasicMaterial({
        color: 0x0000ff
    });
    var position = 0;
    var noOfBlocks = 0;//number of "blocks" to draw
    var noOfLines = 0;//number of line to draw
    noOfBlocks = scheme[position++];
    for (var i = 0; i < noOfBlocks; i++)
    {
        noOfLines = scheme[position++];

        for (var j = 0; j < noOfLines; j++)
        {
            var geometry = new THREE.Geometry();
            geometry.vertices.push(new THREE.Vector3(scheme[position++] + x, scheme[position++] + y, 0));
            geometry.vertices.push(new THREE.Vector3(scheme[position++] + x, scheme[position++] + y, 0));
            position = position - 2; //get to last point in list
            var line = new THREE.Line(geometry, material);
            obj.add(line);
        }
        position = position + 2;//move to value which define number of lines to draw
    }

    var SCALE = 0.05;

    obj.scale.set(SCALE, SCALE,SCALE);
    obj.rotateX(Math.PI);//rotate 180 degree
    obj.translateY(-5.5);
    this._scene.add(obj)

}



NeuronDrawer.prototype.drawConnection = function(id, connections, einfos){

    var destination = connections[id].getDestination();
    var numberOfConnToDest = 0;
    var numberOfDrawenCon = 0;

    //check how many conenctions have the same destination
    for(var i = 0; i < connections.length; i++){
        if(connections[i].getDestination() == destination)
            numberOfConnToDest++;
    }

    //check how many connections had been drawen
    for(var i = 0; i < id; i++){
        if(connections[i].getDestination() == destination)
            numberOfDrawenCon++;
    }

    var positionY = (numberOfDrawenCon + 1) * 100 / (numberOfConnToDest + 1);
    var positionX = (numberOfDrawenCon + 1) * 10 / (numberOfConnToDest + 1);

    var x1 = einfos[connections[id].getDestination()].x;
    var y1 = einfos[connections[id].getDestination()].y;

    var x2 = einfos[connections[id].getSource()].x;
    var y2 = einfos[connections[id].getSource()].y;

    var material = new THREE.LineBasicMaterial({
        color: 0x0000ff
    });

    var SCALE = 0.05;

    var geometry = new THREE.Geometry();
    geometry.vertices.push(new THREE.Vector3(100 + x2 * 100, 10 + 50 + y2 * 100, 0));
    geometry.vertices.push(new THREE.Vector3(10 + 100 + x2 * 100 , 10 + 50 + y2 * 100, 0));
    var line = new THREE.Line(geometry, material);
    line.scale.set(SCALE, SCALE, SCALE);
    this._scene.add(line);

    var geometry1 = new THREE.Geometry();
    geometry1.vertices.push(new THREE.Vector3(10 + positionX + x1 * 100, 10 + positionY + y1 * 100, 0));
    geometry1.vertices.push(new THREE.Vector3(10 + 25 + x1 * 100, 10 + positionY + y1 * 100, 0));
    var line1 = new THREE.Line(geometry1, material);

    line1.scale.set(SCALE, SCALE, SCALE);
    this._scene.add(line1);

    //bezposredni poprzednik - to koniec :)
    if (x1 == x2 + 1)//kazdy polozony w kolumnie po lewej ma jedna ukosna linie
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 10 + positionY + y1 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(110 + x2 * 100, 10 + 50 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
    }

    //pierwszy pionowy
    if (y1 >= y2)
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 60 + y2 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 105 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
    }

    else
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 60 + y2 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 15 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
    }


    //poziomy
    if (y1 >= y2)
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 105 + y2 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 105 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
        //g.drawLine((int) ((110 + (x2 * 100)) * zoom), (int) ((105 + y2 * 100) * zoom), (int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((105 + y2 * 100) * zoom));
    }

    else
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(110 + (x2 * 100), 15 + y2 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 15 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
        //g.drawLine((int) ((110 + (x2 * 100)) * zoom), (int) ((15 + y2 * 100) * zoom), (int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((15 + y2 * 100) * zoom));
    }


    //drugi pionowy
    if (y1 >= y2)
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 10 + positionY + y1 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 105 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
        //g.drawLine((int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((10 + polozenieY + y1 * 100) * zoom), (int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((105 + y2 * 100) * zoom));
    }
    else
    {
        var geometry = new THREE.Geometry();
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 10 + positionY + y1 * 100, 0));
        geometry.vertices.push(new THREE.Vector3(10 + positionX + (x1 * 100), 15 + y2 * 100, 0));
        var line = new THREE.Line(geometry, material);
        line.scale.set(SCALE, SCALE, SCALE);
        this._scene.add(line);
        return;
        //g.drawLine((int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((10 + polozenieY + y1 * 100) * zoom), (int) ((10 + polozenieX + (x1 * 100)) * zoom), (int) ((15 + y2 * 100) * zoom));
    }

}

NeuronDrawer.prototype.drawNetwork = function(neurons, connections, layouts){

    for(var i = 0; i < layouts.length; i++){
        this.drawNeuron(layouts[i].x * 100, -layouts[i].y * 100);
    }

    for(var i = 0; i < connections.length; i++)
        this.drawConnection(i, connections, einfos);
}

NeuronDrawer.prototype.debugTest = function()
{

}