/* Copyright 2008-2010 Gephi Authors : Helder Suzuki <heldersuzuki@gephi.org> Website : http://www.gephi.org This file is part of Gephi. DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER. Copyright 2011 Gephi Consortium. All rights reserved. The contents of this file are subject to the terms of either the GNU General Public License Version 3 only ("GPL") or the Common Development and Distribution License("CDDL") (collectively, the "License"). You may not use this file except in compliance with the License. You can obtain a copy of the License at http://gephi.org/about/legal/license-notice/ or /cddl-1.0.txt and /gpl-3.0.txt. See the License for the specific language governing permissions and limitations under the License. When distributing the software, include this License Header Notice in each file and include the License files at /cddl-1.0.txt and /gpl-3.0.txt. If applicable, add the following below the License Header, with the fields enclosed by brackets [] replaced by your own identifying information: "Portions Copyrighted [year] [name of copyright owner]" If you wish your version of this file to be governed by only the CDDL or only the GPL Version 3, indicate your decision by adding "[Contributor] elects to include this software in this distribution under the [CDDL or GPL Version 3] license." If you do not indicate a single choice of license, a recipient has the option to distribute your version of this file under either the CDDL, the GPL Version 3 or to extend the choice of license to its licensees as provided above. However, if you add GPL Version 3 code and therefore, elected the GPL Version 3 license, then the option applies only if the new code is made subject to such option by the copyright holder. Contributor(s): Portions Copyrighted 2011 Gephi Consortium. */ package org.gephi.layout.plugin.force.quadtree; import java.awt.Color; import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.Set; import org.gephi.graph.api.Column; import org.gephi.graph.api.ColumnIterable; import org.gephi.graph.api.Graph; import org.gephi.graph.api.GraphView; import org.gephi.graph.api.Interval; import org.gephi.graph.api.Node; import org.gephi.graph.api.NodeProperties; import org.gephi.graph.api.Table; import org.gephi.graph.api.TextProperties; import org.gephi.graph.spi.LayoutData; /** * @author Helder Suzuki <heldersuzuki@gephi.org> */ public class QuadTree implements Node { private final float posX; private final float posY; private final float size; private float centerMassX; // X and Y position of the center of mass private float centerMassY; private int mass; // Mass of this tree (the number of nodes it contains) private final int maxLevel; private AddBehaviour add; private List<QuadTree> children; private boolean isLeaf; public static final float eps = (float) 1e-6; public static QuadTree buildTree(Graph graph, int maxLevel) { float minX = Float.POSITIVE_INFINITY; float maxX = Float.NEGATIVE_INFINITY; float minY = Float.POSITIVE_INFINITY; float maxY = Float.NEGATIVE_INFINITY; for (Node node : graph.getNodes()) { minX = Math.min(minX, node.x()); maxX = Math.max(maxX, node.x()); minY = Math.min(minY, node.y()); maxY = Math.max(maxY, node.y()); } float size = Math.max(maxY - minY, maxX - minX); QuadTree tree = new QuadTree(minX, minY, size, maxLevel); for (Node node : graph.getNodes()) { tree.addNode(node); } return tree; } public QuadTree(float posX, float posY, float size, int maxLevel) { this.posX = posX; this.posY = posY; this.size = size; this.maxLevel = maxLevel; this.isLeaf = true; mass = 0; add = new FirstAdd(); } @Override public float size() { return size; } private void divideTree() { float childSize = size / 2; children = new ArrayList<>(); children.add(new QuadTree(posX + childSize, posY + childSize, childSize, maxLevel - 1)); children.add(new QuadTree(posX, posY + childSize, childSize, maxLevel - 1)); children.add(new QuadTree(posX, posY, childSize, maxLevel - 1)); children.add(new QuadTree(posX + childSize, posY, childSize, maxLevel - 1)); isLeaf = false; } private boolean addToChildren(NodeProperties node) { for (QuadTree q : children) { if (q.addNode(node)) { return true; } } return false; } private void assimilateNode(NodeProperties node) { centerMassX = (mass * centerMassX + node.x()) / (mass + 1); centerMassY = (mass * centerMassY + node.y()) / (mass + 1); mass++; } public Iterable<QuadTree> getChildren() { return children; } @Override public float x() { return centerMassX; } @Override public float y() { return centerMassY; } public int mass() { return mass; } @Override public float z() { throw new UnsupportedOperationException("Not supported."); } public boolean addNode(NodeProperties node) { if (posX <= node.x() && node.x() <= posX + size && posY <= node.y() && node.y() <= posY + size) { return add.addNode(node); } else { return false; } } /** * @return the isLeaf */ public boolean isIsLeaf() { return isLeaf; } @Override public float r() { throw new UnsupportedOperationException("Not supported."); } @Override public float g() { throw new UnsupportedOperationException("Not supported."); } @Override public float b() { throw new UnsupportedOperationException("Not supported."); } @Override public int getRGBA() { throw new UnsupportedOperationException("Not supported."); } @Override public Color getColor() { throw new UnsupportedOperationException("Not supported."); } @Override public float alpha() { throw new UnsupportedOperationException("Not supported."); } @Override public boolean isFixed() { throw new UnsupportedOperationException("Not supported."); } @Override public <T extends LayoutData> T getLayoutData() { throw new UnsupportedOperationException("Not supported."); } @Override public TextProperties getTextProperties() { throw new UnsupportedOperationException("Not supported."); } @Override public int getStoreId() { throw new UnsupportedOperationException("Not supported."); } @Override public void setX(float x) { throw new UnsupportedOperationException("Not supported."); } @Override public void setY(float y) { throw new UnsupportedOperationException("Not supported."); } @Override public void setZ(float z) { throw new UnsupportedOperationException("Not supported."); } @Override public void setPosition(float x, float y) { throw new UnsupportedOperationException("Not supported."); } @Override public void setPosition(float x, float y, float z) { throw new UnsupportedOperationException("Not supported."); } @Override public void setR(float r) { throw new UnsupportedOperationException("Not supported."); } @Override public void setG(float g) { throw new UnsupportedOperationException("Not supported."); } @Override public void setB(float b) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAlpha(float a) { throw new UnsupportedOperationException("Not supported."); } @Override public void setColor(Color color) { throw new UnsupportedOperationException("Not supported."); } @Override public void setSize(float size) { throw new UnsupportedOperationException("Not supported."); } @Override public void setFixed(boolean fixed) { throw new UnsupportedOperationException("Not supported."); } @Override public void setLayoutData(LayoutData layoutData) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getId() { throw new UnsupportedOperationException("Not supported."); } @Override public String getLabel() { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(String key) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(Column column) { throw new UnsupportedOperationException("Not supported."); } @Override public Object[] getAttributes() { throw new UnsupportedOperationException("Not supported."); } @Override public Set<String> getAttributeKeys() { throw new UnsupportedOperationException("Not supported."); } @Override public ColumnIterable getAttributeColumns() { throw new UnsupportedOperationException("Not supported."); } @Override public Object removeAttribute(String key) { throw new UnsupportedOperationException("Not supported."); } @Override public Object removeAttribute(Column column) { throw new UnsupportedOperationException("Not supported."); } @Override public void setLabel(String label) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(String key, Object value) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(Column column, Object value) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(String key, Object value, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(Column column, Object value, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean addTimestamp(double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean removeTimestamp(double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public double[] getTimestamps() { throw new UnsupportedOperationException("Not supported."); } @Override public void clearAttributes() { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(String key, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(Column column, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(String key, GraphView view) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(Column column, GraphView view) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean hasTimestamp(double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(String key, Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public Object getAttribute(Column column, Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public Iterable<Map.Entry> getAttributes(Column column) { throw new UnsupportedOperationException("Not supported."); } @Override public Object removeAttribute(String key, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public Object removeAttribute(Column column, double timestamp) { throw new UnsupportedOperationException("Not supported."); } @Override public Object removeAttribute(String key, Interval interval) { throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates. } @Override public Object removeAttribute(Column column, Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(String key, Object value, Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public void setAttribute(Column column, Object value, Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean addInterval(Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean removeInterval(Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public boolean hasInterval(Interval interval) { throw new UnsupportedOperationException("Not supported."); } @Override public Interval[] getIntervals() { throw new UnsupportedOperationException("Not supported."); } @Override public Table getTable() { throw new UnsupportedOperationException("Not supported yet."); //To change body of generated methods, choose Tools | Templates. } class FirstAdd implements AddBehaviour { @Override public boolean addNode(NodeProperties node) { mass = 1; centerMassX = node.x(); centerMassY = node.y(); if (maxLevel == 0) { add = new LeafAdd(); } else { add = new SecondAdd(); } return true; } } class SecondAdd implements AddBehaviour { @Override public boolean addNode(NodeProperties node) { divideTree(); add = new RootAdd(); /* This QuadTree represents one node, add it to a child accordingly */ addToChildren(QuadTree.this); return add.addNode(node); } } class LeafAdd implements AddBehaviour { @Override public boolean addNode(NodeProperties node) { assimilateNode(node); return true; } } class RootAdd implements AddBehaviour { @Override public boolean addNode(NodeProperties node) { assimilateNode(node); return addToChildren(node); } } } interface AddBehaviour { public boolean addNode(NodeProperties node); }