// Copyright (c) 2007-2008, Niels Martin Hansen // All rights reserved. // // Redistribution and use in source and binary forms, with or without // modification, are permitted provided that the following conditions are met: // // * Redistributions of source code must retain the above copyright notice, // this list of conditions and the following disclaimer. // * Redistributions in binary form must reproduce the above copyright notice, // this list of conditions and the following disclaimer in the documentation // and/or other materials provided with the distribution. // * Neither the name of the Aegisub Group nor the names of its contributors // may be used to endorse or promote products derived from this software // without specific prior written permission. // // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE // POSSIBILITY OF SUCH DAMAGE. // // ----------------------------------------------------------------------------- // // AEGISUB // // Website: http://aegisub.cellosoft.com // Contact: mailto:jiifurusu@gmail.com // #include #include #include "audio_provider_pcm.h" #include "utils.h" #include "aegisub_endian.h" #include #include #ifndef _WINDOWS #include #include #include #endif PCMAudioProvider::PCMAudioProvider(const wxString &filename) { #ifdef _WINDOWS file_handle = CreateFile( filename.c_str(), FILE_READ_DATA, FILE_SHARE_READ|FILE_SHARE_WRITE, 0, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL|FILE_FLAG_RANDOM_ACCESS, 0); if (file_handle == INVALID_HANDLE_VALUE) { wxLogWarning(_T("PCM audio provider: Could not open audio file for reading (%d)"), GetLastError()); throw _T("PCM audio provider: Could not open audio file for reading"); } LARGE_INTEGER li_file_size = {0}; if (!GetFileSizeEx(file_handle, &li_file_size)) { CloseHandle(file_handle); throw _T("PCM audio provider: Failed getting file size"); } file_size = li_file_size.QuadPart; file_mapping = CreateFileMapping( file_handle, 0, PAGE_READONLY, 0, 0, 0); if (file_mapping == 0) { CloseHandle(file_handle); throw _T("PCM audio provider: Failed creating file mapping"); } current_mapping = 0; #else file_handle = open(filename.mb_str(*wxConvFileName), O_RDONLY); if (file_handle == -1) { throw _T("PCM audio provider: Could not open audio file for reading"); } struct stat filestats = {0}; if (fstat(file_handle, &filestats)) { close(file_handle); throw _T("PCM audio provider: Could not stat file to get size"); } file_size = filestats.st_size; current_mapping = 0; #endif } PCMAudioProvider::~PCMAudioProvider() { #ifdef _WINDOWS if (current_mapping) { UnmapViewOfFile(current_mapping); } CloseHandle(file_mapping); CloseHandle(file_handle); #else if (current_mapping) { munmap(current_mapping, mapping_length); } close(file_handle); #endif } char * PCMAudioProvider::EnsureRangeAccessible(int64_t range_start, int64_t range_length) { if (range_start + range_length > file_size) { throw _T("PCM audio provider: Attempted to map beyond end of file"); } // Check whether the requested range is already visible if (range_start < mapping_start || range_start+range_length > mapping_start+mapping_length) { // It's not visible, change the current mapping if (current_mapping) { #ifdef _WINDOWS UnmapViewOfFile(current_mapping); #else munmap(current_mapping, mapping_length); #endif } // Align range start on a 1 MB boundary and go 16 MB back mapping_start = (range_start & ~0xFFFFF) - 0x1000000; if (mapping_start < 0) mapping_start = 0; #ifdef _WINDOWS # if defined(_M_X64) || defined(_M_IA64) // Make 2 GB mappings by default on 64 bit archs mapping_length = 0x80000000; # else // Make 256 MB mappings by default on x86 mapping_length = 0x10000000; # endif #else // 256 MB mapping_length = 0x10000000; #endif // Make sure to always make a mapping at least as large as the requested range if (mapping_length < range_length) { if (range_length > (int64_t)(~(size_t)0)) throw _T("PCM audio provider: Requested range larger than max size_t, cannot create view of file"); else mapping_length = range_length; } // But also make sure we don't try to make a mapping larger than the file if (mapping_length > file_size) mapping_length = (size_t)(file_size - mapping_start); // We already checked that the requested range doesn't extend over the end of the file // Hopefully this should ensure that small files are always mapped in their entirety #ifdef _WINDOWS LARGE_INTEGER mapping_start_li; mapping_start_li.QuadPart = mapping_start; current_mapping = MapViewOfFile( file_mapping, // Mapping handle FILE_MAP_READ, // Access type mapping_start_li.HighPart, // Offset high-part mapping_start_li.LowPart, // Offset low-part mapping_length); // Length of view #else current_mapping = mmap(0, mapping_length, PROT_READ, MAP_PRIVATE, file_handle, mapping_start); #endif if (!current_mapping) { throw _T("PCM audio provider: Failed mapping a view of the file"); } } assert(range_start >= mapping_start); // Difference between actual current mapping start and requested range start ptrdiff_t rel_ofs = (ptrdiff_t)(range_start - mapping_start); // Calculate a pointer into current mapping for the requested range return ((char*)current_mapping) + rel_ofs; } void PCMAudioProvider::GetAudio(void *buf, int64_t start, int64_t count) { // Read blocks from the file size_t index = 0; while (count > 0 && index < index_points.size()) { // Check if this index contains the samples we're looking for IndexPoint &ip = index_points[index]; if (ip.start_sample <= start && ip.start_sample+ip.num_samples > start) { // How many samples we can maximum take from this block int64_t samples_can_do = ip.num_samples - start + ip.start_sample; if (samples_can_do > count) samples_can_do = count; // Read as many samples we can char *src = EnsureRangeAccessible( ip.start_byte + (start - ip.start_sample) * bytes_per_sample * channels, samples_can_do * bytes_per_sample * channels); memcpy(buf, src, samples_can_do * bytes_per_sample * channels); // Update data buf = (char*)buf + samples_can_do * bytes_per_sample * channels; start += samples_can_do; count -= samples_can_do; } index++; } // If we exhausted all sample sections zerofill the rest if (count > 0) { if (bytes_per_sample == 1) // 8 bit formats are usually unsigned with bias 127 memset(buf, 127, count*channels); else // While everything else is signed memset(buf, 0, count*bytes_per_sample*channels); } } // RIFF WAV PCM provider // Overview of RIFF WAV: class RiffWavPCMAudioProvider : public PCMAudioProvider { private: struct ChunkHeader { char type[4]; uint32_t size; }; struct RIFFChunk { ChunkHeader ch; char format[4]; }; struct fmtChunk { // Skip the chunk header here, it's processed separately uint16_t compression; // compression format used -- 0x0001 = PCM uint16_t channels; uint32_t samplerate; uint32_t avg_bytes_sec; // can't always be trusted uint16_t block_align; uint16_t significant_bits_sample; // Here was supposed to be some more fields but we don't need them // and just skipping by the size of the struct wouldn't be safe // either way, as the fields can depend on the compression. }; static bool CheckFourcc(char *str1, char *str2) { assert(str1); assert(str2); return (str1[0] == str2[0]) && (str1[1] == str2[1]) && (str1[2] == str2[2]) && (str1[3] == str2[3]); } public: RiffWavPCMAudioProvider(const wxString &_filename) : PCMAudioProvider(_filename) { filename = _filename; // Read header // This should throw an exception if the mapping fails void *filestart = EnsureRangeAccessible(0, sizeof(RIFFChunk)); assert(filestart); RIFFChunk &header = *(RIFFChunk*)filestart; // Check magic values if (!CheckFourcc(header.ch.type, "RIFF")) throw _T("RIFF PCM WAV audio provider: File is not a RIFF file"); if (!CheckFourcc(header.format, "WAVE")) throw _T("RIFF PCM WAV audio provider: File is not a RIFF WAV file"); // Count how much more data we can have in the entire file // The first 4 bytes are already eaten by the header.format field uint32_t data_left = Endian::LittleToMachine(header.ch.size) - 4; // How far into the file we have processed. // Must be incremented by the riff chunk size fields. uint32_t filepos = sizeof(header); bool got_fmt_header = false; // Inherited from AudioProvider num_samples = 0; // Continue reading chunks until out of data while (data_left) { ChunkHeader &ch = *(ChunkHeader*)EnsureRangeAccessible(filepos, sizeof(ChunkHeader)); // Update counters data_left -= sizeof(ch); filepos += sizeof(ch); if (CheckFourcc(ch.type, "fmt ")) { if (got_fmt_header) throw _T("RIFF PCM WAV audio provider: Invalid file, multiple 'fmt ' chunks"); got_fmt_header = true; fmtChunk &fmt = *(fmtChunk*)EnsureRangeAccessible(filepos, sizeof(fmtChunk)); if (Endian::LittleToMachine(fmt.compression) != 1) throw _T("RIFF PCM WAV audio provider: Can't use file, not PCM encoding"); // Set stuff inherited from the AudioProvider class sample_rate = Endian::LittleToMachine(fmt.samplerate); channels = Endian::LittleToMachine(fmt.channels); bytes_per_sample = (Endian::LittleToMachine(fmt.significant_bits_sample) + 7) / 8; // round up to nearest whole byte } else if (CheckFourcc(ch.type, "data")) { // This won't pick up 'data' chunks inside 'wavl' chunks // since the 'wavl' chunks wrap those. if (!got_fmt_header) throw _T("RIFF PCM WAV audio provider: Found 'data' chunk before 'fmt ' chunk, file is invalid."); int64_t samples = Endian::LittleToMachine(ch.size) / bytes_per_sample; int64_t frames = samples / channels; IndexPoint ip; ip.start_sample = num_samples; ip.num_samples = frames; ip.start_byte = filepos; index_points.push_back(ip); num_samples += frames; } // Support wavl (wave list) chunks too? // Update counters // Make sure they're word aligned data_left -= (Endian::LittleToMachine(ch.size) + 1) & ~1; filepos += (Endian::LittleToMachine(ch.size) + 1) & ~1; } } }; // Mix down any number of channels to mono class DownmixingAudioProvider : public AudioProvider { private: AudioProvider *provider; int src_channels; public: DownmixingAudioProvider(AudioProvider *source) { filename = source->GetFilename(); channels = 1; // target src_channels = source->GetChannels(); num_samples = source->GetNumSamples(); bytes_per_sample = source->GetBytesPerSample(); sample_rate = source->GetSampleRate(); // We now own this provider = source; if (!(bytes_per_sample == 1 || bytes_per_sample == 2)) throw _T("Downmixing Audio Provider: Can only downmix 8 and 16 bit audio"); } ~DownmixingAudioProvider() { delete provider; } void GetAudio(void *buf, int64_t start, int64_t count) { if (count == 0) return; // We can do this ourselves if (start >= num_samples) { if (bytes_per_sample == 1) // 8 bit formats are usually unsigned with bias 127 memset(buf, 127, count); else // While everything else is signed memset(buf, 0, count*bytes_per_sample); return; } // So alloc some temporary memory for this // Depending on use, this might be made faster by using // a pre-allocced block of memory...? char *tmp = new char[count*bytes_per_sample*src_channels]; provider->GetAudio(tmp, start, count); // Now downmix // Just average the samples over the channels (really bad if they're out of phase!) // XXX: Assuming here that sample data are in machine endian, an upstream provider should ensure that if (bytes_per_sample == 1) { uint8_t *src = (uint8_t *)tmp; uint8_t *dst = (uint8_t *)buf; while (count > 0) { int sum = 0; for (int c = 0; c < src_channels; c++) sum += *(src++); *(dst++) = (uint8_t)(sum / src_channels); count--; } } else if (bytes_per_sample == 2) { int16_t *src = (int16_t *)tmp; int16_t *dst = (int16_t *)buf; while (count > 0) { int sum = 0; for (int c = 0; c < src_channels; c++) sum += *(src++); *(dst++) = (int16_t)(sum / src_channels); count--; } } // Done downmixing, free the work buffer delete[] tmp; } }; AudioProvider *CreatePCMAudioProvider(const wxString &filename) { AudioProvider *provider = 0; // Try Microsoft/IBM RIFF WAV first try { provider = new RiffWavPCMAudioProvider(filename); } catch (const wxChar *msg) { provider = 0; wxLogDebug(_T("Creating PCM WAV reader failed with message: %s\nProceeding to try other providers."), msg); } catch (...) { provider = 0; } if (provider && provider->GetChannels() > 1) { // Can't feed non-mono audio to the rest of the program. // Create a downmixing proxy and if it fails, don't provide PCM. try { provider = new DownmixingAudioProvider(provider); } catch (...) { delete provider; provider = 0; } } return provider; }