package org.apache.cassandra.utils;
/*
*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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.
*
*/
import org.junit.Test;
import java.lang.reflect.Constructor;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.io.*;
import static org.junit.Assert.*;
import org.apache.cassandra.db.TypeSizes;
import org.apache.cassandra.io.ISerializer;
import org.apache.cassandra.io.IVersionedSerializer;
import org.apache.cassandra.io.util.DataOutputBuffer;
import org.apache.cassandra.io.util.DataOutputPlus;
public class IntervalTreeTest
{
@Test
public void testSearch() throws Exception
{
List<Interval<Integer, Void>> intervals = new ArrayList<Interval<Integer, Void>>();
intervals.add(Interval.<Integer, Void>create(-300, -200));
intervals.add(Interval.<Integer, Void>create(-3, -2));
intervals.add(Interval.<Integer, Void>create(1,2));
intervals.add(Interval.<Integer, Void>create(3,6));
intervals.add(Interval.<Integer, Void>create(2,4));
intervals.add(Interval.<Integer, Void>create(5,7));
intervals.add(Interval.<Integer, Void>create(1,3));
intervals.add(Interval.<Integer, Void>create(4,6));
intervals.add(Interval.<Integer, Void>create(8,9));
intervals.add(Interval.<Integer, Void>create(15,20));
intervals.add(Interval.<Integer, Void>create(40,50));
intervals.add(Interval.<Integer, Void>create(49,60));
IntervalTree<Integer, Void, Interval<Integer, Void>> it = IntervalTree.build(intervals);
assertEquals(3, it.search(Interval.<Integer, Void>create(4,4)).size());
assertEquals(4, it.search(Interval.<Integer, Void>create(4, 5)).size());
assertEquals(7, it.search(Interval.<Integer, Void>create(-1,10)).size());
assertEquals(0, it.search(Interval.<Integer, Void>create(-1,-1)).size());
assertEquals(5, it.search(Interval.<Integer, Void>create(1,4)).size());
assertEquals(2, it.search(Interval.<Integer, Void>create(0,1)).size());
assertEquals(0, it.search(Interval.<Integer, Void>create(10,12)).size());
List<Interval<Integer, Void>> intervals2 = new ArrayList<Interval<Integer, Void>>();
//stravinsky 1880-1971
intervals2.add(Interval.<Integer, Void>create(1880, 1971));
//Schoenberg
intervals2.add(Interval.<Integer, Void>create(1874, 1951));
//Grieg
intervals2.add(Interval.<Integer, Void>create(1843, 1907));
//Schubert
intervals2.add(Interval.<Integer, Void>create(1779, 1828));
//Mozart
intervals2.add(Interval.<Integer, Void>create(1756, 1828));
//Schuetz
intervals2.add(Interval.<Integer, Void>create(1585, 1672));
IntervalTree<Integer, Void, Interval<Integer, Void>> it2 = IntervalTree.build(intervals2);
assertEquals(0, it2.search(Interval.<Integer, Void>create(1829, 1842)).size());
List<Void> intersection1 = it2.search(Interval.<Integer, Void>create(1907, 1907));
assertEquals(3, intersection1.size());
intersection1 = it2.search(Interval.<Integer, Void>create(1780, 1790));
assertEquals(2, intersection1.size());
}
@Test
public void testIteration()
{
List<Interval<Integer, Void>> intervals = new ArrayList<Interval<Integer, Void>>();
intervals.add(Interval.<Integer, Void>create(-300, -200));
intervals.add(Interval.<Integer, Void>create(-3, -2));
intervals.add(Interval.<Integer, Void>create(1,2));
intervals.add(Interval.<Integer, Void>create(3,6));
intervals.add(Interval.<Integer, Void>create(2,4));
intervals.add(Interval.<Integer, Void>create(5,7));
intervals.add(Interval.<Integer, Void>create(1,3));
intervals.add(Interval.<Integer, Void>create(4,6));
intervals.add(Interval.<Integer, Void>create(8,9));
intervals.add(Interval.<Integer, Void>create(15,20));
intervals.add(Interval.<Integer, Void>create(40,50));
intervals.add(Interval.<Integer, Void>create(49,60));
IntervalTree<Integer, Void, Interval<Integer, Void>> it = IntervalTree.build(intervals);
Collections.sort(intervals, it.minOrdering);
List<Interval<Integer, Void>> l = new ArrayList<Interval<Integer, Void>>();
for (Interval<Integer, Void> i : it)
l.add(i);
assertEquals(intervals, l);
}
@Test
public void testSerialization() throws Exception
{
List<Interval<Integer, String>> intervals = new ArrayList<Interval<Integer, String>>();
intervals.add(Interval.<Integer, String>create(-300, -200, "a"));
intervals.add(Interval.<Integer, String>create(-3, -2, "b"));
intervals.add(Interval.<Integer, String>create(1,2, "c"));
intervals.add(Interval.<Integer, String>create(1,3, "d"));
intervals.add(Interval.<Integer, String>create(2,4, "e"));
intervals.add(Interval.<Integer, String>create(3,6, "f"));
intervals.add(Interval.<Integer, String>create(4,6, "g"));
intervals.add(Interval.<Integer, String>create(5,7, "h"));
intervals.add(Interval.<Integer, String>create(8,9, "i"));
intervals.add(Interval.<Integer, String>create(15,20, "j"));
intervals.add(Interval.<Integer, String>create(40,50, "k"));
intervals.add(Interval.<Integer, String>create(49,60, "l"));
IntervalTree<Integer, String, Interval<Integer, String>> it = IntervalTree.build(intervals);
IVersionedSerializer<IntervalTree<Integer, String, Interval<Integer, String>>> serializer = IntervalTree.serializer(
new ISerializer<Integer>()
{
public void serialize(Integer i, DataOutputPlus out) throws IOException { out.writeInt(i); }
public Integer deserialize(DataInput in) throws IOException { return in.readInt(); }
public long serializedSize(Integer i, TypeSizes s) { return 4; }
},
new ISerializer<String>()
{
public void serialize(String v, DataOutputPlus out) throws IOException { out.writeUTF(v); }
public String deserialize(DataInput in) throws IOException { return in.readUTF(); }
public long serializedSize(String v, TypeSizes s) { return v.length(); }
},
(Constructor<Interval<Integer, String>>) (Object) Interval.class.getConstructor(Object.class, Object.class, Object.class)
);
DataOutputBuffer out = new DataOutputBuffer();
serializer.serialize(it, out, 0);
DataInputStream in = new DataInputStream(new ByteArrayInputStream(out.toByteArray()));
IntervalTree<Integer, String, Interval<Integer, String>> it2 = serializer.deserialize(in, 0);
List<Interval<Integer, String>> intervals2 = new ArrayList<Interval<Integer, String>>();
for (Interval<Integer, String> i : it2)
intervals2.add(i);
assertEquals(intervals, intervals2);
}
}