Added Interests Similarity based weight computation

Renamed graphalytics serializer with weights to extended

Author: ArnauPrat

src/main/java/ldbc/snb/datagen/objects/similarity/InterestsSimilarity.java

@@ -2,11 +2,21 @@
 
 import ldbc.snb.datagen.objects.Person;
 
+import java.util.Set;
+import java.util.TreeSet;
+
 /**
  * Created by aprat on 22/01/16.
  */
 public class InterestsSimilarity implements Person.PersonSimilarity {
     public float Similarity(Person personA, Person personB) {
-        return 0;
+        Set<Integer> union = new TreeSet<Integer>(personA.interests());
+        union.addAll(personB.interests());
+        union.add(personA.mainInterest());
+        union.add(personB.mainInterest());
+        Set<Integer> intersection = new TreeSet<Integer>(personA.interests());
+        intersection.retainAll(personB.interests());
+        if(personA.mainInterest() == personB.mainInterest()) intersection.add(personA.mainInterest());
+        return union.size() > 0 ? intersection.size() / (float)union.size() : 0;
     }
 }

src/main/java/ldbc/snb/datagen/serializer/graphalytics/CSVPersonSerializerWithWeights.java renamed to src/main/java/ldbc/snb/datagen/serializer/graphalytics/CSVPersonSerializerExtended.java

@@ -50,11 +50,12 @@
 import java.util.ArrayList;
 import java.util.Iterator;
 
-public class CSVPersonSerializerWithWeights extends PersonSerializer {
+public class CSVPersonSerializerExtended extends PersonSerializer {
 
     private HDFSCSVWriter [] writers;
 
     private enum FileNames {
+        PERSON("person"),
         PERSON_KNOWS_PERSON("person_knows_person");
 
         private final String name;
@@ -67,7 +68,7 @@ public String toString() {
         }
     }
 
-    public CSVPersonSerializerWithWeights() {
+    public CSVPersonSerializerExtended() {
     }
 
     public void initialize(Configuration conf, int reducerId) {
@@ -78,9 +79,15 @@ public void initialize(Configuration conf, int reducerId) {
         }
 
         ArrayList<String> arguments = new ArrayList<String>();
+        arguments.add("id");
+        arguments.add("creationDate");
+        writers[FileNames.PERSON.ordinal()].writeEntry(arguments);
+
+        arguments.clear();
         arguments.clear();
         arguments.add("Person.id");
         arguments.add("Person.id");
+        arguments.add("CreationDate");
         arguments.add("Weight");
         writers[FileNames.PERSON_KNOWS_PERSON.ordinal()].writeEntry(arguments);
 
@@ -96,7 +103,10 @@ public void close() {
 
     @Override
     protected void serialize(Person p) {
-
+        ArrayList<String> arguments = new ArrayList<String>();
+        arguments.add(Long.toString(p.accountId()));
+        arguments.add(Dictionaries.dates.formatDateTime(p.creationDate()));
+        writers[FileNames.PERSON.ordinal()].writeEntry(arguments);
     }
 
     @Override
@@ -111,6 +121,7 @@ protected void serialize(Person p, Knows knows) {
         ArrayList<String> arguments = new ArrayList<String>();
         arguments.add(Long.toString(p.accountId()));
         arguments.add(Long.toString(knows.to().accountId()));
+        arguments.add(Dictionaries.dates.formatDateTime(knows.creationDate()));
         arguments.add(Float.toString(knows.weight()));
         writers[FileNames.PERSON_KNOWS_PERSON.ordinal()].writeEntry(arguments);
     }