gri_wavfile.cc

来自「这是用python语言写的一个数字广播的信号处理工具包。利用它」· CC 代码 · 共 313 行

/* -*- c++ -*- */
/*
 * Copyright 2004,2008 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * GNU Radio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#include <gri_wavfile.h>
#include <cstring>
#include <stdint.h>

# define VALID_COMPRESSION_TYPE 0x0001

// WAV files are always little-endian, so we need some byte switching macros

// FIXME: These need to be refactored into a separate endianess header file
// as they duplicate routines defined in usrp/host/lib/legacy/usrp_bytesex.h

#ifdef WORDS_BIGENDIAN

#ifdef HAVE_BYTESWAP_H
#include <byteswap.h>
#else
#warning Using non-portable code (likely wrong other than ILP32).

static inline short int
bswap_16 (unsigned short int x)
{
  return ((((x) >> 8) & 0xff) | (((x) & 0xff) << 8));
}

static inline unsigned int
bswap_32 (unsigned int x)
{
  return ((((x) & 0xff000000) >> 24) | (((x) & 0x00ff0000) >>  8)	\
	  | (((x) & 0x0000ff00) <<  8) | (((x) & 0x000000ff) << 24));
}
#endif // HAVE_BYTESWAP_H

static inline uint32_t
host_to_wav(uint32_t x)
{
  return bswap_32(x);
}

static inline uint16_t
host_to_wav(uint16_t x)
{
  return bswap_16(x);
}

static inline int16_t
host_to_wav(int16_t x)
{
  return bswap_16(x);
}

static inline uint32_t
wav_to_host(uint32_t x)
{
  return bswap_32(x);
}

static inline uint16_t
wav_to_host(uint16_t x)
{
  return bswap_16(x);
}

static inline int16_t
wav_to_host(int16_t x)
{
  return bswap_16(x);
}

#else

static inline uint32_t
host_to_wav(uint32_t x)
{
  return x;
}

static inline uint16_t
host_to_wav(uint16_t x)
{
  return x;
}

static inline int16_t
host_to_wav(int16_t x)
{
  return x;
}

static inline uint32_t
wav_to_host(uint32_t x)
{
  return x;
}

static inline uint16_t
wav_to_host(uint16_t x)
{
  return x;
}

static inline int16_t
wav_to_host(int16_t x)
{
  return x;
}

#endif // WORDS_BIGENDIAN


bool
gri_wavheader_parse(FILE *fp,
		    unsigned int &sample_rate_o,
		    int &nchans_o,
		    int &bytes_per_sample_o,
		    int &first_sample_pos_o,
		    unsigned int &samples_per_chan_o)
{
  // _o variables take return values
  char str_buf[8] = {0};
  
  uint32_t file_size;
  uint32_t fmt_hdr_skip;
  uint16_t compression_type;
  uint16_t nchans;
  uint32_t sample_rate;
  uint32_t avg_bytes_per_sec;
  uint16_t block_align;
  uint16_t bits_per_sample;
  uint32_t chunk_size;
  
  size_t fresult;

  fresult = fread(str_buf, 1, 4, fp);
  if (fresult != 4 || strncmp(str_buf, "RIFF", 4) || feof(fp)) {
    return false;
  }
  
  fread(&file_size, 1, 4, fp);
  
  fresult = fread(str_buf, 1, 8, fp);
  if (fresult != 8 || strncmp(str_buf, "WAVEfmt ", 8) || feof(fp)) {
    return false;
  }
  
  fread(&fmt_hdr_skip, 1, 4, fp);
  
  fread(&compression_type, 1, 2, fp);
  if (wav_to_host(compression_type) != VALID_COMPRESSION_TYPE) {
    return false;
  }
  
  fread(&nchans,            1, 2, fp);
  fread(&sample_rate,       1, 4, fp);
  fread(&avg_bytes_per_sec, 1, 4, fp);
  fread(&block_align,       1, 2, fp);
  fread(&bits_per_sample,   1, 2, fp);
  
  if (ferror(fp)) {
    return false;
  }
  
  fmt_hdr_skip    = wav_to_host(fmt_hdr_skip);
  nchans          = wav_to_host(nchans);
  sample_rate     = wav_to_host(sample_rate);
  bits_per_sample = wav_to_host(bits_per_sample);
  
  if (bits_per_sample != 8 && bits_per_sample != 16) {
    return false;
  }
  
  fmt_hdr_skip -= 16;
  if (fmt_hdr_skip) {
    fseek(fp, fmt_hdr_skip, SEEK_CUR);
  }
  
  // data chunk
  fresult = fread(str_buf, 1, 4, fp);
  if (strncmp(str_buf, "data", 4)) {
    return false;
  }

  fread(&chunk_size, 1, 4, fp);
  if (ferror(fp)) {
    return false;
  }
  
  // More byte swapping
  chunk_size = wav_to_host(chunk_size);
  
  // Output values
  sample_rate_o      = (unsigned) sample_rate;
  nchans_o           = (int) nchans;
  bytes_per_sample_o = (int) (bits_per_sample / 8);
  first_sample_pos_o = (int) ftell(fp);
  samples_per_chan_o = (unsigned) (chunk_size / (bytes_per_sample_o * nchans));
  return true;
}


short int
gri_wav_read_sample(FILE *fp, int bytes_per_sample)
{
  int16_t buf = 0;
  fread(&buf, bytes_per_sample, 1, fp);
  
  return (short) wav_to_host(buf);
}


bool
gri_wavheader_write(FILE *fp,
		    unsigned int sample_rate,
		    int nchans,
		    int bytes_per_sample)
{
  const int header_len = 44;
  char wav_hdr[header_len] = "RIFF\0\0\0\0WAVEfmt \0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0data\0\0\0";
  uint16_t nchans_f        = (uint16_t) nchans;
  uint32_t sample_rate_f   = (uint32_t) sample_rate;
  uint16_t block_align     = bytes_per_sample * nchans;
  uint32_t avg_bytes       = sample_rate * block_align;
  uint16_t bits_per_sample = bytes_per_sample * 8;
  
  nchans_f        = host_to_wav(nchans_f);
  sample_rate_f   = host_to_wav(sample_rate_f);
  block_align     = host_to_wav(block_align);
  avg_bytes       = host_to_wav(avg_bytes);
  bits_per_sample = host_to_wav(bits_per_sample);
  
  wav_hdr[16] = 0x10; // no extra bytes
  wav_hdr[20] = 0x01; // no compression
  memcpy((void *) (wav_hdr + 22), (void *) &nchans_f,        2);
  memcpy((void *) (wav_hdr + 24), (void *) &sample_rate_f,   4);
  memcpy((void *) (wav_hdr + 28), (void *) &avg_bytes,       4);
  memcpy((void *) (wav_hdr + 32), (void *) &block_align,     2);
  memcpy((void *) (wav_hdr + 34), (void *) &bits_per_sample, 2);
  
  fwrite(&wav_hdr, 1, header_len, fp);
  if (ferror(fp)) {
    return false;
  }
  
  return true;
}


void
gri_wav_write_sample(FILE *fp, short int sample, int bytes_per_sample)
{
  void *data_ptr;
  unsigned char buf_8bit;
  int16_t       buf_16bit;
  
  if (bytes_per_sample == 1) {
    buf_8bit = (unsigned char) sample;
    data_ptr = (void *) &buf_8bit;
  } else {
    buf_16bit = host_to_wav((int16_t) sample);
    data_ptr  = (void *) &buf_16bit;
  }
  
  fwrite(data_ptr, 1, bytes_per_sample, fp);
}


bool
gri_wavheader_complete(FILE *fp, unsigned int byte_count)
{
  uint32_t chunk_size = (uint32_t) byte_count;
  chunk_size = host_to_wav(chunk_size);
  
  fseek(fp, 40, SEEK_SET);
  fwrite(&chunk_size, 1, 4, fp);
  
  chunk_size = (uint32_t) byte_count + 36; // fmt chunk and data header
  chunk_size = host_to_wav(chunk_size);
  fseek(fp, 4, SEEK_SET);
  
  fwrite(&chunk_size, 1, 4, fp);
  
  if (ferror(fp)) {
    return false;
  }
  
  return true;
}