#ifndef _PARSE_H
#define _PARSE_H

#include <string>
#include <iostream>
// #include <fstream>
#include <vector>
#include <set>

#include "const.h"

struct World
{
  std::vector<std::vector<int> > car, foot; // adj-lists [from][to]
  std::vector< std::vector<bool> > mCar, mFoot; // connection-matrices [from][to]
  std::vector<std::pair<int,int> > coords;
  
  // state
  int n, loot, smell, me, hq, maxCoord, robberStartPos, robberCertain;
  bool robberPosKnown;
  std::vector<int> bankMoney;
  std::vector<int> bankPos;
  std::vector< std::pair<int,int> > evidence; // [location,time]
  std::vector< std::pair<int,ptype> > cops;
  std::pair<int,ptype> robber; int robberId;
  std::vector<std::set<int> > borgInfo;    // [owner] is set of owned
  std::set<int> myBorg;                    // robots I own
  // cop move history: [world][cop]=move
  std::vector<std::vector<std::pair<int,ptype> > > moveHist;

  // For each cop, we have info on how dirty we know them to be
  std::vector<dirtinessfactor> dcInfo;	// dcInfo = dirty_cop_info
  
  void print(std::ostream& out)
  {
    out << "Car Map" << std::endl;
    for (unsigned int r=0; r<car.size(); ++r)
    {
      out << "\t" << r << ":  ";
      for (unsigned int c=0; c<car[r].size(); ++c)
        out << car[r][c] << " ";
      out << std::endl;
    }
  }

  std::pair<int,ptype> self() { return me < COPS ? cops[me] : robber; }
  std::set<int> smellyNodes(std::vector<std::vector<nodePair> > pureFootMap);
};

struct guess
{
  int botId, locId, certainty, world;
  ptype type;

  guess() {}
  guess( int bot, int loc, ptype typ, int w )
    : botId(bot), locId(loc), world(w), type(typ) {}
  guess( int bot, int loc, ptype typ, int w, int val )
    : botId(bot), locId(loc), certainty(val), world(w), type(typ) {}

  std::pair<int,ptype> pos() { return std::pair<int,ptype>(locId, type); }

  bool betrayal( World& w )
  {
    if ( botId >= COPS )
      return false;
    if ( w.cops[botId].first != locId || w.cops[botId].second != type )
      return true;
    return false;
  }
};

void sendRegister( std::string bot, ptype type );
void recvWorld( World& world );
bool recvTurn( World& world ); // to simplify state machine stuff returns false for game-over
void sendRobberMove( int where, ptype type, int whoami, bool bribedLastTurn,
                     std::vector<int>& copsForBribe );
void sendCleanCopMove( std::vector<guess>& moveList, std::vector<int> &accuse);
void sendDirtyCopMove(std::vector<guess>& moveList );
void sendBribe(bool amBribing);	// must pass in whether we chose to bribe or not
std::vector<int> recvOfferedCops();
int recvVote();
void sendVote( std::vector<int>& votes );
void recvPlan( std::vector<std::vector<guess> >& plan );
void sendPlan( std::vector<guess>& info );
void recvInform( std::vector<std::vector<guess> >& info );
void sendInform( const std::vector<guess>& info );

void initParse();

std::string getLocName(int locID);
int getLocID(std::string locName);

#ifdef DUMP_TO_FILE
extern std::ofstream debug;
#else
#define debug cerr
#endif

#endif