tikhomirov@13: /*
tikhomirov@74:  * Copyright (c) 2010-2011 TMate Software Ltd
tikhomirov@74:  *  
tikhomirov@74:  * This program is free software; you can redistribute it and/or modify
tikhomirov@74:  * it under the terms of the GNU General Public License as published by
tikhomirov@74:  * the Free Software Foundation; version 2 of the License.
tikhomirov@74:  *
tikhomirov@74:  * This program is distributed in the hope that it will be useful,
tikhomirov@74:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
tikhomirov@74:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
tikhomirov@74:  * GNU General Public License for more details.
tikhomirov@74:  *
tikhomirov@74:  * For information on how to redistribute this software under
tikhomirov@74:  * the terms of a license other than GNU General Public License
tikhomirov@102:  * contact TMate Software at support@hg4j.com
tikhomirov@2:  */
tikhomirov@74: package org.tmatesoft.hg.repo;
tikhomirov@74: 
tikhomirov@218: import static org.tmatesoft.hg.repo.HgRepository.TIP;
tikhomirov@218: 
tikhomirov@232: import java.io.ByteArrayOutputStream;
tikhomirov@157: import java.io.IOException;
tikhomirov@218: import java.util.ArrayList;
tikhomirov@218: import java.util.Arrays;
tikhomirov@326: import java.util.HashMap;
tikhomirov@326: import java.util.Map;
tikhomirov@157: 
tikhomirov@366: import org.tmatesoft.hg.core.HgException;
tikhomirov@354: import org.tmatesoft.hg.core.HgInvalidControlFileException;
tikhomirov@74: import org.tmatesoft.hg.core.Nodeid;
tikhomirov@157: import org.tmatesoft.hg.internal.DataAccess;
tikhomirov@266: import org.tmatesoft.hg.internal.DigestHelper;
tikhomirov@320: import org.tmatesoft.hg.internal.EncodingHelper;
tikhomirov@232: import org.tmatesoft.hg.internal.Experimental;
tikhomirov@312: import org.tmatesoft.hg.internal.IterateControlMediator;
tikhomirov@218: import org.tmatesoft.hg.internal.Lifecycle;
tikhomirov@304: import org.tmatesoft.hg.internal.Pool2;
tikhomirov@77: import org.tmatesoft.hg.internal.RevlogStream;
tikhomirov@312: import org.tmatesoft.hg.util.CancelSupport;
tikhomirov@232: import org.tmatesoft.hg.util.Path;
tikhomirov@312: import org.tmatesoft.hg.util.ProgressSupport;
tikhomirov@74: 
tikhomirov@2: 
tikhomirov@2: /**
tikhomirov@2:  *
tikhomirov@74:  * @author Artem Tikhomirov
tikhomirov@74:  * @author TMate Software Ltd.
tikhomirov@2:  */
tikhomirov@2: public class HgManifest extends Revlog {
tikhomirov@218: 	private RevisionMapper revisionMap;
tikhomirov@285: 	
tikhomirov@285: 	public enum Flags {
tikhomirov@285: 		Exec, Link;
tikhomirov@285: 		
tikhomirov@285: 		static Flags parse(String flags) {
tikhomirov@285: 			if ("x".equalsIgnoreCase(flags)) {
tikhomirov@285: 				return Exec;
tikhomirov@285: 			}
tikhomirov@285: 			if ("l".equalsIgnoreCase(flags)) {
tikhomirov@285: 				return Link;
tikhomirov@285: 			}
tikhomirov@285: 			if (flags == null) {
tikhomirov@285: 				return null;
tikhomirov@285: 			}
tikhomirov@285: 			throw new IllegalStateException(flags);
tikhomirov@285: 		}
tikhomirov@285: 
tikhomirov@285: 		static Flags parse(byte[] data, int start, int length) {
tikhomirov@285: 			if (length == 0) {
tikhomirov@285: 				return null;
tikhomirov@285: 			}
tikhomirov@285: 			if (length == 1) {
tikhomirov@285: 				if (data[start] == 'x') {
tikhomirov@285: 					return Exec;
tikhomirov@285: 				}
tikhomirov@285: 				if (data[start] == 'l') {
tikhomirov@285: 					return Link;
tikhomirov@285: 				}
tikhomirov@285: 				// FALL THROUGH
tikhomirov@285: 			}
tikhomirov@285: 			throw new IllegalStateException(new String(data, start, length));
tikhomirov@285: 		}
tikhomirov@285: 
tikhomirov@285: 		String nativeString() {
tikhomirov@285: 			if (this == Exec) {
tikhomirov@285: 				return "x";
tikhomirov@285: 			}
tikhomirov@285: 			if (this == Link) {
tikhomirov@285: 				return "l";
tikhomirov@285: 			}
tikhomirov@285: 			throw new IllegalStateException(toString());
tikhomirov@285: 		}
tikhomirov@285: 	}
tikhomirov@2: 
tikhomirov@13: 	/*package-local*/ HgManifest(HgRepository hgRepo, RevlogStream content) {
tikhomirov@21: 		super(hgRepo, content);
tikhomirov@13: 	}
tikhomirov@13: 
tikhomirov@218: 	/**
tikhomirov@218: 	 * 
tikhomirov@218: 	 * @param start changelog (not manifest!) revision to begin with
tikhomirov@218: 	 * @param end changelog (not manifest!) revision to stop, inclusive.
tikhomirov@218: 	 * @param inspector can't be <code>null</code>
tikhomirov@218: 	 */
tikhomirov@366: 	public void walk(int start, int end, final Inspector inspector) throws /*FIXME HgInvalidRevisionException,*/ HgInvalidControlFileException {
tikhomirov@196: 		if (inspector == null) {
tikhomirov@196: 			throw new IllegalArgumentException();
tikhomirov@196: 		}
tikhomirov@218: 		int start0 = fromChangelog(start);
tikhomirov@218: 		int end0 = fromChangelog(end);
tikhomirov@300: 		if (end0 < start0) {
tikhomirov@300: 			// there are tool-constructed repositories that got order of changeset revisions completely different from that of manifest
tikhomirov@300: 			int x = end0;
tikhomirov@300: 			end0 = start0;
tikhomirov@300: 			start0 = x;
tikhomirov@300: 		}
tikhomirov@218: 		content.iterate(start0, end0, true, new ManifestParser(inspector));
tikhomirov@19: 	}
tikhomirov@218: 	
tikhomirov@254: 	/**
tikhomirov@254: 	 * "Sparse" iteration of the manifest
tikhomirov@254: 	 * 
tikhomirov@254: 	 * @param inspector
tikhomirov@367: 	 * @param revisionIndexes local indexes of changesets to visit
tikhomirov@254: 	 */
tikhomirov@367: 	public void walk(final Inspector inspector, int... revisionIndexes) throws HgInvalidControlFileException{
tikhomirov@367: 		if (inspector == null || revisionIndexes == null) {
tikhomirov@254: 			throw new IllegalArgumentException();
tikhomirov@254: 		}
tikhomirov@368: 		int[] manifestRevs = toManifestRevisionIndexes(revisionIndexes);
tikhomirov@368: 		content.iterate(manifestRevs, true, new ManifestParser(inspector));
tikhomirov@254: 	}
tikhomirov@254: 	
tikhomirov@232: 	// manifest revision number that corresponds to the given changeset
tikhomirov@368: 	/*package-local*/ int fromChangelog(int changesetRevisionIndex) throws HgInvalidControlFileException {
tikhomirov@368: 		if (HgInternals.wrongRevisionIndex(changesetRevisionIndex)) {
tikhomirov@368: 			throw new IllegalArgumentException(String.valueOf(changesetRevisionIndex));
tikhomirov@218: 		}
tikhomirov@368: 		if (changesetRevisionIndex == HgRepository.WORKING_COPY || changesetRevisionIndex == HgRepository.BAD_REVISION) {
tikhomirov@257: 			throw new IllegalArgumentException("Can't use constants like WORKING_COPY or BAD_REVISION");
tikhomirov@257: 		}
tikhomirov@257: 		// revisionNumber == TIP is processed by RevisionMapper 
tikhomirov@218: 		if (revisionMap == null) {
tikhomirov@218: 			revisionMap = new RevisionMapper(getRepo());
tikhomirov@218: 			content.iterate(0, TIP, false, revisionMap);
tikhomirov@218: 		}
tikhomirov@368: 		return revisionMap.at(changesetRevisionIndex);
tikhomirov@218: 	}
tikhomirov@218: 	
tikhomirov@232: 	/**
tikhomirov@232: 	 * Extracts file revision as it was known at the time of given changeset.
tikhomirov@232: 	 * 
tikhomirov@367: 	 * @param changelogRevisionIndex local changeset index 
tikhomirov@232: 	 * @param file path to file in question
tikhomirov@232: 	 * @return file revision or <code>null</code> if manifest at specified revision doesn't list such file
tikhomirov@232: 	 */
tikhomirov@326: 	@Experimental(reason="Perhaps, HgDataFile shall own this method, or get a delegate?")
tikhomirov@367: 	public Nodeid getFileRevision(int changelogRevisionIndex, final Path file) throws HgInvalidControlFileException{
tikhomirov@367: 		return getFileRevisions(file, changelogRevisionIndex).get(changelogRevisionIndex);
tikhomirov@326: 	}
tikhomirov@326: 	
tikhomirov@326: 	// XXX package-local, IntMap, and HgDataFile getFileRevisionAt(int... localChangelogRevisions)
tikhomirov@326: 	@Experimental(reason="@see #getFileRevision")
tikhomirov@367: 	public Map<Integer, Nodeid> getFileRevisions(final Path file, int... changelogRevisionIndexes) throws HgInvalidControlFileException{
tikhomirov@326: 		// FIXME need tests
tikhomirov@367: 		int[] manifestRevisionIndexes = toManifestRevisionIndexes(changelogRevisionIndexes);
tikhomirov@367: 		final HashMap<Integer,Nodeid> rv = new HashMap<Integer, Nodeid>(changelogRevisionIndexes.length);
tikhomirov@367: 		content.iterate(manifestRevisionIndexes, true, new RevlogStream.Inspector() {
tikhomirov@232: 			
tikhomirov@366: 			public void next(int revisionNumber, int actualLen, int baseRevision, int linkRevision, int parent1Revision, int parent2Revision, byte[] nodeid, DataAccess data) throws HgException {
tikhomirov@232: 				ByteArrayOutputStream bos = new ByteArrayOutputStream();
tikhomirov@232: 				try {
tikhomirov@232: 					byte b;
tikhomirov@232: 					while (!data.isEmpty() && (b = data.readByte()) != '\n') {
tikhomirov@232: 						if (b != 0) {
tikhomirov@232: 							bos.write(b);
tikhomirov@232: 						} else {
tikhomirov@232: 							String fname = new String(bos.toByteArray());
tikhomirov@232: 							bos.reset();
tikhomirov@232: 							if (file.toString().equals(fname)) {
tikhomirov@232: 								byte[] nid = new byte[40];  
tikhomirov@232: 								data.readBytes(nid, 0, 40);
tikhomirov@326: 								rv.put(linkRevision, Nodeid.fromAscii(nid, 0, 40));
tikhomirov@232: 								break;
tikhomirov@326: 							} else {
tikhomirov@326: 								data.skip(40);
tikhomirov@232: 							}
tikhomirov@232: 							// else skip to the end of line
tikhomirov@232: 							while (!data.isEmpty() && (b = data.readByte()) != '\n')
tikhomirov@232: 								;
tikhomirov@232: 						}
tikhomirov@232: 					}
tikhomirov@232: 				} catch (IOException ex) {
tikhomirov@366: 					throw new HgException(ex);
tikhomirov@232: 				}
tikhomirov@232: 			}
tikhomirov@232: 		});
tikhomirov@326: 		return rv;
tikhomirov@232: 	}
tikhomirov@326: 
tikhomirov@326: 
tikhomirov@367: 	private int[] toManifestRevisionIndexes(int[] changelogRevisionIndexes) throws HgInvalidControlFileException {
tikhomirov@368: 		int[] manifestRevs = new int[changelogRevisionIndexes.length];
tikhomirov@326: 		boolean needsSort = false;
tikhomirov@367: 		for (int i = 0; i < changelogRevisionIndexes.length; i++) {
tikhomirov@367: 			final int manifestRevisionIndex = fromChangelog(changelogRevisionIndexes[i]);
tikhomirov@368: 			manifestRevs[i] = manifestRevisionIndex;
tikhomirov@368: 			if (i > 0 && manifestRevs[i-1] > manifestRevisionIndex) {
tikhomirov@326: 				needsSort = true;
tikhomirov@326: 			}
tikhomirov@326: 		}
tikhomirov@326: 		if (needsSort) {
tikhomirov@368: 			Arrays.sort(manifestRevs);
tikhomirov@326: 		}
tikhomirov@368: 		return manifestRevs;
tikhomirov@326: 	}
tikhomirov@326: 
tikhomirov@19: 	public interface Inspector {
tikhomirov@222: 		boolean begin(int mainfestRevision, Nodeid nid, int changelogRevision);
tikhomirov@285: 		/**
tikhomirov@285: 		 * @deprecated switch to {@link Inspector2#next(Nodeid, Path, Flags)}
tikhomirov@285: 		 */
tikhomirov@285: 		@Deprecated
tikhomirov@19: 		boolean next(Nodeid nid, String fname, String flags);
tikhomirov@222: 		boolean end(int manifestRevision);
tikhomirov@2: 	}
tikhomirov@267: 	
tikhomirov@285: 	@Experimental(reason="Explore Path alternative for filenames and enum for flags")
tikhomirov@285: 	public interface Inspector2 extends Inspector {
tikhomirov@285: 		boolean next(Nodeid nid, Path fname, Flags flags);
tikhomirov@267: 	}
tikhomirov@326: 
tikhomirov@268: 	/**
tikhomirov@268: 	 * When Pool uses Strings directly,
tikhomirov@268: 	 * ManifestParser creates new String instance with new char[] value, and does byte->char conversion.
tikhomirov@268: 	 * For cpython repo, walk(0..10k), there are over 16 million filenames, of them only 3020 unique.
tikhomirov@268: 	 * This means there are 15.9 million useless char[] instances and byte->char conversions  
tikhomirov@268: 	 * 
tikhomirov@285: 	 * When String (Path) is wrapped into {@link PathProxy}, there's extra overhead of byte[] representation
tikhomirov@285: 	 * of the String, but these are only for unique Strings (Paths) (3020 in the example above). Besides, I save
tikhomirov@268: 	 * useless char[] and byte->char conversions. 
tikhomirov@268: 	 */
tikhomirov@285: 	private static class PathProxy {
tikhomirov@267: 		private byte[] data;
tikhomirov@268: 		private int start; 
tikhomirov@268: 		private final int hash, length;
tikhomirov@285: 		private Path result;
tikhomirov@267: 
tikhomirov@285: 		public PathProxy(byte[] data, int start, int length) {
tikhomirov@267: 			this.data = data;
tikhomirov@267: 			this.start = start;
tikhomirov@267: 			this.length = length;
tikhomirov@267: 
tikhomirov@268: 			// copy from String.hashCode(). In fact, not necessarily match result of String(data).hashCode
tikhomirov@268: 			// just need some nice algorithm here
tikhomirov@267: 			int h = 0;
tikhomirov@267: 			byte[] d = data;
tikhomirov@267: 			for (int i = 0, off = start, len = length; i < len; i++) {
tikhomirov@267: 				h = 31 * h + d[off++];
tikhomirov@267: 			}
tikhomirov@267: 			hash = h;
tikhomirov@267: 		}
tikhomirov@267: 
tikhomirov@267: 		@Override
tikhomirov@267: 		public boolean equals(Object obj) {
tikhomirov@285: 			if (false == obj instanceof PathProxy) {
tikhomirov@267: 				return false;
tikhomirov@267: 			}
tikhomirov@285: 			PathProxy o = (PathProxy) obj;
tikhomirov@267: 			if (o.result != null && result != null) {
tikhomirov@267: 				return result.equals(o.result);
tikhomirov@267: 			}
tikhomirov@268: 			if (o.length != length || o.hash != hash) {
tikhomirov@267: 				return false;
tikhomirov@267: 			}
tikhomirov@267: 			for (int i = 0, x = o.start, y = start; i < length; i++) {
tikhomirov@267: 				if (o.data[x++] != data[y++]) {
tikhomirov@267: 					return false;
tikhomirov@267: 				}
tikhomirov@267: 			}
tikhomirov@267: 			return true;
tikhomirov@267: 		}
tikhomirov@267: 		@Override
tikhomirov@267: 		public int hashCode() {
tikhomirov@267: 			return hash;
tikhomirov@267: 		}
tikhomirov@267: 		
tikhomirov@285: 		public Path freeze() {
tikhomirov@267: 			if (result == null) {
tikhomirov@320: 				result = Path.create(EncodingHelper.fromManifest(data, start, length));
tikhomirov@268: 				// release reference to bigger data array, make a copy of relevant part only
tikhomirov@285: 				// use original bytes, not those from String above to avoid cache misses due to different encodings 
tikhomirov@268: 				byte[] d = new byte[length];
tikhomirov@268: 				System.arraycopy(data, start, d, 0, length);
tikhomirov@268: 				data = d;
tikhomirov@268: 				start = 0;
tikhomirov@267: 			}
tikhomirov@267: 			return result;
tikhomirov@267: 		}
tikhomirov@267: 	}
tikhomirov@196: 
tikhomirov@312: 	private static class ManifestParser implements RevlogStream.Inspector, Lifecycle {
tikhomirov@196: 		private final Inspector inspector;
tikhomirov@285: 		private final Inspector2 inspector2;
tikhomirov@304: 		private Pool2<Nodeid> nodeidPool, thisRevPool;
tikhomirov@304: 		private final Pool2<PathProxy> fnamePool;
tikhomirov@266: 		private byte[] nodeidLookupBuffer = new byte[20]; // get reassigned each time new Nodeid is added to pool
tikhomirov@312: 		private final ProgressSupport progressHelper;
tikhomirov@312: 		private IterateControlMediator iterateControl;
tikhomirov@262: 		
tikhomirov@196: 		public ManifestParser(Inspector delegate) {
tikhomirov@196: 			assert delegate != null;
tikhomirov@196: 			inspector = delegate;
tikhomirov@285: 			inspector2 = delegate instanceof Inspector2 ? (Inspector2) delegate : null;
tikhomirov@304: 			nodeidPool = new Pool2<Nodeid>();
tikhomirov@304: 			fnamePool = new Pool2<PathProxy>();
tikhomirov@304: 			thisRevPool = new Pool2<Nodeid>();
tikhomirov@312: 			progressHelper = ProgressSupport.Factory.get(delegate);
tikhomirov@196: 		}
tikhomirov@268: 		
tikhomirov@366: 		public void next(int revisionNumber, int actualLen, int baseRevision, int linkRevision, int parent1Revision, int parent2Revision, byte[] nodeid, DataAccess da) throws HgException {
tikhomirov@196: 			try {
tikhomirov@312: 				if (!inspector.begin(revisionNumber, new Nodeid(nodeid, true), linkRevision)) {
tikhomirov@312: 					iterateControl.stop();
tikhomirov@312: 					return;
tikhomirov@312: 				}
tikhomirov@372: 				if (!da.isEmpty()) {
tikhomirov@372: 					// although unlikely, manifest entry may be empty, when all files have been deleted from the repository
tikhomirov@372: 					Path fname = null;
tikhomirov@372: 					Flags flags = null;
tikhomirov@372: 					Nodeid nid = null;
tikhomirov@372: 					int i;
tikhomirov@372: 					byte[] data = da.byteArray();
tikhomirov@372: 					for (i = 0; i < actualLen; i++) {
tikhomirov@372: 						int x = i;
tikhomirov@372: 						for( ; data[i] != '\n' && i < actualLen; i++) {
tikhomirov@372: 							if (fname == null && data[i] == 0) {
tikhomirov@372: 								PathProxy px = fnamePool.unify(new PathProxy(data, x, i - x));
tikhomirov@372: 								// if (cached = fnamePool.unify(px))== px then cacheMiss, else cacheHit
tikhomirov@372: 								// cpython 0..10k: hits: 15 989 152, misses: 3020
tikhomirov@372: 								fname = px.freeze();
tikhomirov@372: 								x = i+1;
tikhomirov@372: 							}
tikhomirov@196: 						}
tikhomirov@372: 						if (i < actualLen) {
tikhomirov@372: 							assert data[i] == '\n'; 
tikhomirov@372: 							int nodeidLen = i - x < 40 ? i-x : 40; // if > 40, there are flags
tikhomirov@372: 							DigestHelper.ascii2bin(data, x, nodeidLen, nodeidLookupBuffer); // ignore return value as it's unlikely to have NULL in manifest
tikhomirov@372: 							nid = new Nodeid(nodeidLookupBuffer, false); // this Nodeid is for pool lookup only, mock object
tikhomirov@372: 							Nodeid cached = nodeidPool.unify(nid);
tikhomirov@372: 							if (cached == nid) {
tikhomirov@372: 								// buffer now belongs to the cached nodeid
tikhomirov@372: 								nodeidLookupBuffer = new byte[20];
tikhomirov@372: 							} else {
tikhomirov@372: 								nid = cached; // use existing version, discard the lookup object
tikhomirov@372: 							} // for cpython 0..10k, cache hits are 15 973 301, vs 18871 misses.
tikhomirov@372: 							thisRevPool.record(nid); // memorize revision for the next iteration. 
tikhomirov@372: 							if (nodeidLen + x < i) {
tikhomirov@372: 								// 'x' and 'l' for executable bits and symlinks?
tikhomirov@372: 								// hg --debug manifest shows 644 for each regular file in my repo
tikhomirov@372: 								// for cpython 0..10k, there are 4361062 flag checks, and there's only 1 unique flag
tikhomirov@372: 								flags = Flags.parse(data, x + nodeidLen, i-x-nodeidLen);
tikhomirov@372: 							} else {
tikhomirov@372: 								flags = null;
tikhomirov@372: 							}
tikhomirov@372: 							boolean good2go;
tikhomirov@372: 							if (inspector2 == null) {
tikhomirov@372: 								String flagString = flags == null ? null : flags.nativeString();
tikhomirov@372: 								good2go = inspector.next(nid, fname.toString(), flagString);
tikhomirov@372: 							} else {
tikhomirov@372: 								good2go = inspector2.next(nid, fname, flags);
tikhomirov@372: 							}
tikhomirov@372: 							if (!good2go) {
tikhomirov@372: 								iterateControl.stop();
tikhomirov@372: 								return;
tikhomirov@372: 							}
tikhomirov@372: 						}
tikhomirov@372: 						nid = null;
tikhomirov@372: 						fname = null;
tikhomirov@372: 						flags = null;
tikhomirov@196: 					}
tikhomirov@196: 				}
tikhomirov@312: 				if (!inspector.end(revisionNumber)) {
tikhomirov@312: 					iterateControl.stop();
tikhomirov@312: 					return;
tikhomirov@312: 				}
tikhomirov@262: 				//
tikhomirov@262: 				// keep only actual file revisions, found at this version 
tikhomirov@262: 				// (next manifest is likely to refer to most of them, although in specific cases 
tikhomirov@262: 				// like commit in another branch a lot may be useless)
tikhomirov@262: 				nodeidPool.clear();
tikhomirov@304: 				Pool2<Nodeid> t = nodeidPool;
tikhomirov@262: 				nodeidPool = thisRevPool;
tikhomirov@264: 				thisRevPool = t;
tikhomirov@313: 				iterateControl.checkCancelled();
tikhomirov@312: 				progressHelper.worked(1);
tikhomirov@196: 			} catch (IOException ex) {
tikhomirov@366: 				throw new HgException(ex);
tikhomirov@196: 			}
tikhomirov@196: 		}
tikhomirov@312: 
tikhomirov@312: 		public void start(int count, Callback callback, Object token) {
tikhomirov@312: 			CancelSupport cs = CancelSupport.Factory.get(inspector, null);
tikhomirov@312: 			iterateControl = new IterateControlMediator(cs, callback);
tikhomirov@312: 			progressHelper.start(count);
tikhomirov@312: 		}
tikhomirov@312: 
tikhomirov@312: 		public void finish(Object token) {
tikhomirov@312: 			progressHelper.done();
tikhomirov@312: 		}
tikhomirov@196: 	}
tikhomirov@218: 	
tikhomirov@218: 	private static class RevisionMapper implements RevlogStream.Inspector, Lifecycle {
tikhomirov@218: 		
tikhomirov@218: 		private final int changelogRevisions;
tikhomirov@218: 		private int[] changelog2manifest;
tikhomirov@218: 		private final HgRepository repo;
tikhomirov@218: 
tikhomirov@218: 		public RevisionMapper(HgRepository hgRepo) {
tikhomirov@218: 			repo = hgRepo;
tikhomirov@218: 			changelogRevisions = repo.getChangelog().getRevisionCount();
tikhomirov@218: 		}
tikhomirov@218: 
tikhomirov@257: 		// respects TIP
tikhomirov@218: 		public int at(int revisionNumber) {
tikhomirov@257: 			if (revisionNumber == TIP) {
tikhomirov@257: 				revisionNumber = changelogRevisions - 1;
tikhomirov@257: 			}
tikhomirov@218: 			if (changelog2manifest != null) {
tikhomirov@218: 				return changelog2manifest[revisionNumber];
tikhomirov@218: 			}
tikhomirov@218: 			return revisionNumber;
tikhomirov@218: 		}
tikhomirov@218: 
tikhomirov@324: 		// XXX likely can be replaced with Revlog.RevisionInspector
tikhomirov@218: 		public void next(int revisionNumber, int actualLen, int baseRevision, int linkRevision, int parent1Revision, int parent2Revision, byte[] nodeid, DataAccess data) {
tikhomirov@218: 			if (changelog2manifest != null) {
tikhomirov@220: 				// next assertion is not an error, rather assumption check, which is too development-related to be explicit exception - 
tikhomirov@220: 				// I just wonder if there are manifests that have two entries pointing to single changeset. It seems unrealistic, though -
tikhomirov@220: 				// changeset records one and only one manifest nodeid
tikhomirov@220: 				assert changelog2manifest[linkRevision] == -1 : String.format("revision:%d, link:%d, already linked to revision:%d", revisionNumber, linkRevision, changelog2manifest[linkRevision]);
tikhomirov@218: 				changelog2manifest[linkRevision] = revisionNumber;
tikhomirov@218: 			} else {
tikhomirov@218: 				if (revisionNumber != linkRevision) {
tikhomirov@218: 					changelog2manifest = new int[changelogRevisions];
tikhomirov@218: 					Arrays.fill(changelog2manifest, -1);
tikhomirov@218: 					for (int i = 0; i < revisionNumber; changelog2manifest[i] = i, i++)
tikhomirov@218: 						;
tikhomirov@218: 					changelog2manifest[linkRevision] = revisionNumber;
tikhomirov@218: 				}
tikhomirov@218: 			}
tikhomirov@218: 		}
tikhomirov@218: 		
tikhomirov@218: 		public void start(int count, Callback callback, Object token) {
tikhomirov@218: 			if (count != changelogRevisions) {
tikhomirov@218: 				assert count < changelogRevisions; // no idea what to do if manifest has more revisions than changelog
tikhomirov@218: 				// the way how manifest may contain more revisions than changelog, as I can imagine, is a result of  
tikhomirov@218: 				// some kind of an import tool (e.g. from SVN or CVS), that creates manifest and changelog independently.
tikhomirov@218: 				// Note, it's pure guess, I didn't see such repository yet (although the way manifest revisions
tikhomirov@218: 				// in cpython repo are numbered makes me think aforementioned way) 
tikhomirov@218: 				changelog2manifest = new int[changelogRevisions];
tikhomirov@218: 				Arrays.fill(changelog2manifest, -1);
tikhomirov@218: 			}
tikhomirov@218: 		}
tikhomirov@218: 
tikhomirov@218: 		public void finish(Object token) {
tikhomirov@218: 			if (changelog2manifest == null) {
tikhomirov@218: 				return;
tikhomirov@218: 			}
tikhomirov@218: 			// I assume there'd be not too many revisions we don't know manifest of
tikhomirov@218: 			ArrayList<Integer> undefinedChangelogRevision = new ArrayList<Integer>();
tikhomirov@218: 			for (int i = 0; i < changelog2manifest.length; i++) {
tikhomirov@218: 				if (changelog2manifest[i] == -1) {
tikhomirov@218: 					undefinedChangelogRevision.add(i);
tikhomirov@218: 				}
tikhomirov@218: 			}
tikhomirov@218: 			for (int u : undefinedChangelogRevision) {
tikhomirov@354: 				try {
tikhomirov@366: 					Nodeid manifest = repo.getChangelog().range(u, u).get(0).manifest();
tikhomirov@366: 					// FIXME calculate those missing effectively (e.g. cache and sort nodeids to speed lookup
tikhomirov@366: 					// right away in the #next (may refactor ParentWalker's sequential and sorted into dedicated helper and reuse here)
tikhomirov@385: 					if (manifest.isNull()) {
tikhomirov@385: 						repo.getContext().getLog().warn(getClass(), "Changeset %d has no associated manifest entry", u);
tikhomirov@385: 						// keep -1 in the changelog2manifest map. FIXME rest of the code shall accomodate to the fact manifest revision may be missing
tikhomirov@385: 					} else {
tikhomirov@385: 						changelog2manifest[u] = repo.getManifest().getRevisionIndex(manifest);
tikhomirov@385: 					}
tikhomirov@354: 				} catch (HgInvalidControlFileException ex) {
tikhomirov@354: 					// FIXME need to propagate the error up to client  
tikhomirov@354: 					repo.getContext().getLog().error(getClass(), ex, null);
tikhomirov@354: 				}
tikhomirov@218: 			}
tikhomirov@218: 		}
tikhomirov@218: 	}
tikhomirov@2: }