| /* |
| * Copyright (C) 2012 The Android Open Source Project |
| * |
| * 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 org.drrickorang.loopback; |
| |
| import java.io.FileDescriptor; |
| import java.io.FileOutputStream; |
| import java.io.IOException; |
| import java.util.Arrays; |
| |
| import android.content.Context; |
| import android.net.Uri; |
| import android.os.ParcelFileDescriptor; |
| import android.util.Log; |
| |
| |
| /** |
| * This class is used to save the results to a .wav file. |
| */ |
| |
| public class AudioFileOutput { |
| private static final String TAG = "AudioFileOutput"; |
| |
| private Uri mUri; |
| private Context mContext; |
| private FileOutputStream mOutputStream; |
| private final int mSamplingRate; |
| |
| |
| public AudioFileOutput(Context context, Uri uri, int samplingRate) { |
| mContext = context; |
| mUri = uri; |
| mSamplingRate = samplingRate; |
| } |
| |
| |
| public boolean writeData(double[] data) { |
| return writeRingBufferData(data, 0, data.length); |
| } |
| |
| /** |
| * Writes recorded wav data to file |
| * endIndex <= startIndex: Writes [startIndex, data.length) then [0, endIndex) |
| * endIndex > startIndex : Writes [startIndex, endIndex) |
| * Returns true on successful write to file |
| */ |
| public boolean writeRingBufferData(double[] data, int startIndex, int endIndex) { |
| |
| boolean status = false; |
| ParcelFileDescriptor parcelFileDescriptor = null; |
| try { |
| parcelFileDescriptor = |
| mContext.getContentResolver().openFileDescriptor(mUri, "w"); |
| FileDescriptor fileDescriptor = parcelFileDescriptor.getFileDescriptor(); |
| mOutputStream = new FileOutputStream(fileDescriptor); |
| log("Done creating output stream"); |
| int sampleCount = endIndex - startIndex; |
| if (sampleCount <= 0) { |
| sampleCount += data.length; |
| } |
| writeHeader(sampleCount); |
| |
| if (endIndex > startIndex) { |
| writeDataBuffer(data, startIndex, endIndex); |
| } else { |
| writeDataBuffer(data, startIndex, data.length); |
| writeDataBuffer(data, 0, endIndex); |
| } |
| |
| mOutputStream.close(); |
| status = true; |
| parcelFileDescriptor.close(); |
| } catch (Exception e) { |
| mOutputStream = null; |
| log("Failed to open wavefile" + e); |
| } finally { |
| try { |
| if (parcelFileDescriptor != null) { |
| parcelFileDescriptor.close(); |
| } |
| } catch (Exception e) { |
| e.printStackTrace(); |
| log("Error closing ParcelFile Descriptor"); |
| } |
| } |
| return status; |
| } |
| |
| private void writeHeader(int samples) { |
| if (mOutputStream != null) { |
| try { |
| int channels = 1; |
| int blockAlignment = 2; |
| int bitsPerSample = 16; |
| byte[] chunkSize = new byte[4]; |
| byte[] dataSize = new byte[4]; |
| int tempChunkSize = (samples * 2) + 36; |
| chunkSize[3] = (byte) (tempChunkSize >> 24); |
| chunkSize[2] = (byte) (tempChunkSize >> 16); |
| chunkSize[1] = (byte) (tempChunkSize >> 8); |
| chunkSize[0] = (byte) tempChunkSize; |
| int tempDataSize = samples * 2; |
| dataSize[3] = (byte) (tempDataSize >> 24); |
| dataSize[2] = (byte) (tempDataSize >> 16); |
| dataSize[1] = (byte) (tempDataSize >> 8); |
| dataSize[0] = (byte) tempDataSize; |
| |
| byte[] header = new byte[] { |
| 'R', 'I', 'F', 'F', |
| chunkSize[0], chunkSize[1], chunkSize[2], chunkSize[3], |
| 'W', 'A', 'V', 'E', |
| 'f', 'm', 't', ' ', |
| 16, 0, 0, 0, |
| 1, 0, // PCM |
| (byte) channels, 0, // number of channels |
| (byte) mSamplingRate, (byte) (mSamplingRate >> 8), 0, 0, // sample rate |
| 0, 0, 0, 0, // byte rate |
| (byte) (channels * blockAlignment), |
| 0, // block alignment |
| (byte) bitsPerSample, |
| 0, // bits per sample |
| 'd', 'a', 't', 'a', |
| dataSize[0], dataSize[1], dataSize[2], dataSize[3], |
| }; |
| mOutputStream.write(header); |
| log("Done writing header"); |
| } catch (IOException e) { |
| Log.e(TAG, "Error writing header " + e); |
| } |
| } |
| } |
| |
| |
| private void writeDataBuffer(double [] data, int startIndex, int end) { |
| if (mOutputStream != null) { |
| try { |
| int bufferSize = 1024; //blocks of 1024 samples |
| byte [] buffer = new byte[bufferSize * 2]; |
| |
| for (int ii = startIndex; ii < end; ii += bufferSize) { |
| //clear buffer |
| Arrays.fill(buffer, (byte) 0); |
| int bytesUsed = 0; |
| for (int jj = 0; jj < bufferSize; jj++) { |
| int index = ii + jj; |
| if (index >= end) |
| break; |
| int value = (int) Math.round(data[index] * Short.MAX_VALUE); |
| byte ba = (byte) (0xFF & (value >> 8)); //little-endian |
| byte bb = (byte) (0xFF & (value)); |
| buffer[(jj * 2) + 1] = ba; |
| buffer[jj * 2] = bb; |
| bytesUsed += 2; |
| } |
| mOutputStream.write(buffer, 0, bytesUsed); |
| } |
| log("Done writing data"); |
| } catch (IOException e) { |
| Log.e(TAG, "Error writing data " + e); |
| } |
| } |
| } |
| |
| |
| private static void log(String msg) { |
| Log.v(TAG, msg); |
| } |
| |
| } |