core/org.eclipse.stem.gis/src/org/eclipse/stem/gis/shp/ShxInputStream.java

/*******************************************************************************
 * Copyright (c) 2012 IBM Corporation and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBM Corporation - initial API and implementation
 *******************************************************************************/
package org.eclipse.stem.gis.shp;

import java.io.BufferedInputStream;
import java.io.EOFException;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.io.EndianUtils;

public class ShxInputStream extends ShpInputStream
{
	private static final List<Integer> offsets = new ArrayList<Integer>();
	public ShxInputStream(InputStream is) throws IOException
	{
		super(is);
		readIndexFile();
	}
	
	
	private void readIndexFile() throws IOException
	{
		try {
			while (true) {
				int offset = EndianUtils.swapInteger(readInt()); // record offset in words
				EndianUtils.swapInteger(readInt()); // content length
				
				offsets.add(new Integer(offset));
			}
		} catch (EOFException ee) {
			// no more records
		}
	}
	
	public int getOffsetForRecord(int recordNumber)
	{
		return offsets.get(recordNumber+1);
	}
	
	

	@Override
	public ShpRecord readNextRecord() throws IOException 
	{
		return null;
	}

	@Override
	public ShpHeader getHeader() 
	{
		return super.getHeader();
	}
	
	
	public static void main(String[] args) throws IOException
	{
		ShxInputStream is = new ShxInputStream(new BufferedInputStream(
				new FileInputStream(new File("data/global.shx"))));
		
		System.out.println(is.getHeader());
	}
	
	
}