package com.zmh.daily.spring;

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

/**
 * 春聯(lián)
 *
 * @version v1.0
 * @Author: sunshinezhang
 * @Date: 2022/1/25 3:27 下午
 */
public class Couplet {
    //設(shè)置生成文件存放文件夾
    final static String imageFolderPath = "/Users/sunshinezhang/Desktop/1/";
    //設(shè)置背景圖
    final static String sourceFilePath = "/Users/sunshinezhang/Desktop/1/2.png";
    //設(shè)置字體
    final static String fontName = "宋體";


    public static void main(String[] args) {
        write();
    }

    public static void write() {
        BufferedImage image = new BufferedImage(640 * 4, 640 + 2 * 640 * 7, BufferedImage.TYPE_INT_RGB);
        Graphics g = image.createGraphics();
        g.drawImage(writeH("辭舊迎新"), 0, 0, 640 * 4, 640, null);
        g.drawImage(writeV("舊歲又添幾個喜"), 0, 640, 640 * 2, 2 * 640 * 7, null);
        g.drawImage(writeV("新年更上一層樓"), 2 * 640, 640, 640 * 2, 2 * 640 * 7, null);
        //輸出文件
        try {
            ImageIO.write(image, "png", new File(imageFolderPath + "春聯(lián).png"));
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    //橫批
    public static BufferedImage writeH(String text) {
        char[] ch = text.toCharArray();
        BufferedImage image = new BufferedImage(640 * ch.length, 640, BufferedImage.TYPE_INT_RGB);
        Graphics g = image.createGraphics();
        for (int i = 0; i < ch.length; i++) {
            char c = ch[i];
            g.drawImage(fontMark(String.valueOf(c)), 640 * i, 0, 640, 640, null);
        }
        g.dispose();
        return image;
    }

    //上下聯(lián)
    public static BufferedImage writeV(String text) {
        char[] ch = text.toCharArray();
        BufferedImage image = new BufferedImage(2 * 640, 2 * 640 * ch.length, BufferedImage.TYPE_INT_RGB);
        Graphics g = image.createGraphics();
        for (int i = 0; i < ch.length; i++) {
            char c = ch[i];
            g.drawImage(fontMark(String.valueOf(c)), 0, 2 * 640 * i, 2 * 640, 2 * 640, null);
        }
        g.dispose();
        return image;
    }


    public static BufferedImage fontMark(String content) {
        try {
            BufferedImage image = ImageIO.read(new File(sourceFilePath));
            return fontMark(image, content);
        } catch (IOException e) {
            return null;
        }
    }

    //文字標記
    public static BufferedImage fontMark(BufferedImage bufImg, String content) {
        Font font = new Font(fontName, Font.BOLD, 350);
        Graphics2D graphics = bufImg.createGraphics();
        graphics.setColor(Color.black);
        graphics.setFont(font);
        graphics.drawString(content, 230, 540);
        graphics.dispose();
        System.out.println(content);
        return bufImg;
    }
}