import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

public class ImageWatermark {

    public static void main(String[] args) {
        try {
            // 1. 加載圖像
            File inputFile = new File("path/to/your/image.jpg");  // 修改為你的圖片路徑
            BufferedImage image = ImageIO.read(inputFile);

            // 2. 添加文字水印
            String textWatermark = "Sample Watermark";
            BufferedImage textWatermarkedImage = addTextWatermark(image, textWatermark);

            // 3. 添加圖片水印
            File watermarkImageFile = new File("path/to/your/watermark_image.png");  // 修改為水印圖片路徑
            BufferedImage watermarkImage = ImageIO.read(watermarkImageFile);
            BufferedImage finalImage = addImageWatermark(textWatermarkedImage, watermarkImage);

            // 4. 保存添加水印后的圖像
            File outputFile = new File("path/to/save/watermarked_image.png");  // 輸出文件路徑
            ImageIO.write(finalImage, "PNG", outputFile);  // 保存為PNG格式

            System.out.println("Watermark added and image saved successfully!");

        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    // 添加文字水印
    private static BufferedImage addTextWatermark(BufferedImage originalImage, String watermarkText) {
        Graphics2D g2d = (Graphics2D) originalImage.getGraphics();

        // Set font and transparency
        Font font = new Font("Arial", Font.BOLD, 50);
        g2d.setFont(font);
        g2d.setColor(new Color(255, 255, 255, 128));  // 白色且半透明

        // Get the watermark's position (center the watermark on the image)
        FontMetrics fontMetrics = g2d.getFontMetrics();
        int x = (originalImage.getWidth() - fontMetrics.stringWidth(watermarkText)) / 2;
        int y = originalImage.getHeight() / 2;

        // Draw watermark text
        g2d.drawString(watermarkText, x, y);

        // Dispose the graphics context
        g2d.dispose();

        return originalImage;
    }

    // 添加圖片水印
    private static BufferedImage addImageWatermark(BufferedImage originalImage, BufferedImage watermarkImage) {
        Graphics2D g2d = (Graphics2D) originalImage.getGraphics();

        // Get the watermark image's position (bottom-right corner)
        int x = originalImage.getWidth() - watermarkImage.getWidth() - 10;  // 10px margin from the right edge
        int y = originalImage.getHeight() - watermarkImage.getHeight() - 10;  // 10px margin from the bottom edge

        // Draw watermark image with transparency
        g2d.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.5f));  // Set transparency
        g2d.drawImage(watermarkImage, x, y, null);

        // Dispose the graphics context
        g2d.dispose();

        return originalImage;
    }
}