comparison toolboxes/bioakustik_tools/sp/seq_wavwrite.m @ 0:e9a9cd732c1e tip

first hg version after svn

+function seq_wavwrite(in_files,wavefile,varargin)
+% Combines wav-data from folder or specific files in one file. 
+%
+% seq_wavwrite(path_string,wavefile) combines all .wav files in
+%   path_string in the file wavefile.
+%   i.e.: seq_wavwrite('C:\wav\','C:\combined.wav') 
+%
+%   The file format will depend on the first file processed, the other
+%   files will be transfered to this format by resampling, dithering and
+%   channel repetition/rejection. You may use some of the extra-options 'Fs','nbits'
+%   or 'channels' to override these settings.
+%   i.e.: seq_wavwrite('C:\wav\','C:\combined.wav','Fs',44100,'nbits',16,'channels',1) 
+%   will produce a mono file with 44.1 kHz samle rate and 16bits per sample
+%   
+% seq_wavwrite(files_cellarray,wavefile) only combines the files specified
+%   in files_cellarray.
+%   i.e.: files={'C:\wav\test1.wav','C:\other_wav\test2.wav'};
+%         seq_wavwrite(files,'C:\combined.wav');
+%
+%   You may want to copy only some parts of the files.
+%   Therefore use the extra-option 'sequences':
+%   seq_wavwrite(files,'C:\combined','sequences',segments);
+%   ,where segments is an cell array the same size as the files_cellarray,
+%   witch contains the position of the parts for every file.
+%   Every cell row contains a struct array with the following fields:
+%       abs_startspl and abs_stopspl
+%           or
+%       abs_startms and abs_stopms
+%   You may also specify the channels that are to be copied in the field
+%       channels
+%       i.e.: files={'C:\wav\test1.wav','C:\other_wav\test2.wav'};
+%             segsforfile1(1).abs_startspl=1;
+%             segsforfile1(1).abs_stopspl=44100;
+%             segsforfile1(2).abs_startspl=88200;
+%             segsforfile1(2).abs_stopspl=200000;
+%             segsforfile2(1).abs_startms=1;
+%             segsforfile2(1).abs_stopms=2000;
+%             segsforfile2(1).channels=[1 2]; <- use the first two channels
+%             segments={segsforfile1,segsforfile2};
+%             seq_wavwrite(files,'C:\combined','sequences',segments);
+%
+%   If you want to copy specific files as a whole, just omit their abs_...
+%   values.
+%
+%   seq_wavwrite uses blockwise file processing to be able to copy large
+%   amounts of data. The option 'max_chunksize' allows you to specify the
+%   blocksize in samples. Keep in mind that in multichannel mode the actual
+%   blocksize will be chunksize times channels.
+%   i.e.: seq_wavwrite('C:\wav\','C:\combined.wav','max_chunksize',44100*60) 
+
+% Parse inputs:
+
+[slash,leftargs]=process_options(varargin,'systemslash','\');
+
+if ischar(in_files)
+    data_names=dir(strcat(in_files,slash,'*.wav'));
+    in_files=strcat(in_files,{data_names.name});
+end
+
+[tmp_sig,tmp_Fs,tmp_nbits]=wavread(char(in_files{1}),[1 2]);
+tmp_channels=size(tmp_sig,2);
+
+def_max_chunk_size = 44100*60*2;%chunksize nearly one minute at 44,1 khz sample rate
+[sequences,Fs,nbits,channels,max_chunk_size]=process_options(leftargs,'sequences',[],'Fs',...
+    tmp_Fs,'nbits',tmp_nbits,'channels',tmp_channels,'max_chunksize',def_max_chunk_size);
+
+if ischar(in_files) && ~isempty(sequences)
+    warning('segment parameters ignored in directory-input mode')
+    sequences=[];
+end
+
+% Determine number of bytes in chunks
+% (not including pad bytes, if needed):
+% ----------------------------------
+%  'RIFF'           4 bytes
+%  size             4 bytes
+%  'WAVE'           4 bytes
+%  'fmt '           4 bytes
+%  size             4 bytes
+% <wave-format>     14 bytes
+% <format_specific> 2 bytes (PCM)
+%  'data'           4 bytes
+%  size             4 bytes
+% <wave-data>       N bytes
+% ----------------------------------
+
+bytes_per_sample = ceil(nbits/8);
+fmt_cksize  = 16;               % Don't include 'fmt ' or its size field
+
+% Open file for output:
+[fid,err] = OpenWaveWrite(wavefile);
+error(err);
+try
+    % Prepare basic chunk structure fields:
+    ck=[]; ck.fid=fid; ck.filename = wavefile;
+
+    fwrite(fid,zeros(1,20),'uchar'); %skip previous chunks
+    % Write <wave-format>:
+    fmt.filename        = wavefile;
+    if nbits == 32,
+        fmt.wFormatTag  = 3;            % Data encoding format (1=PCM, 3=Type 3 32-bit)
+    else
+        fmt.wFormatTag  = 1;            
+    end
+    fmt.nSamplesPerSec  = Fs;           % Samples per second
+    fmt.nAvgBytesPerSec = channels*bytes_per_sample*Fs; % Avg transfer rate
+    fmt.nBlockAlign     = channels*bytes_per_sample;    % Block alignment
+    fmt.nBitsPerSample  = nbits;        % standard <PCM-format-specific> info
+    fmt.nChannels       = channels;     % Number of channels
+    error(write_wavefmt(fid,fmt));
+    
+    fwrite(fid,zeros(1,8),'uchar'); %skip following chunks
+
+    % Write all audio data
+    sample_sum=0;
+    for filei=1:size(in_files,2)
+        resamplewarn=0;
+        channelwarn=0;
+        if ~isempty(sequences)&& ~isempty(sequences{filei})
+            numsegs=size(sequences{filei},2);
+        else numsegs=1;
+        end
+        for seqi=1:numsegs;
+            tmp_fsiz=wavread(char(in_files{filei}),'size');
+            tmp_fsiz=tmp_fsiz(1);
+            [y,tmp_fs,null]=wavread(char(in_files{filei}),[1 2]);%read data
+            if ~isempty(sequences) && ~isempty(sequences{filei})
+                if isfield(sequences{filei}(seqi),'abs_startspl')
+                    spl_seq=[sequences{filei}(seqi).abs_startspl sequences{filei}(seqi).abs_stopspl];
+                elseif isfield(sequences{filei}(seqi),'abs_startms')
+                    spl_seq=floor([sequences{filei}(seqi).abs_startms sequences{filei}(seqi).abs_stopms].*tmp_fs./1000);
+                else
+                    spl_seq=[1 tmp_fsiz];
+                end
+                if (spl_seq(1)< 1) || (spl_seq(2) > tmp_fsiz)
+                    warning('correcting segment range, not necessary critical in miliseconds-mode')
+                    spl_seq(1)=max(spl_seq(1),1);
+                    spl_seq(2)=min(spl_seq(2),tmp_fsiz);
+                end
+            else
+                 spl_seq=[1 tmp_fsiz];
+            end
+            win_start=spl_seq(1);
+            win_stop=(min(spl_seq(2),spl_seq(1)+max_chunk_size-1));
+            while win_stop <= spl_seq(2)
+                [y,tmp_fs,null]=wavread(char(in_files{filei}),[win_start win_stop]);%read data
+                if (size(y,2) > 1) &&  ~isempty(sequences) && isfield(sequences{filei}(seqi),'channels') %choose channel
+                    if size(y,2) >= max(sequences{filei}(seqi).channels)
+                        y=y(:,sequences{filei}(seqi).channels);
+                    else
+                        if ~channelwarn
+                            warning('ignoring errorneous channel field');
+                            channelwarn=1;
+                        end
+                    end
+                end
+                if (tmp_fs ~= Fs) %resample data if necessary
+                    if ~resamplewarn
+                        fprintf('seq_wavwrite.m: resampling from %d to %d Hz. \n',tmp_fs,Fs);
+                        resamplewarn=1;
+                    end
+                    y=resample(y,Fs,tmp_fs);
+                end
+                [samples,akt_channels] = size(y);
+                if akt_channels > channels % if necessary make equivalent channelnum
+                    y=y(:,1:channels);
+                elseif akt_channels < channels
+                    y=[y repmat(y(:,end),1,channels-akt_channels)];
+                end
+                error(write_wavedat(fid,fmt,y));
+                sample_sum=sample_sum+samples;
+                
+                if win_stop == spl_seq(2), break;
+                end
+                win_start=win_start+max_chunk_size;
+                win_stop=(min(spl_seq(2),win_stop+max_chunk_size));
+            end
+        end
+    end
+    clear y;
+
+    total_samples    = sample_sum * channels;
+    total_bytes      = total_samples * bytes_per_sample;
+    data_cksize = total_bytes; 
+    
+    riff_cksize = 36+total_bytes;  
+    
+    % Determine pad bytes:
+    % Determine if a pad-byte must be appended to data chunk:
+    if rem(data_cksize, 2) ~= 0,
+       fwrite(fid,0,'uchar');
+    end 
+    data_pad    = rem(data_cksize,2);
+    riff_cksize = riff_cksize + data_pad; % + fmt_pad, always 0
+
+    % Write RIFF chunk:
+    fseek(fid,0,'bof');
+    ck.ID   = 'RIFF';
+    ck.Size = riff_cksize;
+    error(write_ckinfo(ck));
+    
+    % Write WAVE subchunk:
+    ck.ID   = 'WAVE';
+    ck.Size = [];  % Indicate a subchunk (no chunk size)
+    error(write_ckinfo(ck));
+    
+    % Write <fmt-ck>:
+    ck.ID   = 'fmt ';
+    ck.Size = fmt_cksize;
+    error(write_ckinfo(ck));
+
+    % Write <data-ck>:
+    fseek(fid,36,'bof');
+    ck.ID   = 'data';
+    ck.Size = data_cksize;
+    error(write_ckinfo(ck));
+    err='';
+catch
+    err=lasterr;
+end
+% Close file:
+fclose(fid);
+
+error(err);
+% end of wavwrite()
+
+
+% ------------------------------------------------------------------------
+% Private functions:
+% ------------------------------------------------------------------------
+
+
+% ------------------------------------------------------------------------
+function [fid,err] = OpenWaveWrite(wavefile)
+% OpenWaveWrite
+%   Open WAV file for writing.
+%   If filename does not contain an extension, add ".wav"
+
+fid = [];
+err = '';
+if ~isstr(wavefile),
+   err='Wave file name must be a string.'; return;
+end
+if isempty(findstr(wavefile,'.')),
+  wavefile=[wavefile '.wav'];
+end
+% Open file, little-endian:
+[fid,err] = fopen(wavefile,'wb','l');
+
+return
+
+
+% ------------------------------------------------------------------------
+function err = write_ckinfo(ck)
+% WRITE_CKINFO: Writes next RIFF chunk, but not the chunk data.
+%   Assumes the following fields in ck:
+%         .fid   File ID to an open file
+%         .ID    4-character string chunk identifier
+%         .Size  Size of chunk (empty if subchunk)
+%
+%
+%   Expects an open FID pointing to first byte of chunk header,
+%   and a chunk structure.
+%   ck.fid, ck.ID, ck.Size, ck.Data
+
+errmsg = ['Failed to write ' ck.ID ' chunk to WAVE file: ' ck.filename];
+err    = '';
+
+if (fwrite(ck.fid, ck.ID, 'char') ~= 4),
+   err=errmsg; return;
+end
+
+if ~isempty(ck.Size),
+  % Write chunk size:
+  if (fwrite(ck.fid, ck.Size, 'uint32') ~= 1),
+     err=errmsg; return;
+  end
+end
+
+return
+
+% ------------------------------------------------------------------------
+function err = write_wavefmt(fid, fmt)
+% WRITE_WAVEFMT: Write WAVE format chunk.
+%   Assumes fid points to the wave-format subchunk.
+%   Requires chunk structure to be passed, indicating
+%   the length of the chunk.
+
+errmsg = ['Failed to write WAVE format chunk to file' fmt.filename];
+err    = '';
+
+% Create <wave-format> data:
+if (fwrite(fid, fmt.wFormatTag,      'uint16') ~= 1) | ...
+   (fwrite(fid, fmt.nChannels,       'uint16') ~= 1) | ...
+   (fwrite(fid, fmt.nSamplesPerSec,  'uint32' ) ~= 1) | ...
+   (fwrite(fid, fmt.nAvgBytesPerSec, 'uint32' ) ~= 1) | ...
+   (fwrite(fid, fmt.nBlockAlign,     'uint16') ~= 1),
+   err=errmsg; return;
+end
+
+% Write format-specific info:
+if fmt.wFormatTag==1 | fmt.wFormatTag==3,
+  % Write standard <PCM-format-specific> info:
+  if (fwrite(fid, fmt.nBitsPerSample, 'uint16') ~= 1),
+     err=errmsg; return;
+  end
+  
+else
+  err='Unknown data format.';
+end
+
+return
+
+
+% -----------------------------------------------------------------------
+function y = PCM_Quantize(x, fmt)
+% PCM_Quantize:
+%   Scale and quantize input data, from [-1, +1] range to
+%   either an 8-, 16-, or 24-bit data range.
+
+% Clip data to normalized range [-1,+1]:
+ClipMsg  = ['Data clipped during write to file:' fmt.filename];
+ClipWarn = 0;
+
+% Determine slope (m) and bias (b) for data scaling:
+nbits = fmt.nBitsPerSample;
+m = 2.^(nbits-1);
+
+switch nbits
+case 8,
+   b=128;
+case {16,24},
+   b=0;
+otherwise,
+   error('Invalid number of bits specified.');
+end
+
+y = round(m .* x + b);
+
+% Determine quantized data limits, based on the
+% presumed input data limits of [-1, +1]:
+ylim = [-1 +1];
+qlim = m * ylim + b;
+qlim(2) = qlim(2)-1;
+
+% Clip data to quantizer limits:
+i = find(y < qlim(1));
+if ~isempty(i),
+   warning(ClipMsg); ClipWarn=1;
+   y(i) = qlim(1);
+end
+
+i = find(y > qlim(2));
+if ~isempty(i),
+   if ~ClipWarn, warning(ClipMsg); end
+   y(i) = qlim(2);
+end
+
+return
+
+
+% -----------------------------------------------------------------------
+function err = write_wavedat(fid,fmt,data)
+% WRITE_WAVEDAT: Write WAVE data chunk
+%   Assumes fid points to the wave-data chunk
+%   Requires <wave-format> structure to be passed.
+
+err = '';
+
+if fmt.wFormatTag==1 | fmt.wFormatTag==3,
+   % PCM Format
+   
+   % 32-bit Type 3 is normalized, so no scaling needed.
+   if fmt.nBitsPerSample ~= 32,
+       data = PCM_Quantize(data, fmt);
+   end
+   
+   switch fmt.nBitsPerSample
+   case 8,
+      dtype='uchar'; % unsigned 8-bit
+   case 16,
+      dtype='int16'; % signed 16-bit
+   case 24,
+	  dtype='bit24'; % signed 24-bit
+   case 32,
+      dtype='float'; % normalized 32-bit floating point
+   otherwise,
+      err = 'Invalid number of bits specified.'; return;
+   end
+   
+   % Write data, one row at a time (one sample from each channel):
+   [samples,channels] = size(data);
+   total_samples = samples*channels;
+   
+   if (fwrite(fid, reshape(data',total_samples,1), dtype) ~= total_samples),
+      err = 'Failed to write PCM data samples.'; return;
+   end
+   
+   
+else
+  % Unknown wave-format for data.
+  err = 'Unsupported data format.';
+end
+
+return
+
+% end of wavwrite.m