cScenarioResource.java

package sporemodder.file.simulator;

import java.io.IOException;
import java.io.PrintWriter;

import emord.filestructures.FileStream;
import emord.filestructures.MemoryStream;
import emord.filestructures.StreamReader;
import emord.filestructures.StreamWriter;
import sporemodder.MainApp;
import sporemodder.file.ResourceKey;
import sporemodder.file.dbpf.DBPFItem;
import sporemodder.file.dbpf.DatabasePackedFile;
import sporemodder.file.prop.PropertyList;

public class cScenarioResource {

	// for testing
	private byte[] propData;
	
	private final PropertyList propertyList = new PropertyList();
	private MemoryStream dataStream;
	private final cScenarioResourceAttributes scenario = new cScenarioResourceAttributes();
	
	public void read(StreamReader stream) throws Exception {
		
		long magic = stream.readLEUInt();
		if (magic != 0xB523D4F9L) {
			throw new IOException("Incorrect magic number.");
		}
		
		int version = stream.readLEInt();
		
		if (version < 3) {
			throw new IOException("Unsupported version.");
		}
		
		propertyList.read(stream);
		
		long endPos = stream.getFilePointer();
		
		stream.seek(8);
		propData = new byte[(int) (endPos - 8)];
		stream.read(propData);
		stream.seek(endPos);
		
		int dataVersion = stream.readLEInt();
		int dbpfSize = stream.readLEInt();
		
		byte[] dbpfData = new byte[dbpfSize];
		stream.read(dbpfData);
		
		MemoryStream dbpfStream = new MemoryStream(dbpfData); 
		DatabasePackedFile dbpf = new DatabasePackedFile();
		dbpf.read(dbpfStream);
		
		dataStream = dbpf.getItem(new ResourceKey(0x1897C18, 0x1897C18, 0x1897C18)).processFile(dbpfStream);
		
		scenario.read(dataStream);
	}
	
	public void write(StreamWriter stream, boolean convertData) throws IOException {
		stream.writeLEUInt(0xB523D4F9L);
		stream.writeLEInt(17);
		
		if (propData != null) {
			stream.write(propData);
		}
		else {
			propertyList.write(stream);
		}
		
		// Data version
		stream.writeLEInt(17);
		
		// Write the DBPF
		MemoryStream dbpfStream = new MemoryStream(); 
		DatabasePackedFile dbpf = new DatabasePackedFile();
		
		DBPFItem item = new DBPFItem();
		item.name.setGroupID(0x1897C18);
		item.name.setInstanceID(0x1897C18);
		item.name.setTypeID(0x1897C18);
		
		if (convertData) {
			dataStream = new MemoryStream();
			//TODO write data
		}
		
		dbpf.writeHeader(dbpfStream);
		dbpf.writeFile(dbpfStream, item, dataStream.toByteArray(), true);
		dbpf.writeIndex(dbpfStream);
		
		// Rewrite the DBPF header
		dbpfStream.seek(0);
		dbpf.writeHeader(dbpfStream);
		
		// DBPF size
		stream.writeLEUInt(dbpfStream.length());
		stream.write(dbpfStream.toByteArray());
	}
	
	
	public String printData() {
		StringBuilder sb = new StringBuilder();
		String tabulation = "\t";
		
		sb.append('{');
		sb.append('\n');
		
		scenario.print(sb, tabulation);
		
		sb.append('}');
		
		return sb.toString();
	}
	
	public String printDataXML() {
		StringBuilder sb = new StringBuilder();
		String tabulation = "\t";
		
		sb.append("<cScenarioResource>\n");
		
		scenario.printXML(sb, tabulation);
		
		sb.append("</cScenarioResource>\n");
		
		return sb.toString();
	}
	
	
	private static void extractTest() throws Exception {
	
		MainApp.testInit();
		
//		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\extracted\\";
//		String path = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\adventureImages_1~\\0x1999746C.0x366A930D";
		
		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\real extracted\\";
		String path = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\adventureImages_1~\\0x199C129D.0x366A930D";
		
//		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\re extracted\\";
//		String path = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\adventureImages_1~\\SRNS_PlanetTemplate_100.0x366A930D";
		
//		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\small_planet_extracted\\";
//		String path = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\adventureImages_1~\\SRNS_PlanetTemplate_100_original.0x366A930D";
		
		try (FileStream stream = new FileStream(path, "r")) {
			
			cScenarioResource resource = new cScenarioResource();
			resource.read(stream);
			
			try (PrintWriter writer = new PrintWriter(outputPath + "planet.prop.prop_t")) {
				writer.write(resource.propertyList.toArgScript());
			}
			
			resource.dataStream.writeToFile(outputPath + "data.0x1897C18");
			
			try (PrintWriter writer = new PrintWriter(outputPath + "data.json")) {
				writer.write(resource.printData());
			}
		}
	}
	
	public static void main(String[] args) throws Exception {
		MainApp.testInit();
//		String path = "E:\\Eric\\SporeModder\\Projects\\Spore_EP1_Data.package.unpacked\\adventureImages_1~\\#0D19B6DA.#366A930D";
//		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\scenarioTest\\#0D19B6DA\\";
		
		String path = "E:\\Eric\\SporeModder\\Projects\\Spore_EP1_Data.package.unpacked\\adventureImages_1~\\#3C546825.#366A930D";
		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\scenarioTest\\temple #3C546825\\";
		
//		String path = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\EditorSaves\\adventureImages_1~\\0x1999746C.0x366A930D";
//		String outputPath = "E:\\Eric\\Eclipse Projects\\SporeModder FX\\Projects\\Terrain Test\\extracted\\";
		
		try (FileStream stream = new FileStream(path, "r")) {
			
			cScenarioResource resource = new cScenarioResource();
			resource.read(stream);
			
			try (PrintWriter writer = new PrintWriter(outputPath + "planet.prop.prop_t")) {
				writer.write(resource.propertyList.toArgScript());
			}
			
			resource.dataStream.writeToFile(outputPath + "data.0x1897C18");
			
			try (PrintWriter writer = new PrintWriter(outputPath + "data.xml")) {
				writer.write(resource.printDataXML());
			}
		}
	}
}