0ad/source/tools/rmgen/output.cpp

#include "stdafx.h"
#include "rmgen.h"
#include "output.h"
#include "map.h"
#include "object.h"
#include "pmp_file.h"
#include <iomanip>

using namespace std;

// Sea level elevation
const float SEA_LEVEL = 20.0f;

void OutputObject(Map* m, Object* e, ostringstream& xml) {
	float height = m->getExactHeight(e->x, e->y) + SEA_LEVEL;

	if(e->isEntity()) {
		xml << "\
		<Entity>\n\
			<Template>" << e->name << "</Template>\n\
			<Player>" << e->player << "</Player>\n\
			<Position x=\"" << 4*e->x << "\" y=\"" << height << "\" z=\"" << 4*e->y << "\" />\n\
			<Orientation angle=\"" << e->orientation << "\" />\n\
		</Entity>\n";
	}
	else {
		xml << "\
		<Nonentity>\n\
			<Actor>" << e->name << "</Actor>\n\
			<Position x=\"" << 4*e->x << "\" y=\"" << height << "\" z=\"" << 4*e->y << "\" />\n\
			<Orientation angle=\"" << e->orientation << "\" />\n\
		</Nonentity>\n";
	}
}

void OutputObjects(ostringstream& xml, Map* m, bool entities) {
	for(int i=0; i<m->objects.size(); i++) {
		if(m->objects[i]->isEntity() == entities) {
			OutputObject(m, m->objects[i], xml);
		}
	}

	for(int x=0; x<m->size; x++) {
		for(int y=0; y<m->size; y++) {
			vector<Object*>& vec = m->terrainObjects[x][y];
			for(int i=0; i<vec.size(); i++) {
				if(vec[i]->isEntity() == entities) {
					OutputObject(m, vec[i], xml);
				}
			}
		}
	}
}

void OutputXml(Map* m, FILE* f) {
	ostringstream xml;
	xml << "\
<?xml version=\"1.0\" encoding=\"utf-8\" standalone=\"no\"?>\n\
<Scenario>\n\
	<Environment>\n\
		<SunColour r=\"1\" g=\"1\" b=\"1\" />\n\
		<SunElevation angle=\"0.72\" />\n\
		<SunRotation angle=\"0.5\" />\n\
		<TerrainAmbientColour r=\"0.5\" g=\"0.5\" b=\"0.5\" />\n\
		<UnitsAmbientColour r=\"0.52\" g=\"0.52\" b=\"0.52\" />\n\
		<Water><WaterBody><Height>" << SEA_LEVEL-0.1f << "</Height></WaterBody></Water>\n\
	</Environment>\n\
	<Entities>\n";
	OutputObjects(xml, m, true);		// print entities
	xml << "\
	</Entities>\n\
	<Nonentities>\n";
	OutputObjects(xml, m, false);		// print nonentities
	xml << "\
	</Nonentities>\n\
</Scenario>\n";

	fprintf(f, "%s", xml.str().c_str());
}


void OutputPmp(Map* m, FILE* f) {
/*
struct String {
	u32 length;
	char data[length]; // not NUL-terminated
};
struct PMP {
	char header[4]; // == "PSMP"
	u32 version; // == 4
	u32 data_size; // == filesize-12
	

	u32 map_size; // number of patches (16x16 tiles) per side

	u16 heightmap[(mapsize*16 + 1)^2]; // (squared, not xor) - vertex heights

	u32 num_texture_textures;
	String texture_textures[num_texture_textures]; // filenames (no path), e.g. "cliff1.dds"

	Tile tiles[(mapsize*16)^2];
};*/
	int size = m->size;
	int numTextures = m->idToName.size();

	// header
	fwrite("PSMP", sizeof(char), 4, f);

	// file format version
	u32 version = 4;
	fwrite(&version, sizeof(u32), 1, f);

	// data size (write 0 for now, calculate it at the end)
	int temp = 0;
	fwrite(&temp, sizeof(u32), 1, f);

	// map size in patches
	u32 sizeInPatches = size/16;
	fwrite(&sizeInPatches, sizeof(u32), 1, f);

	// heightmap
	u16* heightmap = new u16[(size+1)*(size+1)];
	for(int x=0; x<size+1; x++) {
		for(int y=0; y<size+1; y++) {
			int intHeight = (int) ((m->height[x][y]+SEA_LEVEL) * 256.0f / 0.35f);
			if(intHeight > 0xFFFF) {
				intHeight = 0xFFFF;
			}
			else if(intHeight < 0) {
				intHeight = 0;
			}
			heightmap[y*(size+1)+x] = intHeight;
		}
	}
	fwrite(heightmap, sizeof(u16), (size+1)*(size+1), f);

	// num texture textures
	fwrite(&numTextures, sizeof(u32), 1, f);

	// texture names
	for(int i=0; i<numTextures; i++) {
		string fname = m->idToName[i] + ".dds";
		int len = fname.length();
		fwrite(&len, sizeof(u32), 1, f);
		fwrite(fname.c_str(), sizeof(char), fname.length(), f);
	}

	
	// texture; note that this is an array of 16x16 patches for some reason
	Tile* tiles = new Tile[size*size];
	for(int x=0; x<size; x++) {
		for(int y=0; y<size; y++) {
			int patchX = x/16, patchY = y/16;
			int offX = x%16, offY = y%16;
			Tile& t = tiles[ (patchY*size/16 + patchX)*16*16 + (offY*16 + offX) ];
			t.texture1 = m->texture[x][y];
			t.texture2 = 0xFFFF;
			t.priority = 0;
		}
	}
	fwrite(tiles, sizeof(Tile), size*size, f);



	// data size (file size - 12)
	fseek(f, 0, SEEK_END);
	int fsize = ftell(f);
	u32 dataSize = fsize-12;
	fseek(f, 8, SEEK_SET);
	fwrite(&dataSize, sizeof(u32), 1, f);
}

void OutputMap(Map* m, const string& outputName) {
	string xmlName = outputName + ".xml";
	FILE* xmlFile = fopen(xmlName.c_str(), "w");
	if(!xmlFile) {
		cerr << "Cannot open " << xmlName << endl;
		Shutdown(1);
	}
	OutputXml(m, xmlFile);
	fclose(xmlFile);

	string pmpName = outputName + ".pmp";
	FILE* pmpFile = fopen(pmpName.c_str(), "wb");
	if(!pmpFile) {
		cerr << "Cannot open " << pmpName << endl;
		Shutdown(1);
	}
	OutputPmp(m, pmpFile);
	fclose(pmpFile);

	cout << "Map outputted to " << outputName << ".*" << endl;
}