001    /*
002     * Copyright (c) Andrey Kuznetsov. All Rights Reserved.
003     *
004     * http://uio.imagero.com
005     *
006     * Redistribution and use in source and binary forms, with or without
007     * modification, are permitted provided that the following conditions are met:
008     *
009     *  o Redistributions of source code must retain the above copyright notice,
010     *    this list of conditions and the following disclaimer.
011     *
012     *  o Redistributions in binary form must reproduce the above copyright notice,
013     *    this list of conditions and the following disclaimer in the documentation
014     *    and/or other materials provided with the distribution.
015     *
016     *  o Neither the name of Andrey Kuznetsov nor the names of
017     *    its contributors may be used to endorse or promote products derived
018     *    from this software without specific prior written permission.
019     *
020     * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
021     * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
022     * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
023     * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
024     * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
025     * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
026     * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
027     * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
028     * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
029     * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE,
030     * EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
031     */
032    package com.imagero.uio.bio;
033    
034    import java.io.OutputStream;
035    import java.io.IOException;
036    
037    /**
038     * OutputStream with shared IOController.
039     *
040     * @author Andrey Kuznetsov
041     */
042    public class IOCOutputStream extends OutputStream {
043    
044        FixedSizeByteBuffer buffer;
045        BufferIndex bufferIndex;
046        BufferPosition bufferPosition;
047        IOController controller;
048    
049        StreamPosition streamPosition = new StreamPosition();
050        long offset;
051    
052        long mark;
053    
054        public IOCOutputStream(IOController controller) {
055            this.controller = controller;
056        }
057    
058        public IOCOutputStream(IOController controller, long offset) {
059            this.controller = controller;
060            this.offset = offset;
061            bufferPosition = new BufferPosition(controller.bufferSize);
062            seek(0);
063        }
064    
065        public void seek(long offset) {
066            streamPosition.pos = offset + this.offset;
067        }
068    
069        protected void prepareBufferForWriting() throws IOException {
070            BufferIndex index = controller.getBufferIndex(streamPosition.pos);
071    
072            if (!index.equals(bufferIndex) || buffer == null || buffer.buf == null) {
073                bufferIndex = index;
074                buffer = controller.getBuffer(streamPosition.pos, false);
075            }
076            bufferPosition.pos = (int) ((streamPosition.pos) % controller.bufferSize);
077            buffer.changed = true;
078        }
079    
080        private void checkBuffer() throws IOException {
081            if (buffer == null || !(bufferPosition.available() > 0)) {
082                prepareBufferForWriting();
083            }
084        }
085    
086        public void write(int b) throws IOException {
087            checkBuffer();
088            buffer.write(b, bufferPosition);
089            streamPosition.pos++;
090        }
091    
092        public void write(byte b[]) throws IOException {
093            write(b, 0, b.length);
094        }
095    
096        public void write(byte b[], int offset, int length) throws IOException {
097            while (length > 0) {
098                checkBuffer();
099                int written = buffer.write(b, offset, length, bufferPosition);
100                length -= written;
101                offset += written;
102                streamPosition.pos += written;
103            }
104        }
105    
106        public void flush() throws IOException {
107            if (controller != null) {
108                controller.sync();
109            }
110        }
111    
112        public void close() throws IOException {
113            flush();
114            controller = null;
115        }
116    }