FryingPan/DTLib/rWAVAudio.cpp

   1 /*
   2  * FryingPan - Amiga CD/DVD Recording Software (User Interface and supporting Libraries only)
   3  * Copyright (C) 2001-2011 Tomasz Wiszkowski Tomasz.Wiszkowski at gmail.com
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU Lesser General Public License
   7  * as published by the Free Software Foundation; either version 2.1
   8  * of the License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  * GNU Lesser General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU Lesser General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18  */
  19
  20 #include "rWAVAudio.h"
  21 #include <libclass/dos.h>
  22 #include <libclass/exec.h>
  23 #include <libclass/utility.h>
  24 #include <exec/lists.h>
  25 #include <Generic/HookT.h>
  26
  27 #define FLIP16(a) ((((a)&0xff00)>>8) | (((a)&0xff)<<8))
  28
  29 #define MKID(a,b,c,d) (((a&255)<<24)|((b&255)<<16)|((c&255)<<8)|(d&255))
  30 #define ID_RIFF MKID('R','I','F','F')
  31 #define ID_WAVE MKID('W','A','V','E')
  32 #define ID_fmt  MKID('f','m','t',' ')
  33 #define ID_data MKID('d','a','t','a')
  34
  35 struct WaveHdr
  36 {
  37    unsigned short    format;
  38    unsigned short    channels;
  39    unsigned long     frequency;
  40    unsigned long     bytesPerSecond;
  41    unsigned short    bytesPerSample;
  42    unsigned short    resolution;
  43 };
  44
  45 IFileReader         *rWAVAudio::openRead(const char* sFile, EDtError &rc)
  46 {
  47    rWAVAudio *pSkel = 0;
  48    if (true == rWAVAudio::checkFile(sFile, rc))
  49    {
  50       pSkel = new rWAVAudio(sFile, rc);
  51    }
  52    return pSkel;
  53 }
  54
  55 rWAVAudio::rWAVAudio(const char *sName, EDtError &rc)
  56    : FileReader(sName, rc)
  57 {
  58    if (rc != DT_OK)
  59       return;
  60
  61    readHeaders();
  62    setBlockSize(2352);
  63    setLittleEndian(true);
  64    setType(Data_Audio);
  65 }
  66
  67 const char          *rWAVAudio::static_getName()
  68 {
  69    return "WAV Audio Track";
  70 }
  71
  72 bool                 rWAVAudio::static_isAudio()
  73 {
  74    return true;
  75 }
  76
  77 bool                 rWAVAudio::static_isData()
  78 {
  79    return false;
  80 }
  81
  82 bool                 rWAVAudio::static_isSession()
  83 {
  84    return false;
  85 }
  86
  87 bool                 rWAVAudio::isAudio()
  88 {
  89    return static_isAudio();
  90 }
  91
  92 bool                 rWAVAudio::isData()
  93 {
  94    return static_isData();
  95 }
  96
  97 const char          *rWAVAudio::getName()
  98 {
  99    return static_getName();
 100 }
 101
 102 void                 rWAVAudio::readHeaders()
 103 {
 104    unsigned int     buff;
 105    int              c=0;
 106
 107    _D(Lvl_Info, "%s: Reading headers", (int)__PRETTY_FUNCTION__);
 108    DOS->Seek(getFile(), 0, OFFSET_BEGINNING);
 109
 110    if (DOS->Read(getFile(), &buff, 4)!=4)
 111    {
 112       _D(Lvl_Info, "%s: File seems to be too small.. reading failed.", (int)__PRETTY_FUNCTION__);
 113       return;
 114    }
 115    buff = L2BE(buff);
 116    if (buff != ID_RIFF)
 117    {
 118       _D(Lvl_Info, "%s: RIFF Wave file header not present (got: %08lx)", (int)__PRETTY_FUNCTION__, buff);
 119       return;
 120    }
 121
 122    DOS->Seek(getFile(), 4, OFFSET_CURRENT);
 123
 124    if (DOS->Read(getFile(), &buff, 4)!=4)
 125    {
 126       _D(Lvl_Info, "%s: File truncated..?", (int)__PRETTY_FUNCTION__);
 127       return;
 128    }
 129    buff = L2BE(buff);
 130    if (buff != ID_WAVE)
 131    {
 132       _D(Lvl_Info, "%s: This RIFF is not Wave... (%08lx)", (int)__PRETTY_FUNCTION__, buff);
 133       return;
 134    }
 135
 136    for (c=0; c<2;)
 137    {
 138       if (DOS->Read(getFile(), &buff, 4)!=4)
 139       {
 140          _D(Lvl_Info, "%s: File truncated...", (int)__PRETTY_FUNCTION__);
 141          return;
 142       }
 143
 144       _D(Lvl_Info, "%s: Read tag: %08lx", (int)__PRETTY_FUNCTION__, buff);
 145
 146       buff = L2BE(buff);
 147
 148       if (buff == ID_fmt)
 149       {
 150          _D(Lvl_Info, "%s: format tag located.", (int)__PRETTY_FUNCTION__);
 151          c++;
 152       }
 153       else if (buff == ID_data)
 154       {
 155          _D(Lvl_Info, "%s: data tag located.", (int)__PRETTY_FUNCTION__);
 156          c++;
 157          if (DOS->Read(getFile(), &buff, 4)!=4)
 158          {
 159             _D(Lvl_Info, "%s: file truncated.", (int)__PRETTY_FUNCTION__);
 160             return;
 161          }
 162
 163          setDataOffset(DOS->Seek(getFile(), 0, OFFSET_CURRENT));
 164          buff = L2LE(buff);
 165          setDataSize(buff);
 166          DOS->Seek(getFile(), -4, OFFSET_CURRENT);
 167       }
 168       if (DOS->Read(getFile(), &buff, 4)!=4)
 169       {
 170          _D(Lvl_Info, "%s: Failed to read chunk size. File truncated.", (int)__PRETTY_FUNCTION__);
 171          return;
 172       }
 173       buff = L2LE(buff);
 174       _D(Lvl_Info,"%s: chunk size = %ld bytes.", (int)__PRETTY_FUNCTION__, buff);
 175       DOS->Seek(getFile(), buff, OFFSET_CURRENT);
 176    }
 177
 178    return;
 179 }
 180
 181 bool                 rWAVAudio::checkHeader(WaveHdr* pHdr, EDtError &rc)
 182 {
 183    if (W2LE(1)     != pHdr->format)
 184    {
 185       rc = DT_FileFormatNotSupported;
 186       return false;
 187    }
 188    if (W2LE(2)     != pHdr->channels)
 189    {
 190       rc = DT_WrongChannelCount;
 191       return false;
 192    }
 193    if (L2LE(44100) != pHdr->frequency)
 194    {
 195       rc = DT_WrongFrequency;
 196       return false;
 197    }
 198    if (W2LE(16)    != pHdr->resolution)
 199    {
 200       rc = DT_WrongResolution;
 201       return false;
 202    }
 203    rc = DT_OK;
 204    return true;
 205 }
 206
 207 bool                 rWAVAudio::checkFile(const char* sFileName, EDtError &rc)
 208 {
 209    unsigned int    buff;
 210    int             c=0, len;
 211    BPTR            fh;
 212
 213    rc = DT_UnableToOpenFile;
 214
 215    fh = DOS->Open(const_cast<char*>(sFileName), MODE_OLDFILE);
 216    if (0 == fh)
 217       return false;
 218
 219    DOS->Seek(fh, 0, OFFSET_BEGINNING);
 220
 221    if (DOS->Read(fh, &buff, 4)!=4)
 222    {
 223       DOS->Close(fh);
 224       rc = DT_FileMalformed;
 225       return false;
 226    }
 227    buff = L2BE(buff);
 228    if (buff != ID_RIFF)
 229    {
 230       DOS->Close(fh);
 231       rc = DT_InvalidFormat;
 232       return false;
 233    }
 234
 235    DOS->Seek(fh, 4, OFFSET_CURRENT);
 236
 237    if (DOS->Read(fh, &buff, 4)!=4)
 238    {
 239       DOS->Close(fh);
 240       rc = DT_FileMalformed;
 241       return false;
 242    }
 243    buff = L2BE(buff);
 244    if (buff != ID_WAVE)
 245    {
 246       DOS->Close(fh);
 247       rc = DT_InvalidFormat;
 248       return false;
 249    }
 250
 251    for (c=0;c<2;)
 252    {
 253       if (DOS->Read(fh, &buff, 4)!=4)
 254       {
 255          rc = DT_FileMalformed;
 256          DOS->Close(fh);
 257          return false;
 258       }
 259
 260       buff = L2BE(buff);
 261       if (buff == ID_fmt)
 262       {
 263          WaveHdr *wave_hdr = 0;
 264          bool     bCorrect = false;
 265
 266          buff = 0;
 267          DOS->Read(fh, &buff, 4);
 268          len = L2LE(buff);
 269          wave_hdr = (WaveHdr*)new char[len];
 270
 271          if ((NULL != wave_hdr) && (len == DOS->Read(fh, wave_hdr, len)))
 272          {
 273             bCorrect = checkHeader(wave_hdr, rc);
 274             delete [] ((char*)wave_hdr);
 275          }
 276          c++;
 277
 278          if (false == bCorrect)
 279          {
 280             DOS->Close(fh);
 281             return false;
 282          }
 283          continue;
 284       }
 285       else if (buff == ID_data)
 286       {
 287          c++;
 288       }
 289
 290       if (DOS->Read(fh, &buff, 4)!=4)
 291       {
 292          rc = DT_FileMalformed;
 293          DOS->Close(fh);
 294          return false;
 295       }
 296
 297       buff = L2LE(buff);
 298       DOS->Seek(fh, buff, OFFSET_CURRENT);
 299    }
 300
 301    DOS->Close(fh);
 302    return true;
 303 }
 304
 305