Question

嗨现在我有以下方法用于一次读取一个文件与具有此方法的类在同一目录中：

private byte[][] getDoubleByteArrayOfFile(String fileName, Region region)
    throws IOException
{
    BufferedImage image = ImageIO.read(getClass().getResource(fileName));
    byte[][] alphaInputData =
        new byte[region.getInputXAxisLength()][region.getInputYAxisLength()];
    for (int x = 0; x < alphaInputData.length; x++)
    {
        for (int y = 0; y < alphaInputData[x].length; y++)
        {
            int color = image.getRGB(x, y);
            alphaInputData[x][y] = (byte)(color >> 23);
        }
    }
    return alphaInputData;
}

我想知道我怎么能这样做，以便不是将“fileName”作为参数，而是将目录名作为参数，然后遍历该目录中的所有文件并对其执行相同的操作。谢谢！

Answer 1

如果您使用的是Java 7，那么您需要查看NIO.2。

具体来说，请查看Listing a Directory's Contents部分。

Path dir = Paths.get("/directory/path");
try (DirectoryStream<Path> stream = Files.newDirectoryStream(dir)) {
    for (Path file: stream) {
        getDoubleByteArrayOfFile(file.getFileName(), someRegion);
    }
} catch (IOException | DirectoryIteratorException x) {
    // IOException can never be thrown by the iteration.
    // In this snippet, it can only be thrown by newDirectoryStream.
    System.err.println(x);
}

Answer 2

以下是一个可能有帮助的简单示例：

private ArrayList<byte[][]> getDoubleByteArrayOfDirectory(String dirName,
    Region region) throws IOException {
       ArrayList<byte[][]> results = new ArrayList<byte[][]>();
       File directory = new File(dirName);
       if (!directory.isDirectory()) return null //or handle however you wish
       for (File file : directory.listFiles()) {
           results.add(getDoubleByteArrayOfFile(file.getName()), region);
       }  
       return results;
}

不完全是你要求的，因为它包装你的旧方法而不是重写它，但我觉得它更清洁一点，让你可以选择仍然处理单个文件。请务必根据您的实际要求调整返回类型以及如何处理region（很难从问题中得知）。

Answer 3

这很简单，使用File#listFiles()返回指定文件中的文件列表，该列表必须是目录。要确保文件是目录，只需使用File#isDirectory()。在您决定如何返回字节缓冲区时会出现问题。由于该方法返回2d缓冲区，因此必须使用3d字节缓冲区数组，或者在这种情况下，List似乎是最佳选择，因为相关目录中将存在未知数量的文件。



    private List getDoubleByteArrayOfDirectory(String directory, Region region) throws IOException {
      File directoryFile = new File(directory);

      if(!directoryFile.isDirectory()) {
        throw new IllegalArgumentException("path must be a directory");
      }

      List results = new ArrayList();

      for(File temp : directoryFile.listFiles()) {
        if(temp.isDirectory()) {
          results.addAll(getDoubleByteArrayOfDirectory(temp.getPath(), region));
        }else {
          results.add(getDoubleByteArrayOfFile(temp.getPath(), region));
        }
      }
      return results;
    }

Answer 4

您可以参阅list and listFiles文档了解如何执行此操作。

Answer 5

我们也可以使用递归来处理带子目录的目录。这里我将逐个删除文件，您可以调用任何其他函数来处理它。

public static void recursiveProcess(File file) {
    //to end the recursive loop
    if (!file.exists())
        return;

    //if directory, go inside and call recursively
    if (file.isDirectory()) {
        for (File f : file.listFiles()) {
            //call recursively
            recursiveProcess(f);
        }
    }
    //call processing function, for example here I am deleting
    file.delete();
    System.out.println("Deleted (Processed) file/folder: "+file.getAbsolutePath());
}

迭代java中的目录

5 个答案: