/*
* Copyright 2013 Alex Kasko (alexkasko.com)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alexkasko.unsafe.offheapstruct;
import com.alexkasko.unsafe.bytearray.ByteArrayTool;
import com.alexkasko.unsafe.offheap.OffHeapUtils;
import org.junit.Test;
import java.util.Arrays;
import java.util.Random;
import static junit.framework.Assert.assertEquals;
import static junit.framework.Assert.assertTrue;
/**
* User: alexkasko
* Date: 3/5/13
*/
public class OffHeapStructBinarySearchTest {
// private static final int LENGTH = 1000000;
private static final int LENGTH = 10000;
@Test
public void testLong() {
ByteArrayTool bat = ByteArrayTool.get();
byte[] buf = new byte[8];
OffHeapStructArray oha = null;
try {
oha = new OffHeapStructArray(LENGTH, 8);
Random random = new Random(42);
long[] arr = new long[LENGTH];
long val4242 = -1;
long val4243 = -1;
for (int i = 0; i < LENGTH; i++) {
long ra = random.nextLong();
arr[i] = ra;
bat.putLong(buf, 0, ra);
oha.set(i, buf);
if (4242 == i) val4242 = ra;
if (4243 == i) val4243 = ra;
}
Arrays.sort(arr);
OffHeapStructSorter.sortByLongKey(oha, 0);
long ind4242e = Arrays.binarySearch(arr, val4242);
long ind4242a = OffHeapStructBinarySearch.binarySearchByLongKey(oha, val4242, 0);
long ind4243e = Arrays.binarySearch(arr, val4243);
long ind4243a = OffHeapStructBinarySearch.binarySearchByLongKey(oha, val4243, 0);
assertEquals(ind4242e, ind4242a);
assertEquals(ind4243e, ind4243a);
} finally {
OffHeapUtils.free(oha);
}
}
@Test
public void testLongRange() {
ByteArrayTool bat = ByteArrayTool.get();
byte[] buf = new byte[8];
OffHeapStructArray oha = null;
try {
oha = new OffHeapStructArray(6, 8);
bat.putLong(buf, 0, 41);
oha.set(0, buf);
bat.putLong(buf, 0, 41);
oha.set(1, buf);
bat.putLong(buf, 0, 42);
oha.set(2, buf);
bat.putLong(buf, 0, 42);
oha.set(3, buf);
bat.putLong(buf, 0, 42);
oha.set(4, buf);
bat.putLong(buf, 0, 43);
oha.set(5, buf);
OffHeapStructBinarySearch.IndexRange range41 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByLongKey(oha, 41, 0, range41);
assertTrue(range41.isNotEmpty());
assertEquals("Start fail", 0, range41.getFromIndex());
assertEquals("Start fail", 1, range41.getToIndex());
OffHeapStructBinarySearch.IndexRange range42 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByLongKey(oha, 42, 0, range42);
assertTrue(range42.isNotEmpty());
assertEquals("Middle fail", 2, range42.getFromIndex());
assertEquals("Middle fail", 4, range42.getToIndex());
OffHeapStructBinarySearch.IndexRange range43 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByLongKey(oha, 43, 0, range43);
assertTrue(range43.isNotEmpty());
assertEquals("End fail", 5, range43.getFromIndex());
assertEquals("End fail", 5, range43.getToIndex());
OffHeapStructBinarySearch.IndexRange range44 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByLongKey(oha, 44, 0, range44);
assertTrue(range44.isEmpty());
assertTrue("Empty fail", range44.getFromIndex() == range44.getToIndex());
assertTrue("Empty fail", range44.getFromIndex() < 0);
assertTrue("Empty fail", range44.getToIndex() < 0);
} finally {
OffHeapUtils.free(oha);
}
}
@Test
public void testInt() {
ByteArrayTool bat = ByteArrayTool.get();
byte[] buf = new byte[8];
OffHeapStructArray oha = null;
try {
oha = new OffHeapStructArray(LENGTH, 8);
Random random = new Random(42);
int[] arr = new int[LENGTH];
int val4242 = -1;
int val4243 = -1;
for (int i = 0; i < LENGTH; i++) {
int ra = random.nextInt();
arr[i] = ra;
bat.putLong(buf, 0, ra);
oha.set(i, buf);
if (4242 == i) val4242 = ra;
if (4243 == i) val4243 = ra;
}
Arrays.sort(arr);
OffHeapStructSorter.sortByIntKey(oha, 0);
long ind4242e = Arrays.binarySearch(arr, val4242);
long ind4242a = OffHeapStructBinarySearch.binarySearchByIntKey(oha, val4242, 0);
long ind4243e = Arrays.binarySearch(arr, val4243);
long ind4243a = OffHeapStructBinarySearch.binarySearchByIntKey(oha, val4243, 0);
assertEquals(ind4242e, ind4242a);
assertEquals(ind4243e, ind4243a);
} finally {
OffHeapUtils.free(oha);
}
}
@Test
public void testIntRange() {
ByteArrayTool bat = ByteArrayTool.get();
byte[] buf = new byte[8];
OffHeapStructArray oha = null;
try {
oha = new OffHeapStructArray(6, 8);
bat.putLong(buf, 0, 41);
oha.set(0, buf);
bat.putLong(buf, 0, 41);
oha.set(1, buf);
bat.putLong(buf, 0, 42);
oha.set(2, buf);
bat.putLong(buf, 0, 42);
oha.set(3, buf);
bat.putLong(buf, 0, 42);
oha.set(4, buf);
bat.putLong(buf, 0, 43);
oha.set(5, buf);
OffHeapStructBinarySearch.IndexRange range41 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByIntKey(oha, 41, 0, range41);
assertTrue(range41.isNotEmpty());
assertEquals("Start fail", 0, range41.getFromIndex());
assertEquals("Start fail", 1, range41.getToIndex());
OffHeapStructBinarySearch.IndexRange range42 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByIntKey(oha, 42, 0, range42);
assertTrue(range42.isNotEmpty());
assertEquals("Middle fail", 2, range42.getFromIndex());
assertEquals("Middle fail", 4, range42.getToIndex());
OffHeapStructBinarySearch.IndexRange range43 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByIntKey(oha, 43, 0, range43);
assertTrue(range43.isNotEmpty());
assertEquals("End fail", 5, range43.getFromIndex());
assertEquals("End fail", 5, range43.getToIndex());
OffHeapStructBinarySearch.IndexRange range44 = new OffHeapStructBinarySearch.IndexRange();
OffHeapStructBinarySearch.binarySearchRangeByIntKey(oha, 44, 0, range44);
assertTrue(range44.isEmpty());
assertTrue("Empty fail", range44.getFromIndex() == range44.getToIndex());
assertTrue("Empty fail", range44.getFromIndex() < 0);
assertTrue("Empty fail", range44.getToIndex() < 0);
} finally {
OffHeapUtils.free(oha);
}
}
}