From: Matt Joiner <anacrolix@gmail.com>
Date: Thu, 21 Aug 2014 17:42:38 +0000 (+1000)
Subject: Extract function generating all chunk specs for a piece
X-Git-Tag: v1.0.0~1623
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=2410c3eddd0585e5751481f01788f5fb4cac1b16;p=btrtrc.git

Extract function generating all chunk specs for a piece
---

diff --git a/torrent.go b/torrent.go
index 178f29ae..9fdb79d5 100644
--- a/torrent.go
+++ b/torrent.go
@@ -358,24 +358,30 @@ func (t *torrent) bitfield() (bf []bool) {
 	return
 }
 
+func (t *torrent) pieceChunks(piece int) (css []chunkSpec) {
+	css = make([]chunkSpec, 0, (t.PieceLength(pp.Integer(piece))+chunkSize-1)/chunkSize)
+	var cs chunkSpec
+	for left := t.PieceLength(pp.Integer(piece)); left != 0; left -= cs.Length {
+		cs.Length = left
+		if cs.Length > chunkSize {
+			cs.Length = chunkSize
+		}
+		css = append(css, cs)
+		cs.Begin += cs.Length
+	}
+	return
+}
+
 func (t *torrent) pendAllChunkSpecs(index pp.Integer) {
 	piece := t.Pieces[index]
 	if piece.PendingChunkSpecs == nil {
 		piece.PendingChunkSpecs = make(
 			map[chunkSpec]struct{},
-			(t.Info.PieceLength+chunkSize-1)/chunkSize)
-	}
-	c := chunkSpec{
-		Begin: 0,
+			(t.PieceLength(index)+chunkSize-1)/chunkSize)
 	}
-	cs := piece.PendingChunkSpecs
-	for left := pp.Integer(t.PieceLength(index)); left != 0; left -= c.Length {
-		c.Length = left
-		if c.Length > chunkSize {
-			c.Length = chunkSize
-		}
-		cs[c] = struct{}{}
-		c.Begin += c.Length
+	pcss := piece.PendingChunkSpecs
+	for _, cs := range t.pieceChunks(int(index)) {
+		pcss[cs] = struct{}{}
 	}
 	t.PiecesByBytesLeft.ValueChanged(piece.bytesLeftElement)
 	return