Temp master

src/main/java/Sentiment.java

@@ -29,18 +29,25 @@ public static void main(String[] args) {
             String  jsonContents = Util.readFile("src/modResults.json");
             Results jsonWrapper = mapper.readValue(new File(Constants.HOME_ROUTE), Results.class);
             ArrayList<Email> cleanedDataSet = Util.cleanInput(jsonWrapper.getResults());
+            ArrayList<String> testCases = new ArrayList(Arrays.asList("<ariellacentlivre@gmail.com>"));
 //            Util.sortByDate(cleanedDataSet);
 
 //            SentimentAnalysis sentiment = new SentimentAnalysis();
 //            sentiment.stripInBoundEmails(cleanedDataSet, "<ariellacentlivre@gmail.com>");
 //            sentiment.sentimentScoreIntervals(cleanedDataSet);
 //            sentiment.setScore(cleanedDataSet);
             emailAnalysis emailChain = new emailAnalysis();
+<<<<<<< HEAD
+            emailChain.returnOutgoingEmails(cleanedDataSet,testCases );
+//            emailChain.averageEmailChainReferences(cleanedDataSet);
+//            emailChain.wordFrequency(cleanedDataSet);
+=======
 //            emailChain.averageEmailChainReferences(cleanedDataSet);
             emailChain.wordFrequency(cleanedDataSet);
+>>>>>>> f1a624bb3fd86b1839798c7acfa295145c356677
 
-            ArrayList<Email> sortedEmails = emailChain.sortEmailsByChainLength(cleanedDataSet);
-            emailChain.emailswithReplies(sortedEmails);
+//            ArrayList<Email> sortedEmails = emailChain.sortEmailsByChainLength(cleanedDataSet);
+//            emailChain.emailswithReplies(sortedEmails);
         }
         catch (FileNotFoundException e) {
             e.printStackTrace();

src/main/java/SentimentAnalysis.java

@@ -62,16 +62,7 @@ public ArrayList<Double> sentimentScoreIntervals(ArrayList<Email> emailList) {
         return intervalSentimentScores;
     }
 
-    public ArrayList<Email> stripInBoundEmails(ArrayList<Email> theEmailList, String input) {
-        // Doesnt handle craigslist
-        ArrayList<Email> outBoundEmails = new ArrayList<>();
-        for (Email email : theEmailList) {
-            if (email.getFromEmail().equals(input)) {
-                outBoundEmails.add(email);
-            }
-        }
-        return outBoundEmails;
-    }
+
 
     public static void main(String[] args) {
 

src/main/java/emailAnalysis.java

@@ -17,6 +17,31 @@ public class emailAnalysis {
     public static void main(String[] args) {
 
     }
+    public  ArrayList<Email> returnOutgoingEmails(ArrayList<Email> emailList, ArrayList input) {
+        ArrayList<Email> outgoingEmails = new ArrayList<>();
+
+        for (int i = 0; i < emailList.size(); i++) {
+            Email currentEmail = emailList.get(i);
+            for (int v = 0; v < input.size(); v++) {
+                if (currentEmail.getFromEmail().equals(input.get(v))) {
+                    outgoingEmails.add(currentEmail);
+                }
+            }
+        }
+
+        return outgoingEmails;
+    }
+
+    public ArrayList<Email> firstEmailinChain (ArrayList<Email> emailList) {
+        ArrayList<Email> firstEmails = new ArrayList<Email>();
+        for (int i = 0; i < emailList.size(); i++) {
+            Email currentEmail = emailList.get(i);
+            if (currentEmail.getReferences().length == 0) {
+                firstEmails.add(currentEmail);
+            }
+        }
+        return firstEmails;
+    }
 
     public int firstInChain(ArrayList<Email> emailList) {
         ArrayList<Email> firstInChain  = new ArrayList();
@@ -31,6 +56,28 @@ public int firstInChain(ArrayList<Email> emailList) {
         return firstInChain.size();
     }
 
+    public int firstEmails (ArrayList<Email> emailList) {
+        int numberFirstEmails = 0;
+        ArrayList<Email> cleanedDataSet = new ArrayList();
+
+        for (int i = 0; i < emailList.size(); i++) {
+            Email currentEmail = emailList.get(i);
+            for (int v = 0; v < cleanedDataSet.size(); v++) {
+                Email currentCleanedEmail = cleanedDataSet.get(v);
+                if (!currentEmail.getReplyTo().equals(currentCleanedEmail.getReplyTo())) {
+                    cleanedDataSet.add(currentEmail);
+                }
+            }
+        }
+
+        numberFirstEmails = cleanedDataSet.size();
+
+        return numberFirstEmails;
+    }
+
+
+
+
     public ArrayList<Email> sortEmailsByChainLength (ArrayList<Email> theEmailList) {
         // Broken should be bubble sort
         for (int i = 1; i < theEmailList.size(); i++) {

src/main/java/util.java

@@ -44,6 +44,21 @@ public  static ArrayList<Email> cleanInput(ArrayList<Email> theEmailList) {
         return cleanedDataSet;
     }
 
+    public static ArrayList<Email> returnOutgoingEmails(ArrayList<Email> emailList, ArrayList input) {
+        ArrayList<Email> outgoingEmails = new ArrayList<>();
+
+        for (int i = 0; i < emailList.size(); i++) {
+            Email currentEmail = emailList.get(i);
+            for (int v = 0; v < input.size(); v++) {
+                if (currentEmail.getFromEmail() == input.get(v)) {
+                    outgoingEmails.add(currentEmail);
+                }
+            }
+        }
+
+        return outgoingEmails;
+    }
+
     public static void outputToJSON(HashMap<String, ArrayList> wordFrequency) {
         Gson gson = new Gson();
         String json = gson.toJson(wordFrequency);