de.learnlib.algorithms.ttt.base.AbstractBaseDTNode Maven / Gradle / Ivy

Go to download
/* Copyright (C) 2013-2019 TU Dortmund
 * This file is part of LearnLib, http://www.learnlib.de/.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.learnlib.algorithms.ttt.base;

import java.util.Iterator;

import de.learnlib.datastructure.discriminationtree.iterators.DiscriminationTreeIterators;
import de.learnlib.datastructure.discriminationtree.model.AbstractTemporaryIntrusiveDTNode;
import de.learnlib.datastructure.list.IntrusiveListElem;
import net.automatalib.words.Word;

public abstract class AbstractBaseDTNode
        extends AbstractTemporaryIntrusiveDTNode, D, TTTState, IncomingList, AbstractBaseDTNode>
        implements IntrusiveListElem> {

    private final IncomingList incoming = new IncomingList<>();

    public AbstractBaseDTNode() {
        this(null, null);
    }

    public AbstractBaseDTNode(AbstractBaseDTNode parent, D parentEdgeLabel) {
        super(parent, parentEdgeLabel, null);
    }

    public TTTState anySubtreeState() {
        AbstractBaseDTNode curr = this;
        while (!curr.isLeaf()) {
            curr = curr.anyChild();
        }
        return curr.data;
    }

    @Override
    public AbstractBaseDTNode anyChild() {
        assert isInner();
        return children.values().iterator().next();
    }

    public Iterable> subtreeStates() {
        return this::subtreeStatesIterator;
    }

    public Iterator> subtreeStatesIterator() {
        return DiscriminationTreeIterators.transformingLeafIterator(this, AbstractBaseDTNode::getData);
    }

    public IncomingList getIncoming() {
        return incoming;
    }

    public Iterator> subtreeNodesIterator() {
        return DiscriminationTreeIterators.nodeIterator(this);
    }

    /**
     * Updates the {@link TTTTransition#nonTreeTarget} attribute to point to this node for all transitions in the
     * incoming list.
     */
    void updateIncoming() {
        for (TTTTransition trans : incoming) {
            trans.nonTreeTarget = this;
        }
    }
}