Changeset 24398 for gs3-extensions/audioDB/trunk/src/src/java/org/greenstone/gsdl3/util/AudioDBDocInfo.java

Timestamp:

2011-08-12T16:40:17+12:00 (13 years ago)

Author:

davidb

Message:

Latest round of changes to code, in preparing for demo to Goldsmiths

File:

• gs3-extensions/audioDB/trunk/src/src/java/org/greenstone/gsdl3/util/AudioDBDocInfo.java (modified) (1 diff)

gs3-extensions/audioDB/trunk/src/src/java/org/greenstone/gsdl3/util/AudioDBDocInfo.java

@@ -19 +19 @@
 package org.greenstone.gsdl3.util;
 
+import java.util.Vector;
+import java.lang.Comparable;
 
-public class AudioDBDocInfo
+public class AudioDBDocInfo implements Comparable<AudioDBDocInfo>
 {
-    public String oid_ = null;
-    public float rank_ = 0;
+    public String oid_;
+    public Vector<Double> rankVector_;
+    public Vector<Integer> offsetVector_;
 
-    public AudioDBDocInfo(String doc_oid, float rank)
+    public AudioDBDocInfo(String doc_oid, double rank, int offset)
     {
     oid_ = doc_oid;
-    rank_ = rank;
+    
+    rankVector_ = new Vector<Double>();
+    rankVector_.add (rank);
+
+    offsetVector_ = new Vector<Integer>();
+    offsetVector_.add(offset);
     }
 
-    public String toString()
+
+    public AudioDBDocInfo(String doc_oid, Vector<Double> rankVector, Vector<Integer> offsetVector)
     {
-    return "" + oid_ + " (" + rank_ + ")";
+    oid_ = doc_oid;
+    rankVector_ = rankVector;
+    offsetVector_ = offsetVector;
+    }
+
+    public String getDocID()
+    {
+    return oid_;
+    }
+
+    public double getTopRank()
+    {
+    return rankVector_.get(0);
+    }
+
+
+    public String getOffsetList() 
+    {
+    StringBuffer all_offsets = new StringBuffer();
+    boolean first = true;
+
+    for (int i=0; i<offsetVector_.size(); i++) {
+        int offset = offsetVector_.get(i);
+        String offsetStr = Integer.toString(offset);
+
+        if (first) {
+        first = false;
+        }
+        else {
+        all_offsets.append(",");
+        }
+
+        all_offsets.append(offsetStr);
+    }
+
+    return all_offsets.toString();
+    }
+
+    public int compareTo(AudioDBDocInfo di)
+    {
+    // based on first entry in rank Vector
+
+    // embodies a descending sort order
+    double lrank = rankVector_.get(0);
+    double rrank = di.rankVector_.get(0);
+
+    if (lrank<rrank) {
+        return 1;
+    }
+    else if (lrank>rrank) {
+        return -1;
+    }
+    else {
+        return 0;
+    }
     }
 }