audio_convert.cpp

#include <iostream>
#include <ostream>
#include <stdio.h>
#include <math.h>

#include "audio_convert.h"
#include "tools_global.h"


using namespace std;


void cAudioConvert::sWavHeader::prepareEndian() {
	#if __BYTE_ORDER == __BIG_ENDIAN
	_BSWAP(chunkSize);
	_BSWAP(lengthFormatData);
	_BSWAP(format);
	_BSWAP(channels);
	_BSWAP(sampleRate);
	_BSWAP(byteRate);
	_BSWAP(bytesPerSample);
	_BSWAP(bitsPerSampleChannel);
	_BSWAP(dataSize);
	#endif
}


cAudioConvert::cAudioConvert() {
	srcDstType = _src;
	formatType = _format_raw;
	fileHandle = NULL;
	destAudio = NULL;
	oggQuality = 0.4;
	headerIsWrited = false;
	onlyGetAudioInfo = false;
}

cAudioConvert::~cAudioConvert() {
	if(fileHandle) {
		fclose(fileHandle);
	}
}

cAudioConvert::eResult cAudioConvert::getAudioInfo() {
	onlyGetAudioInfo = true;
	if(readWav() == _rslt_ok) {
		 formatType = _format_wav;
		 onlyGetAudioInfo = false;
		 return(_rslt_ok);
	}
	if(fileHandle) {
		fclose(fileHandle);
		fileHandle = NULL;
	}
	if(readOgg() == _rslt_ok) {
		 formatType = _format_ogg;
		 onlyGetAudioInfo = false;
		 return(_rslt_ok);
	}
	onlyGetAudioInfo = false;
	return(_rslt_unknown_format);
}

string cAudioConvert::jsonAudioInfo() {
	JsonExport json_export;
	json_export.add("format", formatType == _format_raw ? "raw" :
				  formatType == _format_wav ? "wav" :
				  formatType == _format_ogg ? "ogg" : "unknown");
	json_export.add("sample_rate", audioInfo.sampleRate);
	json_export.add("channels", audioInfo.channels);
	json_export.add("bits_per_sample", audioInfo.bitsPerSample);
	json_export.add("bytes_per_sample", audioInfo.bitsPerSample / 8);
	return(json_export.getJson());
}

cAudioConvert::eResult cAudioConvert::readRaw(sAudioInfo *audioInfo) {
	if(!fileHandle) {
		fileHandle = fopen(fileName.c_str(), "r");
		if(!fileHandle) {
			return(_rslt_open_for_read_failed);
		}
	}
	this->audioInfo = *audioInfo;
	unsigned readbuffer_size = 1024;
	u_char *readbuffer = new u_char[readbuffer_size];
        int read_length;
	eResult rslt_write = _rslt_ok;
	while((read_length = fread(readbuffer, 1, readbuffer_size, fileHandle)) > 0) {
		rslt_write = write(readbuffer, readbuffer_size);
		if(rslt_write != _rslt_ok) {
			break;
		}
	}
	if(rslt_write == _rslt_ok) {
		rslt_write = write(NULL, 0);
	}
	delete [] readbuffer;
	return(rslt_write);
}

cAudioConvert::eResult cAudioConvert::readWav() {
	if(!fileHandle) {
		fileHandle = fopen(fileName.c_str(), "r");
		if(!fileHandle) {
			return(_rslt_open_for_read_failed);
		}
	}
	sWavHeader wavHeader;
	if(!readWavHeader(&wavHeader)) {
		return(_rslt_wav_read_header_failed);
	} else {
		if(!wavHeader.checkHeader()) {
			return(_rslt_wav_bad_header);
		}
		wavHeader.setAudioInfo(&audioInfo);
		if(onlyGetAudioInfo) {
			return(_rslt_ok);
		}
	}
	unsigned readbuffer_size = 1024;
	u_char *readbuffer = new u_char[readbuffer_size];
        int read_length;
	eResult rslt_write = _rslt_ok;
	while((read_length = fread(readbuffer, 1, readbuffer_size, fileHandle)) > 0) {
		rslt_write = write(readbuffer, readbuffer_size);
		if(rslt_write != _rslt_ok) {
			break;
		}
	}
	if(rslt_write == _rslt_ok) {
		rslt_write = write(NULL, 0);
	}
	delete [] readbuffer;
	return(rslt_write);
}

bool cAudioConvert::readWavHeader(sWavHeader *wavHeader) {
	wavHeader->null();
	size_t readSize = fread(wavHeader, 1, sizeof(sWavHeader), fileHandle);
	if(readSize == sizeof(sWavHeader)) {
		wavHeader->prepareAfterRead();
		return(true);
	}
	return(false);
}

cAudioConvert::eResult cAudioConvert::writeWavHeader(long int size) {
	if(size == -1) {
		fseek(fileHandle, 0, SEEK_END);
		size = ftello(fileHandle);
	}
	sWavHeader wavHeader;
	wavHeader.init();
	wavHeader.setFromAudioInfo(&audioInfo);
	wavHeader.setFileSize(size);
	wavHeader.prepareBeforeWrite();
	if(size != 0) {
		fseek(fileHandle, 0, SEEK_SET);
	}
	return(write((u_char*)&wavHeader, sizeof(sWavHeader)));
}

cAudioConvert::eResult cAudioConvert::writeWavData(u_char *data, unsigned datalen) {
	return(fwrite(data, 1, datalen, fileHandle) == datalen ?
		_rslt_ok :
		_rslt_write_failed);
}

cAudioConvert::eResult cAudioConvert::writeWavEnd() {
	return(writeWavHeader(-1));
}

cAudioConvert::eResult cAudioConvert::readOgg() {
	if(!fileHandle) {
		fileHandle = fopen(fileName.c_str(), "r");
		if(!fileHandle) {
			return(_rslt_open_for_read_failed);
		}
	}
 
	sOggDecode oggDecode(4096);

	ogg_sync_init(&ogg.oy); /* Now we can read pages */

	while(1) { /* we repeat if the bitstream is chained */
		ogg.eos = 0;

		/* grab some data at the head of the stream. We want the first page
		   (which is guaranteed to be small and only contain the Vorbis
		   stream initial header) We need the first page to get the stream
		   serialno. */

		/* submit a 4k block to libvorbis' Ogg layer */
		oggDecode.sync_buffer = ogg_sync_buffer(&ogg.oy, oggDecode.sync_buffer_size);
		unsigned read_bytes = fread(oggDecode.sync_buffer, 1, oggDecode.sync_buffer_size, fileHandle);
		ogg_sync_wrote(&ogg.oy, read_bytes);

		/* Get the first page. */
		if(ogg_sync_pageout(&ogg.oy, &ogg.og) != 1) {
			/* have we simply run out of data?  If so, we're done. */
			if(read_bytes < oggDecode.sync_buffer_size) break;
			return(_rslt_ogg_bad_bitstream);
		}

		/* Get the serial number and set up the rest of decode. */
		/* serialno first; use it to set up a logical stream */
		ogg_stream_init(&ogg.os, ogg_page_serialno(&ogg.og));

		/* extract the initial header from the first page and verify that the
		   Ogg bitstream is in fact Vorbis data */

		/* I handle the initial header first instead of just having the code
		   read all three Vorbis headers at once because reading the initial
		   header is an easy way to identify a Vorbis bitstream and it's
		   useful to see that functionality seperated out. */

		vorbis_info_init(&ogg.vi);
		vorbis_comment_init(&ogg.vc);
		if(ogg_stream_pagein(&ogg.os, &ogg.og) < 0){
			return(_rslt_ogg_bad_first_page);
		}

		if(ogg_stream_packetout(&ogg.os, &ogg.op) != 1){
			return(_rslt_ogg_bad_initial_header_packet);
		}

		if(vorbis_synthesis_headerin(&ogg.vi, &ogg.vc, &ogg.op) < 0){
			return(_rslt_ogg_missing_vorbis_audiodata);
		}

		/* At this point, we're sure we're Vorbis. We've set up the logical
		   (Ogg) bitstream decoder. Get the comment and codebook headers and
		   set up the Vorbis decoder */

		/* The next two packets in order are the comment and codebook headers.
		   They're likely large and may span multiple pages. Thus we read
		   and submit data until we get our two packets, watching that no
		   pages are missing. If a page is missing, error out; losing a
		   header page is the only place where missing data is fatal. */

		int i = 0;
		while(i < 2) {
			while(i < 2) {
				int result=ogg_sync_pageout(&ogg.oy, &ogg.og);
				if(result==0) break; /* Need more data */
				/* Don't complain about missing or corrupt data yet. We'll
				   catch it at the packet output phase */
				if(result == 1) {
					ogg_stream_pagein(&ogg.os, &ogg.og); /* we can ignore any errors here
								       as they'll also become apparent
								       at packetout */
					while(i < 2) {
						result=ogg_stream_packetout(&ogg.os, &ogg.op);
						if(result == 0) break;
						if(result < 0){
							/* Uh oh; data at some point was corrupted or missing!
							   We can't tolerate that in a header.  Die. */
							return(_rslt_ogg_corrupt_secondary_header);
						}
						result = vorbis_synthesis_headerin(&ogg.vi, &ogg.vc, &ogg.op);
						if(result<0){
							return(_rslt_ogg_corrupt_secondary_header);
						}
						i++;
					}
				}
			}
			/* no harm in not checking before adding more */
			oggDecode.sync_buffer = ogg_sync_buffer(&ogg.oy, oggDecode.sync_buffer_size);
			unsigned read_bytes = fread(oggDecode.sync_buffer, 1, oggDecode.sync_buffer_size, fileHandle);
			if(read_bytes == 0 && i < 2){
				return(_rslt_ogg_missing_vorbis_headers);
			}
			ogg_sync_wrote(&ogg.oy, read_bytes);
		}

		/* Throw the comments plus a few lines about the bitstream we're
		   decoding */
		{
			char **ptr = ogg.vc.user_comments;
			while(*ptr){
				/*
				fprintf(stderr, "%s\n", *ptr);
				*/
				if(!comment.empty()) {
					comment += "\n";
				}
				comment += *ptr;
				++ptr;
			}
			audioInfo.channels = ogg.vi.channels;
			audioInfo.sampleRate = ogg.vi.rate;
			audioInfo.bitsPerSample = 16;
			if(onlyGetAudioInfo) {
				return(_rslt_ok);
			}
			/*
			fprintf(stderr, "\nBitstream is %d channel, %ldHz\n", ogg.vi.channels, ogg.vi.rate);
			fprintf(stderr, "Encoded by: %s\n\n", ogg.vc.vendor);
			*/
		}

		int convsize = oggDecode.sync_buffer_size / ogg.vi.channels;
		oggDecode.conv_buffer = new ogg_int16_t[convsize];

		/* OK, got and parsed all three headers. Initialize the Vorbis
		   packet->PCM decoder. */
		if(vorbis_synthesis_init(&ogg.vd, &ogg.vi) == 0) { /* central decode state */
			vorbis_block_init(&ogg.vd, &ogg.vb);          /* local state for most of the decode
								so multiple block decodes can
								proceed in parallel. We could init
								multiple vorbis_block structures
								for vd here */

			/* The rest is just a straight decode loop until end of stream */
			while(!ogg.eos) {
				while(!ogg.eos) {
					int result = ogg_sync_pageout(&ogg.oy, &ogg.og);
					if(result == 0) break; /* need more data */
					if(result < 0){ /* missing or corrupt data at this page position */
						/*
						fprintf(stderr, "Corrupt or missing data in bitstream; "
							"continuing...\n");
						*/
					} else {
						ogg_stream_pagein(&ogg.os, &ogg.og); /* can safely ignore errors at
									       this point */
						while(1) {
							result=ogg_stream_packetout(&ogg.os, &ogg.op);

							if(result==0) break; /* need more data */
							if(result<0) { /* missing or corrupt data at this page position */
							  /* no reason to complain; already complained above */
							} else {
								/* we have a packet.  Decode it */
								float **pcm;
								int samples;

								if(vorbis_synthesis(&ogg.vb, &ogg.op) == 0) /* test for success! */
								  vorbis_synthesis_blockin(&ogg.vd, &ogg.vb);
								/*

								**pcm is a multichannel float vector.  In stereo, for
								example, pcm[0] is left, and pcm[1] is right.  samples is
								the size of each channel.  Convert the float values
								(-1.<=range<=1.) to whatever PCM format and write it out */

								while((samples = vorbis_synthesis_pcmout(&ogg.vd, &pcm)) > 0){
									int j;
									/*
									int clipflag = 0;
									*/
									int bout = (samples < convsize ? samples : convsize);

									/* convert floats to 16 bit signed ints (host order) and
									   interleave */
									for(i = 0; i < ogg.vi.channels; i++){
										ogg_int16_t *ptr = oggDecode.conv_buffer + i;
										float *mono=pcm[i];
										for(j = 0; j < bout; j++){
							    
											int val = floor(mono[j]*32767.f+.5f);
								  
											/* might as well guard against clipping */
											if(val > 32767){
												val = 32767;
												/*
												clipflag = 1;
												*/
											}
											if(val<-32768){
												val = -32768;
												/*
												clipflag = 1;
												*/
											}
											*ptr = val;
											ptr += ogg.vi.channels;
										}
									}

									/*
									if(clipflag)
										fprintf(stderr, "Clipping in frame %ld\n", (long)(ogg.vd.sequence));
									*/

									eResult rslt_write = write((u_char*)oggDecode.conv_buffer, 2 * ogg.vi.channels * bout);
									if(rslt_write != _rslt_ok) {
										return(rslt_write);
									}
									//fwrite(convbuffer ,2*ogg.vi.channels, bout, stdout);

									vorbis_synthesis_read(&ogg.vd, bout); /* tell libvorbis how
													    many samples we
													    actually consumed */
								}
							}
						}
						if(ogg_page_eos(&ogg.og)) ogg.eos = 1;
					}
				}
				if(!ogg.eos) {
					oggDecode.sync_buffer = ogg_sync_buffer(&ogg.oy, oggDecode.sync_buffer_size);
					unsigned read_bytes = fread(oggDecode.sync_buffer, 1, oggDecode.sync_buffer_size, fileHandle);
					ogg_sync_wrote(&ogg.oy, read_bytes);
					if(read_bytes == 0) ogg.eos = 1;
				}
			}

			/* ogg_page and ogg_packet structs always point to storage in
			   libvorbis.  They're never freed or manipulated directly */

			vorbis_block_clear(&ogg.vb);
			vorbis_dsp_clear(&ogg.vd);
		} else {
			/*
			fprintf(stderr,"Error: Corrupt header during playback initialization.\n");
			*/
		}

		/* clean up this logical bitstream; before exit we see if we're
		   followed by another [chained] */

		ogg_stream_clear(&ogg.os);
		vorbis_comment_clear(&ogg.vc);
		vorbis_info_clear(&ogg.vi);  /* must be called last */
	}
	
	eResult rslt_write = write(NULL, 0);
	if(rslt_write != _rslt_ok) {
		return(rslt_write);
	}

	return(_rslt_ok);
}


cAudioConvert::eResult cAudioConvert::writeOggHeader() {
	ogg.eos = 0;
	
	vorbis_info_init(&ogg.vi);

	if(vorbis_encode_init_vbr(&ogg.vi, audioInfo.channels, audioInfo.sampleRate, oggQuality)) {
		return(_rslt_ogg_failed_encode_initialization);
	}
	
	if(!comment.empty()) {
		vorbis_comment_init(&ogg.vc);
		vorbis_comment_add_tag(&ogg.vc, "ENCODER", comment.c_str());
	}

	/* set up the analysis state and auxiliary encoding storage */
	vorbis_analysis_init(&ogg.vd, &ogg.vi);
	vorbis_block_init(&ogg.vd, &ogg.vb);

	/* set up our packet->stream encoder */
	/* pick a random serial number; that way we can more likely build
	   chained streams just by concatenation */
	srand(time(NULL));
	ogg_stream_init(&ogg.os, rand());

	/* Vorbis streams begin with three headers; the initial header (with
	   most of the codec setup parameters) which is mandated by the Ogg
	   bitstream spec.  The second header holds any comment fields.  The
	   third header holds the bitstream codebook.  We merely need to
	   make the headers, then pass them to libvorbis one at a time;
	   libvorbis handles the additional Ogg bitstream constraints */

	ogg_packet header;
	ogg_packet header_comm;
	ogg_packet header_code;

	vorbis_analysis_headerout(&ogg.vd, &ogg.vc, &header, &header_comm, &header_code);
	ogg_stream_packetin(&ogg.os, &header); /* automatically placed in its own page */
	ogg_stream_packetin(&ogg.os, &header_comm);
	ogg_stream_packetin(&ogg.os, &header_code);

	/* This ensures the actual
	 * audio data will start on a new page, as per spec
	 */
	while(ogg_stream_flush(&ogg.os ,&ogg.og) != 0) {
		eResult rslt_write = write(ogg.og.header, ogg.og.header_len);
		if(rslt_write != _rslt_ok) {
			return(rslt_write);
		}
		rslt_write = write(ogg.og.body, ogg.og.body_len);
		if(rslt_write != _rslt_ok) {
			return(rslt_write);
		}
		/*
		fwrite(ogg.og.header, 1, ogg.og.header_len, stdout);
		fwrite(ogg.og.body, 1, ogg.og.body_len, stdout);
		*/
	}
	
	return(_rslt_ok);
}

cAudioConvert::eResult cAudioConvert::writeOggData(u_char *data, unsigned datalen) {
	/* expose the buffer to submit data */
	float **analysis_buffer = vorbis_analysis_buffer(&ogg.vd, datalen);

	/* uninterleave samples */
	signed char *_data = (signed char*)data;
	for(unsigned i = 0; i < datalen / 4; i++){
		analysis_buffer[0][i] = ((_data[i*4+1]<<8)|
					(0x00ff&(int)_data[i*4]))/32768.f;
		analysis_buffer[1][i] = ((_data[i*4+3]<<8)|
					(0x00ff&(int)_data[i*4+2]))/32768.f;
	}

	/* tell the library how much we actually submitted */
	vorbis_analysis_wrote(&ogg.vd, datalen / 4);
	
	return(_writeOgg());
}

cAudioConvert::eResult cAudioConvert::writeOggEnd() {
	vorbis_analysis_wrote(&ogg.vd, 0);
	return(_writeOgg());
}

cAudioConvert::eResult cAudioConvert::_writeOgg() {
	/* vorbis does some data preanalysis, then divvies up blocks for
	   more involved (potentially parallel) processing.  Get a single
	   block for encoding now */
	while(vorbis_analysis_blockout(&ogg.vd, &ogg.vb) == 1) {

		/* analysis, assume we want to use bitrate management */
		vorbis_analysis(&ogg.vb, NULL);
		vorbis_bitrate_addblock(&ogg.vb);

		while(vorbis_bitrate_flushpacket(&ogg.vd, &ogg.op)) {

			/* weld the packet into the bitstream */
			ogg_stream_packetin(&ogg.os, &ogg.op);

			/* write out pages (if any) */
			while(ogg_stream_pageout(&ogg.os, &ogg.og) != 0) {
				eResult rslt_write = write(ogg.og.header, ogg.og.header_len);
				if(rslt_write != _rslt_ok) {
					return(rslt_write);
				}
				rslt_write = write(ogg.og.body, ogg.og.body_len);
				if(rslt_write != _rslt_ok) {
					return(rslt_write);
				}
				/*
				fwrite(ogg.og.header, 1, ogg.og.header_len, stdout);
				fwrite(ogg.og.body, 1, ogg.og.body_len, stdout);
				*/

				/* this could be set above, but for illustrative purposes, I do
				   it here (to show that vorbis does know where the stream ends) */

				if(ogg_page_eos(&ogg.og)) break;;
			}
		}
	}
	return(_rslt_ok);
}

cAudioConvert::eResult cAudioConvert::write(u_char *data, unsigned datalen) {
	if(destAudio) {
		eResult rslt = _rslt_ok;
		if(datalen) {
			if(!headerIsWrited) {
				destAudio->audioInfo = audioInfo;
				switch(destAudio->formatType) {
				case _format_raw:
					break;
				case _format_wav:
					rslt = destAudio->writeWavHeader();
					break;
				case _format_ogg:
					rslt = destAudio->writeOggHeader();
					break;
				}
				headerIsWrited = true;
			}
			if(rslt == _rslt_ok) {
				switch(destAudio->formatType) {
				case _format_raw:
					rslt = destAudio->write(data, datalen);
					break;
				case _format_wav:
					rslt = destAudio->writeWavData(data, datalen);
					break;
				case _format_ogg:
					rslt = destAudio->writeOggData(data, datalen);
					break;
				}
				headerIsWrited = true;
			}
		} else {
			switch(destAudio->formatType) {
			case _format_raw:
				break;
			case _format_wav:
				rslt = destAudio->writeWavEnd();
				break;
			case _format_ogg:
				rslt = destAudio->writeOggEnd();
				break;
			}
		}
	}
	if(srcDstType == _dst && !fileName.empty()) {
		if(!fileHandle) {
			fileHandle = fopen(fileName.c_str(), "w");
			if(!fileHandle) {
				return(_rslt_open_for_write_failed);
			}
		}
		if(fileHandle) {
			return(fwrite(data, 1, datalen, fileHandle) == datalen ?
				_rslt_ok :
				_rslt_write_failed);
		}
	}
	return(_rslt_ok);
}

void cAudioConvert::test() {
 
	{
	cAudioConvert info;
	info.fileName = "/home/jumbox/Plocha/ac/1781060762.ogg";
	info.getAudioInfo();
	cout << info.jsonAudioInfo() <<  endl;
	}
 
	{
	cAudioConvert info;
	info.fileName = "/home/jumbox/Plocha/ac/1781060762.wav";
	info.getAudioInfo();
	cout << info.jsonAudioInfo() <<  endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/1781060762.ogg";
	cAudioConvert dst;
	dst.formatType = _format_wav;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/1781060762-2.wav";
	src.destAudio = &dst;
	cout << "1: " << src.readOgg() << endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/1781060762-2.wav";
	cAudioConvert dst;
	dst.formatType = _format_ogg;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/1781060762-2.ogg";
	src.destAudio = &dst;
	cout << "2: " << src.readWav() << endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/1781060762-2.wav";
	cAudioConvert dst;
	dst.formatType = _format_raw;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/1781060762-2.raw";
	src.destAudio = &dst;
	cout << "3: " << src.readWav() << endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/1781060762-2.raw";
	cAudioConvert dst;
	dst.formatType = _format_wav;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/1781060762-3.wav";
	src.destAudio = &dst;
	sAudioInfo ai;
	ai.sampleRate = 8000;
	ai.channels = 2;
	ai.bitsPerSample = 16;
	cout << "4: " << src.readRaw(&ai) << endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/1781060762-2.raw";
	cAudioConvert dst;
	dst.formatType = _format_ogg;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/1781060762-3.ogg";
	src.destAudio = &dst;
	sAudioInfo ai;
	ai.sampleRate = 8000;
	ai.channels = 2;
	ai.bitsPerSample = 16;
	cout << "5: " << src.readRaw(&ai) << endl;
	}
	
	{
	cAudioConvert src;
	src.fileName = "/home/jumbox/Plocha/ac/test.raw";
	cAudioConvert dst;
	dst.formatType = _format_wav;
	dst.srcDstType = _dst;
	dst.fileName = "/home/jumbox/Plocha/ac/test.wav";
	src.destAudio = &dst;
	sAudioInfo ai;
	ai.sampleRate = 8000;
	ai.channels = 2;
	ai.bitsPerSample = 16;
	cout << "6: " << src.readRaw(&ai) << endl;
	}
}