TrieWordNode.java

/*******************************************************************************
 * Copyright (c) 2012-present Jakub Kováč, Jozef Brandýs, Katarína Kotrlová,
 * Pavol Lukča, Ladislav Pápay, Viktor Tomkovič, Tatiana Tóthová
 *
 * 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 algvis.ds.trie;

import algvis.core.DataStructure;
import algvis.core.Node;
import algvis.core.NodeColor;
import algvis.core.history.HashtableStoreSupport;
import algvis.ui.Fonts;
import algvis.ui.view.View;

import java.awt.Color;
import java.awt.FontMetrics;
import java.util.Hashtable;

public class TrieWordNode extends Node {
    private String s = "";

    public TrieWordNode(DataStructure D, String s, int x, int y, NodeColor c) {
        super(D, 0, x, y);
        setS(s);
        setColor(c);
        this.D = D;
    }

    public TrieWordNode(DataStructure D, String s) {
        this(D, s, 0, 0, NodeColor.BLACK);
    }

    public String getS() {
        return s;
    }

    void setS(String s) {
        this.s = s;
    }

    /**
     * 
     * @param x
     *            how many chars will be cut from the beginning of the string
     */
    String cut(int x) {
        s = s.substring(x, s.length());
        return s;
    }

    @Override
    public void goNextTo(Node v) {
        goTo(v.tox + DataStructure.minsepx / 3, v.toy);
    }

    public void setAndGoNextTo(String s, Node v) {
        setS(s);
        goNextTo(v);
    }

    public String cutOneAndGoNextTo(Node v) {
        final String result = cut(1);
        goNextTo(v);
        return result;
    }

    @Override
    public void draw(View v) {
        if (s.compareTo("") == 0) {
            return;
        }
        final FontMetrics fm = v.getGraphics().getFontMetrics();
        final int width = (fm.stringWidth(s) + 4) / 2;
        final int height = (fm.getHeight() + 4) / 2;
        v.setColor(getColor().bgColor);
        v.fillRoundRectangle(x + width, y - height / 2, width, height, 3, 3);
        v.setColor(Color.BLACK);
        v.drawRoundRectangle(x + width, y - height / 2, width, height, 3, 3);
        v.setColor(getColor().fgColor);
        v.drawString(s, x + width, y - height / 2, Fonts.TYPEWRITER);
    }

    @Override
    public void storeState(Hashtable<Object, Object> state) {
        super.storeState(state);
        HashtableStoreSupport.store(state, hash + "s", s);
    }

    @Override
    public void restoreState(Hashtable<?, ?> state) {
        super.restoreState(state);
        final Object s = state.get(hash + "s");
        if (s != null) {
            this.s = (String) HashtableStoreSupport.restore(s);
        }
    }
}