allow appending to CSV file and use biunit answer

- src/NameGame.h: add append option to writeRecord to allow updating one
  big CSV file.
- add speaker and hearer type (i.e. committed) to CSV file.
- src/NameGame.cc (initMemory): change answers to biunit i.e. -1 and 1
  for easier analysis.
- (runRound): write speaker/hearer type.
- (writeRecord): add append mode to allow updating a single CSV file.
- had to change to fstream with in/out because using an ofstream always
  overwrote the file.
- only write the header line when at the beginning of the file.
- add speaker/hearer type.
- src/main.cc: clear record again in each loop and go into append mode
  for every iteration after the first one.

Signed-off-by: Aiden Woodruff <woodra@rpi.edu>

3 files changed, 21 insertions, 13 deletions

diff --git a/src/NameGame.cc b/src/NameGame.cc
index 5dfcbc7..2dd64c0 100644
--- a/src/NameGame.cc
+++ b/src/NameGame.cc
@@ -59,7 +59,7 @@ void NameGame::initMemory() {
                 auto& rDat = graph.GetNDat(rId);
                 rDat.Val1 = true;
                 rDat.Val2.Clr();
-                rDat.Val2.Add(2);
+                rDat.Val2.Add(1);
                 ids.erase(ids.begin() + r);
         }
         // Establish norm (fully memory of strategy 1) for remaining nodes.
@@ -67,7 +67,7 @@ void NameGame::initMemory() {
                 auto& d = graph.GetNDat(id);
                 d.Val1 = false;
                 d.Val2.Clr();
-                while (d.Val2.Len() < memlen) d.Val2.Add(1);
+                while (d.Val2.Len() < memlen) d.Val2.Add(-1);
         }
 }
 
@@ -115,7 +115,8 @@ void NameGame::runRound(int r) {
         // Speaker updates memory.
         update_memory(speaker.GetId(), h_strategy);
         strategy_record.push_back({
-                r, speaker.GetId(), hearer.GetId(), s_strategy, h_strategy
+                r, speaker.GetId(), speaker().Val1,
+                hearer.GetId(), hearer().Val1, s_strategy, h_strategy
         });
 }
 
@@ -147,14 +148,19 @@ void NameGame::clearRecord() {
         strategy_record.clear();
 }
 
-void NameGame::writeRecord(const char* fname) {
-        std::ofstream f(fname);
-        f << "group_size,cm_size,memlen,round,"
-                "speaker,hearer,speaker_strategy,hearer_strategy\n";
+void NameGame::writeRecord(const char* fname, bool append) {
+        auto mode = std::ios::in | std::ios::out;
+        if (append) mode |= std::ios::ate;
+        else mode |= std::ios::trunc;
+        std::fstream f(fname, mode);
+        if (!append || f.tellp() == 0) {
+                f << "group_size,cm_size,memlen,round,speaker,speaker_type,"
+                        "hearer,hearer_type,speaker_strategy,hearer_strategy\n";
+        }
         for (const auto& r : strategy_record) {
                 f << gsize << ',' << cmsize << ',' << memlen << ',' << r.round << ','
-                        << r.speaker << ',' << r.hearer << ',' << r.s_strategy << ','
-                        << r.h_strategy << '\n';
+                        << r.speaker << ',' << r.s_cm << ',' << r.hearer << ',' << r.h_cm << ','
+                        << r.s_strategy << ',' << r.h_strategy << '\n';
         }
 }
 
diff --git a/src/NameGame.h b/src/NameGame.h
index abd9d6d..eae45f3 100644
--- a/src/NameGame.h
+++ b/src/NameGame.h
@@ -19,7 +19,7 @@ public:
         void initMemory();
         void clearRecord();
         void run(int rounds);
-        void writeRecord(const char* fname);
+        void writeRecord(const char* fname, bool append = false);
 
 protected:
         int best_move(int nId) const;
@@ -32,7 +32,7 @@ private:
         std::uniform_int_distribution<> dist;
         TNodeNet<TPair<TBool, TIntV>> graph;
         struct Record {
-                int round, speaker, hearer, s_strategy, h_strategy;
+                int round, speaker, s_cm, hearer, h_cm, s_strategy, h_strategy;
         };
         std::vector<Record> strategy_record;
 }; // class NameGame
diff --git a/src/main.cc b/src/main.cc
index 2c56584..7a5f2bc 100644
--- a/src/main.cc
+++ b/src/main.cc
@@ -5,14 +5,16 @@
 int main(int argc, char* argv[]) {
         int group_size = 100;
         tp::NameGame namegame(group_size, 0);
-        for (int cmperc = 18; cmperc < 28; cmperc += 2) {
+        int cmperc_start = 18;
+        for (int cmperc = cmperc_start; cmperc < 28; cmperc += 2) {
                 int cmsize = group_size * cmperc / 100;
+                namegame.clearRecord();
                 namegame.setCMsize(cmsize);
                 namegame.initMemory();
                 std::cout << "CM " << cmsize << " / " << group_size << std::endl;
                 namegame.run(1000);
+                namegame.writeRecord("strategy_record.csv", cmperc != cmperc_start);
         }
-        namegame.writeRecord("strategy_record.csv");
         return 0;
 }