Преобразования BufferedImage в ByteBuffer

Question

Я пытаюсь преобразовать Буферизированное изображение в ByteBuffer, но я получаю это исключение

java.awt.image.DataBufferInt cannot be cast to java.awt.image.DataBufferByte 

может кто-то пожалуйста, помогите мне и предложить хороший метод преобразования.

Источник:

public static ByteBuffer convertImageData(BufferedImage bi) 
{ 
    byte[] pixelData = ((DataBufferByte) bi.getRaster().getDataBuffer()).getData(); 
    //  return ByteBuffer.wrap(pixelData); 
    ByteBuffer buf = ByteBuffer.allocateDirect(pixelData.length); 
    buf.order(ByteOrder.nativeOrder()); 
    buf.put(pixelData); 
    buf.flip(); 
    return buf; 
} 

это мой объект

ByteBuffer buf = convertImageData(image); 

Answer 1

Если вы хотите, чтобы кодировать данные изображения, вы можете использовать ImageIO здесь. Что-то вроде этого:

public static ByteBuffer convertImageData(BufferedImage bi) { 
    ByteArrayOutputStream out = new ByteArrayOutputStream(); 
    try { 
     ImageIO.write(bi, "png", out); 
     return ByteBuffer.wrap(out.toByteArray()); 
    } catch (IOException ex) { 
     //TODO 
    } 
    return null; 
} 

Here is the list поддерживаемых форматов.

Answer 2

Вы не можете просто бросить произвольное DataBuffer к DataBufferByte, вы должны убедиться, что это на самом деле правильный тип:

ByteBuffer byteBuffer; 
DataBuffer dataBuffer = bi.getRaster().getDataBuffer(); 

if (dataBuffer instanceof DataBufferByte) { 
    byte[] pixelData = ((DataBufferByte) dataBuffer).getData(); 
    byteBuffer = ByteBuffer.wrap(pixelData); 
} 
else if (dataBuffer instanceof DataBufferUShort) { 
    short[] pixelData = ((DataBufferUShort) dataBuffer).getData(); 
    byteBuffer = ByteBuffer.allocate(pixelData.length * 2); 
    byteBuffer.asShortBuffer().put(ShortBuffer.wrap(pixelData)); 
} 
else if (dataBuffer instanceof DataBufferShort) { 
    short[] pixelData = ((DataBufferShort) dataBuffer).getData(); 
    byteBuffer = ByteBuffer.allocate(pixelData.length * 2); 
    byteBuffer.asShortBuffer().put(ShortBuffer.wrap(pixelData)); 
} 
else if (dataBuffer instanceof DataBufferInt) { 
    int[] pixelData = ((DataBufferInt) dataBuffer).getData(); 
    byteBuffer = ByteBuffer.allocate(pixelData.length * 4); 
    byteBuffer.asIntBuffer().put(IntBuffer.wrap(pixelData)); 
} 
else { 
    throw new IllegalArgumentException("Not implemented for data buffer type: " + dataBuffer.getClass()); 
} 

Если ваш BufferedImage является одним из стандартных типов (кроме BufferedImage.TYPE_*TYPE_CUSTOM) вышеуказанное должно работать.

Обратите внимание, что специальные подклассы DatBuffer могут существовать и могут хранить пиксели в нескольких банках с различным порядком байтов, может быть чередованием каналов (а не стандартным чередованием пикселей) и т. Д. Таким образом, приведенный выше код еще не является полностью общим.

Если вы передать эти ByteBuffer S в машинный код, используя allocateDirect(..) и копирование пикселов над может быть быстрее, в противном случае я думаю, с помощью wrap(..) будет делать как для простого кода и быть более эффективными.