/*
* Autopsy Forensic Browser
*
* Copyright 2013 Basis Technology Corp.
* Contact: carrier <at> sleuthkit <dot> org
*
* 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 org.sleuthkit.autopsy.timeline.ui.detailview.tree;
import java.net.URL;
import java.util.Arrays;
import java.util.Comparator;
import java.util.ResourceBundle;
import javafx.application.Platform;
import javafx.beans.Observable;
import javafx.collections.ObservableList;
import javafx.fxml.FXML;
import javafx.scene.control.ComboBox;
import javafx.scene.control.SelectionMode;
import javafx.scene.control.Tooltip;
import javafx.scene.control.TreeCell;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeView;
import javafx.scene.image.ImageView;
import javafx.scene.layout.BorderPane;
import javafx.scene.layout.StackPane;
import javafx.scene.shape.Rectangle;
import org.sleuthkit.autopsy.timeline.FXMLConstructor;
import org.sleuthkit.autopsy.timeline.TimeLineController;
import org.sleuthkit.autopsy.timeline.TimeLineView;
import org.sleuthkit.autopsy.timeline.events.AggregateEvent;
import org.sleuthkit.autopsy.timeline.events.FilteredEventsModel;
import org.sleuthkit.autopsy.timeline.ui.detailview.AggregateEventNode;
import org.sleuthkit.autopsy.timeline.ui.detailview.DetailViewPane;
/**
* Display two trees. one shows all folders (groups) and calls out folders with
* images. the user can select folders with images to see them in the main
* GroupListPane The other shows folders with hash set hits.
*/
public class NavPanel extends BorderPane implements TimeLineView {
private TimeLineController controller;
private FilteredEventsModel filteredEvents;
@FXML
private ResourceBundle resources;
@FXML
private URL location;
private DetailViewPane detailViewPane;
/**
* TreeView for folders with hash hits
*/
@FXML
private TreeView< NavTreeNode> eventsTree;
@FXML
private ComboBox<Comparator<TreeItem<NavTreeNode>>> sortByBox;
public NavPanel() {
FXMLConstructor.construct(this, "NavPanel.fxml");
}
public void setChart(DetailViewPane detailViewPane) {
this.detailViewPane = detailViewPane;
detailViewPane.setSelectionModel(eventsTree.getSelectionModel());
setRoot();
detailViewPane.getAggregatedEvents().addListener((Observable observable) -> {
setRoot();
});
detailViewPane.getSelectedNodes().addListener((Observable observable) -> {
eventsTree.getSelectionModel().clearSelection();
detailViewPane.getSelectedNodes().forEach((AggregateEventNode t) -> {
eventsTree.getSelectionModel().select(((NavTreeItem) eventsTree.getRoot()).findTreeItemForEvent(t.getEvent()));
});
});
}
private void setRoot() {
RootItem root = new RootItem();
final ObservableList<AggregateEvent> aggregatedEvents = detailViewPane.getAggregatedEvents();
synchronized (aggregatedEvents) {
for (AggregateEvent agg : aggregatedEvents) {
root.insert(agg);
}
}
Platform.runLater(() -> {
eventsTree.setRoot(root);
});
}
@Override
public void setController(TimeLineController controller) {
this.controller = controller;
setModel(controller.getEventsModel());
}
@Override
public void setModel(FilteredEventsModel filteredEvents) {
this.filteredEvents = filteredEvents;
}
@FXML
void initialize() {
assert sortByBox != null : "fx:id=\"sortByBox\" was not injected: check your FXML file 'NavPanel.fxml'.";
sortByBox.getItems().setAll(Arrays.asList(TreeComparator.Description, TreeComparator.Count));
sortByBox.getSelectionModel().select(TreeComparator.Description);
sortByBox.getSelectionModel().selectedItemProperty().addListener((Observable o) -> {
((RootItem) eventsTree.getRoot()).resort(sortByBox.getSelectionModel().getSelectedItem());
});
eventsTree.setShowRoot(false);
eventsTree.setCellFactory((TreeView<NavTreeNode> p) -> new EventTreeCell());
eventsTree.getSelectionModel().setSelectionMode(SelectionMode.MULTIPLE);
}
/** A tree cell to display {@link NavTreeNode}s. Shows the description, and
* count, as well a a "legend icon" for the event type. */
private static class EventTreeCell extends TreeCell<NavTreeNode> {
@Override
protected void updateItem(NavTreeNode item, boolean empty) {
super.updateItem(item, empty);
if (item != null) {
final String text = item.getDescription() + " (" + item.getCount() + ")";
setText(text);
setTooltip(new Tooltip(text));
Rectangle rect = new Rectangle(24, 24);
rect.setArcHeight(5);
rect.setArcWidth(5);
rect.setStrokeWidth(2);
rect.setStroke(item.getType().getColor());
rect.setFill(item.getType().getColor().deriveColor(0, 1, 1, 0.1));
setGraphic(new StackPane(rect, new ImageView(item.getType().getFXImage())));
} else {
setText(null);
setTooltip(null);
setGraphic(null);
}
}
}
}