JTree не фильтрует расширения - PullRequest
Новая
       20

JTree не фильтрует расширения

0 голосов
/ 10 ноября 2018

Я застрял на этой проблеме:

У меня есть JTree, который показывает весь каталог и его дочерние элементы, но мне нужно отфильтровать его, чтобы показать только каталоги и файлы с некоторыми расширениями (например, ".m", ".py"). 

package classes;

import java.io.File;
import java.io.FilenameFilter;
import java.util.Iterator;
import java.util.Vector;
import javax.swing.event.TreeModelEvent;
import javax.swing.event.TreeModelListener;
import javax.swing.tree.TreeModel;
import javax.swing.tree.TreePath;

public class FileTree implements TreeModel{
private File root;
private Vector listeners = new Vector();

public FileTree(File rootDirectory){
    root = rootDirectory;
}

@Override
public Object getRoot(){
    return root;
}

@Override
public Object getChild(Object parent, int index){
    File directory = (File) parent;
    String[] children = directory.list(new FilenameFilter(){
        @Override
        public boolean accept(File dir, String name){
            return name.endsWith(".m") || dir.isDirectory();
        }           
    });
    return new FileTree.TreeFile(directory, children[index]);
}

@Override
public int getChildCount(Object parent){
    File file = (File) parent;
    if (file.isDirectory()){
        String[] fileList = file.list();          
        if (fileList != null){
            return file.list().length;
        }
    }
    return 0;
}

@Override
public boolean isLeaf(Object node){
    File file = (File) node;
    return file.isFile();
}

@Override
public int getIndexOfChild(Object parent, Object child){
    File directory = (File) parent;
    File file = (File) child;
    String[] children = directory.list();
    for (int i = 0; i < children.length; i++){
        if (file.getName().equals(children[i])){
            return i;
        }
    }
    return -1;
}

@Override
public void valueForPathChanged(TreePath path, Object value){
    File oldFile = (File) path.getLastPathComponent();
    String fileParentPath = oldFile.getParent();
    String newFileName = (String) value;
    File targetFile = new File(fileParentPath, newFileName);
    oldFile.renameTo(targetFile);
    File parent = new File(fileParentPath);
    int[] changedChildrenIndices = {getIndexOfChild(parent, targetFile)};
    Object[] changedChildren = {targetFile};
    fireTreeNodesChanged(path.getParentPath(), changedChildrenIndices, changedChildren);
}

private void fireTreeNodesChanged(TreePath parentPath, int[] indices, Object[] children){
    TreeModelEvent event = new TreeModelEvent(this, parentPath, indices, children);
    Iterator iterator = listeners.iterator();
    TreeModelListener listener = null;
    while (iterator.hasNext()){
        listener = (TreeModelListener) iterator.next();
        listener.treeNodesChanged(event);
    }
}

@Override
public void addTreeModelListener(TreeModelListener listener){
    listeners.add(listener);
}

@Override
public void removeTreeModelListener(TreeModelListener listener){
    listeners.remove(listener);
}

private class TreeFile extends File{

    public TreeFile(File parent, String child){
        super(parent, child);
    }

    @Override
    public String toString(){
        return getName();
    }
  }
}

И в моем main() методе я называю это 

JTree tree = new JTree();
tree.setModel(new FileTree(new File("\\C:\\")));
treePanel.setViewportView(tree);

Я думаю, что проблема в getChild() методе, однако я просто могу понять, как ее решить.

Существуют ли различия между file.list() и file.listFiles() методами? Потому что я уже использовал оба и до сих пор не понимаю.

РЕДАКТИРОВАТЬ: после следующих предложений @Thomas, теперь он показывает мне эту ошибку:

PS: Когда я печатаю переменную "children" в методе getChild() для отладки, он возвращает мне все каталоги (их 14) в "C: \".

Исключение в потоке "AWT-EventQueue-0" java.lang.ArrayIndexOutOfBoundsException: 14

Обновленные методы: 

    @Override
public int getIndexOfChild(Object parent, Object child){
    File directory = (File) parent;
    File file = (File) child;
    String[] children = directory.list(new FilenameFilter(){
        @Override
        public boolean accept(File dir, String name){
            return name.endsWith(".m") || new File(dir, name).isDirectory();
        }
    });
    for (int i = 0; i < children.length; i++){
        if (file.getName().equals(children[i])){
            return i;
        }
    }
    return -1;
}

и 

    @Override
public int getChildCount(Object parent){
    File file = (File) parent;
    if (file.isDirectory()){
        String[] fileList = file.list(new FilenameFilter(){
            @Override
            public boolean accept(File dir, String name){
                return name.endsWith(".m") || new File(dir, name).isDirectory();
            }
        });          
        if (fileList != null){
            return file.list().length;
        }
    }
    return 0;
}

1 Ответ

0 голосов
/ 10 ноября 2018

Проблема в вашей реализации FilenameFilter.

Javadoc для метода FilenameFilter#accept (https://docs.oracle.com/javase/8/docs/api/java/io/FilenameFilter.html#accept-java.io.File-java.lang.String-)) описывает параметры:

  • dir - каталог, в котором был найден файл.
  • name - имя файла.

Из этого описания ясно, что ваш метод приема: 

    public boolean accept(File dir, String name){
        return name.endsWith(".m") || dir.isDirectory();
    }

всегда будет возвращать true, поскольку dir всегда является каталогом.

Если вы хотите проверить, является ли запись name в каталоге dir каталогом, вам придется переписать ваш метод accept как 

    public boolean accept(File dir, String name){
        return name.endsWith(".m") || new File(dir, name).isDirectory();
    }

Это даст вам только каталоги и файлы с именем, оканчивающимся на «.m».

Но теперь вам также нужно изменить методы getChildCount() и getIndexOfChild() - потому что в настоящее время они основаны на нефильтрованных дочерних списках.

...