/*
 * LabelLayoutRTree.java Copyright (C) 2019. Daniel H. Huson
 *
 *  (Some files contain contributions from other authors, who are then mentioned separately.)
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package jloda.swing.graphview;

import jloda.graph.Node;
import jloda.swing.util.BasicSwing;
import jloda.swing.util.RTree;

import java.awt.*;

/**
 * layout labels using an RTree
 * Daniel Huson, 9.2012
 */
public class LabelLayoutRTree {
    /**
     * layout nodes for drawing
     *
     * @param graphView
     * @param gc
     */
    public void layout(GraphView graphView, Graphics gc) {
        Transform trans = graphView.trans;

        RTree<Node> rTree = new RTree<>();
        // add all nodes to avoid:
        Point center = new Point();
        int count = 0;
        for (Node v = graphView.getGraph().getFirstNode(); v != null; v = v.getNext()) {
            NodeView nv = graphView.getNV(v);
            String label = nv.getLabel();
            if (label != null && nv.getLabelVisible()) {
                nv.setLabelAngle(0);
                Rectangle bbox = nv.getBox(trans);
                //  if (nv.getHeight() > 1 || nv.getWidth() > 1) {
                rTree.add(bbox, v);
                //  }
                center.x += bbox.x + bbox.width / 2;
                center.y += bbox.y + bbox.height / 2;
                count++;
                if (count > 500)
                    return;
                if (nv.getLabelLayout() != NodeView.LAYOUT) {
                    Rectangle rect = nv.getLabelRect(trans);
                    if (rect != null)
                        rTree.add(rect, v);
                }
            }
        }
        if (count > 0) {
            center.x /= count;
            center.y /= count;
        }
        for (Node v = graphView.getGraph().getFirstNode(); v != null; v = v.getNext()) {
            NodeView nv = graphView.getNV(v);
            String label = nv.getLabel();
            if (label != null && nv.getLabelVisible() && nv.getLabelLayout() == NodeView.LAYOUT) {
                Dimension labelSize = BasicSwing.getStringSize(gc, label, nv.getFont()).getSize();
                Point location = graphView.trans.w2d(nv.getLocation());
                boolean left = (location.x < center.x);
                // location.x += nv.getWidth() / 2;
                location.y -= labelSize.getHeight() / 2;
                location = rTree.addCloseTo(v.getId(), location, nv.getWidth() / 2, nv.getHeight() / 2, left, labelSize.getSize(), v);
                nv.setLabelPosition(location.x, location.y + labelSize.height, graphView.trans);
            }
        }
        //rTree.draw(gc);
        rTree.clear();
    }
}