Загрузка скриншота в Mat - PullRequest
Новая
       10

Загрузка скриншота в Mat

0 голосов
/ 10 ноября 2018

Я использую робота, чтобы сделать снимок экрана. Чтобы избежать ненужного ввода-вывода записи BufferedImage на диск и последующей загрузки его обратно в Mat, я пытаюсь загрузить BufferedImage непосредственно в Mat со следующим кодом. 

public static Mat screenShot() throws AWTException, IOException {

    Robot r = new Robot();      
    Rectangle capture =  new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()); 
    BufferedImage Image = r.createScreenCapture(capture); 
    Mat mat = new Mat(Image.getHeight(), Image.getWidth(), CvType.CV_8UC1);     
    byte[] data = ((DataBufferByte) Image.getRaster().getDataBuffer()).getData();
    mat.put(0, 0, data);
    return mat;

}

Я получаю эту ошибку: 

Exception in thread "AWT-EventQueue-0" java.lang.ClassCastException: java.awt.image.DataBufferInt cannot be cast to java.awt.image.DataBufferByte

Как мне обойти эту проблему?

1 Ответ

0 голосов
/ 10 ноября 2018

Я нашел обходной путь в этой ветке , ответ ультрафиолета решает эту проблему.

Рабочий код: 

public static Mat screenShot() throws AWTException, IOException {

    Robot r = new Robot();      
    Rectangle capture =  new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()); 
    BufferedImage Image = r.createScreenCapture(capture);       
    Mat mat = BufferedImage2Mat(Image);

    return mat;

}

public static Mat BufferedImage2Mat(BufferedImage image) throws IOException {

    ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
    ImageIO.write(image, "jpg", byteArrayOutputStream);
    byteArrayOutputStream.flush();
    return Imgcodecs.imdecode(new MatOfByte(byteArrayOutputStream.toByteArray()), Imgcodecs.CV_LOAD_IMAGE_UNCHANGED);

}
...