ArcFlags.java

package edu.kit.pse.ws2013.routekit.models;

import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.ByteOrder;
import java.nio.MappedByteBuffer;
import java.nio.channels.FileChannel;
import java.nio.channels.FileChannel.MapMode;
import java.util.Arrays;

/**
 * Stores Arc-Flags for a precalculated graph.
 */
public class ArcFlags {

	/**
	 * The Arc-Flags of the graph (as int bit vectors). The initial partition
	 * (0<sup>th</sup> partition) corresponds to the least-significant bit.
	 */
	private int[] flags;

	/**
	 * Creates a new {@link ArcFlags} object with the given Arc-Flags.
	 * 
	 * @param flags
	 *            The Arc-Flags.
	 */
	public ArcFlags(int[] flags) {
		this.flags = flags;
	}

	/**
	 * Returns the Arc-Flags of the given turn. The initial partition
	 * (0<sup>th</sup> partition) corresponds to the least-significant bit.
	 * 
	 * @param turn
	 *            The ID of the turn.
	 * @return The Arc-Flags for the turn.
	 */
	public int getFlag(int turn) {
		return flags[turn];
	}

	public void save(File f) throws IOException {
		try (RandomAccessFile raf = new RandomAccessFile(f, "rw");
				FileChannel fc = raf.getChannel()) {
			MappedByteBuffer mbb = fc.map(MapMode.READ_WRITE, 0,
					(1 + flags.length) * 4);
			mbb.order(ByteOrder.BIG_ENDIAN).asIntBuffer().put(flags.length)
					.put(flags);
			mbb.force();
		}
	}

	public static ArcFlags load(File f) throws IOException {
		try (FileInputStream fis = new FileInputStream(f);
				DataInputStream dis = new DataInputStream(fis);
				FileChannel fc = fis.getChannel()) {
			int length = dis.readInt();
			MappedByteBuffer mbb = fc.map(MapMode.READ_ONLY, 4, length * 4);
			int[] flags = new int[length];
			mbb.order(ByteOrder.BIG_ENDIAN).asIntBuffer().get(flags);
			return new ArcFlags(flags);
		}
	}

	@Override
	public boolean equals(Object obj) {
		if (obj instanceof ArcFlags) {
			return Arrays.equals(flags, ((ArcFlags) obj).flags);
		}
		return false;
	}

	@Override
	public int hashCode() {
		return flags[0];
	}
}