#include <iostream>
#include <string>
#include <fstream>
#include <sstream>
#include <vector>
#include <map>
#include "Structures.h"
#include "ParseTest.h"
#include "util.h"
#include "Algorithms.h"

using namespace std;

class GenerateProbeRatings {
public:
    GenerateProbeRatings() {
    }

    void initialize(vector<Entry>& entries, bool isProbe) {
    }
  
    void processMovieRatings(int movieId, vector<Entry>& entries) {
        std::copy(entries.begin(), entries.end(), back_inserter(entriesMap[movieId]));
        entries.clear();
    }

    char getRating(const Entry& e) {
        vector<Entry>::const_iterator it;
        it = std::find_if(entriesMap[e.getMovieId()].begin(), entriesMap[e.getMovieId()].end(), ResultComparator(e));
        if (it != entriesMap[e.getMovieId()].end()) {
            return it->getRating();
        } else {
            cout << "Couldn't find entry for " << e << endl;
            abort();
        }
    }

    void postProcess(vector<Entry>& entries) {
        ofstream outFile(probeResultsFileName.c_str());
        ifstream inFile(probeFileName.c_str());
        string line;
        getline(inFile, line);
        int movieId;
        bool changedId;
        while (!inFile.eof()) {
            Entry e = parseChallengeLine(line, movieId, changedId, false);
            if (changedId) {
                outFile << movieId << ":" << endl;
            } else {
                outFile << (int) getRating(e) << endl;
            }
            getline(inFile, line);
        }
    }

    std::map<int, vector<Entry> > entriesMap;
};

int main() {
    vector<Entry> entries;
    bool doProbe = true;
    ProcessorWrapper<CalculateUserSimilarities> processor(entries, doProbe);
    cout << "sizeof(Entry) is " << sizeof(Entry) << endl;
    //entries.reserve(100480507);
    std::map<int, vector<Entry> > probeEntriesMap;
    if (doProbe) {
        appendEntriesFromProbeFile(probeEntriesMap);
    }
    for (int i = 1; i <= numMovies; i++) {
        if (i % 250 == 0) {
            cout << "Working on movie " << i << " - got " << entries.size() << " entries so far" << endl;
        }
        appendEntriesFromMovieFile(i, entries, true);
        if (doProbe) {
            filterEntries(entries, probeEntriesMap);
        }
        processor.processMovieRatings(i, entries);
    }
    processor.postProcess(entries);
    return 0;
}