cannam@85: /*
cannam@85: ** Copyright (C) 1999-2011 Erik de Castro Lopo <erikd@mega-nerd.com>
cannam@85: ** Copyright (C) 2008 George Blood Audio
cannam@85: **
cannam@85: ** All rights reserved.
cannam@85: **
cannam@85: ** Redistribution and use in source and binary forms, with or without
cannam@85: ** modification, are permitted provided that the following conditions are
cannam@85: ** met:
cannam@85: **
cannam@85: **     * Redistributions of source code must retain the above copyright
cannam@85: **       notice, this list of conditions and the following disclaimer.
cannam@85: **     * Redistributions in binary form must reproduce the above copyright
cannam@85: **       notice, this list of conditions and the following disclaimer in
cannam@85: **       the documentation and/or other materials provided with the
cannam@85: **       distribution.
cannam@85: **     * Neither the author nor the names of any contributors may be used
cannam@85: **       to endorse or promote products derived from this software without
cannam@85: **       specific prior written permission.
cannam@85: **
cannam@85: ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
cannam@85: ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
cannam@85: ** TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
cannam@85: ** PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
cannam@85: ** CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
cannam@85: ** EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
cannam@85: ** PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
cannam@85: ** OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
cannam@85: ** WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
cannam@85: ** OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
cannam@85: ** ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
cannam@85: */
cannam@85: 
cannam@85: #include <stdio.h>
cannam@85: #include <stdlib.h>
cannam@85: #include <string.h>
cannam@85: #include <ctype.h>
cannam@85: #include <stdint.h>
cannam@85: 
cannam@85: #include <sndfile.h>
cannam@85: 
cannam@85: #include "common.h"
cannam@85: 
cannam@85: #define	 BUFFER_LEN	4096
cannam@85: 
cannam@85: #define	MIN(x,y)	((x) < (y) ? (x) : (y))
cannam@85: 
cannam@85: void
cannam@85: sfe_copy_data_fp (SNDFILE *outfile, SNDFILE *infile, int channels)
cannam@85: {	static double	data [BUFFER_LEN], max ;
cannam@85: 	int		frames, readcount, k ;
cannam@85: 
cannam@85: 	frames = BUFFER_LEN / channels ;
cannam@85: 	readcount = frames ;
cannam@85: 
cannam@85: 	sf_command (infile, SFC_CALC_SIGNAL_MAX, &max, sizeof (max)) ;
cannam@85: 
cannam@85: 	if (max < 1.0)
cannam@85: 	{	while (readcount > 0)
cannam@85: 		{	readcount = sf_readf_double (infile, data, frames) ;
cannam@85: 			sf_writef_double (outfile, data, readcount) ;
cannam@85: 			} ;
cannam@85: 		}
cannam@85: 	else
cannam@85: 	{	sf_command (infile, SFC_SET_NORM_DOUBLE, NULL, SF_FALSE) ;
cannam@85: 
cannam@85: 		while (readcount > 0)
cannam@85: 		{	readcount = sf_readf_double (infile, data, frames) ;
cannam@85: 			for (k = 0 ; k < readcount * channels ; k++)
cannam@85: 				data [k] /= max ;
cannam@85: 			sf_writef_double (outfile, data, readcount) ;
cannam@85: 			} ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	return ;
cannam@85: } /* sfe_copy_data_fp */
cannam@85: 
cannam@85: void
cannam@85: sfe_copy_data_int (SNDFILE *outfile, SNDFILE *infile, int channels)
cannam@85: {	static int	data [BUFFER_LEN] ;
cannam@85: 	int		frames, readcount ;
cannam@85: 
cannam@85: 	frames = BUFFER_LEN / channels ;
cannam@85: 	readcount = frames ;
cannam@85: 
cannam@85: 	while (readcount > 0)
cannam@85: 	{	readcount = sf_readf_int (infile, data, frames) ;
cannam@85: 		sf_writef_int (outfile, data, readcount) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	return ;
cannam@85: } /* sfe_copy_data_int */
cannam@85: 
cannam@85: /*==============================================================================
cannam@85: */
cannam@85: 
cannam@85: static int
cannam@85: merge_broadcast_info (SNDFILE * infile, SNDFILE * outfile, int format, const METADATA_INFO * info)
cannam@85: {	SF_BROADCAST_INFO_2K binfo ;
cannam@85: 	int infileminor ;
cannam@85: 
cannam@85: 	memset (&binfo, 0, sizeof (binfo)) ;
cannam@85: 
cannam@85: 	if ((SF_FORMAT_TYPEMASK & format) != SF_FORMAT_WAV)
cannam@85: 	{	printf ("Error : This is not a WAV file and hence broadcast info cannot be added to it.\n\n") ;
cannam@85: 		return 1 ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	infileminor = SF_FORMAT_SUBMASK & format ;
cannam@85: 
cannam@85: 	switch (infileminor)
cannam@85: 	{	case SF_FORMAT_PCM_16 :
cannam@85: 		case SF_FORMAT_PCM_24 :
cannam@85: 		case SF_FORMAT_PCM_32 :
cannam@85: 			break ;
cannam@85: 
cannam@85: 		default :
cannam@85: 			printf (
cannam@85: 				"Warning : The EBU Technical Recommendation R68-2000 states that the only\n"
cannam@85: 				"          allowed encodings are Linear PCM and MPEG3. This file is not in\n"
cannam@85: 				"          the right format.\n\n"
cannam@85: 				) ;
cannam@85: 			break ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (sf_command (infile, SFC_GET_BROADCAST_INFO, &binfo, sizeof (binfo)) == 0)
cannam@85: 	{	if (infile == outfile)
cannam@85: 		{	printf (
cannam@85: 				"Error : Attempting in-place broadcast info update, but file does not\n"
cannam@85: 				"        have a 'bext' chunk to modify. The solution is to specify both\n"
cannam@85: 				"        input and output files on the command line.\n\n"
cannam@85: 				) ;
cannam@85: 			return 1 ;
cannam@85: 			} ;
cannam@85: 		} ;
cannam@85: 
cannam@85: #define REPLACE_IF_NEW(x) \
cannam@85: 		if (info->x != NULL) \
cannam@85: 		{	memset (binfo.x, 0, sizeof (binfo.x)) ; \
cannam@85: 			memcpy (binfo.x, info->x, MIN (strlen (info->x), sizeof (binfo.x))) ; \
cannam@85: 			} ;
cannam@85: 
cannam@85: 	REPLACE_IF_NEW (description) ;
cannam@85: 	REPLACE_IF_NEW (originator) ;
cannam@85: 	REPLACE_IF_NEW (originator_reference) ;
cannam@85: 	REPLACE_IF_NEW (origination_date) ;
cannam@85: 	REPLACE_IF_NEW (origination_time) ;
cannam@85: 	REPLACE_IF_NEW (umid) ;
cannam@85: 
cannam@85: 	/* Special case for Time Ref. */
cannam@85: 	if (info->time_ref != NULL)
cannam@85: 	{	uint64_t ts = atoll (info->time_ref) ;
cannam@85: 
cannam@85: 		binfo.time_reference_high = (ts >> 32) ;
cannam@85: 		binfo.time_reference_low = (ts & 0xffffffff) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	/* Special case for coding_history because we may want to append. */
cannam@85: 	if (info->coding_history != NULL)
cannam@85: 	{	if (info->coding_hist_append)
cannam@85: 		{	int slen = strlen (binfo.coding_history) ;
cannam@85: 
cannam@85: 			while (slen > 1 && isspace (binfo.coding_history [slen - 1]))
cannam@85: 				slen -- ;
cannam@85: 
cannam@85: 			memcpy (binfo.coding_history + slen, info->coding_history, sizeof (binfo.coding_history) - slen) ;
cannam@85: 			}
cannam@85: 		else
cannam@85: 		{	size_t slen = MIN (strlen (info->coding_history), sizeof (binfo.coding_history)) ;
cannam@85: 
cannam@85: 			memset (binfo.coding_history, 0, sizeof (binfo.coding_history)) ;
cannam@85: 			memcpy (binfo.coding_history, info->coding_history, slen) ;
cannam@85: 			binfo.coding_history_size = slen ;
cannam@85: 			} ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (sf_command (outfile, SFC_SET_BROADCAST_INFO, &binfo, sizeof (binfo)) == 0)
cannam@85: 	{	printf ("Error : Setting of broadcast info chunks failed.\n\n") ;
cannam@85: 		return 1 ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	return 0 ;
cannam@85: } /* merge_broadcast_info*/
cannam@85: 
cannam@85: static void
cannam@85: update_strings (SNDFILE * outfile, const METADATA_INFO * info)
cannam@85: {
cannam@85: 	if (info->title != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_TITLE, info->title) ;
cannam@85: 
cannam@85: 	if (info->copyright != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_COPYRIGHT, info->copyright) ;
cannam@85: 
cannam@85: 	if (info->artist != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_ARTIST, info->artist) ;
cannam@85: 
cannam@85: 	if (info->comment != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_COMMENT, info->comment) ;
cannam@85: 
cannam@85: 	if (info->date != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_DATE, info->date) ;
cannam@85: 
cannam@85: 	if (info->album != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_ALBUM, info->album) ;
cannam@85: 
cannam@85: 	if (info->license != NULL)
cannam@85: 		sf_set_string (outfile, SF_STR_LICENSE, info->license) ;
cannam@85: 
cannam@85: } /* update_strings */
cannam@85: 
cannam@85: 
cannam@85: 
cannam@85: void
cannam@85: sfe_apply_metadata_changes (const char * filenames [2], const METADATA_INFO * info)
cannam@85: {	SNDFILE *infile = NULL, *outfile = NULL ;
cannam@85: 	SF_INFO sfinfo ;
cannam@85: 	METADATA_INFO tmpinfo ;
cannam@85: 	int error_code = 0 ;
cannam@85: 
cannam@85: 	memset (&sfinfo, 0, sizeof (sfinfo)) ;
cannam@85: 	memset (&tmpinfo, 0, sizeof (tmpinfo)) ;
cannam@85: 
cannam@85: 	if (filenames [1] == NULL)
cannam@85: 		infile = outfile = sf_open (filenames [0], SFM_RDWR, &sfinfo) ;
cannam@85: 	else
cannam@85: 	{	infile = sf_open (filenames [0], SFM_READ, &sfinfo) ;
cannam@85: 
cannam@85: 		/* Output must be WAV. */
cannam@85: 		sfinfo.format = SF_FORMAT_WAV | (SF_FORMAT_SUBMASK & sfinfo.format) ;
cannam@85: 		outfile = sf_open (filenames [1], SFM_WRITE, &sfinfo) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (infile == NULL)
cannam@85: 	{	printf ("Error : Not able to open input file '%s' : %s\n", filenames [0], sf_strerror (infile)) ;
cannam@85: 		error_code = 1 ;
cannam@85: 		goto cleanup_exit ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (outfile == NULL)
cannam@85: 	{	printf ("Error : Not able to open output file '%s' : %s\n", filenames [1], sf_strerror (outfile)) ;
cannam@85: 		error_code = 1 ;
cannam@85: 		goto cleanup_exit ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (info->has_bext_fields && merge_broadcast_info (infile, outfile, sfinfo.format, info))
cannam@85: 	{	error_code = 1 ;
cannam@85: 		goto cleanup_exit ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	if (infile != outfile)
cannam@85: 	{	int infileminor = SF_FORMAT_SUBMASK & sfinfo.format ;
cannam@85: 
cannam@85: 		/* If the input file is not the same as the output file, copy the data. */
cannam@85: 		if ((infileminor == SF_FORMAT_DOUBLE) || (infileminor == SF_FORMAT_FLOAT))
cannam@85: 			sfe_copy_data_fp (outfile, infile, sfinfo.channels) ;
cannam@85: 		else
cannam@85: 			sfe_copy_data_int (outfile, infile, sfinfo.channels) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	update_strings (outfile, info) ;
cannam@85: 
cannam@85: cleanup_exit :
cannam@85: 
cannam@85: 	if (outfile != NULL && outfile != infile)
cannam@85: 		sf_close (outfile) ;
cannam@85: 
cannam@85: 	if (infile != NULL)
cannam@85: 		sf_close (infile) ;
cannam@85: 
cannam@85: 	if (error_code)
cannam@85: 		exit (error_code) ;
cannam@85: 
cannam@85: 	return ;
cannam@85: } /* sfe_apply_metadata_changes */
cannam@85: 
cannam@85: /*==============================================================================
cannam@85: */
cannam@85: 
cannam@85: typedef struct
cannam@85: {	const char	*ext ;
cannam@85: 	int			len ;
cannam@85: 	int			format ;
cannam@85: } OUTPUT_FORMAT_MAP ;
cannam@85: 
cannam@85: static OUTPUT_FORMAT_MAP format_map [] =
cannam@85: {
cannam@85: 	{	"aif",		3,	SF_FORMAT_AIFF	},
cannam@85: 	{	"wav", 		0,	SF_FORMAT_WAV	},
cannam@85: 	{	"au",		0,	SF_FORMAT_AU	},
cannam@85: 	{	"caf",		0,	SF_FORMAT_CAF	},
cannam@85: 	{	"flac",		0,	SF_FORMAT_FLAC	},
cannam@85: 	{	"snd",		0,	SF_FORMAT_AU	},
cannam@85: 	{	"svx",		0,	SF_FORMAT_SVX	},
cannam@85: 	{	"paf",		0,	SF_ENDIAN_BIG | SF_FORMAT_PAF	},
cannam@85: 	{	"fap",		0,	SF_ENDIAN_LITTLE | SF_FORMAT_PAF	},
cannam@85: 	{	"gsm",		0,	SF_FORMAT_RAW	},
cannam@85: 	{	"nist", 	0,	SF_FORMAT_NIST	},
cannam@85: 	{	"htk",		0,	SF_FORMAT_HTK	},
cannam@85: 	{	"ircam",	0,	SF_FORMAT_IRCAM	},
cannam@85: 	{	"sf",		0, 	SF_FORMAT_IRCAM	},
cannam@85: 	{	"voc",		0, 	SF_FORMAT_VOC	},
cannam@85: 	{	"w64", 		0, 	SF_FORMAT_W64	},
cannam@85: 	{	"raw",		0,	SF_FORMAT_RAW	},
cannam@85: 	{	"mat4", 	0,	SF_FORMAT_MAT4	},
cannam@85: 	{	"mat5", 	0, 	SF_FORMAT_MAT5 	},
cannam@85: 	{	"mat",		0, 	SF_FORMAT_MAT4 	},
cannam@85: 	{	"pvf",		0, 	SF_FORMAT_PVF 	},
cannam@85: 	{	"sds",		0, 	SF_FORMAT_SDS 	},
cannam@85: 	{	"sd2",		0, 	SF_FORMAT_SD2 	},
cannam@85: 	{	"vox",		0, 	SF_FORMAT_RAW 	},
cannam@85: 	{	"xi",		0, 	SF_FORMAT_XI 	},
cannam@85: 	{	"wve",		0,	SF_FORMAT_WVE	},
cannam@85: 	{	"oga",		0,	SF_FORMAT_OGG	},
cannam@85: 	{	"mpc",		0,	SF_FORMAT_MPC2K	},
cannam@85: 	{	"rf64",		0,	SF_FORMAT_RF64	},
cannam@85: } ; /* format_map */
cannam@85: 
cannam@85: int
cannam@85: sfe_file_type_of_ext (const char *str, int format)
cannam@85: {	char	buffer [16], *cptr ;
cannam@85: 	int		k ;
cannam@85: 
cannam@85: 	format &= SF_FORMAT_SUBMASK ;
cannam@85: 
cannam@85: 	if ((cptr = strrchr (str, '.')) == NULL)
cannam@85: 		return 0 ;
cannam@85: 
cannam@85: 	strncpy (buffer, cptr + 1, 15) ;
cannam@85: 	buffer [15] = 0 ;
cannam@85: 
cannam@85: 	for (k = 0 ; buffer [k] ; k++)
cannam@85: 		buffer [k] = tolower ((buffer [k])) ;
cannam@85: 
cannam@85: 	if (strcmp (buffer, "gsm") == 0)
cannam@85: 		return SF_FORMAT_RAW | SF_FORMAT_GSM610 ;
cannam@85: 
cannam@85: 	if (strcmp (buffer, "vox") == 0)
cannam@85: 		return SF_FORMAT_RAW | SF_FORMAT_VOX_ADPCM ;
cannam@85: 
cannam@85: 	for (k = 0 ; k < (int) (sizeof (format_map) / sizeof (format_map [0])) ; k++)
cannam@85: 	{	if (format_map [k].len > 0 && strncmp (buffer, format_map [k].ext, format_map [k].len) == 0)
cannam@85: 			return format_map [k].format | format ;
cannam@85: 		else if (strcmp (buffer, format_map [k].ext) == 0)
cannam@85: 			return format_map [k].format | format ;
cannam@85: 		} ;
cannam@85: 
cannam@85: 	/* Default if all the above fails. */
cannam@85: 	return (SF_FORMAT_WAV | SF_FORMAT_PCM_24) ;
cannam@85: } /* sfe_file_type_of_ext */
cannam@85: 
cannam@85: void
cannam@85: sfe_dump_format_map (void)
cannam@85: {	SF_FORMAT_INFO	info ;
cannam@85: 	int k ;
cannam@85: 
cannam@85: 	for (k = 0 ; k < ARRAY_LEN (format_map) ; k++)
cannam@85: 	{	info.format = format_map [k].format ;
cannam@85: 		sf_command (NULL, SFC_GET_FORMAT_INFO, &info, sizeof (info)) ;
cannam@85: 		printf ("        %-10s : %s\n", format_map [k].ext, info.name == NULL ? "????" : info.name) ;
cannam@85: 		} ;
cannam@85: 
cannam@85: } /* sfe_dump_format_map */
cannam@85: 
cannam@85: const char *
cannam@85: program_name (const char * argv0)
cannam@85: {	const char * tmp ;
cannam@85: 
cannam@85: 	tmp = strrchr (argv0, '/') ;
cannam@85: 	argv0 = tmp ? tmp + 1 : argv0 ;
cannam@85: 
cannam@85: 	tmp = strrchr (argv0, '/') ;
cannam@85: 	argv0 = tmp ? tmp + 1 : argv0 ;
cannam@85: 
cannam@85: 	/* Remove leading libtool name mangling. */
cannam@85: 	if (strstr (argv0, "lt-") == argv0)
cannam@85: 		return argv0 + 3 ;
cannam@85: 
cannam@85: 	return argv0 ;
cannam@85: } /* program_name */