main.cpp

#include <iostream>
#include <json/json.h>
#include <getopt.h>
#include <fstream>
#include "Finder.h"

using std::string;
using std::vector;
using Json::Reader;
using Json::Writer;
using std::ifstream;
using std::cout;
using std::endl;

using namespace icfp2015;

int main(int argc, char **argv) {
    vector<string> ofPower;
    int opt;
    long memLimit;
    long timeLimit;
    string fName;
    Json::Reader x;
    Json::Value root;
    bool printUnits = false, showScore = false, showPower = false, runOne = false;
    int printField = 0;

    while ((opt = getopt(argc, argv, "f:t:m:p:FUSPO")) >= 0) {
        switch (opt) {
            case 'O':
                runOne = true;
                break;
            case 'P':
                showPower = true;
                break;
            case 'F':
                ++printField;
                break;
            case 'U':
                printUnits = true;
                break;
            case 'S':
                showScore = true;
                break;
            case 'f':
                fName = optarg;
                break;
            case 't':
                timeLimit = strtol(optarg, NULL, 10);
                break;
            case 'm':
                memLimit = strtol(optarg, NULL, 10);
                break;
            case 'p':
                ofPower.push_back(optarg);
                break;
            default:
                if (opt > 0) {
                    printf("unidentified arg '%c'\n", (char) opt);
                } else {
                    printf("no args?\n");
                }
                return 1;
        }
    }

    {
        ifstream file(fName);
        x.parse(file, root, false);
    }

    long id = root["id"].asInt64();

    if (printField || printUnits || showPower || showScore)
        glbLog.Setup(id);

    Field f(root);
    if (printField)
        f.print(string("Initial field status"));

    Units u(root);
    if (printUnits) {
        for (int i = 0; i < u.num(); ++i) {
            Unit tmp = u[i];
            glbLog() << "------- unit " << i << endl;
            for (int j = 0; j < 6; ++j) {
                tmp.print();
                tmp.rotate(true);
                if (tmp == u[i]) break;
            }
        }
    }

    RNGSeeds s(root);

    int maxUnits = root["sourceLength"].asInt();
    Finder k(ofPower);
    if (showPower) {
        k.print();
    }
    Json::Value outRoot(Json::arrayValue);
    long allScores = 0;

    for (RNG &gen : s.list) {
        long seed = gen.seed();
        Solver slv(gen, f, u, k.Words());

        int baseScore = (int) slv.Run(printField, maxUnits);

        int powerScore = k.FormatSolution(slv.getSol(), id, seed, outRoot);

        if (showScore) {
            glbLog() << "solution score: " << baseScore + powerScore << endl;
            allScores += baseScore + powerScore;
        }
        if (runOne)
            break;
    }

    if (showScore) {
        glbLog() << "board score: " << allScores / s.list.size() << endl;
    }

    {
        Json::FastWriter fw;
        printf("%s", fw.write(outRoot).c_str());
    }

    return 0;
}