lipsync/Boare.Lib.Media/WaveWriter.cs

/*
 * WaveWriter.cs
 * Copyright (c) 2009 kbinani
 *
 * This file is part of Boare.Lib.Media.
 *
 * Boare.Lib.Media is free software; you can redistribute it and/or
 * modify it under the terms of the BSD License.
 *
 * Boare.Lib.Media 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.
 */
#if JAVA
package org.kbinani.media;

import java.io.*;
import org.kbinani.*;
#else
using System;
using bocoree;
using bocoree.java.io;

namespace Boare.Lib.Media {
#endif

#if JAVA
    public class WaveWriter{
#else
    public class WaveWriter : IDisposable{
#endif
        private int m_channel = 1;
        private int m_bit_per_sample;
        private int m_sample_rate;
        private int m_total_samples = 0;
        private RandomAccessFile m_stream = null;
        private String m_path = "";

        public WaveWriter( String path ) 
#if JAVA

            throws IOException
        {
            this ( path, 2, 16, 44100 );
#else
            : this( path, 2, 16, 44100 ) {
#endif
        }

        public WaveWriter( String path, int channel, int bit_per_sample, int sample_rate ) 
#if JAVA
            throws IOException
#endif
        {
            m_path = path;
            m_stream = new RandomAccessFile( m_path, "rw" );
            m_channel = channel;
            m_bit_per_sample = bit_per_sample;
            m_sample_rate = sample_rate;
            m_total_samples = 0;
            writeHeader();
        }

#if !JAVA
        public void Dispose() {
            close();
        }
#endif

        /// <summary>
        /// Writes header of WAVE file
        /// </summary>
        private void writeHeader()
#if JAVA
            throws IOException
#endif
        {
            // RIFF
            m_stream.writeByte( 0x52 ); // loc=0x00
            m_stream.writeByte( 0x49 );
            m_stream.writeByte( 0x46 );
            m_stream.writeByte( 0x46 );

            // ファイルサイズ - 8最後に記入
            m_stream.writeByte( 0x00 ); // loc=0x04
            m_stream.writeByte( 0x00 );
            m_stream.writeByte( 0x00 );
            m_stream.writeByte( 0x00 );

            // WAVE
            m_stream.writeByte( 0x57 ); // loc=0x08
            m_stream.writeByte( 0x41 );
            m_stream.writeByte( 0x56 );
            m_stream.writeByte( 0x45 );

            // fmt 
            m_stream.writeByte( 0x66 ); // loc=0x0c
            m_stream.writeByte( 0x6d );
            m_stream.writeByte( 0x74 );
            m_stream.writeByte( 0x20 );

            // fmt チャンクのサイズ
            m_stream.writeByte( 0x12 ); // loc=0x10
            m_stream.writeByte( 0x00 );
            m_stream.writeByte( 0x00 );
            m_stream.writeByte( 0x00 );

            // format ID
            m_stream.writeByte( 0x01 ); // loc=0x14
            m_stream.writeByte( 0x00 );

            // チャンネル数
            if ( m_channel == 1 ) {
                m_stream.writeByte( 0x01 ); // loc=0x16
                m_stream.writeByte( 0x00 );
            } else {
                m_stream.writeByte( 0x02 ); //loc=0x16
                m_stream.writeByte( 0x00 );
            }

            // サンプリングレート
            byte[] buf = PortUtil.getbytes_uint32_le( m_sample_rate );
            writeByteArray( m_stream, buf, 4 ); // 0x18

            // データ速度
            int block_size = (int)(m_bit_per_sample / 8 * (int)m_channel);
            int data_rate = m_sample_rate * block_size;
            buf = PortUtil.getbytes_uint32_le( data_rate );
            writeByteArray( m_stream, buf, 4 );//loc=0x1c

            // ブロックサイズ
            buf = PortUtil.getbytes_uint16_le( block_size );
            writeByteArray( m_stream, buf, 2 ); //0x20

            // サンプルあたりのビット数
            buf = PortUtil.getbytes_uint16_le( m_bit_per_sample );
            writeByteArray( m_stream, buf, 2 ); //loc=0x22

            // 拡張部分
            m_stream.writeByte( 0x00 ); //loc=0x24
            m_stream.writeByte( 0x00 );

            // data
            m_stream.writeByte( 0x64 ); //loc=0x26
            m_stream.writeByte( 0x61 );
            m_stream.writeByte( 0x74 );
            m_stream.writeByte( 0x61 );

            // size of data chunk
            int size = block_size * m_total_samples;
            buf = PortUtil.getbytes_uint32_le( size );
            writeByteArray( m_stream, buf, 4 );
        }

        public void close()
#if JAVA
            throws IOException
#endif
        {
            if ( m_stream != null ) {
                // 最後にWAVEチャンクのサイズ
                int position = (int)m_stream.getFilePointer();
                m_stream.seek( 4 );
                byte[] buf = PortUtil.getbytes_uint32_le( position - 8 );
                writeByteArray( m_stream, buf, 4 );

                // size of data chunk
                int block_size = (int)(m_bit_per_sample / 8 * (int)m_channel);
                int size = block_size * m_total_samples;
                m_stream.seek( 42 );
                buf = PortUtil.getbytes_uint32_le( size );
                writeByteArray( m_stream, buf, 4 );

                m_stream.close();
            }
        }

        public int getSampleRate(){
            return m_sample_rate;
        }

        public void append( float[] L )
#if JAVA
            throws IOException
#endif
        {
            int total = L.Length;
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)((L[i] + 1.0f) * 127.5f) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        int b = (int)((L[i] + 1.0f) * 127.5f);
                        m_stream.writeByte( b );
                        m_stream.writeByte( b );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        public void append( double[] L ) 
#if JAVA
            throws IOException
#endif
        {
            int total = L.Length;
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)((L[i] + 1.0) * 127.5) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        int b = (int)((L[i] + 1.0) * 127.5);
                        m_stream.writeByte( b );
                        m_stream.writeByte( b );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768.0) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768.0) );
                        writeByteArray( m_stream, buf, 2 );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        public void append( float[] L, float[] R ) 
#if JAVA
            throws IOException
#endif
        {
            int total = Math.Min( L.Length, R.Length );
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (byte)((L[i] + R[i] + 2.0f) * 63.75f) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (byte)((L[i] + 1.0f) * 127.5f) );
                        m_stream.writeByte( (byte)((R[i] + 1.0f) * 127.5f) );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)((L[i] + R[i]) * 16384f) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                        buf = PortUtil.getbytes_int16_le( (short)(R[i] * 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        public void append( double[] L, double[] R )
#if JAVA
            throws IOException
#endif
        {
            int total = Math.Min( L.Length, R.Length );
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)((L[i] + R[i] + 2.0) * 63.75) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)((L[i] + 1.0) * 127.5) );
                        m_stream.writeByte( (int)((R[i] + 1.0) * 127.5) );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)((L[i] + R[i]) * 16384.0) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 32768.0) );
                        writeByteArray( m_stream, buf, 2 );
                        buf = PortUtil.getbytes_int16_le( (short)(R[i] * 32768.0) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        public void append( byte[] L, byte[] R ) 
#if JAVA
            throws IOException
#endif
        {
            int total = Math.Min( L.Length, R.Length );
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( 0xff & ((L[i] + R[i]) / 2) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( 0xff & L[i] );
                        m_stream.writeByte( 0xff & R[i] );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)((L[i] + R[i]) * 128.5f - 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)(L[i] * 257f - 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                        buf = PortUtil.getbytes_int16_le( (short)(R[i] * 257f - 32768f) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        public void append( short[] L, short[] R ) 
#if JAVA
            throws IOException
#endif
        {
            int total = Math.Min( L.Length, R.Length );
            if ( m_bit_per_sample == 8 ) {
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)(((L[i] + R[i]) / 2f + 32768f) / 255f) );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        m_stream.writeByte( (int)((L[i] + 32768f) / 255f) );
                        m_stream.writeByte( (int)((R[i] + 32768f) / 255f) );
                    }
                }
            } else {
                byte[] buf;
                if ( m_channel == 1 ) {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( (short)((L[i] + R[i]) / 2) );
                        writeByteArray( m_stream, buf, 2 );
                    }
                } else {
                    for ( int i = 0; i < total; i++ ) {
                        buf = PortUtil.getbytes_int16_le( L[i] );
                        writeByteArray( m_stream, buf, 2 );
                        buf = PortUtil.getbytes_int16_le( R[i] );
                        writeByteArray( m_stream, buf, 2 );
                    }
                }
            }
            m_total_samples += (int)total;
        }

        private static void writeByteArray( RandomAccessFile fs, byte[] dat, int limit )
#if JAVA
            throws IOException
#endif
        {
            fs.write( dat, 0, (dat.Length > limit) ? limit : dat.Length );
            if ( dat.Length < limit ) {
                for ( int i = 0; i < limit - dat.Length; i++ ) {
                    fs.writeByte( 0x00 );
                }
            }
        }
    }

#if !JAVA
}
#endif