package org.fxd.utils;

import org.opencv.core.Mat;
import org.opencv.imgproc.Imgproc;
import org.opencv.videoio.VideoCapture;

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

/**
 * 將圖片轉成text文件輸出
 * @author Litluecat
 */
public class Image2Text {

  static {
    //opencv_java410.dll的所在地址，自己換成自己dll的地址
    System.load("D:\\Sofeware\\opencv\\build\\java\\x64\\opencv_java410.dll");
  }

  /** 此處設置灰度字符*/
  private static char[] cs = new char[] {'0','1','.',' '};

  public static void main(String[] args) throws IOException {
    //String imageUrl = "E:\\1.png";
    // 設置JTextArea，用于展示字符
    JTextArea textArea = getJFrame("XXTop", "宋體", 0, 2, 900, 900);
    // 將圖片轉成字符
    //textArea.setText(image2Text(imageUrl,1).toString());
    //將視頻轉成字符
    video2TextByOpenCV(textArea, "E:\\1.mp4", 1);
  }

  /**
   * 生成JFrame窗口，并返回JTextArea對象
   * @param frameTitle 窗口標題
   * @param fontName 文本框中字體類型
   * @param fontStyle 文本框中字體格式
   * @param fontSize 文本框中字體大小
   * @param JFWideth 窗口寬度
   * @param JFHeight 窗口高度
   * @return
   */
  public static JTextArea getJFrame(String frameTitle,String fontName, int fontStyle, int fontSize, int JFWideth, int JFHeight){
    JFrame frame = new JFrame();
    //創(chuàng)建一個窗口對象
    JPanel panel = new JPanel();
    JTextArea textArea = new JTextArea();
    // name:字體類型 style:0,表示字形;1,粗體;2,斜體  size:字體大小
    textArea.setFont(new Font(fontName,fontStyle,fontSize));
    panel.setLayout(new GridLayout());
    //當TextArea里的內容過長時生成滾動條
    panel.add(new JScrollPane(textArea));
    frame.add(panel);
    //對窗口設置標題
    frame.setTitle(frameTitle);
    //設置窗口的大小
    frame.setSize(JFWideth,JFHeight);
    //設置窗口界面的關閉按鈕真的生效(也可以直接傳一個3進去，JFrame.EXIT_ON_CLOSE==3，效果一樣)
    frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
    //設置窗口居中顯示
    frame.setLocationRelativeTo(null);
    //設置窗口置頂顯示
    frame.setAlwaysOnTop(true);
    //設置窗口顯示
    frame.setVisible(true);
    return textArea;
  }

  /**
   * OpenCV-4.1.0 從視頻文件中讀取
   * @param textArea JTextArea文本框對象
   * @param targetImgUrl 視頻地址
   * @param img2TextSzie 圖片轉文本縮小比例
   */
  public static void video2TextByOpenCV(JTextArea textArea, String targetImgUrl, int img2TextSzie) {
    VideoCapture capture=new VideoCapture();
    //1 讀取視頻文件的路徑
    capture.open(targetImgUrl);
    if(!capture.isOpened()){
      System.out.println("讀取視頻文件失?。?);
      return;
    }
    Mat video=new Mat();
    while(capture.isOpened()) {
      //2 視頻文件的視頻寫入 Mat video 中
      capture.read(video);
      try{
        textArea.setText(image2TextByOpenCV(video,img2TextSzie).toString());
      }catch (Exception e){
        System.out.println("圖片識別異常");
        break;
      }
    }
  }

  /**
   *基于openCV實現
   * @param image 待處理Mat圖片(視頻中的某一幀)
   * @param size 對結果進行縮小，1為不縮小
   */
  public static StringBuffer image2TextByOpenCV(Mat image,int size) throws Exception{
    StringBuffer text = null;
    Mat img_mat = new Mat();
    //mat表示要要轉換的圖片[Mat類型],img_mat表示轉換后的圖片
    Imgproc.cvtColor(image, img_mat, Imgproc.COLOR_RGB2GRAY);
    int rows = img_mat.rows();
    int cols = img_mat.cols();
    // 圖片轉字符串后的數組
    char[][] css = new char[rows/size + 1][cols/size + 1];
    for(int x=0; x<rows; x+=size){
      for (int y=0; y<cols; y+=size){
        //獲得圖片的灰度值 x,y 表示像素點的位置(姑且理解為像素點吧)
        int gray = (int) img_mat.get(x, y)[0];
        // 得到灰度值
        int index = Math.round((gray * cs.length -1) / 255);
        css[x/size][y/size] = cs[index];
      }
    }
    StringBuffer sb = new StringBuffer();
    // 開始拼接內容
    for (int x = 0; x < css.length; x++) {
      for (int y = 0; y < css[0].length; y++) {
        sb.append(css[x][y]);
      }
      sb.append("\r\n");
    }
    return sb;
  }

  /**
   * @param imageUrl 圖片地址
   * @param size 對結果進行縮小，1為不縮小
   */
  public static StringBuffer image2Text(String imageUrl,int size){
    StringBuffer text = null;
    try {
      BufferedImage image = ImageIO.read(new File(imageUrl));
      int width = image.getWidth();
      int height = image.getHeight();
      // 圖片轉字符串后的數組
      char[][] css = new char[width/size + 1][height/size + 1];
      for (int x = 0; x < width; x+=size) {
        for (int y = 0; y < height; y+=size) {
          int rgb = image.getRGB(x, y);
          Color c = new Color(rgb);
          // 得到灰度值
          int cc = (c.getRed() + c.getGreen() + c.getBlue()) / 3;
          css[x/size][y/size] = cs[(int) ((cc * cs.length - 1) / 255)];
        }
      }
      StringBuffer sb = new StringBuffer();
      // 開始拼接內容
      for (int y = 0; y < css[0].length; y++) {
        for (int x = 0; x < css.length; x++) {
          sb.append(css[x][y]);
        }
        sb.append("\r\n");
      }
      text = sb;
    } catch (IOException e) {
      System.out.println(e);
    }
    return text;
  }

}