package org.simantics.proconf.g3d.tools;

import java.util.ArrayList;

import com.jme.scene.Node;
import com.jme.scene.Spatial;

public class ScenegraphLockTraverser {
	Node root;
	boolean lock;
	public ScenegraphLockTraverser(Node root, boolean lock) {
		this.root = root;
		this.lock = lock;
		lock(root);
	}
	
	private void lock(Spatial spatial) {
		if (lock)
			spatial.lock();
		else
			spatial.unlock();
		if (spatial instanceof Node) {
			Node node = (Node)spatial;
			ArrayList<Spatial> children = node.getChildren();
    		for (Spatial s : children) {
    			s.lock();
    			lock(s);
    		}
		}
		
	}
}