SliceWriter.java

Go to the documentation of this file.

package theba.core.io;

import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.RandomAccessFile;

public class SliceWriter extends SliceReader {

        public SliceWriter(final String filename, int width, int height, int depth,
                        int cacheSize) {
                super(width, height, depth, cacheSize);
                this.filename = filename;
                file = null;
                try {
                        file = new RandomAccessFile(filename, "rw");
                } catch (FileNotFoundException e) {
                        e.printStackTrace();
                }
        }

        @Override
        public synchronized void putSlice(short[] data, int pos) {
                short[] cached = getSlice(pos);
                System.arraycopy(data, 0, cached, 0, cached.length);
        }

        public static void main(String[] args) {
                short[] data = new short[10 * 10];
                data[0] = 0;
                data[1] = 255;
                data[2] = 512;
                System.out.println(data[0]);
                System.out.println(data[1]);
                System.out.println(data[2]);
                SliceWriter sw = new SliceWriter("test", 10, 10, 5, 10);
                sw.putSlice(data, 0);
                sw.putSlice(data, 4);
                sw.flush();

                try {
                        sw.getFile().close();
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }

                SliceReader r = new SliceReader("test", 10, 10, 1, 10);
                data = r.getSlice(0);
                System.out.println(data[0]);
                System.out.println(data[1]);
                System.out.println(data[2]);
                System.out.println(data[0]);
                System.out.println(data[1]);
                System.out.println(data[2]);
        }

        @Override
        public synchronized void updateCache() {
                while (queue.size() > cacheSize) {
                        int deletedSlice = queue.removeFirst();
                        short[] sliceCache = cache.remove(deletedSlice);
                        if (sliceCache != null) {
                                try {
                                        file.seek(sliceCache.length * 2 * deletedSlice);
                                        byte[] out = DataConversion.shortToBytes(sliceCache);
                                        file.write(out, 0, out.length);
                                } catch (Exception e) {
                                        e.printStackTrace();
                                }
                                preallocatedDataArrays.add(sliceCache);
                        } else
                                System.err.println("UpdateCache: this slce was not cached "
                                                + deletedSlice);
                }
        }

        @Override
        public synchronized void flush() {
                while (queue.size() > 0) {
                        int deletedSlice = queue.removeFirst();
                        short[] sliceCache = cache.remove(deletedSlice);
                        if (sliceCache != null) {
                                try {
                                        file.seek(sliceCache.length * 2 * deletedSlice);
                                        byte[] out = DataConversion.shortToBytes(sliceCache);
                                        file.write(out, 0, out.length);
                                } catch (Exception e) {
                                        e.printStackTrace();
                                }
                                preallocatedDataArrays.add(sliceCache);
                        } else
                                System.err.println("Flush : this slice was not cached: "
                                                + deletedSlice);
                }
        }

        @Override
        public synchronized void close() {
                try {
                        file.close();
                } catch (IOException e) {
                        e.printStackTrace();
                }
        }
}