ImageUtil.cpp

#include "ImageUtil.h"

namespace sc2util {

	bool isSet(const sc2::GameInfo & info, const sc2::ImageData & grid, const sc2::Point2D & point)
	{
		sc2::Point2DI pointI((int)point.x, (int)point.y);
		return isSet(info, grid, pointI);
	}

	bool isSet(const sc2::GameInfo & info, const sc2::ImageData & grid, const sc2::Point2DI & pointI)
	{
		if (pointI.x < 0 || pointI.x >= grid.width || pointI.y < 0 || pointI.y >= grid.height)
		{
			return false;
		}
		if (grid.bits_per_pixel == 1) {
			div_t idx = div(pointI.x + pointI.y * grid.width, 8);
			bool dst = (grid.data[idx.quot] >> (7 - idx.rem)) & 1;
			return dst;
		}
		else {
			// Image data is stored with an upper left origin.			
			char dst = grid.data[pointI.x + pointI.y * grid.width];
			return dst;
		}
	}

	bool set(sc2::GameInfo & info, sc2::ImageData & grid, const sc2::Point2D & point, bool b)
	{
		sc2::Point2DI pointI((int)point.x, (int)point.y);
		return set(info, grid, pointI, b);
	}

	bool set(sc2::GameInfo & info, sc2::ImageData & grid, const sc2::Point2DI & pointI, bool b)
	{
		if (pointI.x < 0 || pointI.x >= grid.width || pointI.y < 0 || pointI.y >= grid.height)
		{
			return false;
		}
		if (grid.bits_per_pixel == 1) {
			div_t idx = div(pointI.x + pointI.y * grid.width, 8);
			char mask = 1 << (7 - idx.rem);
			if (b) {
				grid.data[idx.quot] |= mask; 
			}
			else {
				grid.data[idx.quot] ^= ~mask;				
			}
			return true;
		}
		else {
			// Image data is stored with an upper left origin.			
			grid.data[pointI.x + pointI.y * grid.width] = b;
			return true;
		}
	}

	bool setBuildingAt(sc2::GameInfo & info, sc2::ImageData & grid, const sc2::Point2D & pos, int footprint, bool b)
	{
		using namespace sc2;
		float fx = pos.x - (float)footprint / 2.0f;
		float fy = pos.y - (float)footprint / 2.0f;

		Point2D point(fx, fy);
		auto inzone = sc2::Point2DI((int)point.x, (int)point.y);
		for (int y = 0; y < footprint; y++) {
			for (int x = 0; x < footprint; x++) {
				set(info, grid, Point2DI(inzone.x + x, inzone.y + y), !b);
			}
		}
		return true;
	}
}