CopySelectionCommand.java

package evopaint.commands;

import java.awt.Graphics2D;
import java.awt.Point;
import java.awt.Rectangle;
import java.awt.image.BufferedImage;

import evopaint.Configuration;
import evopaint.Selection;
import evopaint.gui.util.IOverlay;
import evopaint.pixel.Pixel;
import evopaint.pixel.rulebased.RuleBasedPixel;

public class CopySelectionCommand extends AbstractCommand {

	private final Configuration config;
	private SelectionCopyOverlay overlay;
	private Point location;
	private boolean dragging = false;
	private Rectangle rect;
	
	public CopySelectionCommand(Configuration config) {
		this.config = config;
	}
	
	public void setLocation(Point p) {
		location = p;
	}
	
	@Override
	public void execute() {
		if (config.mainFrame.getShowcase().getActiveSelection() == null) return;
		if (dragging == false) {
			Selection activeSelection = config.mainFrame.getShowcase().getActiveSelection();
			rect = activeSelection.getRectangle();
			overlay = new SelectionCopyOverlay(rect.width, rect.height);
			for(int x = 0; x < rect.width; x++) {
				for (int y = 0; y < rect.height; y++) {
					Pixel pixel = config.world.get(rect.x + x, rect.y + y);
					if (pixel == null) continue;
					overlay.setRGB(x, y, pixel.getPixelColor().getInteger());
				}
			}
			
			config.mainFrame.getShowcase().subscribe(overlay);
			dragging = true;
		}
		else {
			config.mainFrame.getShowcase().unsubscribe(overlay);
			
			for(int x = 0; x < overlay.getWidth(); x++) {
				for (int y = 0; y < overlay.getHeight(); y++) {
					RuleBasedPixel pixel = config.world.get(location.x + x, location.y + y);
					if (pixel != null)
						pixel.getPixelColor().setInteger(overlay.getRGB(x, y));
				}
			}
			dragging = false;
		}
	}
	
		
	private class SelectionCopyOverlay extends BufferedImage implements IOverlay {

		public SelectionCopyOverlay(int width, int height) {
			super(width, height, BufferedImage.TYPE_INT_RGB);
		}

		@Override
		public void paint(Graphics2D g2) {
			g2.drawImage(this, location.x, location.y, config.mainFrame.getShowcase());
		}
	}
}