将short []转换为灰度图像

Question

我正在使用aparapi编写Buddhabrot分形生成器。我将其中的OpenCL部分工作了，得到了一个表示每个像素的单维数组。我将最终图像的尺寸作为最终整数，并编写了代码以获取该数组中任意点的索引。我想将其另存为图像，并且尝试将BufferedImage与TYPE_USHORT_GRAY一起使用。这是我到目前为止的内容：

    BufferedImage image=new BufferedImage(VERTICAL_PIXELS, HORIZONTAL_PIXELS, BufferedImage.TYPE_USHORT_GRAY);
    for(int i=0; i<VERTICAL_PIXELS; i++)
        for(int k=0; k<HORIZONTAL_PIXELS; k++)
            image.setRGB(k, i, normalized[getArrayIndex(k,i,HORIZONTAL_PIXELS)]);

问题是，我不知道将RGB设置为什么。我需要做什么？

Answer 1

这里的问题是setRGB()想要0xRRGGBB颜色值。BufferedImage喜欢假装图像是RGB，无论数据是以什么形式存储的。实际上，您可以使用DataBufferShort（getTile(0, 0).getDataBuffer()）来获取内部信息，但是弄清楚它的布局方式可能很棘手。

如果您已经有像素short[]，则一个简单的解决方案可能是将像素复制到int[]而不是将其塞入MemoryImageSource：

int[] buffer = /* pixels */;

ColorModel model = new ComponentColorModel(
   ColorSpace.getInstance(ColorSpace.CS_GRAY), new int[] { 16 }, 
   false, true, Transparency.OPAQUE, DataBuffer.TYPE_USHORT);

Image image = Toolkit.getDefaultToolkit().createImage(
   new MemoryImageSource(VERTICAL_PIXELS, HORIZONTAL_PIXELS, 
                         model, buffer, 0, VERTICAL_PIXELS));

这种方法的优点是您可以控制基础像素阵列。你可以更改该数组并调用newPixels()你的MemoryImageSource，它会实时更新。除了灰度以外，它还使您能够完全定义自己的调色板：

int[] cmap = new int[65536];
for(int i = 0; i < 65536; ++i) {

    cmap[i] = (((i % 10000) * 256 / 10000) << 16) 
            | (((i % 20000) * 256 / 20000) << 8)
            | (((i % 40000) * 256 / 40000) << 0);
}
ColorModel model = new IndexColorModel(16, 65536, cmap, 0, false, -1, DataBuffer.TYPE_USHORT);

如果您只想在屏幕上显示图像，则此方法效果很好：

JFrame frame = new JFrame();
frame.getContentPane().add(new JLabel(new ImageIcon(image)));
frame.pack();
frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
frame.setVisible(true);

但是，如果您想将其写到文件中并保留每像素一个短的格式（例如，加载到Matlab中），那么您就不走运了。最好的办法是将其绘制为，BufferedImage然后用保存，ImageIO将另存为RGB。

如果最后肯定需要a BufferedImage，则另一种方法是自己应用调色板，计算RGB值，然后将它们复制到图像中：

short[] data = /* your data */;
int[] cmap = /* as above */;
int[] rgb = new int[data.length];

for(int i = i; i < rgb.length; ++i) {
   rgb[i] = cmap[data[i]];
}

BufferedImage image = new BufferedImage(
   VERTICAL_PIXELS, HORIZONTAL_PIXELS, 
   BufferedImage.TYPE_INT_RGB);

image.setRGB(0, 0, VERTICAL_PIXELS, HORIZONTAL_PIXELS,
   pixels, 0, VERTICAL_PIXELS);