forked from mia/Aegisub
269 lines
6.9 KiB
C++
269 lines
6.9 KiB
C++
// Copyright (c) 2005-2006, Rodrigo Braz Monteiro
|
|
// 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:zeratul@cellosoft.com
|
|
//
|
|
|
|
|
|
///////////
|
|
// Headers
|
|
#include <wx/wxprec.h>
|
|
#include <wx/thread.h>
|
|
#include "audio_provider_ram.h"
|
|
#include "audio_provider_hd.h"
|
|
#include "audio_provider_pcm.h"
|
|
#include "audio_provider_convert.h"
|
|
#ifdef WITH_AVISYNTH
|
|
#include "audio_provider_avs.h"
|
|
#endif
|
|
#ifdef WITH_FFMPEG
|
|
#include "audio_provider_lavc.h"
|
|
#endif
|
|
#include "options.h"
|
|
#include "audio_display.h"
|
|
|
|
|
|
///////////////
|
|
// Constructor
|
|
AudioProvider::AudioProvider() {
|
|
raw = NULL;
|
|
}
|
|
|
|
|
|
//////////////
|
|
// Destructor
|
|
AudioProvider::~AudioProvider() {
|
|
// Clear buffers
|
|
delete[] raw;
|
|
}
|
|
|
|
|
|
//////////////////////////
|
|
// Get number of channels
|
|
int AudioProvider::GetChannels() {
|
|
return channels;
|
|
}
|
|
|
|
|
|
//////////////////////////
|
|
// Get number of samples
|
|
int64_t AudioProvider::GetNumSamples() {
|
|
return num_samples;
|
|
}
|
|
|
|
|
|
///////////////////
|
|
// Get sample rate
|
|
int AudioProvider::GetSampleRate() {
|
|
return sample_rate;
|
|
}
|
|
|
|
|
|
////////////////////////
|
|
// Get bytes per sample
|
|
int AudioProvider::GetBytesPerSample() {
|
|
return bytes_per_sample;
|
|
}
|
|
|
|
|
|
////////////////
|
|
// Get filename
|
|
wxString AudioProvider::GetFilename() {
|
|
return filename;
|
|
}
|
|
|
|
|
|
////////////////
|
|
// Get waveform
|
|
void AudioProvider::GetWaveForm(int *min,int *peak,int64_t start,int w,int h,int samples,float scale) {
|
|
// Setup
|
|
int channels = GetChannels();
|
|
int n = w * samples;
|
|
for (int i=0;i<w;i++) {
|
|
peak[i] = 0;
|
|
min[i] = h;
|
|
}
|
|
|
|
// Prepare waveform
|
|
int cur;
|
|
int curvalue;
|
|
|
|
// Prepare buffers
|
|
int needLen = n*channels*bytes_per_sample;
|
|
if (raw) {
|
|
if (raw_len < needLen) {
|
|
delete[] raw;
|
|
raw = NULL;
|
|
}
|
|
}
|
|
if (!raw) {
|
|
raw_len = needLen;
|
|
raw = new char[raw_len];
|
|
}
|
|
|
|
if (bytes_per_sample == 1) {
|
|
// Read raw samples
|
|
unsigned char *raw_char = (unsigned char*) raw;
|
|
GetAudio(raw,start,n);
|
|
int amplitude = int(h*scale);
|
|
|
|
// Calculate waveform
|
|
for (int i=0;i<n;i++) {
|
|
cur = i/samples;
|
|
curvalue = h - (int(raw_char[i*channels])*amplitude)/0xFF;
|
|
if (curvalue > h) curvalue = h;
|
|
if (curvalue < 0) curvalue = 0;
|
|
if (curvalue < min[cur]) min[cur] = curvalue;
|
|
if (curvalue > peak[cur]) peak[cur] = curvalue;
|
|
}
|
|
}
|
|
|
|
if (bytes_per_sample == 2) {
|
|
// Read raw samples
|
|
short *raw_short = (short*) raw;
|
|
GetAudio(raw,start,n);
|
|
int half_h = h/2;
|
|
int half_amplitude = int(half_h * scale);
|
|
|
|
// Calculate waveform
|
|
for (int i=0;i<n;i++) {
|
|
cur = i/samples;
|
|
curvalue = half_h - (int(raw_short[i*channels])*half_amplitude)/0x8000;
|
|
if (curvalue > h) curvalue = h;
|
|
if (curvalue < 0) curvalue = 0;
|
|
if (curvalue < min[cur]) min[cur] = curvalue;
|
|
if (curvalue > peak[cur]) peak[cur] = curvalue;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/////////////////////////
|
|
// Get audio with volume
|
|
void AudioProvider::GetAudioWithVolume(void *buf, int64_t start, int64_t count, double volume) {
|
|
GetAudio(buf,start,count);
|
|
if (volume == 1.0) return;
|
|
|
|
if (bytes_per_sample == 2) {
|
|
// Read raw samples
|
|
short *buffer = (short*) buf;
|
|
int value;
|
|
|
|
// Modify
|
|
for (int64_t i=0;i<count;i++) {
|
|
value = (int)(buffer[i]*volume+0.5);
|
|
if (value < -0x8000) value = -0x8000;
|
|
if (value > 0x7FFF) value = 0x7FFF;
|
|
buffer[i] = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
////////////////
|
|
// Get provider
|
|
AudioProvider *AudioProviderFactoryManager::GetAudioProvider(wxString filename, int cache) {
|
|
// Prepare provider
|
|
AudioProvider *provider = NULL;
|
|
|
|
// Try a PCM provider first
|
|
provider = CreatePCMAudioProvider(filename);
|
|
if (provider) {
|
|
if (provider->GetBytesPerSample() == 2 && provider->GetSampleRate() >= 32000) return provider;
|
|
return new ConvertAudioProvider(provider);
|
|
}
|
|
|
|
// List of providers
|
|
wxArrayString list = GetFactoryList(Options.AsText(_T("Audio provider")));
|
|
|
|
// None available
|
|
if (list.Count() == 0) throw _T("No audio providers are available.");
|
|
|
|
// Get provider
|
|
wxString error;
|
|
for (unsigned int i=0;i<list.Count();i++) {
|
|
try {
|
|
AudioProvider *prov = GetFactory(list[i])->CreateProvider(filename.c_str());
|
|
if (prov) {
|
|
provider = prov;
|
|
break;
|
|
}
|
|
}
|
|
catch (wxString err) { error += list[i] + _T(" factory: ") + err + _T("\n"); }
|
|
catch (const wxChar *err) { error += list[i] + _T(" factory: ") + wxString(err) + _T("\n"); }
|
|
catch (...) { error += list[i] + _T(" factory: Unknown error\n"); }
|
|
}
|
|
|
|
// Failed
|
|
if (!provider) throw error;
|
|
|
|
// Give it a conversor if needed
|
|
if (provider->GetBytesPerSample() != 2 || provider->GetSampleRate() < 32000) provider = new ConvertAudioProvider(provider);
|
|
|
|
// Change provider to RAM/HD cache if needed
|
|
if (cache == -1) cache = Options.AsInt(_T("Audio Cache"));
|
|
if (cache) {
|
|
AudioProvider *final = NULL;
|
|
|
|
// Convert to RAM
|
|
if (cache == 1) final = new RAMAudioProvider(provider);
|
|
|
|
// Convert to HD
|
|
if (cache == 2) final = new HDAudioProvider(provider);
|
|
|
|
// Reassign
|
|
if (final) {
|
|
delete provider;
|
|
provider = final;
|
|
}
|
|
}
|
|
|
|
// Return
|
|
return provider;
|
|
}
|
|
|
|
|
|
///////////////////////////
|
|
// Register all providers
|
|
void AudioProviderFactoryManager::RegisterProviders() {
|
|
#ifdef WITH_AVISYNTH
|
|
RegisterFactory(new AvisynthAudioProviderFactory(),_T("Avisynth"));
|
|
#endif
|
|
#ifdef WITH_FFMPEG
|
|
RegisterFactory(new LAVCAudioProviderFactory(),_T("FFMPEG"));
|
|
#endif
|
|
}
|
|
|
|
|
|
//////////
|
|
// Static
|
|
template <class AudioProviderFactory> std::map<wxString,AudioProviderFactory*>* FactoryManager<AudioProviderFactory>::factories=NULL;
|