1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44 package org.eclipse.jgit.internal.storage.file;
45
46 import java.io.BufferedOutputStream;
47 import java.io.DataOutput;
48 import java.io.IOException;
49 import java.io.OutputStream;
50 import java.security.DigestOutputStream;
51 import java.text.MessageFormat;
52
53 import com.googlecode.javaewah.EWAHCompressedBitmap;
54
55 import org.eclipse.jgit.internal.JGitText;
56 import org.eclipse.jgit.internal.storage.file.PackBitmapIndexBuilder.StoredEntry;
57 import org.eclipse.jgit.lib.Constants;
58 import org.eclipse.jgit.util.io.SafeBufferedOutputStream;
59
60
61
62
63
64
65 public class PackBitmapIndexWriterV1 {
66 private final DigestOutputStream out;
67 private final DataOutput dataOutput;
68
69
70
71
72
73
74
75 public PackBitmapIndexWriterV1(final OutputStream dst) {
76 out = new DigestOutputStream(dst instanceof BufferedOutputStream ? dst
77 : new SafeBufferedOutputStream(dst),
78 Constants.newMessageDigest());
79 dataOutput = new SimpleDataOutput(out);
80 }
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97 public void write(PackBitmapIndexBuilder bitmaps, byte[] packDataChecksum)
98 throws IOException {
99 if (bitmaps == null || packDataChecksum.length != 20)
100 throw new IllegalStateException();
101
102 writeHeader(bitmaps.getOptions(), bitmaps.getBitmapCount(),
103 packDataChecksum);
104 writeBody(bitmaps);
105 writeFooter();
106
107 out.flush();
108 }
109
110 private void writeHeader(
111 int options, int bitmapCount, byte[] packDataChecksum)
112 throws IOException {
113 out.write(PackBitmapIndexV1.MAGIC);
114 dataOutput.writeShort(1);
115 dataOutput.writeShort(options);
116 dataOutput.writeInt(bitmapCount);
117 out.write(packDataChecksum);
118 }
119
120 private void writeBody(PackBitmapIndexBuilder bitmaps) throws IOException {
121 writeBitmap(bitmaps.getCommits());
122 writeBitmap(bitmaps.getTrees());
123 writeBitmap(bitmaps.getBlobs());
124 writeBitmap(bitmaps.getTags());
125 writeBitmaps(bitmaps);
126 }
127
128 private void writeBitmap(EWAHCompressedBitmap bitmap) throws IOException {
129 bitmap.serialize(dataOutput);
130 }
131
132 private void writeBitmaps(PackBitmapIndexBuilder bitmaps)
133 throws IOException {
134 int bitmapCount = 0;
135 for (StoredEntry entry : bitmaps.getCompressedBitmaps()) {
136 writeBitmapEntry(entry);
137 bitmapCount++;
138 }
139
140 int expectedBitmapCount = bitmaps.getBitmapCount();
141 if (expectedBitmapCount != bitmapCount)
142 throw new IOException(MessageFormat.format(
143 JGitText.get().expectedGot,
144 String.valueOf(expectedBitmapCount),
145 String.valueOf(bitmapCount)));
146 }
147
148 private void writeBitmapEntry(StoredEntry entry) throws IOException {
149
150 dataOutput.writeInt((int) entry.getObjectId());
151 out.write(entry.getXorOffset());
152 out.write(entry.getFlags());
153 writeBitmap(entry.getBitmap());
154 }
155
156 private void writeFooter() throws IOException {
157 out.on(false);
158 out.write(out.getMessageDigest().digest());
159 }
160 }