1 files changed, 122 insertions, 0 deletions

@@ -0,0 +1,122 @@
+#include "vendor/csv.h"
+#include "identifier.h"
+#include "histogram.h"
+#include <rawr.h>
+#include <cstdlib>
+#include <ctime>
+#include <map>
+#include <string>
+#include <iostream>
+#include <sstream>
+
+
+
+using speakerstore = identifier<std::string>;
+using speaker_id = speakerstore::key_type;
+
+
+struct speaker_data {
+
+  std::string name;
+  histogram<speaker_id> nextSpeaker;
+  rawr chain;
+
+};
+
+
+
+
+int main(int, char**)
+{
+  srand(time(NULL));
+  rand(); rand(); rand(); rand();
+
+  speakerstore speakers;
+  std::map<speaker_id, speaker_data> speakerData;
+  histogram<speaker_id> allSpeakers;
+
+
+
+  io::CSVReader<2,io::trim_chars<' ', '\t'>,io::double_quote_escape<',', '"'>> in("../dialogue.csv");
+  std::string speaker;
+  std::string line;
+
+  bool hadPrev = false;
+  speaker_id prevSpeaker;
+
+  while (in.read_row(speaker, line))
+  {
+    speaker_id spId = speakers.add(speaker);
+    speaker_data& myData = speakerData[spId];
+    myData.name = speaker;
+
+    allSpeakers.add(spId);
+
+    if (hadPrev && prevSpeaker != spId)
+    {
+      speaker_data& psd = speakerData[prevSpeaker];
+      psd.nextSpeaker.add(spId);
+    }
+
+    myData.chain.addCorpus(line);
+
+    hadPrev = true;
+    prevSpeaker = spId;
+  }
+
+  for (auto& sp : speakerData)
+  {
+    sp.second.chain.compile(4);
+    sp.second.nextSpeaker.compile();
+  }
+
+  std::cout << "Speakers:" << std::endl;
+  for (auto& sp : speakerData)
+  {
+    std::cout << "  " << sp.second.name << std::endl;
+  }
+  std::cout << std::endl;
+
+  allSpeakers.compile();
+
+  for (;;)
+  {
+    speaker_id curSpeaker = allSpeakers.next();
+
+    std::ostringstream theEnd;
+
+    for (int i = 0; i < 5; i++)
+    {
+      speaker_data& curSd = speakerData.at(curSpeaker);
+
+      //std::ostringstream thisLine;
+
+      if (curSd.name != "")
+      {
+        theEnd << curSd.name << ": ";
+      }
+
+      theEnd << curSd.chain.randomSentence(1);
+
+      /*if (i > 0 && theEnd.str().length() + thisLine.str().length() > 280)
+      {
+        break;
+      }*/
+
+      theEnd << std::endl;
+      //theEnd << thisLine.str();
+
+      curSpeaker = curSd.nextSpeaker.next();
+    }
+
+    std::string output = theEnd.str();
+    output.resize(280);
+    output = output.substr(0, output.find_last_of('\n'));
+    std::cout << output;
+
+    std::cout << std::endl;
+    std::cout << std::endl;
+
+    getc(stdin);
+  }
+}