/***************************************************************************************\
* Mortal Kombat 3 (MAME) AutoSplitter.                                                  *
* By piratesephiroth.                                                                   *
\***************************************************************************************/

state("mame"){}
state("mame64"){}

startup
{
    refreshRate = 80; // to be on the safe side
    settings.Add("ladderSplit",false,"Split at tower");
    settings.Add("onlyShao",false,"Ignore everything, split only at Shao Kahn's defeat");
    settings.Add("diagCantReset",false,"Don't reset timer after leaving the Diagnostics Menu");
}

init
{
    Action ScanMemoryAndUpdateAddresses = () =>
    {
        int offset = -0xc5c0;
        string sig = "4D 3C 2B 1A 00 00 ?? 00";
        long gstate = 0xc12c;
        long p1State = 0xc130;
        long p2State = 0xc2a6;
        long p1rounds = 0xc14e;
        long p2rounds = 0xc2c4;
        long ladderPos = 0xc450;
        long ladderSel = 0xc44f;
        
        if (game.MainWindowTitle.Contains("[mk3r10]") || game.MainWindowTitle.Contains("[mk3p40]"))
        {
            // magic at 0xe410
            offset -= 0x1e50;
            gstate -= 0xf0;
            p1State -= 0xf0;
            p2State -= 0xf0;
            p1rounds -= 0xf0;
            p2rounds -= 0xf0;
            ladderPos -= 0xf0;
            ladderSel -= 0xf0;

            print("detected r1.x");
        }
        
        print("Scanning memory...");
        foreach (var page in game.MemoryPages(true))
        {
            var scanner = new SignatureScanner(game, page.BaseAddress, (int)page.RegionSize);
            var ptr = scanner.Scan(new SigScanTarget(offset, sig));
            if (ptr != IntPtr.Zero)
            { 
                vars.membase = (long)ptr;
                print("membase found: " + game.ProcessName + ".exe+0x" + vars.membase.ToString("X"));
                vars.scanNeeded = false;
                break;
            }
        }
        
        vars.watchers = new MemoryWatcherList
        {
            new MemoryWatcher<byte>((IntPtr)(vars.membase + gstate)) { Name = "gameState" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + p1State)) { Name = "p1State" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + p2State)) { Name = "p2State" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + p1rounds)) { Name = "p1RoundsWon" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + p2rounds)) { Name = "p2RoundsWon" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + ladderPos)) { Name = "ladderPos" },
            new MemoryWatcher<byte>((IntPtr)(vars.membase + ladderSel)) { Name = "ladderSel" },
        };
     };
    
    vars.ScanMemoryAndUpdateAddresses = ScanMemoryAndUpdateAddresses;
    vars.scanNeeded = true;
    vars.matchWon = false;
    vars.disableReset = false;
    
    game.Refresh();
    if (!game.MainWindowTitle.Contains("Mortal Kombat 3"))
    {
        throw new Exception("Couldn't find process!");
    }
}

update
{
    // rom was unloaded? find the membase again
    game.Refresh();
    if (!game.MainWindowTitle.Contains("Mortal Kombat 3"))
    {
        vars.scanNeeded = true;
        return false;
    }

    if (vars.scanNeeded )
    {
        vars.ScanMemoryAndUpdateAddresses();
    }
    
    vars.watchers.UpdateAll(game);
    
    if (settings["diagCantReset"])
    {
        // if Diag menu, disable timer reset
        if (!vars.disableReset && vars.watchers["gameState"].Current == 8)
        {
            print("DIAG MENU, RESET DISABLED");
            vars.disableReset = true;
        }
    }
}

start
{
    // start timer only at the "Select your Destiny" screen
    if (vars.watchers["gameState"].Current != 13)
    {
        return false;
    }
    //print("TOWER SELECT");
    // start timer after selecting the tower
    if (vars.watchers["ladderSel"].Current == 255)
    {
        print("START TIMER");
        return true;
    }
}

reset
{
    // reset timer if game is booting up
    if (vars.watchers["gameState"].Current == 0 && vars.watchers["gameState"].Old != 0)
    {
        if (vars.disableReset)
        {
            vars.disableReset = false;
            print("RESET ENABLED AGAIN");
            return false;
        }
        
        print("RESET TIMER");
        return true;
    }
}

split
{
    // tower
    if (settings["ladderSplit"] && !settings["onlyShao"])
    {
        if (vars.watchers["ladderPos"].Current > vars.watchers["ladderPos"].Old)
        {
            print("TOWER SPLIT");
            return true;
        }
    }
    
    // won the match
    if (vars.watchers["p1State"].Current == 1 || vars.watchers["p1State"].Old == 1)
    {
        if (vars.watchers["p1RoundsWon"].Current == 2 && (vars.watchers["p1RoundsWon"].Current != vars.watchers["p1RoundsWon"].Old)) {
            vars.matchWon = true;
            print ("P1 WON");
        }
    }
    
    if (vars.watchers["p2State"].Current == 1 || vars.watchers["p2State"].Old == 1)
    {
        if (vars.watchers["p2RoundsWon"].Current == 2 && (vars.watchers["p2RoundsWon"].Current != vars.watchers["p2RoundsWon"].Old))
        {
            vars.matchWon = true;
            print ("P2 WON");
        }
    }
    
    // find out if it was a regular character or Shao Khan
    if (vars.matchWon)
    {
        if (vars.watchers["gameState"].Current == 5 && !settings["onlyShao"] && !settings["ladderSplit"])
        {
            vars.matchWon = false;
            print("VICTORY SPLIT");
            return true;
        }
        if (vars.watchers["gameState"].Current== 11)
        {
            vars.matchWon = false;
            print("SHAO KAHN'S RULE IS OVER");
            return true;
        }
    }
}