var rows;
var nodes;
var area;
var edges;
var k;
var myMap = new Map();

function V(x,y){
    this.x=x;
    this.y=y;
    this.dx=0;
    this.dy=0;
}

function preload(){
    table = loadTable('3980.csv', 'csv');
}
  
function setup(){
    createCanvas(900, 900);

    print("Hello world!");
    print(table.getRowCount() + ' total rows in table');
    print(table.getColumnCount() + ' total columns in table');

    //cycle through the table
    nodes=[];
    edges=[];
    rows = table.getRowCount();
    for (var r = 0; r < table.getRowCount(); r++){
        edges.push([table.getString(r,0),table.getString(r,1)]);
        for (var c = 0; c < table.getColumnCount(); c++) {
            // print(table.getString(r,c));
            var temp = table.getString(r,c);
            if(!nodes.includes(temp)){
                nodes.push(temp);
                node = new V(50+Math.random()*800,50+Math.random()*800);
                // print(node.x);
                myMap.set(temp,node);
            }
        }
    }
    print(nodes.length);
    print(Math.random()*50);
    print("map",myMap.size);
    
    for(i=0; i<nodes.length; i++){
        console.log(myMap.get(nodes[i]).x);
        console.log(myMap.get(nodes[i]).y);
    }

    //canvas is 900 by 900 but I only take 50-850
    area=800*800;
    edge=rows;
    k=Math.sqrt(area/nodes.length);
    // var t=300;

    for(i=0;i<100;i++){
        //repulsive
        for(j=0;j<nodes.length;j++){
            for(k=0;k<nodes.length;k++){
                if(j!=k){
                    v=myMap.get(nodes[j]);
                    u=myMap.get(nodes[k]);
                    var deltax=v.x-u.x;
                    var deltay=v.y-u.y;
                    var delta=Math.sqrt(deltax**2+deltay**2);
                    v.dx=v.dx+(deltax/delta)*gamma1(delta,k);
                }
            }
        }
        //attractive
        for(l=0;l<edges.length;l++){
            v=myMap.get(edges[l][0]);
            u=myMap.get(edges[l][1]);
            var delx=v.x-u.x;
            var dely=v.y-u.y;
            var del=Math.sqrt(delx**2+dely**2);
            v.dx=v.dx-(delx/del)*alpha1(del,k);
            v.dy=v.dy-(dely/del)*alpha1(del,k);
            u.dx=u.dx+(delx/del)*alpha1(del,k);
            u.dy=u.dy+(dely/del)*alpha1(del,k);
        }
        //limit max to t
        for(m=0;m<nodes.length;m++){
            v=myMap.get(nodes[m]);
            vdmag=Math.sqrt(v.dx**2+v.dy**2);
            // v.x=v.x+(v.dx/vdmag)*Math.min(vdmag,t);
            // v.y=v.y+(v.dy/vdmag)*Math.min(vdmag,t);
            v.x=v.x+(v.dx/vdmag);
            v.y=v.y+(v.dy/vdmag);
            v.x=Math.min(850,Math.max(50,v.x));
            v.y=Math.min(850,Math.max(50,v.y));
        }
        // t=0.9*t;
        // draw();
    }
}

function alpha1(x,k) {
    return x**2/k;
}

function gamma1(x,k) {
    return k**2/x;
}

function draw(){
    clear();
    for(i=0; i<nodes.length; i++){
        // ellipse(240,240,80,80);
        // console.log(myMap.get(nodes[i]).x,myMap.get(nodes[i]).y);
        xcor=myMap.get(nodes[i]).x;
        ycor=myMap.get(nodes[i]).y;
        ellipse(xcor, ycor, 3, 3);
        if((mouseX-xcor)**2<12 && (mouseY-ycor)**2<12){
            strokeWeight(1);
            fill(0,0,0);
            text(i, xcor+3, ycor-3);
        }
    }

    for(i=0; i<edges.length; i++){
        xcor1=myMap.get(edges[i][0]).x;
        ycor1=myMap.get(edges[i][0]).y;
        xcor2=myMap.get(edges[i][1]).x;
        ycor2=myMap.get(edges[i][1]).y;
        stroke(0,0,0);
        strokeWeight(1);
        line(xcor1, ycor1, xcor2, ycor2);
        if((mouseX-xcor1)**2<12 && (mouseY-ycor1)**2<12){
            stroke(255, 115, 25);
            strokeWeight(2);
            line(xcor1, ycor1, xcor2, ycor2);
            strokeWeight(1);
            fill(0,0,0);
            text(edges[i][1],xcor2+3, ycor2-3);
        }
        if((mouseX-xcor2)**2<12 && (mouseY-ycor2)**2<12){
            stroke(255, 115, 25);
            strokeWeight(2);
            line(xcor1, ycor1, xcor2, ycor2);
            strokeWeight(1);
            fill(0,0,0);
            text(edges[i][0],xcor1+3, ycor1-3);
        }
    }

}