From d37e4f1f4e7e6bb26ed83b81821b6a4df09a3b24 Mon Sep 17 00:00:00 2001
From: clarkohw <clark_oh-willeke@brown.edu>
Date: Tue, 20 Apr 2021 10:51:04 -0400
Subject: added link, gain

---
 src/main/java/edu/brown/cs/student/term/Main.java             |  7 ++++++-
 .../edu/brown/cs/student/term/profit/ProfitCalculation.java   | 11 ++++++++---
 2 files changed, 14 insertions(+), 4 deletions(-)

(limited to 'src/main/java/edu')

diff --git a/src/main/java/edu/brown/cs/student/term/Main.java b/src/main/java/edu/brown/cs/student/term/Main.java
index 6ba2393..d1e5604 100644
--- a/src/main/java/edu/brown/cs/student/term/Main.java
+++ b/src/main/java/edu/brown/cs/student/term/Main.java
@@ -189,12 +189,17 @@ public final class Main {
       double gains = profit.calculateGainsSingle(holder_id);
       double sp500PercentGain = profit.compareToSP500();
 
+      double percentGain = 100 * (gains / profit.getMoneyInput());
+      if (profit.getMoneyInput() == 0) {
+        percentGain = 0;
+      }
+
       Map<String, Object> res = new HashMap<>();
       res.put("holder_id", holder_id);
       res.put("moneyIn", profit.getMoneyInput());
       res.put("moneyOut", profit.getMoneyInput() + gains);
       res.put("holdings", holdings);
-      res.put("percentGain", 100 * (gains / profit.getMoneyInput()));
+      res.put("percentGain", percentGain);
       res.put("SP500", (1 + sp500PercentGain) * profit.getMoneyInput());
       res.put("percentSP500", 100 * sp500PercentGain);
       System.err.println("LOG: Returning to GUI " + res);
diff --git a/src/main/java/edu/brown/cs/student/term/profit/ProfitCalculation.java b/src/main/java/edu/brown/cs/student/term/profit/ProfitCalculation.java
index d5fa872..4b19899 100644
--- a/src/main/java/edu/brown/cs/student/term/profit/ProfitCalculation.java
+++ b/src/main/java/edu/brown/cs/student/term/profit/ProfitCalculation.java
@@ -299,17 +299,22 @@ public class ProfitCalculation {
     if (!tablesFilled) {
       organizeOrders(id);
       getRealizedGains();
+      getUnrealizedGains();
       tablesFilled = true;
     }
 
 
-    double realizedGains = 0;
+    double gains = 0;
 
     for (double value : realizedGainsMap.values()) {
-      realizedGains += value;
+      gains += value;
+    }
+
+    for (double value: unrealizedGainsMap.values()) {
+      gains += value;
     }
 
-    return realizedGains;
+    return gains;
   }
 
   public List<StockHolding> getHoldingsList(Integer id) {
-- 
cgit v1.2.3-70-g09d2