import nub.core.Graph;

import nub.core.Node;

import nub.primitives.Vector;

import nub.processing.Scene;

import processing.core.PApplet;

import processing.core.PGraphics;

import processing.event.MouseEvent;

/**

* This example modifies the visual representation on a Node

* depending on the scene it is drawn. It demos custom scene behavior

*

* By Sebastian Chaparro

*/

public class VisitExample extends PApplet {

boolean relativeToGraph = true;

int rows = 2, cols = 2;

int n = rows * cols;

Scene[] scenes = new Scene[n];

Scene focus;

int w = 1400, h = 1400;

int[] strokeColors = new int[n];

int[] fillColors = new int[n];

public void settings() {

size(w, h, P3D);

}

@Override

public void setup() {

//create the scenes

for(int i = 0; i < n; i++){

final int idx = i;

scenes[i] = new Scene(createGraphics(w / cols, h / rows, P3D));

scenes[i].setRadius(50);

strokeColors[i] = color(random(255), random(255), random(255));

fillColors[i] = color(random(255), random(255), random(255));

scenes[i].fit();

if(Math.random() < 0.5) scenes[i].togglePerspective();

}

generateRandomNodes(10);

}

@Override

public void draw() {

for(int i = 0; i < n; i++){

focus = scenes[i].hasFocus() ? scenes[i] : focus;

scenes[i].openContext();

scenes[i].context().background(125);

scenes[i].render();

scenes[i].drawAxes();

scenes[i].beginHUD();

sceneHUD(i, scenes[i].context());

scenes[i].endHUD();

scenes[i].closeContext();

scenes[i].image(i / rows * h / rows, (i % cols) * w / cols);

}

}

public void sceneHUD(int i, PGraphics pg) {

pg.pushStyle();

pg.stroke(255);

pg.fill(255);

pg.text("On scene " + i + " radius : " + (int) (scenes[i].radius()), 10, pg.height / 10);

pg.popStyle();

}

public void generateRandomNodes(int n) {

for(int i = 0; i < n; i++){

Node node = new CustomNode();

node.randomize(new Vector(), 150, true);

node.setMagnitude(1);

}

}

class CustomNode extends Node {

//Add some attributes

float radiusInPixels = 50, radius;

int strokeCol, fillCol;

public CustomNode() {

super();

//without modifying Node class

//for(Scene scene : scenes) setVisit(scene, (g , n) -> setupAttributesByGraph(g));

//modifying the node class

for (Scene scene : scenes) setBehavior(scene, this::setupAttributesByGraph);

setShape(pg -> {

pg.pushStyle();

pg.stroke(strokeCol);

pg.fill(fillCol);

pg.box(radius);

pg.popStyle();

});

}

public void setupAttributesByGraph(Graph g) {

if (relativeToGraph) {

//keep the size of the node relative to the graph such that it always occupies 10 pixels

this.radius = this.radiusInPixels * g.sceneToPixelRatio(worldPosition());

//change color based on Graph

for (int i = 0; i < n; i++) {

if (g == scenes[i]) {

this.strokeCol = strokeColors[i];

this.fillCol = fillColors[i];

}

}

} else{

this.radius = 10;

this.strokeCol = color(255);

this.fillCol = color(255,255, 0);

}

}

}

@Override

public void mouseMoved(){

focus.tag();

}

@Override

public void mouseDragged() {

if(mouseButton == LEFT)

focus.spin();

else if(mouseButton == RIGHT)

focus.shift();

else

focus.zoom(focus.mouseDX());

}

@Override

public void mouseWheel(MouseEvent event) {

focus.moveForward(event.getCount() * 40);

}

@Override

public void mouseClicked(MouseEvent event) {

if (event.getCount() == 2)

if (event.getButton() == LEFT)

focus.focus();

else

focus.align();

}

@Override

public void keyPressed() {

if (key == ' ') {

relativeToGraph = !relativeToGraph;

}

}

public static void main(String[] args) {

PApplet.main(new String[]{"VisitExample"});

}

}