org.key_project.keyide.ui.test.util.TreeViewerIterator.java Source code

Introduction

Here is the source code for org.key_project.keyide.ui.test.util.TreeViewerIterator.java

Source

/*******************************************************************************
 * Copyright (c) 2014 Karlsruhe Institute of Technology, Germany
 *                    Technical University Darmstadt, Germany
 *                    Chalmers University of Technology, Sweden
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Technical University Darmstadt - initial API and implementation and/or initial documentation
 *******************************************************************************/

package org.key_project.keyide.ui.test.util;

import java.util.LinkedList;
import java.util.List;

import org.eclipse.jface.viewers.TreeViewer;
import org.eclipse.swt.widgets.Tree;
import org.eclipse.swt.widgets.TreeItem;

// TODO: Document class TreeViewerIterator
public class TreeViewerIterator {
    private List<TreeItem> treeItemList = new LinkedList<TreeItem>();

    public TreeViewerIterator(TreeViewer viewer) {
        viewer.expandAll();
        treeToList(viewer.getTree());
    }

    private void treeToList(Tree tree) {
        TreeItem[] items = tree.getItems();
        iterateTree(items);
    }

    private void iterateTree(TreeItem[] items) {
        for (TreeItem item : items) {
            treeItemList.add(item);
            TreeItem[] children = item.getItems();
            iterateTree(children);
        }
    }

    public boolean hasNext() {
        if (!treeItemList.isEmpty()) {
            return true;
        } else
            return false;
    }

    public TreeItem next() {
        TreeItem item = null;
        if (!treeItemList.isEmpty()) {
            item = treeItemList.get(0);
            treeItemList.remove(0);

        }
        return item;
    }

    public List<TreeItem> getList() {
        return treeItemList;
    }
}