main page modules namespaces classes files
Gecode/J home

ExplorerNode.java

Go to the documentation of this file.

package org.gecode.explorer.cocoa;

import com.apple.cocoa.foundation.*;
import com.apple.cocoa.application.*;

import org.gecode.*;
import org.gecode.explorer.*;


public class ExplorerNode extends SpaceNode {
    
    Node graphicalNode;
    
    private static HiddenSubtreeNode hiddenSubtree = new HiddenSubtreeNode();
    
    public ExplorerNode(Space theSpace) {
        super(theSpace);
//        this.graphicalNode = new UnknownNode();
        this.graphicalNode = new UnknownNode();
    }
    
    public ExplorerNode(int alternative) {
        super(alternative);
//        this.graphicalNode = new UnknownNode();
        this.graphicalNode = new UnknownNode();
    }
    
//    public void mark() {
//        marked = true;
//    }
    
//    public void unmark() {
//        marked = false;
//    }
    
//    public void toggleMarked() {
//        marked = ! marked;
//    }
    
    public int getWidth() {
        return unit;
    }
    
    public Shape getUnknownShape() {
        return graphicalNode.getUnknownShape();
    }
    
    public Shape getHiddenShape() {
        return graphicalNode.getHiddenShape();
    }
    
    public void drawNode(Object gc, int x, int y) {
        if (isHidden()) {
            hiddenSubtree.draw(gc, x, y);
        } else {
            graphicalNode.draw(gc, x, y);
        }
    }
    
    public void move(int offset) {}
    
    public SpaceNode createChild(int i) {
        return new ExplorerNode(i);
    }
    
    public VisualNode findNodeForPoint(int x, int y) {
        Coordinate foundPoint = new Coordinate();
        ExplorerNode foundNode = (ExplorerNode) super.findNode(x, y, foundPoint);
        if (foundNode != null) {
            if (foundNode.isHidden()) {
                if (hiddenSubtree.containsPoint(foundPoint)) {
                    return foundNode;
                }
            } else {
                if (foundNode.graphicalNode.containsPoint(foundPoint)) {
                    return foundNode;
                }
            }
        }
        return null;
    }

    public int getNumberOfChildNodes() {
        if (getStatus() == Undetermined) {
            int numberOfChildren = super.getNumberOfChildNodes();
            switch(getStatus()) {
                case SS_FAILED:
                    graphicalNode = new FailureNode();
                    break;
                case SS_SOLVED:
                    graphicalNode = new SolutionNode();
                    break;
                case SS_BRANCH:
                    graphicalNode = new ChoiceNode();
                    break;
            }
            return numberOfChildren;
        } else {
            return super.getNumberOfChildNodes();
        }
    }

    public void connect(Object gc, int x, int y, int parentX, int parentY) {
        if (! isRoot()) {
            NSAffineTransform parentTransform = new NSAffineTransform();
            parentTransform.translateXYBy(parentX, parentY);
            ExplorerNode parent = (ExplorerNode) getParent();
            NSPoint p1 = parentTransform.transformPoint(parent.graphicalNode.getBotConnector());
            
            NSAffineTransform transform = new NSAffineTransform();
            transform.translateXYBy(x, y);
            NSPoint p2 = transform.transformPoint(graphicalNode.getTopConnector());
            
            NSBezierPath connection = new NSBezierPath();
            connection.moveToPoint(p1);
            connection.lineToPoint(p2);
            connection.setLineWidth(5);
            connection.stroke();
        }
    }    
    
}