import java.awt.BorderLayout;
import java.awt.Graphics;
import java.awt.GridLayout;
import java.awt.image.BufferedImage;
import java.awt.image.DataBuffer;
import java.awt.image.DataBufferUShort;
import java.awt.image.RenderedImage;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JSlider;
import javax.swing.SwingUtilities;
import javax.swing.event.ChangeEvent;
import javax.swing.event.ChangeListener;
class ImagePanel extends JPanel
{
private BufferedImage image;
public ImagePanel()
{
}
public void setImage(BufferedImage image)
{
this.image = image;
repaint();
}
@Override
protected void paintComponent(Graphics g)
{
super.paintComponent(g);
if (image != null)
{
long before = System.nanoTime();
g.drawImage(image, 0, 0, null);
long after = System.nanoTime();
System.out.println("Painting "+(after-before)/1e6+" ms");
}
}
}
public class ShortToByteImageTest extends JPanel
{
public static void main(String[] args)
{
SwingUtilities.invokeLater(new Runnable()
{
@Override
public void run()
{
createAndShowGUI();
}
});
}
private ImagePanel imagePanel;
private RenderedImage shortImage;
private short shortImageData[];
public ShortToByteImageTest()
{
super(new BorderLayout());
shortImage = obtainImageWithShortBufferFromSomewhere();
shortImageData = getShortArray(shortImage);
imagePanel = new ImagePanel();
add(imagePanel, BorderLayout.CENTER);
JPanel controlPanel = new JPanel(new GridLayout(2,0));
JPanel pMin = new JPanel(new BorderLayout());
final JSlider minSlider = new JSlider(
-Short.MAX_VALUE, Short.MAX_VALUE, -Short.MAX_VALUE);
pMin.add(new JLabel("min"), BorderLayout.WEST);
pMin.add(minSlider, BorderLayout.CENTER);
controlPanel.add(pMin);
JPanel pMax = new JPanel(new BorderLayout());
final JSlider maxSlider = new JSlider(
-Short.MAX_VALUE, Short.MAX_VALUE, Short.MAX_VALUE);
pMax.add(new JLabel("max"), BorderLayout.WEST);
pMax.add(maxSlider, BorderLayout.CENTER);
controlPanel.add(pMax);
add(controlPanel, BorderLayout.SOUTH);
ChangeListener changeListener = new ChangeListener()
{
@Override
public void stateChanged(ChangeEvent e)
{
int min = minSlider.getValue();
int max = maxSlider.getValue();
updateImage(shortImage.getWidth(), shortImage.getHeight(), min, max);
}
};
minSlider.addChangeListener(changeListener);
maxSlider.addChangeListener(changeListener);
}
private void updateImage(int w, int h, int min, int max)
{
long before = System.nanoTime();
BufferedImage iImage = null;
boolean combined = false;
combined = true;
if (combined)
{
iImage = createImageFromShortArrayNormalized(w, h, shortImageData, min, max);
}
else
{
byte bData[] = convertShortToByteArray(shortImageData, min, max);
iImage = createImageFromByteArray(w, h, bData);
}
long after = System.nanoTime();
System.out.println("Conversion "+(after-before)/1e6+" ms");
imagePanel.setImage(iImage);
}
private static RenderedImage obtainImageWithShortBufferFromSomewhere()
{
// TODO: Replace this with the code for loading a TIFF from JAI
return createImageWithShortBuffer(1000, 1000);
}
private static RenderedImage createImageWithShortBuffer(int w, int h)
{
BufferedImage image = new BufferedImage(w,h,BufferedImage.TYPE_USHORT_GRAY);
DataBuffer dataBuffer = image.getRaster().getDataBuffer();
DataBufferUShort databufferUShort = (DataBufferUShort)dataBuffer;
short data[] = databufferUShort.getData();
for (int i=0; i<data.length; i++)
{
data[i] = (short)i;
}
return image;
}
private static short[] getShortArray(RenderedImage image)
{
DataBuffer dataBuffer = image.getData().getDataBuffer();
DataBufferUShort databufferUShort = (DataBufferUShort)dataBuffer;
return databufferUShort.getData();
}
private static byte[] convertShortToByteArray(short input[], int min, int max)
{
byte output[] = new byte[input.length];
float normalization = 1.0f / (max - min);
for (int i=0; i<input.length; i++)
{
float f = (input[i]-min)*normalization;
output[i] = (byte)(f * 256);
}
return output;
}
private static BufferedImage createImageFromByteArray(int w, int h, byte input[])
{
int output[] = new int[input.length];
for (int i = 0; i < input.length; i++)
{
byte v = input[i];
int a = 0xFF000000;
int r = (v & 0xFF) << 16;
int g = (v & 0xFF) << 8;
int b = (v & 0xFF) << 0;
output[i] = a | r | g | b;
}
BufferedImage result = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
result.setRGB(0,0, w, h, output, 0, w);
return result;
}
private static BufferedImage createImageFromShortArrayNormalized(int w, int h, short input[], int min, int max)
{
int output[] = new int[input.length];
float normalization = 1.0f / (max - min);
for (int i = 0; i < input.length; i++)
{
float f = (input[i]-min)*normalization;
int v = (int)(f * 256);
int a = 0xFF000000;
int r = (v & 0xFF) << 16;
int g = (v & 0xFF) << 8;
int b = (v & 0xFF) << 0;
output[i] = a | r | g | b;
}
BufferedImage result = new BufferedImage(w, h, BufferedImage.TYPE_INT_ARGB);
result.setRGB(0,0, w, h, output, 0, w);
return result;
}
private static void createAndShowGUI()
{
JFrame f = new JFrame();
f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
f.getContentPane().add(new ShortToByteImageTest());
f.setSize(1000,1000);
f.setVisible(true);
}
}
