Subversion Repositories mkgmap

/*
 * Copyright (C) 2007 Steve Ratcliffe
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License version 2 as
 *  published by the Free Software Foundation.
 *
 *  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.  See the
 *  GNU General Public License for more details.
 *
 *
 * Author: Steve Ratcliffe
 * Create date: Dec 14, 2007
 */
package uk.me.parabola.imgfmt.sys;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;

import uk.me.parabola.imgfmt.ReadFailedException;
import uk.me.parabola.imgfmt.fs.ImgChannel;

/**
 * This is an implementation of ImgChannel that delegates to a regular channel.
 * It can therefore be used to read or write regular files on the file system.
 *
 * @author Steve Ratcliffe
 */
public class FileImgChannel implements ImgChannel {
        private final FileChannel channel;
        private long position;

        public FileImgChannel(String filename, String mode) {
                RandomAccessFile raf;
                try {
                        raf = new RandomAccessFile(filename, mode);
                } catch (FileNotFoundException e) {
                        throw new ReadFailedException("Could not open " + filename, e);
                }
                this.channel = raf.getChannel();
        }

        public FileImgChannel(FileChannel channel) {
                this.channel = channel;
        }

        public int read(ByteBuffer dst) throws IOException {
                int n = channel.read(dst);
                if (n > 0)
                        position += n;
                return n;
        }

        public boolean isOpen() {
                return channel.isOpen();
        }

        public void close() throws IOException {
                channel.close();
        }

        public int write(ByteBuffer src) throws IOException {
                int n = channel.write(src);
                position += n;
                return n;
        }

        /**
         * Get the file position. Note that this is a logical position relative to the
         * beginning of the file (the file within the .img file, not the beginning of the
         * .img file itself).
         *
         * @return The offset in bytes from the beginning of the file.
         */
        public long position() {
                return position;
        }

        /**
         * Set the position within the file.
         *
         * @param pos The position to set.
         */
        public void position(long pos) {
                try {
                        channel.position(pos);
                        position = pos;
                } catch (IOException e) {
                        throw new ReadFailedException("Could not seek", e);
                }
        }
}