lipsync/Boare.Lib.Media/WaveWriter.cs
2024-05-20 00:17:44 +00:00

415 lines
14 KiB
C#

/*
* 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