UFOOutputStream.java

/*
 *  UFO Saved Game Editor
 *  Copyright (C) 2010 Christopher Davoren
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  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.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package net.rubikscomplex.ufosge.util;

import java.io.DataOutputStream;
import java.io.EOFException;
import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;


/**
 *
 * @author Chris Davoren
 */
public class UFOOutputStream extends DataOutputStream {
    public UFOOutputStream(OutputStream out) {
        super(out);
    }

    public void writeUFOByte(int s) throws IOException {
        writeByte((byte)s);
    }
    
    public void writeUFOShort(int s) throws IOException {
        byte lower = (byte)(s & 0xFF);
        byte upper = (byte)((s & 0xFF00) >>> 8);

        writeByte(lower);
        writeByte(upper);
    }

    public void writeUFOInt(int i) throws IOException {
        byte byte1 = (byte)(i & 0xFF);
        byte byte2 = (byte)((i >>> 8) & 0xFF);
        byte byte3 = (byte)((i >>> 16) & 0xFF);
        byte byte4 = (byte)((i >>> 24) & 0xFF);

        writeByte(byte1);
        writeByte(byte2);
        writeByte(byte3);
        writeByte(byte4);
    }

    public void writeUFOString(String str, int maxlen) throws IOException, EOFException {
        byte[] stringBytes;
        int bytesWritten = 0;

        try {
            stringBytes = str.getBytes("US-ASCII");
        }
        catch (UnsupportedEncodingException e) {
            stringBytes = new byte[0];
        }

        while (bytesWritten < stringBytes.length && bytesWritten < maxlen - 1) {
            write(stringBytes[bytesWritten]);
            bytesWritten++;
        }

        // Fill out any leftover space with null characters - there should always be at least one
        for (int i = bytesWritten; i < maxlen; i++) {
            write(0);
        }
    }

}