AnnotatedTree.Util.SemiDoneFileFilter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of AnnotatedTree Show documentation

Annotated constituency treebank library

There is a newer version: 1.0.58

package AnnotatedTree.Util;

import AnnotatedSentence.ViewLayerType;
import AnnotatedTree.ParseNodeDrawable;
import AnnotatedTree.ParseTreeDrawable;
import AnnotatedTree.Processor.Condition.IsLeafNode;
import AnnotatedTree.Processor.NodeDrawableCollector;
import AnnotatedTree.Processor.LayerExist.SemiContainsLayerInformation;

import javax.swing.filechooser.FileFilter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.ArrayList;

public class SemiDoneFileFilter extends FileFilter {

    private final String path;
    private final ViewLayerType viewLayerType;

    /**
     * Constructor for SemiDoneFileFilter. SemiDoneFileFilter is used to filter out the files whose leaf nodes are fully
     * annotated or fully not annotated. The files shown to the user are those that are partially annotated with the
     * given layer info.
     * @param path Path of the files.
     * @param viewLayerType Name of the layer for which annotation check is done.
     */
    public SemiDoneFileFilter(String path, ViewLayerType viewLayerType){
        this.path = path;
        this.viewLayerType = viewLayerType;
    }

    /**
     * Checks if the file satisfies the condition of the filter.
     * @param f File to be checked.
     * @return True if the file satisfies the condition of the filter, false otherwise.
     */
    public boolean accept(File f) {
        File file = new File(path + f.getName());
        try {
            ParseTreeDrawable tree = new ParseTreeDrawable(new FileInputStream(file));
            NodeDrawableCollector nodeDrawableCollector = new NodeDrawableCollector((ParseNodeDrawable) tree.getRoot(), new IsLeafNode());
            ArrayList leafList = nodeDrawableCollector.collect();
            return new SemiContainsLayerInformation(viewLayerType).satisfies(leafList);
        } catch (FileNotFoundException e) {
            return false;
        }
    }

    /**
     * Returns "Semi Done".
     * @return "Semi Done".
     */
    @Override
    public String getDescription() {
        return "Semi Done";
    }
}