package cc.mrbird.febs.mall.test;

import com.sun.image.codec.jpeg.JPEGCodec;
import com.sun.image.codec.jpeg.JPEGEncodeParam;
import com.sun.image.codec.jpeg.JPEGImageEncoder;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.awt.image.ConvolveOp;
import java.awt.image.Kernel;
import java.awt.image.RenderedImage;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

public class imageTest {

    public static void main(String[] args) throws IOException {

//        File originalImage = new File("D:\\image\\out.png");
//        try{
//            RenderedImage rendImage =getGray.gray(ImageIO.read(new File("D:\\image\\input.png")));
//            ImageIO.write(rendImage, "png", originalImage);
//        }catch(IOException e){
//            System.out.println(e);
//        }



        File originalImage = new File("D:\\image\\input.png");
        System.out.println("源文件大小" + originalImage.length());
        File resizedImg1 = new File("D:\\image\\out3.png");
        resize(originalImage, resizedImg1, Integer.parseInt(Long.toString(originalImage.length())), 0.1f);
        System.out.println("0.1转换后文件大小" + resizedImg1.length());

    }
//        ## 遍历文件夹下图片批量压缩
//    public static void main(String[] args) throws IOException {
//        File folder = new File("C:\\Users\\Win\\Desktop\\faceImages");//文件夹路径
//        File[] listOfFiles = folder.listFiles();
//        for (int i = 0; i < listOfFiles.length; i++) {
//            if (listOfFiles[i].isFile()) {
//                File originalImage = listOfFiles[i];
//                System.out.println("源文件大小" + originalImage.length());
//                String fileName = originalImage.getName();
//                File resizedImg1 = new File("C:\\Users\\Win\\Desktop\\productImages\\" + fileName);
//                resize(originalImage, resizedImg1, 64, 0.9f);
//                System.out.println(fileName + "0.8转换后文件大小" + resizedImg1.length());
//            } else if (listOfFiles[i].isDirectory()) {
//                System.out.println("Directory " + listOfFiles[i].getName());
//            }
//        }
//
//
//    }

    /**
     * 缩放图片(压缩图片质量，改变图片尺寸)
     * 若原图宽度小于新宽度，则宽度不变！
     * @param newWidth 新的宽度
     * @param quality  图片质量参数 0.7f 相当于70%质量
     */
    public static void resize(File originalFile, File resizedFile, int newWidth, float quality) throws IOException {
        if (quality > 1) {
        }
        ImageIcon ii = new ImageIcon(originalFile.getCanonicalPath());
        Image i = ii.getImage();
        Image resizedImage = null;
        int iWidth = i.getWidth(null);
        int iHeight = i.getHeight(null);
        if (iWidth < newWidth) {
            newWidth = iWidth;
        }
        if (iWidth > iHeight) {
            resizedImage = i.getScaledInstance(newWidth, (newWidth * iHeight) / iWidth, Image.SCALE_SMOOTH);
        } else {
            resizedImage = i.getScaledInstance((newWidth * iWidth) / iHeight, newWidth, Image.SCALE_SMOOTH);
        }
        Image temp = new ImageIcon(resizedImage).getImage();
        BufferedImage bufferedImage = new BufferedImage(temp.getWidth(null), temp.getHeight(null), BufferedImage.TYPE_INT_RGB);
        Graphics g = bufferedImage.createGraphics();
        g.setColor(Color.white);
        g.fillRect(0, 0, temp.getWidth(null), temp.getHeight(null));
        g.drawImage(temp, 0, 0, null);
        g.dispose();
        float softenFactor = 0.05f;
        float[] softenArray = {0, softenFactor, 0, softenFactor, 1 - (softenFactor * 4), softenFactor, 0, softenFactor, 0};
        Kernel kernel = new Kernel(3, 3, softenArray);
        ConvolveOp cOp = new ConvolveOp(kernel, ConvolveOp.EDGE_NO_OP, null);
        bufferedImage = cOp.filter(bufferedImage, null);
        FileOutputStream out = new FileOutputStream(resizedFile);
        JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(out);
        JPEGEncodeParam param = encoder.getDefaultJPEGEncodeParam(bufferedImage);
        param.setQuality(quality, true);
        encoder.setJPEGEncodeParam(param);
        encoder.encode(bufferedImage);
    }
}