java判断文件是图片还是视频

Java判断文件是图片还是视频

在开发中，有时候我们需要判断一个文件是图片还是视频。本文将介绍如何使用Java判断文件的类型，并给出示例代码。

文件类型判断

在Java中，我们可以通过对文件的扩展名进行判断来确定文件的类型。一般情况下，图片文件的扩展名包括.jpg、.jpeg、.png、.gif等；视频文件的扩展名则包括.mp4、.avi、.mov等。

方法一：通过扩展名判断

我们可以通过获取文件的扩展名，然后根据扩展名来判断文件的类型。

import java.io.File;

public class FileTypeChecker {
    public static void main(String[] args) {
        String filePath = "path/to/file"; // 替换成你的文件路径
        File file = new File(filePath);
        String extension = getFileExtension(file);

        if (isImage(extension)) {
            System.out.println("This file is an image.");
        } else if (isVideo(extension)) {
            System.out.println("This file is a video.");
        } else {
            System.out.println("This file is neither an image nor a video.");
        }
    }

    // 获取文件的扩展名
    public static String getFileExtension(File file) {
        String extension = "";
        String fileName = file.getName();
        int dotIndex = fileName.lastIndexOf(".");
        if (dotIndex > 0 && dotIndex < fileName.length() - 1) {
            extension = fileName.substring(dotIndex + 1).toLowerCase();
        }
        return extension;
    }

    // 判断是否为图片文件
    public static boolean isImage(String extension) {
        return extension.equals("jpg") || extension.equals("jpeg") || extension.equals("png") || extension.equals("gif");
    }

    // 判断是否为视频文件
    public static boolean isVideo(String extension) {
        return extension.equals("mp4") || extension.equals("avi") || extension.equals("mov");
    }
}

以上代码通过getFileExtension函数获取文件的扩展名，然后通过isImage和isVideo函数判断文件类型。如果是图片文件，则输出"This file is an image."；如果是视频文件，则输出"This file is a video."；否则输出"This file is neither an image nor a video."。

方法二：通过文件头判断

除了通过扩展名判断文件类型外，我们还可以通过文件的二进制数据来判断文件类型。每个文件的开头几个字节是文件的"魔数"，不同类型的文件有不同的魔数。

下面是一个示例代码，通过读取文件的二进制数据来判断文件类型：

import java.io.FileInputStream;
import java.io.IOException;

public class FileTypeChecker {
    public static void main(String[] args) {
        String filePath = "path/to/file"; // 替换成你的文件路径
        try {
            FileInputStream fis = new FileInputStream(filePath);
            byte[] header = new byte[10];
            fis.read(header);
            fis.close();

            if (isImage(header)) {
                System.out.println("This file is an image.");
            } else if (isVideo(header)) {
                System.out.println("This file is a video.");
            } else {
                System.out.println("This file is neither an image nor a video.");
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 判断是否为图片文件
    public static boolean isImage(byte[] header) {
        // 根据文件的魔数来判断，这里只是示例，实际情况需要根据具体文件类型来判断
        // 图片文件的魔数通常以特定的字节开头
        return header[0] == 0xFF && header[1] == 0xD8 && header[2] == 0xFF;
    }

    // 判断是否为视频文件
    public static boolean isVideo(byte[] header) {
        // 根据文件的魔数来判断，这里只是示例，实际情况需要根据具体文件类型来判断
        // 视频文件的魔数通常以特定的字节开头
        return header[0] == 0x00 && header[1] == 0x00 && header[2] == 0x00 && header[3] == 0x18;
    }
}

以上代码通过FileInputStream读取文件的前10个字节，并将其保存在header数组中。然后通过isImage和isVideo函数判断文件类型。如果是图片文件，则输出"This file is an image."；