# HG changeset patch
# User Artem Tikhomirov <tikhomirov.artem@gmail.com>
# Date 1361480035 -3600
# Node ID a5fd757d1b5dfb329fbced29c2d887709fe9e5fe
# Parent  093a2022dad50b5fb0b3838758fb92a0234e1ad3
Access to content of annotated files through BlockData interface

diff -r 093a2022dad5 -r a5fd757d1b5d src/org/tmatesoft/hg/internal/AnnotateFacility.java
--- a/src/org/tmatesoft/hg/internal/AnnotateFacility.java	Thu Feb 21 20:25:26 2013 +0100
+++ b/src/org/tmatesoft/hg/internal/AnnotateFacility.java	Thu Feb 21 21:53:55 2013 +0100
@@ -25,7 +25,9 @@
 
 import org.tmatesoft.hg.core.HgIterateDirection;
 import org.tmatesoft.hg.core.Nodeid;
+import org.tmatesoft.hg.internal.AnnotateFacility.BlockData;
 import org.tmatesoft.hg.internal.DiffHelper.LineSequence;
+import org.tmatesoft.hg.internal.DiffHelper.LineSequence.ByteChain;
 import org.tmatesoft.hg.repo.HgDataFile;
 import org.tmatesoft.hg.repo.HgInvalidStateException;
 import org.tmatesoft.hg.util.CancelledException;
@@ -242,10 +244,38 @@
 	@Callback
 	public interface BlockInspectorEx extends BlockInspector { // XXX better name
 		// XXX perhaps, shall pass object instead of separate values for future extension?
-		void start(int originLineCount, int targetLineCount);
+		void start(BlockData originContent, BlockData targetContent);
 		void done();
 	}
 	
+	/**
+	 * Represents content of a block, either as a sequence of bytes or a 
+	 * sequence of smaller blocks (lines), if appropriate (according to usage context).
+	 * 
+	 * This approach allows line-by-line access to content data along with complete byte sequence for the whole block, i.e.
+	 * <pre>
+	 *    BlockData bd = addBlock.addedLines()
+	 *    // bd describes data from the addition completely.
+	 *    // elements of the BlockData are lines
+	 *    bd.elementCount() == addBlock.totalAddedLines();
+	 *    // one cat obtain complete addition with
+	 *    byte[] everythingAdded = bd.asArray();
+	 *    // or iterate line by line
+	 *    for (int i = 0; i < bd.elementCount(); i++) {
+	 *    	 byte[] lineContent = bd.elementAt(i);
+	 *       String line = new String(lineContent, fileEncodingCharset);
+	 *    }
+	 *    where bd.elementAt(0) is the line at index addBlock.firstAddedLine() 
+	 * </pre> 
+	 * 
+	 * LineData or ChunkData? 
+	 */
+	public interface BlockData {
+		BlockData elementAt(int index);
+		int elementCount();
+		byte[] asArray();
+	}
+	
 	public interface Block {
 		int originChangesetIndex();
 		int targetChangesetIndex();
@@ -260,19 +290,20 @@
 		int originStart();
 		int targetStart();
 		int length();
+		BlockData content();
 	}
 	
 	public interface AddBlock extends Block {
 		int insertedAt(); // line index in the old file 
 		int firstAddedLine();
 		int totalAddedLines();
-		String[] addedLines();
+		BlockData addedLines();
 	}
 	public interface DeleteBlock extends Block {
 		int removedAt(); // line index in the new file
 		int firstRemovedLine();
 		int totalRemovedLines();
-		String[] removedLines();
+		BlockData removedLines();
 	}
 	public interface ChangeBlock extends AddBlock, DeleteBlock {
 	}
@@ -298,6 +329,7 @@
 		private EqualBlocksCollector p2MergeCommon;
 		private int csetMergeParent;
 		private IntVector mergeRanges;
+		private ContentBlock originContent, targetContent;
 
 		public BlameBlockInspector(BlockInspector inspector, int originCset, int targetCset) {
 			assert inspector != null;
@@ -315,8 +347,10 @@
 		@Override
 		public void begin(LineSequence s1, LineSequence s2) {
 			super.begin(s1, s2);
+			originContent = new ContentBlock(s1);
+			targetContent = new ContentBlock(s2);
 			if (insp instanceof BlockInspectorEx) {
-				((BlockInspectorEx) insp).start(s1.chunkCount() - 1, s2.chunkCount() - 1);
+				((BlockInspectorEx) insp).start(originContent, targetContent);
 			}
 		}
 		
@@ -353,13 +387,13 @@
 					// how many lines we may reported as changed (don't use more than in range unless it's the very last range)
 					final int s1LinesToBorrow = lastRange ? s1LinesLeft : Math.min(s1LinesLeft, rangeLen);
 					if (s1LinesToBorrow > 0) {
-						BlockImpl2 block = new BlockImpl2(seq1, seq2, s1Start, s1LinesToBorrow, rangeStart, rangeLen, s1Start, rangeStart);
+						ChangeBlockImpl block = new ChangeBlockImpl(originContent, targetContent, s1Start, s1LinesToBorrow, rangeStart, rangeLen, s1Start, rangeStart);
 						block.setOriginAndTarget(rangeOrigin, csetTarget);
 						insp.changed(block);
 						s1ConsumedLines += s1LinesToBorrow;
 						s1Start += s1LinesToBorrow;
 					} else {
-						BlockImpl2 block = getAddBlock(rangeStart, rangeLen, s1Start);
+						ChangeBlockImpl block = getAddBlock(rangeStart, rangeLen, s1Start);
 						block.setOriginAndTarget(rangeOrigin, csetTarget);
 						insp.added(block);
 					}
@@ -368,7 +402,7 @@
 					throw new HgInvalidStateException(String.format("Expected to process %d lines, but actually was %d", s1TotalLines, s1ConsumedLines));
 				}
 			} else {
-				BlockImpl2 block = new BlockImpl2(seq1, seq2, s1From, s1To-s1From, s2From, s2To - s2From, s1From, s2From);
+				ChangeBlockImpl block = new ChangeBlockImpl(originContent, targetContent, s1From, s1To-s1From, s2From, s2To - s2From, s1From, s2From);
 				block.setOriginAndTarget(csetOrigin, csetTarget);
 				insp.changed(block);
 			}
@@ -384,14 +418,14 @@
 					int rangeOrigin = mergeRanges.get(i);
 					int rangeStart = mergeRanges.get(i+1);
 					int rangeLen = mergeRanges.get(i+2);
-					BlockImpl2 block = getAddBlock(rangeStart, rangeLen, insPoint);
+					ChangeBlockImpl block = getAddBlock(rangeStart, rangeLen, insPoint);
 					block.setOriginAndTarget(rangeOrigin, csetTarget);
 					insp.added(block);
 					// indicate insPoint moved down number of lines we just reported
 					insPoint += rangeLen;
 				}
 			} else {
-				BlockImpl2 block = getAddBlock(s2From, s2To - s2From, s1InsertPoint);
+				ChangeBlockImpl block = getAddBlock(s2From, s2To - s2From, s1InsertPoint);
 				block.setOriginAndTarget(csetOrigin, csetTarget);
 				insp.added(block);
 			}
@@ -399,20 +433,20 @@
 		
 		@Override
 		protected void deleted(int s2DeletePoint, int s1From, int s1To) {
-			BlockImpl2 block = new BlockImpl2(seq1, null, s1From, s1To - s1From, -1, -1, -1, s2DeletePoint);
+			ChangeBlockImpl block = new ChangeBlockImpl(originContent, null, s1From, s1To - s1From, -1, -1, -1, s2DeletePoint);
 			block.setOriginAndTarget(csetOrigin, csetTarget);
 			insp.deleted(block);
 		}
 
 		@Override
 		protected void unchanged(int s1From, int s2From, int length) {
-			BlockImpl1 block = new BlockImpl1(s1From, s2From, length);
+			EqualBlockImpl block = new EqualBlockImpl(s1From, s2From, length, targetContent);
 			block.setOriginAndTarget(csetOrigin, csetTarget);
 			insp.same(block);
 		}
 		
-		private BlockImpl2 getAddBlock(int start, int len, int insPoint) {
-			return new BlockImpl2(null, seq2, -1, -1, start, len, insPoint, -1);
+		private ChangeBlockImpl getAddBlock(int start, int len, int insPoint) {
+			return new ChangeBlockImpl(null, targetContent, -1, -1, start, len, insPoint, -1);
 		}
 	}
 	
@@ -438,14 +472,17 @@
 		}
 	}
 
-	static class BlockImpl1 extends BlockImpl implements EqualBlock {
+	static class EqualBlockImpl extends BlockImpl implements EqualBlock {
 		private final int start1, start2;
 		private final int length;
+		private final ContentBlock fullContent;
+		private FilterBlock myContent;
 		
-		BlockImpl1(int blockStartSeq1, int blockStartSeq2, int blockLength) {
+		EqualBlockImpl(int blockStartSeq1, int blockStartSeq2, int blockLength, ContentBlock targetContent) {
 			start1 = blockStartSeq1;
 			start2 = blockStartSeq2;
 			length = blockLength;
+			fullContent = targetContent;
 		}
 
 		public int originStart() {
@@ -460,26 +497,34 @@
 			return length;
 		}
 		
+		public BlockData content() {
+			if (myContent == null) {
+				myContent = new FilterBlock(fullContent, start2, length);
+			}
+			return myContent;
+		}
+		
 		@Override
 		public String toString() {
 			return String.format("@@ [%d..%d) == [%d..%d) @@", start1, start1+length, start2, start2+length);
 		}
 	}
 	
-	static class BlockImpl2 extends BlockImpl implements ChangeBlock {
+	static class ChangeBlockImpl extends BlockImpl implements ChangeBlock {
 		
-		private final LineSequence oldSeq;
-		private final LineSequence newSeq;
+		private final ContentBlock oldContent;
+		private final ContentBlock newContent;
 		private final int s1Start;
 		private final int s1Len;
 		private final int s2Start;
 		private final int s2Len;
 		private final int s1InsertPoint;
 		private final int s2DeletePoint;
+		private FilterBlock addedBlock, removedBlock;
 
-		public BlockImpl2(LineSequence s1, LineSequence s2, int s1Start, int s1Len, int s2Start, int s2Len, int s1InsertPoint, int s2DeletePoint) {
-			oldSeq = s1;
-			newSeq = s2;
+		public ChangeBlockImpl(ContentBlock c1, ContentBlock c2, int s1Start, int s1Len, int s2Start, int s2Len, int s1InsertPoint, int s2DeletePoint) {
+			oldContent = c1;
+			newContent = c2;
 			this.s1Start = s1Start;
 			this.s1Len = s1Len;
 			this.s2Start = s2Start;
@@ -500,8 +545,11 @@
 			return s2Len;
 		}
 
-		public String[] addedLines() {
-			return generateLines(totalAddedLines(), firstAddedLine());
+		public BlockData addedLines() {
+			if (addedBlock == null) {
+				addedBlock = new FilterBlock(newContent, firstAddedLine(), totalAddedLines());
+			}
+			return addedBlock;
 		}
 		
 		public int removedAt() {
@@ -516,16 +564,11 @@
 			return s1Len;
 		}
 
-		public String[] removedLines() {
-			return generateLines(totalRemovedLines(), firstRemovedLine());
-		}
-		
-		private String[] generateLines(int count, int startFrom) {
-			String[] rv = new String[count];
-			for (int i = 0; i < count; i++) {
-				rv[i] = String.format("LINE %d", startFrom + i+1);
+		public BlockData removedLines() {
+			if (removedBlock == null) {
+				removedBlock = new FilterBlock(oldContent, firstRemovedLine(), totalRemovedLines());
 			}
-			return rv;
+			return removedBlock;
 		}
 		
 		@Override
@@ -539,6 +582,77 @@
 			return String.format("@@ -%d,%d +%d,%d @@", firstRemovedLine(), totalRemovedLines(), firstAddedLine(), totalAddedLines());
 		}
 	}
+	
+	private static class SingleLine implements BlockData {
+		private final ByteChain line;
+
+		public SingleLine(ByteChain lineContent) {
+			line = lineContent;
+		}
+
+		public BlockData elementAt(int index) {
+			assert false;
+			return null;
+		}
+
+		public int elementCount() {
+			return 0;
+		}
+
+		public byte[] asArray() {
+			return line.data();
+		}
+	}
+	
+	private static class ContentBlock implements BlockData {
+		private final LineSequence seq;
+
+		public ContentBlock(LineSequence sequence) {
+			seq = sequence;
+		}
+
+		public BlockData elementAt(int index) {
+			return new SingleLine(seq.chunk(index));
+		}
+
+		public int elementCount() {
+			return seq.chunkCount() - 1;
+		}
+
+		public byte[] asArray() {
+			return seq.data(0, seq.chunkCount() - 1);
+		}
+	}
+	
+	private static class FilterBlock implements BlockData {
+		private final ContentBlock contentBlock;
+		private final int from;
+		private final int length;
+
+		public FilterBlock(ContentBlock bd, int startFrom, int len) {
+			assert bd != null;
+			assert startFrom + len < bd.seq.chunkCount(); // there's one extra chunk in the end, so strict less is ok
+			contentBlock = bd;
+			from = startFrom;
+			length = len;
+		}
+
+		public BlockData elementAt(int index) {
+			if (index < 0 || index >= length) {
+				throw new IllegalArgumentException(String.format("Expected value from [0..%d), got %d", length, index));
+			}
+			return contentBlock.elementAt(from + index);
+		}
+
+		public int elementCount() {
+			return length;
+		}
+
+		public byte[] asArray() {
+			return contentBlock.seq.data(from, from + length);
+		}
+	}
+	
 
 	static class EqualBlocksCollector implements DiffHelper.MatchInspector<LineSequence> {
 		private final IntVector matches = new IntVector(10*3, 2*3);
diff -r 093a2022dad5 -r a5fd757d1b5d src/org/tmatesoft/hg/internal/FileAnnotation.java
--- a/src/org/tmatesoft/hg/internal/FileAnnotation.java	Thu Feb 21 20:25:26 2013 +0100
+++ b/src/org/tmatesoft/hg/internal/FileAnnotation.java	Thu Feb 21 21:53:55 2013 +0100
@@ -19,6 +19,7 @@
 import java.util.LinkedList;
 
 import org.tmatesoft.hg.internal.AnnotateFacility.AddBlock;
+import org.tmatesoft.hg.internal.AnnotateFacility.BlockData;
 import org.tmatesoft.hg.internal.AnnotateFacility.ChangeBlock;
 import org.tmatesoft.hg.internal.AnnotateFacility.DeleteBlock;
 import org.tmatesoft.hg.internal.AnnotateFacility.EqualBlock;
@@ -51,9 +52,9 @@
 			delegate = lineInspector;
 		}
 		
-		public void start(int originLineCount, int targetLineCount) {
+		public void start(BlockData originContent, BlockData targetContent) {
 			if (knownLines == null) {
-				knownLines = new boolean[targetLineCount];
+				knownLines = new boolean[targetContent.elementCount()];
 			}
 		}
 
diff -r 093a2022dad5 -r a5fd757d1b5d test/org/tmatesoft/hg/test/TestBlame.java
--- a/test/org/tmatesoft/hg/test/TestBlame.java	Thu Feb 21 20:25:26 2013 +0100
+++ b/test/org/tmatesoft/hg/test/TestBlame.java	Thu Feb 21 21:53:55 2013 +0100
@@ -36,6 +36,7 @@
 import org.tmatesoft.hg.internal.IntVector;
 import org.tmatesoft.hg.internal.AnnotateFacility.AddBlock;
 import org.tmatesoft.hg.internal.AnnotateFacility.Block;
+import org.tmatesoft.hg.internal.AnnotateFacility.BlockData;
 import org.tmatesoft.hg.internal.AnnotateFacility.ChangeBlock;
 import org.tmatesoft.hg.internal.AnnotateFacility.DeleteBlock;
 import org.tmatesoft.hg.internal.AnnotateFacility.EqualBlock;
@@ -194,13 +195,17 @@
 	}
 	
 	private void ccc() throws Exception {
-		HgRepository repo = new HgLookup().detect("/home/artem/hg/test-annotate/");
+		HgRepository repo = new HgLookup().detect("/home/artem/hg/junit-test-repos/test-annotate/");
 		HgDataFile df = repo.getFileNode("file1");
 		AnnotateFacility af = new AnnotateFacility();
 		DiffOutInspector dump = new DiffOutInspector(System.out);
 		dump.needRevisions(true);
 		af.annotate(df, TIP, dump, HgIterateDirection.OldToNew);
 		System.out.println();
+		af.annotate(df, TIP, new LineDumpInspector(true), HgIterateDirection.NewToOld);
+		System.out.println();
+		af.annotate(df, TIP, new LineDumpInspector(false), HgIterateDirection.NewToOld);
+		System.out.println();
 		FileAnnotateInspector fa = new FileAnnotateInspector();
 		af.annotate(df, TIP, fa);
 		for (int i = 0; i < fa.lineRevisions.length; i++) {
@@ -216,7 +221,7 @@
 		new TestBlame().ccc();
 	}
 
-	static class DiffOutInspector implements AnnotateFacility.BlockInspector {
+	private static class DiffOutInspector implements AnnotateFacility.BlockInspector {
 		private final PrintStream out;
 		private boolean dumpRevs;
 		private IntVector reportedRevisionPairs = new IntVector();
@@ -225,8 +230,8 @@
 			out = ps;
 		}
 		
+		// Note, true makes output incompatible with 'hg diff'
 		public void needRevisions(boolean dumpRevs) {
-			// Note, true makes output incompatible with 'hg diff'
 			this.dumpRevs = dumpRevs;
 		}
 		
@@ -257,16 +262,9 @@
 		public void deleted(DeleteBlock block) {
 			printRevs(block);
 			out.printf("@@ -%d,%d +%d,0 @@\n", block.firstRemovedLine() + 1, block.totalRemovedLines(), block.removedAt());
-//			String[] lines = block.removedLines();
-//			assert lines.length == block.totalRemovedLines();
-//			for (int i = 0, ln = block.firstRemovedLine(); i < lines.length; i++, ln++) {
-//				linesOld.put(ln, String.format("%3d:---:%s", ln, lines[i]));
-//			}
 		}
 		
 		public void changed(ChangeBlock block) {
-//			deleted(block);
-//			added(block);
 			printRevs(block);
 			out.printf("@@ -%d,%d +%d,%d @@\n", block.firstRemovedLine() + 1, block.totalRemovedLines(), block.firstAddedLine() + 1, block.totalAddedLines());
 		}
@@ -274,11 +272,6 @@
 		public void added(AddBlock block) {
 			printRevs(block);
 			out.printf("@@ -%d,0 +%d,%d @@\n", block.insertedAt(), block.firstAddedLine() + 1, block.totalAddedLines());
-//			String[] addedLines = block.addedLines();
-//			assert addedLines.length == block.totalAddedLines();
-//			for (int i = 0, ln = block.firstAddedLine(), x = addedLines.length; i < x; i++, ln++) {
-//				linesNew.put(ln, String.format("%3d:+++:%s", ln, addedLines[i]));
-//			}
 		}
 	}
 	
@@ -331,4 +324,44 @@
 		}
 	}
 
+	private static class LineDumpInspector implements AnnotateFacility.BlockInspector {
+		
+		private final boolean lineByLine;
+
+		public LineDumpInspector(boolean lineByLine) {
+			this.lineByLine = lineByLine;
+		}
+
+		public void same(EqualBlock block) {
+		}
+
+		public void added(AddBlock block) {
+			BlockData lines = block.addedLines();
+			printBlock(lines, block.targetChangesetIndex(), block.firstAddedLine(), block.totalAddedLines(), "+++");
+		}
+
+		public void changed(ChangeBlock block) {
+			deleted(block);
+			added(block);
+		}
+
+		public void deleted(DeleteBlock block) {
+			BlockData lines = block.removedLines();
+			assert lines.elementCount() == block.totalRemovedLines();
+			printBlock(lines, block.originChangesetIndex(), block.firstRemovedLine(), block.totalRemovedLines(), "---");
+		}
+		
+		private void printBlock(BlockData lines, int cset, int first, int length, String marker) {
+			assert lines.elementCount() == length;
+			if (lineByLine) {
+				for (int i = 0, ln = first; i < length; i++, ln++) {
+					String line = new String(lines.elementAt(i).asArray());
+					System.out.printf("%3d:%3d:%s:%s", cset, ln, marker, line);
+				}
+			} else {
+				String content = new String(lines.asArray());
+				System.out.printf("%3d:%s:[%d..%d):\n%s", cset, marker, first, first+length, content);
+			}
+		}
+	}
 }