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 int width;
	private int height;
	private int bpp;
	private 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);
	}
}