WorkingTreeModel.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.turro.wd.tree;
 
import java.text.ParseException;
import org.turro.wd.files.Directory;
import org.turro.wd.files.DirectorySet;
import org.turro.wd.files.WorkFile;
import org.turro.wd.files.WorkFileSet;
import org.turro.wd.files.WorkingDirectory;
import org.apache.pivot.wtk.content.TreeNode;
import org.apache.pivot.collections.ArrayList;
import org.apache.pivot.wtk.content.TreeBranch;
 
public class WorkingTreeModel extends ArrayList<TreeNode> {
 
  private WorkingDirectory workingDirectory;
 
  public WorkingTreeModel(WorkingDirectory workingDirectory) throws ParseException {
    this.workingDirectory = workingDirectory;
    createList();
  }
 
  private void createList() throws ParseException {
    clear();
    addDirectories(null, workingDirectory.getDirs());
    addFiles(null, workingDirectory.getFiles());
  }
 
  private void addDirectories(TreeBranch root, DirectorySet dirs) throws ParseException {
    for(Directory d : dirs) {
      TreeDirectory td = new TreeDirectory();
      if(root == null) {
        add(td);
      } else {
        root.add(td);
      }
      td.setDirectory(d);
      addDirectories(td, d.getDirs());
      addFiles(td, d.getFiles());
    }
  }
 
  private void addFiles(TreeBranch root, WorkFileSet files) throws ParseException {
    for(WorkFile f : files) {
      TreeWorkFile twf = new TreeWorkFile();
      if(root == null) {
        add(twf);
      } else {
        root.add(twf);
      }
      twf.setWorkFile(f);
    }
  }
 
}