/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 1997-2009 Sun Microsystems, Inc. All rights reserved.

 *

 * The contents of this file are subject to the terms of either the GNU

 * General Public License Version 2 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://www.netbeans.org/cddl-gplv2.html

 * or nbbuild/licenses/CDDL-GPL-2-CP. 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 file at

 * nbbuild/licenses/CDDL-GPL-2-CP.  Sun designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Sun in the GPL Version 2 section of the License file that

 * accompanied this code. 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]"

 *

 * Contributor(s):

 *

 * The Original Software is NetBeans. The Initial Developer of the Original

 * Software is Sun Microsystems, Inc. Portions Copyright 1997-2007 Sun

 * Microsystems, Inc. All Rights Reserved.

 *

 * If you wish your version of this file to be governed by only the CDDL

 * or only the GPL Version 2, indicate your decision by adding

 * "[Contributor] elects to include this software in this distribution

 * under the [CDDL or GPL Version 2] 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 2 or

 * to extend the choice of license to its licensees as provided above.

 * However, if you add GPL Version 2 code and therefore, elected the GPL

 * Version 2 license, then the option applies only if the new code is

 * made subject to such option by the copyright holder.

 */

package org.netbeans.modules.visual.apichanges;

import org.netbeans.modules.visual.framework.VisualTestCase;

import org.netbeans.api.visual.widget.LabelWidget;

import org.netbeans.api.visual.widget.LayerWidget;

import org.netbeans.api.visual.widget.Widget;

import org.netbeans.api.visual.anchor.AnchorFactory;

import org.netbeans.api.visual.border.BorderFactory;

import org.netbeans.api.visual.graph.GraphScene;

import org.netbeans.api.visual.graph.layout.GraphLayout;

import org.netbeans.api.visual.graph.layout.GraphLayoutFactory;

import org.netbeans.api.visual.widget.ConnectionWidget;

import java.awt.*;

import java.awt.image.BufferedImage;

import org.netbeans.api.visual.graph.layout.GraphLayoutSupport;

import org.netbeans.api.visual.layout.LayoutFactory;

import org.netbeans.api.visual.layout.SceneLayout;

/**

 * Test for #178705 - Enhancement to the TreeGraphLayout (gaps between nodes can be minimized now)

 * @author David Kaspar

 */

public class MinimizeGapTreeGraphLayoutTest extends VisualTestCase {

    public MinimizeGapTreeGraphLayoutTest (String testName) {

        super (testName);

    }

    public void testMinimizeGap () {

        TestScene scene = new TestScene ();

        // hiding the connections to simplify cleaness check

        scene.connLayer.setVisible(false);

        scene.addNode("Root");

        scene.addNode ("1-1");

        scene.addNode ("1-2");

        scene.addNode ("1-3");

        scene.addNode ("2-1");

        scene.addNode ("2-2");

        scene.addNode ("2-3");

        scene.addEdge ("Root", "1-1");

        scene.addEdge ("Root", "1-2");

        scene.addEdge ("Root", "1-3");

        scene.addEdge ("1-2", "2-1");

        scene.addEdge ("1-2", "2-2");

        scene.addEdge ("1-2", "2-3");

        GraphLayout graphLayout = GraphLayoutFactory.createTreeGraphLayout(0, 0, 20, 20, true, true);

        graphLayout.setAnimated(false);

        GraphLayoutSupport.setTreeGraphLayoutRootNode(graphLayout, "Root");

        SceneLayout sceneLayout = LayoutFactory.createSceneGraphLayout(scene, graphLayout);

        sceneLayout.invokeLayout();

        BufferedImage snapshot = takeOneTimeSnapshot(scene);

        BufferedImage clean = clearRegions (snapshot, Color.WHITE,

                new Rectangle (48, -2, 34, 34),

                new Rectangle (-2, 48, 34, 34),

                new Rectangle (48, 48, 34, 34),

                new Rectangle (98, 48, 34, 34),

                new Rectangle (-2, 98, 34, 34),

                new Rectangle (48, 98, 34, 34),

                new Rectangle (98, 98, 34, 34)

        );

        assertCleaness (testCleaness (clean, Color.WHITE), snapshot, clean);

    }

    private final class TestScene extends GraphScene<String,String> {

        private final LayerWidget mainLayer;

        private final LayerWidget connLayer;

        public TestScene() {

            mainLayer = new LayerWidget(this);

            addChild(mainLayer);

            connLayer = new LayerWidget(this);

            addChild(connLayer);

        }

        public void addEdge (String source, String target) {

            String edge = source + " to " + target;

            addEdge (edge);

            setEdgeSource (edge, source);

            setEdgeTarget (edge, target);

        }

        @Override

        protected Widget attachNodeWidget(String node) {

            LabelWidget widget = new LabelWidget(this, node);

            widget.setCheckClipping(true);

            widget.setAlignment(LabelWidget.Alignment.CENTER);

            widget.setVerticalAlignment(LabelWidget.VerticalAlignment.CENTER);

            widget.setPreferredSize(new Dimension(30, 30));

            widget.setBorder(BorderFactory.createLineBorder());

            mainLayer.addChild(widget);

            return widget;

        }

        @Override

        protected Widget attachEdgeWidget(String edge) {

            ConnectionWidget conn = new ConnectionWidget(this);

            connLayer.addChild(conn);

            return conn;

        }

        @Override

        protected void attachEdgeSourceAnchor(String edge, String oldSourceNode, String sourceNode) {

            ConnectionWidget conn = (ConnectionWidget) findWidget (edge);

            Widget source = findWidget (sourceNode);

            conn.setSourceAnchor(source != null ? AnchorFactory.createRectangularAnchor(source) : null);

        }

        @Override

        protected void attachEdgeTargetAnchor(String edge, String oldTargetNode, String targetNode) {

            ConnectionWidget conn = (ConnectionWidget) findWidget (edge);

            Widget target = findWidget (targetNode);

            conn.setTargetAnchor(target != null ? AnchorFactory.createRectangularAnchor(target) : null);

        }

    }

}