com.github.cosycode.bdmp.BdmpHandle Maven / Gradle / Ivy

Go to download
package com.github.cosycode.bdmp;

import com.github.cosycode.common.util.io.IoUtils;
import lombok.extern.slf4j.Slf4j;

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

/**
 * Description :  生成 Bdmp 图片工具类.
 * 因为当前类可能需要手动输入到指定环境, 因此当前类中的代码, 以及其调用到的代码不使用任何其它非JRE官方jar包
 *
 * @author CPF
 * Date: 2020/5/19 15:40
 */
@Slf4j
public class BdmpHandle {

    private BdmpHandle() {
    }

    /**
     * 生成图片宽度为 2 * borderLength +
     *
     * @param file            转换文件路径
     * @param rowPxNum        一行像素个数
     * @param pixelSideLength 像素宽度
     * @param margin          margin长度,
     * @param powerOf2        2的指数幂, 可选值为 1, 2, 4, 8, 以及8的倍数,
     *                        像素颜色类型为 2 ^ powerOf2 个,
     *                        如果为 1, 像素颜色为  2种, 每 1 个bit作为一个像素,
     *                        如果为 2, 像素颜色为  4种, 每 2 个bit作为一个像素存储.
     *                        如果为 4, 像素颜色为 16种, 每 4 个bit作为一个像素存储.
     *                        如果为 8, 像素颜色为256种, 每 8 个bit作为一个像素存储.
     *                        如果为8n, 像素颜色为 2 ^ 8n 种, 每 8n 个bit作为一个像素存储.
     * @throws IOException 写入文件和读取文件流异常
     */
    public static BufferedImage convertFileToBdmp(File file, int rowPxNum, int pixelSideLength, int margin, byte powerOf2) throws IOException {
        BdmpGeneConfig bdmpGeneConfig = new BdmpGeneConfig();
        bdmpGeneConfig.setMargin(margin);
        bdmpGeneConfig.setRowPixelCnt(rowPxNum);
        bdmpGeneConfig.setPixelSideWidth(pixelSideLength);
        bdmpGeneConfig.setPixelSideHeight(pixelSideLength);
        bdmpGeneConfig.setMappingColor(BdmpUtils.getPxType(powerOf2));
        final BdmpSource bdmpSource = BdmpSource.geneByFile(file);
        return PixelPngDrawer.geneRatePixelPng(new BdmpGeneInfo(bdmpGeneConfig, bdmpSource));
    }

    /**
     * @param filePath 文件路径
     * @param savePath 存储路径
     * @param rowPxNum 一行像素个数
     * @param pxWidth  像素宽度
     * @param margin   margin长度
     * @param powerOf2 2的指数幂, 可选值为 1, 2, 4, 8, 以及8的倍数,
     *                 像素颜色类型为 2 ^ powerOf2 个,
     *                 如果为 1, 像素颜色为  2种, 每 1 个bit作为一个像素,
     *                 如果为 2, 像素颜色为  4种, 每 2 个bit作为一个像素存储.
     *                 如果为 4, 像素颜色为 16种, 每 4 个bit作为一个像素存储.
     *                 如果为 8, 像素颜色为256种, 每 8 个bit作为一个像素存储.
     *                 如果为8n, 像素颜色为 2 ^ 8n 种, 每 8n 个bit作为一个像素存储.
     * @throws IOException 写入文件和读取文件流异常
     */
    public static void convertFileToBdmp(String filePath, String savePath, int rowPxNum, int pxWidth, int margin, byte powerOf2) throws IOException {
        File file = new File(filePath);
        final BufferedImage image = convertFileToBdmp(file, rowPxNum, pxWidth, margin, powerOf2);
        // 保存图片 JPEG表示保存格式
        ImageIO.write(image, "png", new FileOutputStream(savePath));
        log.info("pixel end " + filePath);
    }

    /**
     * @param filePath 文件路径
     * @param rowPxNum 一行像素个数
     * @param pxWidth  像素宽度
     * @param margin   margin长度
     * @param powerOf2 2的指数幂, 可选值为 1, 2, 4, 8, 以及8的倍数,
     *                 像素颜色类型为 2 ^ powerOf2 个,
     *                 如果为 1, 像素颜色为  2种, 每 1 个bit作为一个像素,
     *                 如果为 2, 像素颜色为  4种, 每 2 个bit作为一个像素存储.
     *                 如果为 4, 像素颜色为 16种, 每 4 个bit作为一个像素存储.
     *                 如果为 8, 像素颜色为256种, 每 8 个bit作为一个像素存储.
     *                 如果为8n, 像素颜色为 2 ^ 8n 种, 每 8n 个bit作为一个像素存储.
     * @throws IOException 写入文件和读取文件流异常
     */
    public static void convertFileToBdmp(String filePath, int rowPxNum, int pxWidth, int margin, byte powerOf2) throws IOException {
        convertFileToBdmp(filePath, filePath + ".png", rowPxNum, pxWidth, margin, powerOf2);
    }

    /**
     * 将路径指向的 Bdmp 转换为文件并存储到指定文件夹
     *
     * @param picPath     Bdmp 图片路径
     * @param saveDirPath 解析后的文件存储路径
     */
    public static boolean convertBdmpToFile(String picPath, String saveDirPath) throws IOException {
        final BufferedImage image = BdmpUtils.load(picPath);
        final BdmpRecInfo picRecInfo = BdmpRecognizer.resolver(image);
        if (picRecInfo == null) {
            log.warn("未识别出像素图片区域");
            return false;
        }
        boolean check = picRecInfo.check();
        if (!check) {
            log.warn("转换文件失败, MD5值不一样");
            return false;
        }
        if (!(saveDirPath.endsWith("\\") && saveDirPath.endsWith("/"))) {
            saveDirPath += File.separator;
        }
        // 确保存储的文件夹存在
        IoUtils.insureFileDirExist(new File(saveDirPath));
        // 写入文件
        try (FileOutputStream outputStream = new FileOutputStream(new File(saveDirPath + picRecInfo.getBdmpHeader().getTag()))) {
            outputStream.write(picRecInfo.getFileContent());
        }
        return true;
    }

}