BinaryDataWriter.java

/*
 * Class:        BinaryDataWriter
 * Description:  Binary data writer
 * Environment:  Java
 * Software:     SSJ 
 * Copyright (C) 2001  Pierre L'Ecuyer and Universite de Montreal
 * Organization: DIRO, Universite de Montreal
 * @author       David Munger 
 * @since        August 2009
 *
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package umontreal.ssj.util.io;
 
import java.io.*;

public class BinaryDataWriter extends AbstractDataWriter {
   protected DataOutputStream out;

   protected void writeStringData(String s) throws IOException {
      if (s != null) {
         out.writeInt(s.length());
         out.writeBytes(s);
      } else {
         out.writeInt(0);
      }
   }

   protected void writeLabel(String label) throws IOException {
      out.writeByte(TYPECHAR_LABEL);
      writeStringData(label);
   }


   public final static byte TYPECHAR_LABEL = '|';

   public final static byte TYPECHAR_STRING = 'S';

   public final static byte TYPECHAR_INTEGER = 'i';

   public final static byte TYPECHAR_FLOAT = 'f';

   public final static byte TYPECHAR_DOUBLE = 'd';


   public BinaryDataWriter(String filename, boolean append) throws IOException {
      this.out = new DataOutputStream(new FileOutputStream(filename, append));
   }

   public BinaryDataWriter(File file, boolean append) throws IOException {
      this.out = new DataOutputStream(new FileOutputStream(file, append));
   }

   public BinaryDataWriter(String filename) throws IOException {
      this.out = new DataOutputStream(new FileOutputStream(filename));
   }

   public BinaryDataWriter(File file) throws IOException {
      this.out = new DataOutputStream(new FileOutputStream(file));
   }

   public BinaryDataWriter(OutputStream outputStream) throws IOException {
      this.out = new DataOutputStream(outputStream);
   }


   public void write(String label, String s) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_STRING);
      out.writeByte(0);
      writeStringData(s);
   }

   public void write(String label, int a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_INTEGER);
      out.writeByte(0);
      out.writeInt(a);
   }

   public void write(String label, float a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_FLOAT);
      out.writeByte(0);
      out.writeFloat(a);
   }

   public void write(String label, double a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_DOUBLE);
      out.writeByte(0);
      out.writeDouble(a);
   }



   public void write(String label, String[] a, int n) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_STRING);
      out.writeByte(1);
      out.writeInt(n);
      for (int i = 0; i < n; i++)
         writeStringData(a[i]);
   }

   public void write(String label, int[] a, int n) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_INTEGER);
      out.writeByte(1);
      out.writeInt(n);
      for (int i = 0; i < n; i++)
         out.writeInt(a[i]);
   }

   public void write(String label, float[] a, int n) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_FLOAT);
      out.writeByte(1);
      out.writeInt(n);
      for (int i = 0; i < n; i++)
         out.writeFloat(a[i]);
   }

   public void write(String label, double[] a, int n) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_DOUBLE);
      out.writeByte(1);
      out.writeInt(n);
      for (int i = 0; i < n; i++)
         out.writeDouble(a[i]);
   }



   public void write(String label, String[][] a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_STRING);
      out.writeByte(2);
      out.writeInt(a.length);
      out.writeInt(a[0].length);
      for (int i = 0; i < a.length; i++)
         for (int j = 0; j < a[i].length; j++)
            writeStringData(a[i][j]);
   }

   public void write(String label, int[][] a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_INTEGER);
      out.writeByte(2);
      out.writeInt(a.length);
      out.writeInt(a[0].length);
      for (int i = 0; i < a.length; i++)
         for (int j = 0; j < a[i].length; j++)
            out.writeInt(a[i][j]);
   }

   public void write(String label, float[][] a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_FLOAT);
      out.writeByte(2);
      out.writeInt(a.length);
      out.writeInt(a[0].length);
      for (int i = 0; i < a.length; i++)
         for (int j = 0; j < a[i].length; j++)
            out.writeFloat(a[i][j]);
   }

   public void write(String label, double[][] a) throws IOException {
      writeLabel(label);
      out.writeByte(TYPECHAR_DOUBLE);
      out.writeByte(2);
      out.writeInt(a.length);
      out.writeInt(a[0].length);
      for (int i = 0; i < a.length; i++)
         for (int j = 0; j < a[i].length; j++)
            out.writeDouble(a[i][j]);
   }



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