| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "media/audio/sounds/wav_audio_handler.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 #include <cstring> | |
| 9 | |
| 10 #include "base/logging.h" | |
| 11 #include "base/sys_byteorder.h" | |
| 12 #include "media/base/audio_bus.h" | |
| 13 | |
| 14 namespace { | |
| 15 | |
| 16 const char kChunkId[] = "RIFF"; | |
| 17 const char kFormat[] = "WAVE"; | |
| 18 const char kSubchunk1Id[] = "fmt "; | |
| 19 const char kSubchunk2Id[] = "data"; | |
| 20 | |
| 21 // The size of the header of a wav file. The header consists of 'RIFF', 4 bytes | |
| 22 // of total data length, and 'WAVE'. | |
| 23 const size_t kWavFileHeaderSize = 12; | |
| 24 | |
| 25 // The size of a chunk header in wav file format. A chunk header consists of a | |
| 26 // tag ('fmt ' or 'data') and 4 bytes of chunk length. | |
| 27 const size_t kChunkHeaderSize = 8; | |
| 28 | |
| 29 // The minimum size of 'fmt' chunk. | |
| 30 const size_t kFmtChunkMinimumSize = 16; | |
| 31 | |
| 32 // The offsets of 'fmt' fields. | |
| 33 const size_t kAudioFormatOffset = 0; | |
| 34 const size_t kChannelOffset = 2; | |
| 35 const size_t kSampleRateOffset = 4; | |
| 36 const size_t kByteRateOffset = 8; | |
| 37 const size_t kBitsPerSampleOffset = 14; | |
| 38 | |
| 39 // Some constants for audio format. | |
| 40 const int kAudioFormatPCM = 1; | |
| 41 | |
| 42 // Reads an integer from |data| with |offset|. | |
| 43 template<typename T> T ReadInt(const base::StringPiece& data, size_t offset) { | |
| 44 CHECK_LE(offset + sizeof(T), data.size()); | |
| 45 T result; | |
| 46 memcpy(&result, data.data() + offset, sizeof(T)); | |
| 47 #if !defined(ARCH_CPU_LITTLE_ENDIAN) | |
| 48 result = base::ByteSwap(result); | |
| 49 #endif | |
| 50 return result; | |
| 51 } | |
| 52 | |
| 53 } // namespace | |
| 54 | |
| 55 namespace media { | |
| 56 | |
| 57 WavAudioHandler::WavAudioHandler(const base::StringPiece& wav_data) | |
| 58 : num_channels_(0), | |
| 59 sample_rate_(0), | |
| 60 byte_rate_(0), | |
| 61 bits_per_sample_(0) { | |
| 62 CHECK_LE(kWavFileHeaderSize, wav_data.size()) << "wav data is too small"; | |
| 63 CHECK(wav_data.starts_with(kChunkId) && | |
| 64 memcmp(wav_data.data() + 8, kFormat, 4) == 0) | |
| 65 << "incorrect wav header"; | |
| 66 | |
| 67 uint32 total_length = std::min(ReadInt<uint32>(wav_data, 4), | |
| 68 static_cast<uint32>(wav_data.size())); | |
| 69 uint32 offset = kWavFileHeaderSize; | |
| 70 while (offset < total_length) { | |
| 71 const int length = ParseSubChunk(wav_data.substr(offset)); | |
| 72 CHECK_LE(0, length) << "can't parse wav sub-chunk"; | |
| 73 offset += length; | |
| 74 } | |
| 75 } | |
| 76 | |
| 77 WavAudioHandler::~WavAudioHandler() { | |
| 78 } | |
| 79 | |
| 80 bool WavAudioHandler::AtEnd(size_t cursor) const { | |
| 81 return data_.size() <= cursor; | |
| 82 } | |
| 83 | |
| 84 bool WavAudioHandler::CopyTo(AudioBus* bus, | |
| 85 size_t cursor, | |
| 86 size_t* bytes_written) const { | |
| 87 if (!bus) | |
| 88 return false; | |
| 89 if (bus->channels() != num_channels_) { | |
| 90 LOG(ERROR) << "Number of channel mismatch."; | |
| 91 return false; | |
| 92 } | |
| 93 if (AtEnd(cursor)) { | |
| 94 bus->Zero(); | |
| 95 return true; | |
| 96 } | |
| 97 const int remaining_frames = (data_.size() - cursor) / bytes_per_frame_; | |
| 98 const int frames = std::min(bus->frames(), remaining_frames); | |
| 99 bus->FromInterleaved(data_.data() + cursor, frames, bytes_per_sample_); | |
| 100 *bytes_written = frames * bytes_per_frame_; | |
| 101 bus->ZeroFramesPartial(frames, bus->frames() - frames); | |
| 102 return true; | |
| 103 } | |
| 104 | |
| 105 int WavAudioHandler::ParseSubChunk(const base::StringPiece& data) { | |
| 106 if (data.size() < kChunkHeaderSize) | |
| 107 return data.size(); | |
| 108 uint32 chunk_length = ReadInt<uint32>(data, 4); | |
| 109 if (data.starts_with(kSubchunk1Id)) { | |
| 110 if (!ParseFmtChunk(data.substr(kChunkHeaderSize, chunk_length))) | |
| 111 return -1; | |
| 112 } else if (data.starts_with(kSubchunk2Id)) { | |
| 113 if (!ParseDataChunk(data.substr(kChunkHeaderSize, chunk_length))) | |
| 114 return -1; | |
| 115 } else { | |
| 116 LOG(ERROR) << "Unknown data chunk: " << data.substr(0, 4) << "."; | |
| 117 } | |
| 118 return chunk_length + kChunkHeaderSize; | |
| 119 } | |
| 120 | |
| 121 bool WavAudioHandler::ParseFmtChunk(const base::StringPiece& data) { | |
| 122 if (data.size() < kFmtChunkMinimumSize) { | |
| 123 LOG(ERROR) << "Data size " << data.size() << " is too short."; | |
| 124 return false; | |
| 125 } | |
| 126 DCHECK_EQ(ReadInt<uint16>(data, kAudioFormatOffset), kAudioFormatPCM); | |
| 127 num_channels_ = ReadInt<uint16>(data, kChannelOffset); | |
| 128 sample_rate_ = ReadInt<uint32>(data, kSampleRateOffset); | |
| 129 byte_rate_ = ReadInt<uint32>(data, kByteRateOffset); | |
| 130 bits_per_sample_ = ReadInt<uint16>(data, kBitsPerSampleOffset); | |
| 131 bytes_per_sample_ = bits_per_sample_ >> 3; | |
| 132 bytes_per_frame_ = num_channels_ * bytes_per_sample_; | |
| 133 return true; | |
| 134 } | |
| 135 | |
| 136 bool WavAudioHandler::ParseDataChunk(const base::StringPiece& data) { | |
| 137 data_ = data; | |
| 138 return true; | |
| 139 } | |
| 140 | |
| 141 } // namespace media | |
| OLD | NEW |