ImageUtils.java

package Roguelike.Util;

import java.awt.image.BufferedImage;
import java.io.File;

import javax.imageio.ImageIO;

import Roguelike.DungeonGeneration.Symbol;
import Roguelike.Sprite.Sprite;

import com.badlogic.gdx.graphics.Color;
import com.badlogic.gdx.graphics.Pixmap;
import com.badlogic.gdx.graphics.Pixmap.Format;
import com.badlogic.gdx.graphics.Texture;
import com.badlogic.gdx.graphics.g2d.TextureRegion;

public class ImageUtils
{
	public static void writeSymbolGridToFile(Symbol[][] grid, String filename, int tileSize)
	{
		// Get size of image
		int height = grid[0].length * tileSize;
		int width = grid.length * tileSize;

		// Create resources
		Pixmap pixmap = new Pixmap(width, height, Format.RGBA4444);

		// do render
		for (int x = 0; x < grid.length; x++)
		{
			for (int y = 0; y < grid[0].length; y++)
			{
				Symbol s = grid[x][y];

				if (s.character == '#')
				{
					pixmap.setColor(Color.MAROON);
				}
				else if (s.character == '.')
				{
					pixmap.setColor(Color.GREEN);
				}
				else
				{
					pixmap.setColor(Color.BLUE);
				}
				pixmap.fillRectangle(x*tileSize, y*tileSize, tileSize, tileSize);

//				if (s.hasEnvironmentEntity())
//				{
//					Pixmap src = TextureToPixmap(s.getEnvironmentEntity("").sprite.getCurrentTexture());
//					pixmap.drawPixmap(src,
//							0, 0, src.getWidth(), src.getHeight(),
//							x*tileSize, y*tileSize, tileSize, tileSize);
//					src.dispose();
//				}
//
//				if (s.hasGameEntity())
//				{
//					Pixmap src = TextureToPixmap(s.getGameEntity().sprite.getCurrentTexture());
//					pixmap.drawPixmap(src,
//							0, 0, src.getWidth(), src.getHeight(),
//							x*tileSize, y*tileSize, tileSize, tileSize);
//					src.dispose();
//				}
			}
		}

		// Save texture
		try
		{
			BufferedImage bi = pixmapToImage(pixmap);
			File outputfile = new File(filename);
			ImageIO.write(bi, "png", outputfile);

		}
		catch (Exception e)
		{
			e.printStackTrace();
		}

		// dispose of resources
		pixmap.dispose();
	}

	public static BufferedImage pixmapToImage(Pixmap pm)
	{
		BufferedImage image = new BufferedImage(pm.getWidth(), pm.getHeight(), BufferedImage.TYPE_INT_ARGB);
		for (int x = 0; x < pm.getWidth(); x++)
		{
			for (int y = 0; y < pm.getHeight(); y++)
			{
				Color c = new Color();
				Color.rgba8888ToColor(c, pm.getPixel(x, y));

				java.awt.Color cc = new java.awt.Color(c.r, c.g, c.b, c.a);

				image.setRGB(x, y, cc.getRGB());
			}
		}

		return image;
	}

	public static Pixmap textureToPixmap(Texture texture)
	{
		if (!texture.getTextureData().isPrepared())
		{
			texture.getTextureData().prepare();
		}

		Pixmap pixmap = texture.getTextureData().consumePixmap();

		return pixmap;
	}

	public static Pixmap multiplyPixmap( Pixmap image, Pixmap mask )
	{
		Pixmap pixmap = new Pixmap( image.getWidth(), image.getHeight(), Format.RGBA8888 );

		pixmap.setColor( 1, 1, 1, 0 );
		pixmap.fill();

		Color cb = new Color();
		Color ca = new Color();

		float xRatio = (float)mask.getWidth() / (float)image.getWidth();
		float yRatio = (float)mask.getHeight() / (float)image.getHeight();

		for (int x = 0; x < image.getWidth(); x++)
		{
			for (int y = 0; y < image.getHeight(); y++)
			{
				Color.rgba8888ToColor(ca, image.getPixel(x, y));

				int maskX = (int)((float)x * xRatio);
				int maskY = (int)((float)y * yRatio);

				Color.rgba8888ToColor(cb, mask.getPixel(maskX, maskY));

				ca.mul( cb );

				pixmap.drawPixel(x, y, Color.rgba8888(ca));
			}
		}

		return pixmap;
	}

	public static Pixmap addPixmap( Pixmap image, Pixmap mask )
	{
		Pixmap pixmap = new Pixmap( image.getWidth(), image.getHeight(), Format.RGBA8888 );

		pixmap.setColor( 1, 1, 1, 0 );
		pixmap.fill();

		Color cb = new Color();
		Color ca = new Color();

		float xRatio = (float)mask.getWidth() / (float)image.getWidth();
		float yRatio = (float)mask.getHeight() / (float)image.getHeight();

		for (int x = 0; x < image.getWidth(); x++)
		{
			for (int y = 0; y < image.getHeight(); y++)
			{
				Color.rgba8888ToColor(ca, image.getPixel(x, y));

				int maskX = (int)((float)x * xRatio);
				int maskY = (int)((float)y * yRatio);

				Color.rgba8888ToColor(cb, mask.getPixel(maskX, maskY));

				ca.add( cb );

				pixmap.drawPixel(x, y, Color.rgba8888(ca));
			}
		}

		return pixmap;
	}
}