Update fixed getMean not returning correct value

CM2005 Object Oriented Programming/Midterm/Merkelrex-TradingBot/TradingBot.cpp

@@ -26,6 +26,7 @@ void TradingBot::printMenu()
   // 1 print help
   std::cout << "1: Retrieve current orders" << std::endl;
   std::cout << "2: Predict ask and bid rates" << std::endl;
+  std::cout << "3: Print predictions" << std::endl;;
   std::cout << "8: Exit bot" << std::endl;
   std::cout << spacer << std::endl;
   std::cout << "Enter an option: ";
@@ -111,6 +112,7 @@ void TradingBot::predictRates()
   for (const std::string &product : products)
   {
     askMap[product] = getMean(bookAsks, product);
+    std::cout << askMap[product] << std::endl;
   }
   askHistory.push_back(askMap);
 
@@ -143,17 +145,23 @@ void TradingBot::predictRates()
 double TradingBot::getMean(std::vector<OrderBookEntry> orders, std::string product)
 {
   double sum = 0;
+  int count = 0;
   for (const OrderBookEntry &order : orders)
   {
     if (order.product == product)
+    {
       sum += order.price;
+      ++count;
     }
-  return sum / orders.size();
+  }
+  if (count == 0) return 0.0;
+  return sum / count;
 }
 
 // Print predictions if they have been made
 void TradingBot::printPredictions()
 {
+  std::cout.precision(8);
   std::cout << "Asks" << std::endl;
   if (predictedAsks.size() == 0)
     std::cout << "No ask predictions have been made yet" << std::endl;
@@ -173,20 +181,6 @@ void TradingBot::printPredictions()
 
 void TradingBot::testFunc()
 {
-  std::cout << "Asks" << std::endl;
-  for (auto const &ask : predictedAsks)
-  {
-    std::cout << ask.first << " - " << ask.second << std::endl;
-  }
-
-  std::cout << "Bids" << std::endl;
-  for (auto const &bid : predictedBids)
-  {
-    std::cout << bid.first << " - " << bid.second << std::endl;
-  }
-
-  //std::cout << "Ask: " << askHistory[0] << std::endl;
-  return;
     std::cout << "Asks" << std::endl;
   for (int i = 0; i < bookAsks.size(); ++i)
   {
@@ -210,6 +204,21 @@ void TradingBot::testFunc()
   }
 
   return;
+  std::cout << "Asks" << std::endl;
+  for (auto const &ask : predictedAsks)
+  {
+    std::cout << ask.first << " - " << ask.second << std::endl;
+  }
+
+  std::cout << "Bids" << std::endl;
+  for (auto const &bid : predictedBids)
+  {
+    std::cout << bid.first << " - " << bid.second << std::endl;
+  }
+
+  //std::cout << "Ask: " << askHistory[0] << std::endl;
+  return;
+
   std::vector<OrderBookEntry> currentAsks = merkel->getCurrentAsks();
   std::cout.precision(8);
   for (int i = 0; i < currentAsks.size(); ++i)

CM2005 Object Oriented Programming/Midterm/Merkelrex-TradingBot/TradingBot.h

@@ -21,12 +21,18 @@ private:
   void printMenu();
   int getUserOption();
   void processUserOption(int userOption);
+  void printPredictions();
   // Exchange functions
   void retrieveOrders(); // Retrieves current orders from Merkel and stores them in bookBids and BookAsks
-
+  void predictRates();
+  // Helper functions
+  static double getMean(std::vector<OrderBookEntry> orders, std::string product);
   // ===== Variables =====
   MerkelMain *merkel;
   std::vector<OrderBookEntry> bookBids;
   std::vector<OrderBookEntry> bookAsks;
-  double bidHistory = {};
+  std::vector<std::map<std::string, double>> bidHistory;
+  std::vector<std::map<std::string, double>> askHistory;
+  std::map<std::string, double> predictedBids;
+  std::map<std::string, double> predictedAsks;
 };