From a50a4a5eafff9145e98860a7e4fe5df8a752f52a Mon Sep 17 00:00:00 2001
From: Brian Patino <brian.patino@uconn.edu>
Date: Sun, 7 Feb 2016 21:59:33 -0500
Subject: [PATCH] Create Disk View * add sample disk view * attach controller
 to StartPage.fxml

---
 src/main/java/Controller.java     | 38 +++++++++++++++++++++++--------
 src/main/resources/StartPage.fxml |  4 ++--
 2 files changed, 31 insertions(+), 11 deletions(-)

diff --git a/src/main/java/Controller.java b/src/main/java/Controller.java
index a905ef4..0d9f721 100644
--- 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
index d5eb679..17cb614 100644
--- 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">