FilterTreeItem.java

Go to the documentation of this file.

package org.sleuthkit.autopsy.timeline.ui.filtering;

import javafx.beans.Observable;
import javafx.collections.ListChangeListener;
import javafx.collections.MapChangeListener;
import javafx.collections.ObservableMap;
import javafx.scene.control.TreeItem;
import org.sleuthkit.autopsy.timeline.filters.CompoundFilter;
import org.sleuthkit.autopsy.timeline.filters.Filter;

final public class FilterTreeItem extends TreeItem<Filter> {

    public FilterTreeItem(Filter f, ObservableMap<String, Boolean> expansionMap) {
        super(f);

        expansionMap.addListener((MapChangeListener.Change<? extends String, ? extends Boolean> change) -> {
            if (change.getKey() == f.getDisplayName()) {
                setExpanded(expansionMap.get(change.getKey()));
            }
        });

        if (expansionMap.get(f.getDisplayName()) != null) {
            setExpanded(expansionMap.get(f.getDisplayName()));
        }

        expandedProperty().addListener((Observable observable) -> {
            expansionMap.put(f.getDisplayName(), isExpanded());
        });

        if (f instanceof CompoundFilter<?>) {
            CompoundFilter<?> compoundFilter = (CompoundFilter<?>) f;

            for (Filter subFilter : compoundFilter.getSubFilters()) {
                getChildren().add(new FilterTreeItem(subFilter, expansionMap));
            }

            compoundFilter.getSubFilters().addListener((ListChangeListener.Change<? extends Filter> c) -> {
                while (c.next()) {
                    for (Filter subfFilter : c.getAddedSubList()) {
                        setExpanded(true);
                        getChildren().add(new FilterTreeItem(subfFilter, expansionMap));
                    }
                }
            });
        }
    }
}