You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.4 KiB
60 lines
1.4 KiB
package mightypork.gamecore.render;
|
|
|
|
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.nio.ByteBuffer;
|
|
|
|
import javax.imageio.ImageIO;
|
|
|
|
|
|
/**
|
|
* Screenshot object, can be used to extract image or write to file.<br>
|
|
* Screenshot, once taken, can be safely processed in separate thread.
|
|
*
|
|
* @author MightyPork
|
|
*/
|
|
public class Screenshot {
|
|
|
|
private final int width;
|
|
private final int height;
|
|
private final int bpp;
|
|
private final ByteBuffer bytes;
|
|
private BufferedImage image;
|
|
|
|
|
|
public Screenshot(int width, int height, int bpp, ByteBuffer buffer) {
|
|
this.width = width;
|
|
this.height = height;
|
|
this.bpp = bpp;
|
|
this.bytes = buffer;
|
|
}
|
|
|
|
|
|
public BufferedImage getImage()
|
|
{
|
|
if (image != null) return image;
|
|
|
|
image = new BufferedImage(this.width, this.height, BufferedImage.TYPE_INT_RGB);
|
|
|
|
// convert to a buffered image
|
|
for (int x = 0; x < this.width; x++) {
|
|
for (int y = 0; y < this.height; y++) {
|
|
final int i = (x + (this.width * y)) * this.bpp;
|
|
final int r = this.bytes.get(i) & 0xFF;
|
|
final int g = this.bytes.get(i + 1) & 0xFF;
|
|
final int b = this.bytes.get(i + 2) & 0xFF;
|
|
image.setRGB(x, this.height - (y + 1), (0xFF << 24) | (r << 16) | (g << 8) | b);
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
|
|
public void save(File file) throws IOException
|
|
{
|
|
ImageIO.write(getImage(), "PNG", file);
|
|
}
|
|
}
|
|
|