mash.cpp

/*****************************************************************************************************
*  Author: Written by Ryan "Riukuzaki" Cornwall.  Contains parts of code from Cryect's Tools.        *
*  Title: Riu's Zone Masher.  "Smash ADT's together willy-nilly to create new zones!"                *
*                                                                                                    *
*  Features:                                                                                         *
*      Allows for easy renaming of ADT files                                                         *
*      Accepts any number of ADT files                                                               *
*      Attempts to fit all ADTs into a single map                                                    *
*      Automatically adjusts zones to fit new coordinates.                                           *
*      Automatically creates a WDT file that works with the new ADTs.                                *
*****************************************************************************************************/

#include <cstdio>
#include <string>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <fstream>
#include <windows.h> //Used in creating directory
#include <sys/stat.h>
#include <sys/types.h>

#include "thread_pool.hpp"
#include "list.h" //Makes my life so much easier
#include "zoneGroup.h" //Custom class that contains information about a group of ADTs
#include "offsetFix.h" //Used in fixing offsets after movement
#include "createWDT.h" //Used in creating a WDT file for the new map
//#include "createWDL.h" //NOT YET IMPLEMENTED

int mash(list<ZoneGroup> zoneGroupList, std::string outputName, bool offsetFixEnabled, bool wdtCreateEnabled, int zOffset, int wdtXOffset, int wdtYOffset, int wdtZOffset)
{
    std::cout << "Combining ADT groups\n";
    for(int i=1; i < zoneGroupList.getSize(); i++) //Mash stuff!
        zoneGroupList[0].mash(zoneGroupList[i]);

    // Once files are written, we will correct offsets that might have appeared
    // This part of the code was modeled from Cryect's OffsetFix
    if(offsetFixEnabled){
        std::cout << "Correcting offsets\n";

	    synced_stream sync_out;
	    thread_pool pool;


        for(int i=0; i < zoneGroupList[0].getSize(); i++) //Fix each file
        {
            pool.push_task([&,i] {
                std::fstream zoneFile;
                zoneFile.open(zoneGroupList[0][i].c_str(), std::ios::in | std::ios::out | std::ios::binary);
                sync_out.println("Correcting ", zoneGroupList[0][i]);
                OffsetFixData offData;
                offData.offset.x    = zoneGroupList[0].getAdtX(i);
                offData.offset.y    = zoneGroupList[0].getAdtY(i);
                offData.offset.zOff = 1.0 * zOffset;
                offData.offset.wdtXOff = 1.0 * wdtXOffset;
                offData.offset.wdtYOff = 1.0 * wdtYOffset;
                offData.offset.wdtZOff = 1.0 * wdtZOffset;

                //Fix the offsets
                findMCNKs      (zoneFile, offData);
                findMDDFandMODF(zoneFile, offData);
                fixMCNKs       (zoneFile, offData);
                fixDoodads     (zoneFile, offData);
                fixWMOs        (zoneFile, offData);

                //createWDL      (zoneFile); //NOT YET IMPLEMENTED
                zoneFile.close();
            });
        }
        pool.wait_for_tasks();
    }
    //Create a WDT file
    /* This part of the code was checked with Cryect's CreateWDT.  I made several improvements compared to the original. */
    if(wdtCreateEnabled){
        std::cout << "Creating WDT file\n";
        createWDT(zoneGroupList[0]);
    }
    std::cout << "Finished!\n";
    return 0;
}