Create Disk View

* add sample disk view * attach controller to StartPage.fxml
brp14005 · Feb 8, 2016 · a50a4a5 · a50a4a5
1 parent add73ce
commit a50a4a5
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 11 deletions.
diff --git a/src/main/java/Controller.java b/src/main/java/Controller.java
@@ -2,9 +2,16 @@
 
 import javafx.fxml.FXML;
 import javafx.scene.control.MenuItem;
+import javafx.scene.control.TreeCell;
 import javafx.scene.control.TreeItem;
 import javafx.scene.control.TreeView;
+import javafx.util.Callback;
 
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.util.Arrays;
 import java.util.ResourceBundle;
 
 public class Controller{
@@ -13,23 +20,36 @@ public class Controller{
     @FXML private MenuItem fileTag;
     @FXML private MenuItem fileExit;
     @FXML private MenuItem editPreferences;
-    @FXML private TreeView<String> dirTree;
+    @FXML private TreeView<File> dirTree;
 
     @FXML private ResourceBundle resources;
 
-    @FXML
-    private void initialize() {
-//        createTree();
+    @FXML private void initialize() {
+        createTree();
     }
 
     // Create the directory Tree
     private void createTree() {
-        TreeItem<String> root = new TreeItem<String>("Root");
-        TreeItem<String> itemChild = new TreeItem<String>("Child");
-        itemChild.setExpanded(false);
-        root.getChildren().add(itemChild);
-        dirTree.setRoot(root);
+        dirTree.setRoot(new TreeItem<>());
+        dirTree.setShowRoot(false);
+        File[] roots = File.listRoots();
+        for (File root : roots) {
+            TreeItem<File> itemChild = new TreeItem<>(root);
+            itemChild.setExpanded(false);
+            dirTree.getRoot().getChildren().add(itemChild);
+        }
+        /*dirTree.setCellFactory(new Callback<TreeView<File>, TreeCell<File>>() {
+            @Override
+            public TreeCell<File> call(TreeView<File> param) {
+                return new TreeFieldImpl();
+            }
+        });*/
     }
 
+    private class TreeFieldImpl extends TreeCell<File> {
+        public TreeFieldImpl() {
+//            MenuItem
+        }
+    }
 }
 
diff --git a/src/main/resources/StartPage.fxml b/src/main/resources/StartPage.fxml
@@ -10,7 +10,7 @@
 <?import javafx.scene.control.*?>
 <?import javafx.scene.layout.*?>
 
-<BorderPane maxHeight="-Infinity" maxWidth="-Infinity" minHeight="-Infinity" minWidth="-Infinity" prefHeight="480.0" prefWidth="640.0" xmlns="http://javafx.com/javafx/8" xmlns:fx="http://javafx.com/fxml/1">
+<BorderPane maxHeight="-Infinity" maxWidth="-Infinity" minHeight="-Infinity" minWidth="-Infinity" prefHeight="480.0" prefWidth="640.0" xmlns="http://javafx.com/javafx/8" xmlns:fx="http://javafx.com/fxml/1" fx:controller="main.java.Controller">
     <top>
         <MenuBar id="menubar-pane" BorderPane.alignment="CENTER">
             <menus>
@@ -42,7 +42,7 @@
         </MenuBar>
     </top>
     <center>
-        <TreeView id="folder-view-pane" prefHeight="200.0" prefWidth="200.0" BorderPane.alignment="CENTER" />
+        <TreeView id="folder-view-pane" fx:id="dirTree" prefHeight="200.0" prefWidth="200.0" BorderPane.alignment="CENTER" />
     </center>
     <left>
         <BorderPane prefHeight="200.0" prefWidth="300.0" styleClass="pane-general-style" BorderPane.alignment="CENTER">