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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*
* DSI utilities
*
* Copyright (C) 2010-2022 Sebastiano Vigna
*
* This program and the accompanying materials are made available under the
* terms of the GNU Lesser General Public License v2.1 or later,
* which is available at
* http://www.gnu.org/licenses/old-licenses/lgpl-2.1-standalone.html,
* or the Apache Software License 2.0, which is available at
* https://www.apache.org/licenses/LICENSE-2.0.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE.
*
* SPDX-License-Identifier: LGPL-2.1-or-later OR Apache-2.0
*/
package it.unimi.dsi.io;
import static org.junit.Assert.assertEquals;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;
import java.nio.channels.FileChannel.MapMode;
import java.util.Random;
import org.junit.Test;
import it.unimi.dsi.util.SplitMix64Random;
/** Note: this test has little meaning unless you change ByteBufferInputStream.FIRST_SHIFT to 16.
*/
@SuppressWarnings("resource")
public class ByteBufferInputStreamTest {
private static final boolean DEBUG = false;
@Test
public void testStream() throws FileNotFoundException, IOException {
final File f = File.createTempFile(ByteBufferInputStreamTest.class.getName(), "tmp");
f.deleteOnExit();
final int l = 100000;
final long seed = System.currentTimeMillis();
if (DEBUG) System.err.println("Seed: " + seed);
final Random random = new SplitMix64Random(seed);
for(int n = 1; n < 8; n++) {
final FileOutputStream fos = new FileOutputStream(f);
for(int i = 0; i < l * n; i++) fos.write(random.nextInt() & 0xFF);
fos.close();
final FileChannel channel = new FileInputStream(f).getChannel();
ByteBufferInputStream bis = ByteBufferInputStream.map(channel, MapMode.READ_ONLY);
if (n % 2 == 0) bis = bis.copy();
FileInputStream test = new FileInputStream(f);
FileChannel fc = test.getChannel();
int a1, a2, off, len, pos;
final byte b1[] = new byte[32768];
final byte b2[] = new byte[32768];
for(int k = 0; k < l / 10; k++) {
switch (random.nextInt(6)) {
case 0:
if (DEBUG) System.err.println("read()");
a1 = bis.read();
a2 = test.read();
assertEquals(a2, a1);
break;
case 1:
off = random.nextInt(b1.length);
len = random.nextInt(b1.length - off + 1);
a1 = bis.read(b1, off, len);
a2 = test.read(b2, off, len);
if (DEBUG) System.err.println("read(b, " + off + ", " + len + ")");
assertEquals(a2, a1);
for (int i = off; i < off + len; i++)
assertEquals(b2[i], b1[i]);
break;
case 2:
if (DEBUG) System.err.println("available()");
assertEquals(test.available(), bis.available());
break;
case 3:
pos = (int)bis.position();
if (DEBUG) System.err.println("position()=" + pos);
assertEquals((int)fc.position(), pos);
break;
case 4:
pos = random.nextInt(l * n);
bis.position(pos);
if (DEBUG) System.err.println("position(" + pos + ")");
(test = new FileInputStream(f)).skip(pos);
fc = test.getChannel();
break;
case 5:
pos = random.nextInt((int)(l * n - bis.position() + 1));
if (DEBUG) System.err.println("skip(" + pos + ")");
a1 = (int)bis.skip(pos);
a2 = (int)test.skip(pos);
assertEquals(a2, a1);
break;
}
}
test.close();
bis = null;
System.gc(); // Try to get rid of mapped buffers.
channel.close();
}
}
}
|
