java - 计算目录中的图像

Question

我想统计目录中有多少张图片。我有一些来自互联网的代码副本，可以检测目录中的总文件。

import java.io.*;

public class CountFilesInDirectory {
      public static void main(String[] args) {
            File f = new File("/home/pc3/Documents/ffmpeg_temp/");
            int count = 0;
            for (File file : f.listFiles()) {
                    if (file.isFile()) {
                            count++;
                    }
            }
            System.out.println("Number of files: " + count);
    }
}

我想计算一个特定的文件类型，比如 jpg/txt。我应该怎么办？

Answer 1

改变这个

if (file.isFile()) { 
  count++; 
} 

到

if (file.isFile() &&
    (file.getName().endsWith(".txt") || file.getName().endsWith(".jpg"))) { 
  count++; 
} 

这段代码所做的不仅仅是检查 file 表示的实体是否实际上是一个文件(而不是目录)，您还检查它的名称是否以您指定的特定扩展名结尾正在寻找

注意:您可以使用 String.toLowerCase() 方法在比较之前将名称转换为更准确。

更新(基于评论):对于更 OOP 的解决方案，您可以实现 java.io.FilenameFilter 接口(interface)并将其实例传递给 f.listFiles()——这样您就可以在程序的另一部分重复使用该过滤器，而无需重复太多代码。如果使用这种方法，则需要将 endsWith 逻辑移至 FilenameFilter 实现