add strategy_record

- src/NameGame.h: add strategy record as vector of tuples.
- src/NameGame.cc (runRound): remove each round printout but update the
	strategy record.
- fix speaker memory update.
- (clearRecord): add function.
- (writeRecord): add function to output CSV file with record info.
- (NameGame::run): print average strategy.
- src/main.cc: clear memory and run 1000 rounds.
- TASKS.md: update tasks

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

1 files changed, 35 insertions, 7 deletions

diff --git a/src/NameGame.cc b/src/NameGame.cc
index 8ea3dcf..022b951 100644
--- a/src/NameGame.cc
+++ b/src/NameGame.cc
@@ -1,3 +1,4 @@
+#include <fstream>
 #include <iostream>
 #include <map>
 #include <stdexcept>
@@ -75,9 +76,16 @@ void NameGame::run(int rounds) {
         for (int i = 0; i < rounds; ++i) runRound();
         // Track all plays and plays by non-CM.
         // Report average plays.
+        float avg = 0;
+        for (const auto r : strategy_record) {
+                avg += std::get<2>(r) + std::get<3>(r);
+        }
+        avg /= strategy_record.size();
+        std::cout << avg << std::endl;
 }
 
 void NameGame::runRound() {
+        bool verbose_flag = false;
         // Select random speaker and hearer.
         auto speaker = graph.GetNI(dist(rng));
         auto hearer = graph.GetNI(
@@ -86,21 +94,26 @@ void NameGame::runRound() {
 
         // Speaker chooses best strategy.
         int s_strategy = best_move(speaker.GetId());
-        std::cout << "speaker (" << speaker.GetId() << " "
-                << std::boolalpha << speaker().Val1
-                << ") chose: " << s_strategy << std::endl;
+        if (verbose_flag) {
+                std::cout << "speaker (" << speaker.GetId() << " "
+                        << std::boolalpha << speaker().Val1
+                        << ") chose: " << s_strategy << std::endl;
+        }
 
         // Listener updates memory.
         update_memory(hearer.GetId(), s_strategy);
 
         // Listener chooses best strategy.
         int h_strategy = best_move(hearer.GetId());
-        std::cout << "hearer (" << hearer.GetId() << " "
-                << std::boolalpha << hearer().Val1
-                << ") chose: " << h_strategy << std::endl;
+        if (verbose_flag) {
+                std::cout << "hearer (" << hearer.GetId() << " "
+                        << std::boolalpha << hearer().Val1
+                        << ") chose: " << h_strategy << std::endl;
+        }
 
         // Speaker updates memory.
-        update_memory(speaker.GetId(), s_strategy);
+        update_memory(speaker.GetId(), h_strategy);
+        strategy_record.emplace_back(speaker.GetId(), hearer.GetId(), s_strategy, h_strategy);
 }
 
 int NameGame::best_move(int nId) const {
@@ -127,4 +140,19 @@ void NameGame::update_memory(int nId, int strategy) {
         }
 }
 
+void NameGame::clearRecord() {
+        strategy_record.clear();
+}
+
+void NameGame::writeRecord(const char* fname) {
+        std::ofstream f(fname);
+        f << "speaker,hearer,speaker_strategy,hearer_strategy\n";
+        int speaker, hearer, s_strategy, h_strategy;
+        for (const auto& r : strategy_record) {
+                std::tie(speaker, hearer, s_strategy, h_strategy) = r;
+                f << speaker << ',' << hearer << ',' << s_strategy << ','
+                        << h_strategy << '\n';
+        }
+}
+
 } // namespace tp