/*

** Copyright (C) 2004-2017 Erik de Castro Lopo <erikd@mega-nerd.com>

**

** This program is free software; you can redistribute it and/or modify

** it under the terms of the GNU Lesser General Public License as published by

** the Free Software Foundation; either version 2.1 of the License, or

** (at your option) any later version.

**

** This program 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 Lesser General Public License for more details.

**

** You should have received a copy of the GNU Lesser General Public License

** along with this program; if not, write to the Free Software

** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

*/

#include "sfconfig.h"

#include <stdio.h>

#include <string.h>

#include "sndfile.h"

#include "sfendian.h"

#include "common.h"

#define TWOBIT_MARKER (MAKE_MARKER ('2', 'B', 'I', 'T'))

#define AVR_HDR_SIZE  128

/*

** From: hyc@hanauma.Jpl.Nasa.Gov (Howard Chu)

**

** A lot of PD software exists to play Mac .snd files on the ST. One other

** format that seems pretty popular (used by a number of commercial packages)

** is the AVR format (from Audio Visual Research). This format has a 128 byte

** header that looks like this (its actually packed, but thats not portable):

*/

typedef struct

{ int   marker ;  /* 2BIT */

  char  name [8] ;  /* null-padded sample name */

  short mono ;    /* 0 = mono, 0xffff = stereo */

  short rez ;   /* 8 = 8 bit, 16 = 16 bit */

  short sign ;    /* 0 = unsigned, 0xffff = signed */

  short loop ;    /* 0 = no loop, 0xffff = looping sample */

  short midi ;    /* 0xffff = no MIDI note assigned,  */

            /*  0xffXX = single key note assignment */

            /*  0xLLHH = key split, low/hi note */

  int   srate ;   /* sample frequency in hertz */

  int   frames ;  /* sample length in bytes or words (see rez) */

  int   lbeg ;    /* offset to start of loop in bytes or words. */

            /* set to zero if unused */

  int   lend ;    /* offset to end of loop in bytes or words. */

            /* set to sample length if unused */

  short res1 ;    /* Reserved, MIDI keyboard split */

  short res2 ;    /* Reserved, sample compression */

  short res3 ;    /* Reserved */

  char  ext [20] ;  /* Additional filename space, used if (name[7] != 0) */

  char  user [64] ; /* User defined. Typically ASCII message */

} AVR_HEADER ;

/*------------------------------------------------------------------------------

** Private static functions.

*/

static int    avr_close (SF_PRIVATE *psf) ;

static int    avr_read_header (SF_PRIVATE *psf) ;

static int    avr_write_header (SF_PRIVATE *psf, int calc_length) ;

/*------------------------------------------------------------------------------

** Public function.

*/

int

avr_open  (SF_PRIVATE *psf)

{ int   error = 0 ;

  if (psf->file.mode == SFM_READ || (psf->file.mode == SFM_RDWR && psf->filelength > 0))

  { if ((error = avr_read_header (psf)))

      return error ;

    } ;

  if ((SF_CONTAINER (psf->sf.format)) != SF_FORMAT_AVR)

    return  SFE_BAD_OPEN_FORMAT ;

  if (psf->file.mode == SFM_WRITE || psf->file.mode == SFM_RDWR)

  { psf->endian = SF_ENDIAN_BIG ;

    if (avr_write_header (psf, SF_FALSE))

      return psf->error ;

    psf->write_header = avr_write_header ;

    } ;

  psf->container_close = avr_close ;

  psf->blockwidth = psf->bytewidth * psf->sf.channels ;

  error = pcm_init (psf) ;

  return error ;

} /* avr_open */

static int

avr_read_header (SF_PRIVATE *psf)

{ AVR_HEADER  hdr ;

  memset (&hdr, 0, sizeof (hdr)) ;

  psf_binheader_readf (psf, "pmb", 0, &hdr.marker, &hdr.name, sizeof (hdr.name)) ;

  psf_log_printf (psf, "%M\n", hdr.marker) ;

  if (hdr.marker != TWOBIT_MARKER)

    return SFE_AVR_NOT_AVR ;

  psf_log_printf (psf, "  Name        : %s\n", hdr.name) ;

  psf_binheader_readf (psf, "E22222", &hdr.mono, &hdr.rez, &hdr.sign, &hdr.loop, &hdr.midi) ;

  psf->sf.channels = (hdr.mono & 1) + 1 ;

  psf_log_printf (psf, "  Channels    : %d\n  Bit width   : %d\n  Signed      : %s\n",

      (hdr.mono & 1) + 1, hdr.rez, hdr.sign ? "yes" : "no") ;

  switch (arith_shift_left (hdr.rez, 16) + (hdr.sign & 1))

  { case ((8 << 16) + 0) :

      psf->sf.format = SF_FORMAT_AVR | SF_FORMAT_PCM_U8 ;

      psf->bytewidth = 1 ;

      break ;

    case ((8 << 16) + 1) :

      psf->sf.format = SF_FORMAT_AVR | SF_FORMAT_PCM_S8 ;

      psf->bytewidth = 1 ;

      break ;

    case ((16 << 16) + 1) :

      psf->sf.format = SF_FORMAT_AVR | SF_FORMAT_PCM_16 ;

      psf->bytewidth = 2 ;

      break ;

    default :

      psf_log_printf (psf, "Error : bad rez/sign combination.\n") ;

      return SFE_AVR_BAD_REZ_SIGN ;

    } ;

  psf_binheader_readf (psf, "E4444", &hdr.srate, &hdr.frames, &hdr.lbeg, &hdr.lend) ;

  psf->sf.frames = hdr.frames ;

  psf->sf.samplerate = hdr.srate ;

  psf_log_printf (psf, "  Frames      : %D\n", psf->sf.frames) ;

  psf_log_printf (psf, "  Sample rate : %d\n", psf->sf.samplerate) ;

  psf_binheader_readf (psf, "E222", &hdr.res1, &hdr.res2, &hdr.res3) ;

  psf_binheader_readf (psf, "bb", hdr.ext, sizeof (hdr.ext), hdr.user, sizeof (hdr.user)) ;

  psf_log_printf (psf, "  Ext         : %s\n  User        : %s\n", hdr.ext, hdr.user) ;

  psf->endian = SF_ENDIAN_BIG ;

  psf->dataoffset = AVR_HDR_SIZE ;

  psf->datalength = (sf_count_t) hdr.frames * (hdr.rez / 8) ;

  if (psf->fileoffset > 0)

    psf->filelength = AVR_HDR_SIZE + psf->datalength ;

  if (psf_ftell (psf) != psf->dataoffset)

    psf_binheader_readf (psf, "j", psf->dataoffset - psf_ftell (psf)) ;

  psf->blockwidth = psf->sf.channels * psf->bytewidth ;

  if (psf->sf.frames == 0 && psf->blockwidth)

    psf->sf.frames = (psf->filelength - psf->dataoffset) / psf->blockwidth ;

  return 0 ;

} /* avr_read_header */

static int

avr_write_header (SF_PRIVATE *psf, int calc_length)

{ sf_count_t  current ;

  int     sign ;

  if (psf->pipeoffset > 0)

    return 0 ;

  current = psf_ftell (psf) ;

  if (calc_length)

  { psf->filelength = psf_get_filelen (psf) ;

    psf->datalength = psf->filelength - psf->dataoffset ;

    if (psf->dataend)

      psf->datalength -= psf->filelength - psf->dataend ;

    psf->sf.frames = psf->datalength / (psf->bytewidth * psf->sf.channels) ;

    } ;

  /* Reset the current header length to zero. */

  psf->header.ptr [0] = 0 ;

  psf->header.indx = 0 ;

  /*

  ** Only attempt to seek if we are not writng to a pipe. If we are

  ** writing to a pipe we shouldn't be here anyway.

  */

  if (psf->is_pipe == SF_FALSE)

    psf_fseek (psf, 0, SEEK_SET) ;

  psf_binheader_writef (psf, "Emz22", BHWm (TWOBIT_MARKER), BHWz (8),

      BHW2 (psf->sf.channels == 2 ? 0xFFFF : 0), BHW2 (psf->bytewidth * 8)) ;

  sign = ((SF_CODEC (psf->sf.format)) == SF_FORMAT_PCM_U8) ? 0 : 0xFFFF ;

  psf_binheader_writef (psf, "E222", BHW2 (sign), BHW2 (0), BHW2 (0xFFFF)) ;

  psf_binheader_writef (psf, "E4444", BHW4 (psf->sf.samplerate), BHW4 (psf->sf.frames), BHW4 (0), BHW4 (0)) ;

  psf_binheader_writef (psf, "E222zz", BHW2 (0), BHW2 (0), BHW2 (0), BHWz (20), BHWz (64)) ;

  /* Header construction complete so write it out. */

  psf_fwrite (psf->header.ptr, psf->header.indx, 1, psf) ;

  if (psf->error)

    return psf->error ;

  psf->dataoffset = psf->header.indx ;

  if (current > 0)

    psf_fseek (psf, current, SEEK_SET) ;

  return psf->error ;

} /* avr_write_header */

static int

avr_close (SF_PRIVATE *psf)

{

  if (psf->file.mode == SFM_WRITE || psf->file.mode == SFM_RDWR)

    avr_write_header (psf, SF_TRUE) ;

  return 0 ;

} /* avr_close */