Minor change in PiecesTracker API.

frostwire · Feb 4, 2015 · 94e8109 · 94e8109
1 parent 7fd55fd
commit 94e8109
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/src/com/frostwire/jlibtorrent/PiecesTracker.java b/src/com/frostwire/jlibtorrent/PiecesTracker.java
@@ -64,7 +64,7 @@ public void setComplete(int pieceIndex, boolean complete) throws IllegalArgument
         this.complete[pieceIndex] = complete;
     }
 
-    public long getSequentialDownloaded(int fileIndex) {
+    public long getSequentialDownloadedBytes(int fileIndex) {
         int[] pieces = files[fileIndex];
 
         long downloaded = 0;
@@ -81,4 +81,22 @@ public long getSequentialDownloaded(int fileIndex) {
 
         return downloaded;
     }
+
+    public int getSequentialDownloadedPieces(int fileIndex) {
+        int[] pieces = files[fileIndex];
+
+        int count = 0;
+
+        for (int i = 0; i < pieces.length; i++) {
+            int pieceIndex = pieces[i];
+
+            if (complete[pieceIndex]) {
+                count++;
+            } else {
+                break;
+            }
+        }
+
+        return count;
+    }
 }
diff --git a/src/com/frostwire/jlibtorrent/demo/PieceMap.java b/src/com/frostwire/jlibtorrent/demo/PieceMap.java
@@ -33,7 +33,7 @@ public static void main(String[] args) throws Throwable {
         }
 
         for (int i = 0; i < numFiles; i++) {
-            System.out.println("File index (seq)completed: " + tracker.getSequentialDownloaded(i));
+            System.out.println("File index (seq)completed: " + tracker.getSequentialDownloadedBytes(i));
         }
     }
 }