/*
* ------------------------------------------------------------------------
*
* Copyright (C) 2003 - 2013
* University of Konstanz, Germany and
* KNIME GmbH, Konstanz, Germany
* Website: http://www.knime.org; Email: contact@knime.org
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License, Version 3, as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, see <http://www.gnu.org/licenses>.
*
* Additional permission under GNU GPL version 3 section 7:
*
* KNIME interoperates with ECLIPSE solely via ECLIPSE's plug-in APIs.
* Hence, KNIME and ECLIPSE are both independent programs and are not
* derived from each other. Should, however, the interpretation of the
* GNU GPL Version 3 ("License") under any applicable laws result in
* KNIME and ECLIPSE being a combined program, KNIME GMBH herewith grants
* you the additional permission to use and propagate KNIME together with
* ECLIPSE with only the license terms in place for ECLIPSE applying to
* ECLIPSE and the GNU GPL Version 3 applying for KNIME, provided the
* license terms of ECLIPSE themselves allow for the respective use and
* propagation of ECLIPSE together with KNIME.
*
* Additional permission relating to nodes for KNIME that extend the Node
* Extension (and in particular that are based on subclasses of NodeModel,
* NodeDialog, and NodeView) and that only interoperate with KNIME through
* standard APIs ("Nodes"):
* Nodes are deemed to be separate and independent programs and to not be
* covered works. Notwithstanding anything to the contrary in the
* License, the License does not apply to Nodes, you are not required to
* license Nodes under the License, and you are granted a license to
* prepare and propagate Nodes, in each case even if such Nodes are
* propagated with or for interoperation with KNIME. The owner of a Node
* may freely choose the license terms applicable to such Node, including
* when such Node is propagated with or for interoperation with KNIME.
* --------------------------------------------------------------------- *
*
*/
package org.knime.knip.base.data;
import java.awt.image.BufferedImage;
import java.awt.image.DataBuffer;
import java.awt.image.DataBufferByte;
import java.awt.image.DataBufferInt;
import java.awt.image.DataBufferShort;
import java.awt.image.DataBufferUShort;
import java.awt.image.Raster;
import java.awt.image.SampleModel;
import java.awt.image.WritableRaster;
import net.imglib2.display.screenimage.awt.ARGBScreenImage;
import net.imglib2.display.screenimage.awt.ByteAWTScreenImage;
import net.imglib2.display.screenimage.awt.ShortAWTScreenImage;
import net.imglib2.img.basictypeaccess.array.ByteArray;
import net.imglib2.img.basictypeaccess.array.ShortArray;
import net.imglib2.type.numeric.integer.ByteType;
import net.imglib2.type.numeric.integer.ShortType;
import org.knime.knip.core.io.externalization.BufferedDataInputStream;
import org.knime.knip.core.io.externalization.BufferedDataOutputStream;
import org.knime.knip.core.io.externalization.Externalizer;
import org.knime.knip.core.io.externalization.ExternalizerManager;
/**
* Externalizer for the java buffered image object.
*
* @author <a href="mailto:dietzc85@googlemail.com">Christian Dietz</a>
* @author <a href="mailto:horn_martin@gmx.de">Martin Horn</a>
* @author <a href="mailto:michael.zinsmaier@googlemail.com">Michael Zinsmaier</a>
*/
public class BufferedImageExt0 implements Externalizer<BufferedImage> {
/**
* {@inheritDoc}
*/
@Override
public String getId() {
return BufferedImageExt0.class.getSimpleName();
}
/**
* {@inheritDoc}
*/
@Override
public int getPriority() {
return 0;
}
/**
* {@inheritDoc}
*/
@Override
public Class<? extends BufferedImage> getType() {
return BufferedImage.class;
}
/**
* {@inheritDoc}
*/
@Override
public BufferedImage read(final BufferedDataInputStream in) throws Exception {
final int width = in.readInt();
final int height = in.readInt();
final Class<?> clazz = ExternalizerManager.read(in);
final int length = in.readInt();
if (clazz.isAssignableFrom(DataBufferByte.class)) {
final byte[] data = new byte[length];
in.read(data);
return new ByteAWTScreenImage(new ByteType(), new ByteArray(data), new long[]{width, height}).image();
} else if (clazz.isAssignableFrom(DataBufferUShort.class)) {
final short[] data = new short[length];
in.read(data);
return new ShortAWTScreenImage(new ShortType(), new ShortArray(data), new long[]{width, height}).image();
} else if (clazz.isAssignableFrom(DataBufferInt.class)) {
final int[] data = new int[length];
in.read(data);
final SampleModel sampleModel =
ARGBScreenImage.ARGB_COLOR_MODEL.createCompatibleWritableRaster(1, 1).getSampleModel()
.createCompatibleSampleModel(width, height);
final DataBuffer dataBuffer = new DataBufferInt(data, width * height, 0);
final WritableRaster rgbRaster = Raster.createWritableRaster(sampleModel, dataBuffer, null);
return new BufferedImage(ARGBScreenImage.ARGB_COLOR_MODEL, rgbRaster, false, null);
} else {
throw new IllegalArgumentException("Cannot restore BufferedImage from " + clazz.getSimpleName());
}
}
/**
* {@inheritDoc}
*/
@Override
public void write(final BufferedDataOutputStream out, final BufferedImage obj) throws Exception {
out.writeInt(obj.getWidth());
out.writeInt(obj.getHeight());
final DataBuffer buffer = obj.getRaster().getDataBuffer();
ExternalizerManager.write(out, buffer.getClass());
if (buffer instanceof DataBufferByte) {
final byte[] data = ((DataBufferByte)buffer).getData();
out.writeInt(data.length);
out.write(data);
} else if (buffer instanceof DataBufferShort) {
final short[] data = ((DataBufferShort)buffer).getData();
out.writeInt(data.length);
out.write(data);
} else if (buffer instanceof DataBufferInt) {
final int[] data = ((DataBufferInt)buffer).getData();
out.writeInt(data.length);
out.write(data);
} else {
throw new IllegalArgumentException("Cant externalize BufferedImage with "
+ buffer.getClass().getSimpleName());
}
}
}