org.marid.bd.shapes.LinkShape Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of marid-swing-ide Show documentation
Marid Integrated Development Environment
The newest version!
/*
 * Copyright (C) 2014 Dmitry Ovchinnikov
 * Marid, the free data acquisition and visualization software
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see .
 */

package org.marid.bd.shapes;

import org.marid.bd.BlockComponent;
import org.marid.swing.actions.MaridAction;

import javax.swing.*;
import java.awt.*;

/**
 * @author Dmitry Ovchinnikov
 */
public abstract class LinkShape {

    public final BlockComponent.Output output;
    public final BlockComponent.Input input;

    public LinkShape(BlockComponent.Output output, BlockComponent.Input input) {
        this.output = output;
        this.input = input;
    }

    public abstract void update();

    public void paint(Graphics2D g) {
        g.draw(getShape());
    }

    public void paint(Graphics2D g, boolean isCurrent) {
        g.setColor(isCurrent ? SystemColor.activeCaption : getColor());
        paint(g);
    }

    public abstract Shape getShape();

    public boolean isAssociatedWith(BlockComponent blockComponent) {
        return output.getBlockComponent() == blockComponent || input.getBlockComponent() == blockComponent;
    }

    public JPopupMenu popupMenu() {
        final JPopupMenu popupMenu = new JPopupMenu();
        popupMenu.add(new MaridAction("Remove", "remove",
                ev -> output.getBlockComponent().getSchemaEditor().removeLink(this)));
        return popupMenu;
    }

    public boolean isValid() {
        final Class it = input.getInput().getInputType();
        final Class ot = output.getOutput().getOutputType();
        return it.isAssignableFrom(ot) || it.isArray() && it.getComponentType().isAssignableFrom(ot);
    }

    public Color getColor() {
        if (isValid()) {
            return SystemColor.controlDkShadow;
        } else {
            return SystemColor.RED;
        }
    }

    public Class getOutputType() {
        return output.getOutput().getOutputType();
    }

    public Class getInputType() {
        return input.getInput().getInputType();
    }

    @Override
    public int hashCode() {
        return output.hashCode() ^ input.hashCode();
    }

    @Override
    public boolean equals(Object obj) {
        if (obj instanceof LinkShape) {
            final LinkShape that = (LinkShape) obj;
            return this.input == that.input && this.output == that.output;
        } else {
            return false;
        }
    }

    @Override
    public String toString() {
        return String.format("%s:%s - %s:%s",
                System.identityHashCode(output), output.getOutput().getName(),
                System.identityHashCode(input), input.getInput().getName());
    }
}