1 /*
   2  * Copyright (c) 2012, 2014, Oracle and/or its affiliates.
   3  * All rights reserved. Use is subject to license terms.
   4  *
   5  * This file is available and licensed under the following license:
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  *  - Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  *  - Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in
  15  *    the documentation and/or other materials provided with the distribution.
  16  *  - Neither the name of Oracle Corporation nor the names of its
  17  *    contributors may be used to endorse or promote products derived
  18  *    from this software without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  23  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  24  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  27  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  28  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  29  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  30  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  31  */
  32 
  33 package com.oracle.javafx.scenebuilder.kit.fxom.sampledata;
  34 
  35 import javafx.scene.control.TreeItem;
  36 import javafx.scene.control.TreeView;
  37 import javafx.scene.shape.Circle;
  38 import javafx.scene.shape.Rectangle;
  39 
  40 /**
  41  *
  42  */
  43 class TreeViewSampleData extends AbstractSampleData {
  44 
  45     private final TreeItem<String> sampleRoot;
  46 
  47     public TreeViewSampleData() {
  48         int i = 0;
  49         sampleRoot = new TreeItem<>(lorem(i++));
  50         sampleRoot.setExpanded(true);
  51         for (int j = 0; j<10; j++) {
  52             final Rectangle r = new Rectangle(10, 10);
  53             r.setFill(color(i));
  54             TreeItem<String> child = new TreeItem<>(lorem(i++));
  55             child.setExpanded(true);
  56             child.setGraphic(r);
  57             for (int k=0; k<3; k++) {
  58                 final TreeItem<String> child2 = new TreeItem<>(lorem(i++));
  59                 child2.setExpanded(true);
  60                 final Circle c = new Circle(5);
  61                 c.setFill(color(i));
  62                 child2.setGraphic(c);
  63                 child.getChildren().add(child2);
  64             }
  65             sampleRoot.getChildren().add(child);
  66         }
  67     }
  68 
  69     /*
  70      * AbstractSampleData
  71      */
  72 
  73 
  74     @Override
  75     public void applyTo(Object sceneGraphObject) {
  76         assert sceneGraphObject instanceof TreeView;
  77         @SuppressWarnings("unchecked")
  78         final TreeView<String> treeView = (TreeView<String>) sceneGraphObject;
  79         treeView.setRoot(sampleRoot);
  80     }
  81 
  82     @Override
  83     public void removeFrom(Object sceneGraphObject) {
  84         assert sceneGraphObject instanceof TreeView;
  85         @SuppressWarnings("unchecked")
  86         final TreeView<String> treeView = (TreeView<String>) sceneGraphObject;
  87         treeView.setRoot(null);
  88     }
  89 
  90 }