判斷base64中是不是隻包含一種顏色

安详的苦丁茶發表於2024-10-31

列印結果為true說明只有一種顏色

public static void main(String[] args) {
        // 替換為你的圖片路徑
        String filePath = "F:\\image\\a1.png";
        try {
            String base64Image = convertImageToBase64(filePath);
            System.out.println(base64Image);
            boolean isSingleColor = isSingleColor(base64Image);
            System.out.println("影像是否只有一種顏色: " + isSingleColor);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 將影像檔案轉換為 Base64 字串
    public static String convertImageToBase64(String filePath) throws IOException {
        File file = new File(filePath);
        FileInputStream fileInputStream = new FileInputStream(file);
        byte[] bytes = new byte[(int) file.length()];
        fileInputStream.read(bytes);
        fileInputStream.close();
        return Base64.getEncoder().encodeToString(bytes);
    }

    // 判斷 Base64 影像是否只有一種顏色
    public static boolean isSingleColor(String base64Image) {
        try {
            // 解碼 Base64 字串為位元組陣列
            byte[] imageBytes = Base64.getDecoder().decode(base64Image);
            // 從位元組陣列建立 BufferedImage
            BufferedImage image = ImageIO.read(new ByteArrayInputStream(imageBytes));

            if (image == null) {
                throw new IOException("影像為空,請檢查 Base64 字串。");
            }

            // 獲取第一個畫素的顏色
            int firstColor = image.getRGB(0, 0);
            // 檢查所有畫素的顏色
            for (int y = 0; y < image.getHeight(); y++) {
                for (int x = 0; x < image.getWidth(); x++) {
                    if (image.getRGB(x, y) != firstColor) {
                        return false; // 找到不同顏色
                    }
                }
            }
            return true; // 所有畫素顏色相同
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

相關文章