275 lines
7.1 KiB
C++
275 lines
7.1 KiB
C++
/*************************************************************************/
|
|
/* resource_format_wav.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* http://www.godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2016 Juan Linietsky, Ariel Manzur. */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
#include "resource_format_wav.h"
|
|
#include "os/file_access.h"
|
|
#include "scene/resources/sample.h"
|
|
|
|
|
|
RES ResourceFormatLoaderWAV::load(const String &p_path, const String& p_original_path, Error *r_error) {
|
|
if (r_error)
|
|
*r_error=ERR_FILE_CANT_OPEN;
|
|
|
|
Error err;
|
|
FileAccess *file=FileAccess::open(p_path, FileAccess::READ,&err);
|
|
|
|
ERR_FAIL_COND_V( err!=OK, RES() );
|
|
|
|
if (r_error)
|
|
*r_error=ERR_FILE_CORRUPT;
|
|
|
|
/* CHECK RIFF */
|
|
char riff[5];
|
|
riff[4]=0;
|
|
file->get_buffer((uint8_t*)&riff,4); //RIFF
|
|
|
|
if (riff[0]!='R' || riff[1]!='I' || riff[2]!='F' || riff[3]!='F') {
|
|
|
|
file->close();
|
|
memdelete(file);
|
|
ERR_FAIL_V( RES() );
|
|
}
|
|
|
|
|
|
/* GET FILESIZE */
|
|
uint32_t filesize=file->get_32();
|
|
|
|
/* CHECK WAVE */
|
|
|
|
char wave[4];
|
|
|
|
file->get_buffer((uint8_t*)&wave,4); //RIFF
|
|
|
|
if (wave[0]!='W' || wave[1]!='A' || wave[2]!='V' || wave[3]!='E') {
|
|
|
|
|
|
file->close();
|
|
memdelete(file);
|
|
ERR_EXPLAIN("Not a WAV file (no WAVE RIFF Header)")
|
|
ERR_FAIL_V( RES() );
|
|
}
|
|
|
|
bool format_found=false;
|
|
bool data_found=false;
|
|
int format_bits=0;
|
|
int format_channels=0;
|
|
int format_freq=0;
|
|
Sample::LoopFormat loop=Sample::LOOP_NONE;
|
|
int loop_begin=0;
|
|
int loop_end=0;
|
|
|
|
|
|
Ref<Sample> sample( memnew( Sample ) );
|
|
|
|
|
|
while (!file->eof_reached()) {
|
|
|
|
|
|
/* chunk */
|
|
char chunkID[4];
|
|
file->get_buffer((uint8_t*)&chunkID,4); //RIFF
|
|
|
|
/* chunk size */
|
|
uint32_t chunksize=file->get_32();
|
|
uint32_t file_pos=file->get_pos(); //save file pos, so we can skip to next chunk safely
|
|
|
|
if (file->eof_reached()) {
|
|
|
|
//ERR_PRINT("EOF REACH");
|
|
break;
|
|
}
|
|
|
|
if (chunkID[0]=='f' && chunkID[1]=='m' && chunkID[2]=='t' && chunkID[3]==' ' && !format_found) {
|
|
/* IS FORMAT CHUNK */
|
|
|
|
uint16_t compression_code=file->get_16();
|
|
|
|
|
|
if (compression_code!=1) {
|
|
ERR_PRINT("Format not supported for WAVE file (not PCM). Save WAVE files as uncompressed PCM instead.");
|
|
break;
|
|
}
|
|
|
|
format_channels=file->get_16();
|
|
if (format_channels!=1 && format_channels !=2) {
|
|
|
|
ERR_PRINT("Format not supported for WAVE file (not stereo or mono)");
|
|
break;
|
|
|
|
}
|
|
|
|
format_freq=file->get_32(); //sampling rate
|
|
|
|
file->get_32(); // average bits/second (unused)
|
|
file->get_16(); // block align (unused)
|
|
format_bits=file->get_16(); // bits per sample
|
|
|
|
if (format_bits%8) {
|
|
|
|
ERR_PRINT("Strange number of bits in sample (not 8,16,24,32)");
|
|
break;
|
|
}
|
|
|
|
/* Dont need anything else, continue */
|
|
format_found=true;
|
|
}
|
|
|
|
|
|
if (chunkID[0]=='d' && chunkID[1]=='a' && chunkID[2]=='t' && chunkID[3]=='a' && !data_found) {
|
|
/* IS FORMAT CHUNK */
|
|
data_found=true;
|
|
|
|
if (!format_found) {
|
|
ERR_PRINT("'data' chunk before 'format' chunk found.");
|
|
break;
|
|
|
|
}
|
|
|
|
int frames=chunksize;
|
|
|
|
frames/=format_channels;
|
|
frames/=(format_bits>>3);
|
|
|
|
/*print_line("chunksize: "+itos(chunksize));
|
|
print_line("channels: "+itos(format_channels));
|
|
print_line("bits: "+itos(format_bits));
|
|
*/
|
|
sample->create(
|
|
(format_bits==8) ? Sample::FORMAT_PCM8 : Sample::FORMAT_PCM16,
|
|
(format_channels==2)?true:false,
|
|
frames );
|
|
sample->set_mix_rate( format_freq );
|
|
|
|
int len=frames;
|
|
if (format_channels==2)
|
|
len*=2;
|
|
if (format_bits>8)
|
|
len*=2;
|
|
|
|
DVector<uint8_t> data;
|
|
data.resize(len);
|
|
DVector<uint8_t>::Write dataw = data.write();
|
|
void * data_ptr = dataw.ptr();
|
|
|
|
for (int i=0;i<frames;i++) {
|
|
|
|
|
|
for (int c=0;c<format_channels;c++) {
|
|
|
|
|
|
if (format_bits==8) {
|
|
// 8 bit samples are UNSIGNED
|
|
|
|
uint8_t s = file->get_8();
|
|
s-=128;
|
|
int8_t *sp=(int8_t*)&s;
|
|
|
|
int8_t *data_ptr8=&((int8_t*)data_ptr)[i*format_channels+c];
|
|
|
|
*data_ptr8=*sp;
|
|
|
|
} else {
|
|
//16+ bits samples are SIGNED
|
|
// if sample is > 16 bits, just read extra bytes
|
|
|
|
uint32_t data=0;
|
|
for (int b=0;b<(format_bits>>3);b++) {
|
|
|
|
data|=((uint32_t)file->get_8())<<(b*8);
|
|
}
|
|
data<<=(32-format_bits);
|
|
|
|
|
|
int32_t s=data;
|
|
|
|
int16_t *data_ptr16=&((int16_t*)data_ptr)[i*format_channels+c];
|
|
|
|
*data_ptr16=s>>16;
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
dataw=DVector<uint8_t>::Write();
|
|
|
|
sample->set_data(data);
|
|
|
|
|
|
if (file->eof_reached()) {
|
|
file->close();
|
|
memdelete(file);
|
|
ERR_EXPLAIN("Premature end of file.");
|
|
ERR_FAIL_V(RES());
|
|
}
|
|
}
|
|
|
|
if (chunkID[0]=='s' && chunkID[1]=='m' && chunkID[2]=='p' && chunkID[3]=='l') {
|
|
//loop point info!
|
|
|
|
for(int i=0;i<10;i++)
|
|
file->get_32(); // i wish to know why should i do this... no doc!
|
|
|
|
loop=file->get_32()?Sample::LOOP_PING_PONG:Sample::LOOP_FORWARD;
|
|
loop_begin=file->get_32();
|
|
loop_end=file->get_32();
|
|
|
|
}
|
|
file->seek( file_pos+chunksize );
|
|
}
|
|
|
|
sample->set_loop_format(loop);
|
|
sample->set_loop_begin(loop_begin);
|
|
sample->set_loop_end(loop_end);
|
|
|
|
file->close();
|
|
memdelete(file);
|
|
|
|
if (r_error)
|
|
*r_error=OK;
|
|
|
|
|
|
return sample;
|
|
|
|
}
|
|
void ResourceFormatLoaderWAV::get_recognized_extensions(List<String> *p_extensions) const {
|
|
|
|
p_extensions->push_back("wav");
|
|
}
|
|
bool ResourceFormatLoaderWAV::handles_type(const String& p_type) const {
|
|
|
|
return (p_type=="Sample");
|
|
}
|
|
|
|
String ResourceFormatLoaderWAV::get_resource_type(const String &p_path) const {
|
|
|
|
if (p_path.extension().to_lower()=="wav")
|
|
return "Sample";
|
|
return "";
|
|
}
|
|
|