ms_adpcm.c
上传用户:shw771010
上传日期:2022-01-05
资源大小:991k
文件大小:24k
源码类别:

Audio

开发平台:

Unix_Linux

  1. /*
  2. ** Copyright (C) 1999-2009 Erik de Castro Lopo <erikd@mega-nerd.com>
  3. **
  4. ** This program is free software; you can redistribute it and/or modify
  5. ** it under the terms of the GNU Lesser General Public License as published by
  6. ** the Free Software Foundation; either version 2.1 of the License, or
  7. ** (at your option) any later version.
  8. **
  9. ** This program is distributed in the hope that it will be useful,
  10. ** but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12. ** GNU Lesser General Public License for more details.
  13. **
  14. ** You should have received a copy of the GNU Lesser General Public License
  15. ** along with this program; if not, write to the Free Software
  16. ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  17. */
  18. #include "sfconfig.h"
  19. #include <stdio.h>
  20. #include <stdlib.h>
  21. #include <string.h>
  22. #include <math.h>
  23. #include "sndfile.h"
  24. #include "sfendian.h"
  25. #include "common.h"
  26. #include "wav_w64.h"
  27. /* These required here because we write the header in this file. */
  28. #define RIFF_MARKER (MAKE_MARKER ('R', 'I', 'F', 'F'))
  29. #define WAVE_MARKER (MAKE_MARKER ('W', 'A', 'V', 'E'))
  30. #define fmt_MARKER (MAKE_MARKER ('f', 'm', 't', ' '))
  31. #define fact_MARKER (MAKE_MARKER ('f', 'a', 'c', 't'))
  32. #define data_MARKER (MAKE_MARKER ('d', 'a', 't', 'a'))
  33. #define WAVE_FORMAT_MS_ADPCM 0x0002
  34. typedef struct
  35. { int channels, blocksize, samplesperblock, blocks, dataremaining ;
  36. int blockcount ;
  37. sf_count_t samplecount ;
  38. short *samples ;
  39. unsigned char *block ;
  40. #if HAVE_FLEXIBLE_ARRAY
  41. short dummydata [] ; /* ISO C99 struct flexible array. */
  42. #else
  43. short dummydata [0] ; /* This is a hack an might not work. */
  44. #endif
  45. } MSADPCM_PRIVATE ;
  46. /*============================================================================================
  47. ** MS ADPCM static data and functions.
  48. */
  49. static int AdaptationTable [] =
  50. { 230, 230, 230, 230, 307, 409, 512, 614,
  51. 768, 614, 512, 409, 307, 230, 230, 230
  52. } ;
  53. /* TODO : The first 7 coef's are are always hardcode and must
  54.    appear in the actual WAVE file.  They should be read in
  55.    in case a sound program added extras to the list. */
  56. static int AdaptCoeff1 [MSADPCM_ADAPT_COEFF_COUNT] =
  57. { 256, 512, 0, 192, 240, 460, 392
  58. } ;
  59. static int AdaptCoeff2 [MSADPCM_ADAPT_COEFF_COUNT] =
  60. { 0, -256, 0, 64, 0, -208, -232
  61. } ;
  62. /*============================================================================================
  63. ** MS ADPCM Block Layout.
  64. ** ======================
  65. ** Block is usually 256, 512 or 1024 bytes depending on sample rate.
  66. ** For a mono file, the block is laid out as follows:
  67. ** byte purpose
  68. ** 0 block predictor [0..6]
  69. ** 1,2 initial idelta (positive)
  70. ** 3,4 sample 1
  71. ** 5,6 sample 0
  72. ** 7..n packed bytecodes
  73. **
  74. ** For a stereo file, the block is laid out as follows:
  75. ** byte purpose
  76. ** 0 block predictor [0..6] for left channel
  77. ** 1 block predictor [0..6] for right channel
  78. ** 2,3 initial idelta (positive) for left channel
  79. ** 4,5 initial idelta (positive) for right channel
  80. ** 6,7 sample 1 for left channel
  81. ** 8,9 sample 1 for right channel
  82. ** 10,11 sample 0 for left channel
  83. ** 12,13 sample 0 for right channel
  84. ** 14..n packed bytecodes
  85. */
  86. /*============================================================================================
  87. ** Static functions.
  88. */
  89. static int msadpcm_decode_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms) ;
  90. static sf_count_t msadpcm_read_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms, short *ptr, int len) ;
  91. static int msadpcm_encode_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms) ;
  92. static sf_count_t msadpcm_write_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms, const short *ptr, int len) ;
  93. static sf_count_t msadpcm_read_s (SF_PRIVATE *psf, short *ptr, sf_count_t len) ;
  94. static sf_count_t msadpcm_read_i (SF_PRIVATE *psf, int *ptr, sf_count_t len) ;
  95. static sf_count_t msadpcm_read_f (SF_PRIVATE *psf, float *ptr, sf_count_t len) ;
  96. static sf_count_t msadpcm_read_d (SF_PRIVATE *psf, double *ptr, sf_count_t len) ;
  97. static sf_count_t msadpcm_write_s (SF_PRIVATE *psf, const short *ptr, sf_count_t len) ;
  98. static sf_count_t msadpcm_write_i (SF_PRIVATE *psf, const int *ptr, sf_count_t len) ;
  99. static sf_count_t msadpcm_write_f (SF_PRIVATE *psf, const float *ptr, sf_count_t len) ;
  100. static sf_count_t msadpcm_write_d (SF_PRIVATE *psf, const double *ptr, sf_count_t len) ;
  101. static sf_count_t msadpcm_seek (SF_PRIVATE *psf, int mode, sf_count_t offset) ;
  102. static int msadpcm_close (SF_PRIVATE *psf) ;
  103. static void choose_predictor (unsigned int channels, short *data, int *bpred, int *idelta) ;
  104. /*============================================================================================
  105. ** MS ADPCM Read Functions.
  106. */
  107. int
  108. wav_w64_msadpcm_init (SF_PRIVATE *psf, int blockalign, int samplesperblock)
  109. { MSADPCM_PRIVATE *pms ;
  110. unsigned int pmssize ;
  111. int count ;
  112. if (psf->codec_data != NULL)
  113. { psf_log_printf (psf, "*** psf->codec_data is not NULL.n") ;
  114. return SFE_INTERNAL ;
  115. } ;
  116. if (psf->file.mode == SFM_WRITE)
  117. samplesperblock = 2 + 2 * (blockalign - 7 * psf->sf.channels) / psf->sf.channels ;
  118. pmssize = sizeof (MSADPCM_PRIVATE) + blockalign + 3 * psf->sf.channels * samplesperblock ;
  119. if (! (psf->codec_data = malloc (pmssize)))
  120. return SFE_MALLOC_FAILED ;
  121. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  122. memset (pms, 0, pmssize) ;
  123. pms->samples = pms->dummydata ;
  124. pms->block = (unsigned char*) (pms->dummydata + psf->sf.channels * samplesperblock) ;
  125. pms->channels = psf->sf.channels ;
  126. pms->blocksize = blockalign ;
  127. pms->samplesperblock = samplesperblock ;
  128. if (pms->blocksize == 0)
  129. { psf_log_printf (psf, "*** Error : pms->blocksize should not be zero.n") ;
  130. return SFE_INTERNAL ;
  131. } ;
  132. if (psf->file.mode == SFM_READ)
  133. { pms->dataremaining  = psf->datalength ;
  134. if (psf->datalength % pms->blocksize)
  135. pms->blocks = psf->datalength / pms->blocksize + 1 ;
  136. else
  137. pms->blocks = psf->datalength / pms->blocksize ;
  138. count = 2 * (pms->blocksize - 6 * pms->channels) / pms->channels ;
  139. if (pms->samplesperblock != count)
  140. { psf_log_printf (psf, "*** Error : samplesperblock should be %d.n", count) ;
  141. return SFE_INTERNAL ;
  142. } ;
  143. psf->sf.frames = (psf->datalength / pms->blocksize) * pms->samplesperblock ;
  144. psf_log_printf (psf, " bpred   ideltan") ;
  145. msadpcm_decode_block (psf, pms) ;
  146. psf->read_short = msadpcm_read_s ;
  147. psf->read_int = msadpcm_read_i ;
  148. psf->read_float = msadpcm_read_f ;
  149. psf->read_double = msadpcm_read_d ;
  150. } ;
  151. if (psf->file.mode == SFM_WRITE)
  152. { pms->samples = pms->dummydata ;
  153. pms->samplecount = 0 ;
  154. psf->write_short = msadpcm_write_s ;
  155. psf->write_int = msadpcm_write_i ;
  156. psf->write_float = msadpcm_write_f ;
  157. psf->write_double = msadpcm_write_d ;
  158. } ;
  159. psf->codec_close = msadpcm_close ;
  160. psf->seek = msadpcm_seek ;
  161. return 0 ;
  162. } /* wav_w64_msadpcm_init */
  163. static int
  164. msadpcm_decode_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms)
  165. { int chan, k, blockindx, sampleindx ;
  166. short bytecode, bpred [2], chan_idelta [2] ;
  167.     int predict ;
  168.     int current ;
  169.     int idelta ;
  170. pms->blockcount ++ ;
  171. pms->samplecount = 0 ;
  172. if (pms->blockcount > pms->blocks)
  173. { memset (pms->samples, 0, pms->samplesperblock * pms->channels) ;
  174. return 1 ;
  175. } ;
  176. if ((k = psf_fread (pms->block, 1, pms->blocksize, psf)) != pms->blocksize)
  177. psf_log_printf (psf, "*** Warning : short read (%d != %d).n", k, pms->blocksize) ;
  178. /* Read and check the block header. */
  179. if (pms->channels == 1)
  180. { bpred [0] = pms->block [0] ;
  181. if (bpred [0] >= 7)
  182. psf_log_printf (psf, "MS ADPCM synchronisation error (%d).n", bpred [0]) ;
  183. chan_idelta [0] = pms->block [1] | (pms->block [2] << 8) ;
  184. chan_idelta [1] = 0 ;
  185. psf_log_printf (psf, "(%d) (%d)n", bpred [0], chan_idelta [0]) ;
  186. pms->samples [1] = pms->block [3] | (pms->block [4] << 8) ;
  187. pms->samples [0] = pms->block [5] | (pms->block [6] << 8) ;
  188. blockindx = 7 ;
  189. }
  190. else
  191. { bpred [0] = pms->block [0] ;
  192. bpred [1] = pms->block [1] ;
  193. if (bpred [0] >= 7 || bpred [1] >= 7)
  194. psf_log_printf (psf, "MS ADPCM synchronisation error (%d %d).n", bpred [0], bpred [1]) ;
  195. chan_idelta [0] = pms->block [2] | (pms->block [3] << 8) ;
  196. chan_idelta [1] = pms->block [4] | (pms->block [5] << 8) ;
  197. psf_log_printf (psf, "(%d, %d) (%d, %d)n", bpred [0], bpred [1], chan_idelta [0], chan_idelta [1]) ;
  198. pms->samples [2] = pms->block [6] | (pms->block [7] << 8) ;
  199. pms->samples [3] = pms->block [8] | (pms->block [9] << 8) ;
  200. pms->samples [0] = pms->block [10] | (pms->block [11] << 8) ;
  201. pms->samples [1] = pms->block [12] | (pms->block [13] << 8) ;
  202. blockindx = 14 ;
  203. } ;
  204. /*--------------------------------------------------------
  205. This was left over from a time when calculations were done
  206. as ints rather than shorts. Keep this around as a reminder
  207. in case I ever find a file which decodes incorrectly.
  208.     if (chan_idelta [0] & 0x8000)
  209. chan_idelta [0] -= 0x10000 ;
  210.     if (chan_idelta [1] & 0x8000)
  211. chan_idelta [1] -= 0x10000 ;
  212. --------------------------------------------------------*/
  213. /* Pull apart the packed 4 bit samples and store them in their
  214. ** correct sample positions.
  215. */
  216. sampleindx = 2 * pms->channels ;
  217. while (blockindx < pms->blocksize)
  218. { bytecode = pms->block [blockindx++] ;
  219.    pms->samples [sampleindx++] = (bytecode >> 4) & 0x0F ;
  220. pms->samples [sampleindx++] = bytecode & 0x0F ;
  221. } ;
  222. /* Decode the encoded 4 bit samples. */
  223. for (k = 2 * pms->channels ; k < (pms->samplesperblock * pms->channels) ; k ++)
  224. { chan = (pms->channels > 1) ? (k % 2) : 0 ;
  225. bytecode = pms->samples [k] & 0xF ;
  226. /* Compute next Adaptive Scale Factor (ASF) */
  227. idelta = chan_idelta [chan] ;
  228. chan_idelta [chan] = (AdaptationTable [bytecode] * idelta) >> 8 ; /* => / 256 => FIXED_POINT_ADAPTATION_BASE == 256 */
  229. if (chan_idelta [chan] < 16)
  230. chan_idelta [chan] = 16 ;
  231. if (bytecode & 0x8)
  232. bytecode -= 0x10 ;
  233.      predict = ((pms->samples [k - pms->channels] * AdaptCoeff1 [bpred [chan]])
  234. + (pms->samples [k - 2 * pms->channels] * AdaptCoeff2 [bpred [chan]])) >> 8 ; /* => / 256 => FIXED_POINT_COEFF_BASE == 256 */
  235. current = (bytecode * idelta) + predict ;
  236. if (current > 32767)
  237. current = 32767 ;
  238. else if (current < -32768)
  239. current = -32768 ;
  240. pms->samples [k] = current ;
  241. } ;
  242. return 1 ;
  243. } /* msadpcm_decode_block */
  244. static sf_count_t
  245. msadpcm_read_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms, short *ptr, int len)
  246. { int count, total = 0, indx = 0 ;
  247. while (indx < len)
  248. { if (pms->blockcount >= pms->blocks && pms->samplecount >= pms->samplesperblock)
  249. { memset (&(ptr [indx]), 0, (size_t) ((len - indx) * sizeof (short))) ;
  250. return total ;
  251. } ;
  252. if (pms->samplecount >= pms->samplesperblock)
  253. msadpcm_decode_block (psf, pms) ;
  254. count = (pms->samplesperblock - pms->samplecount) * pms->channels ;
  255. count = (len - indx > count) ? count : len - indx ;
  256. memcpy (&(ptr [indx]), &(pms->samples [pms->samplecount * pms->channels]), count * sizeof (short)) ;
  257. indx += count ;
  258. pms->samplecount += count / pms->channels ;
  259. total = indx ;
  260. } ;
  261. return total ;
  262. } /* msadpcm_read_block */
  263. static sf_count_t
  264. msadpcm_read_s (SF_PRIVATE *psf, short *ptr, sf_count_t len)
  265. { MSADPCM_PRIVATE  *pms ;
  266. int readcount, count ;
  267. sf_count_t total = 0 ;
  268. if (! psf->codec_data)
  269. return 0 ;
  270. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  271. while (len > 0)
  272. { readcount = (len > 0x10000000) ? 0x10000000 : (int) len ;
  273. count = msadpcm_read_block (psf, pms, ptr, readcount) ;
  274. total += count ;
  275. len -= count ;
  276. if (count != readcount)
  277. break ;
  278. } ;
  279. return total ;
  280. } /* msadpcm_read_s */
  281. static sf_count_t
  282. msadpcm_read_i (SF_PRIVATE *psf, int *ptr, sf_count_t len)
  283. { MSADPCM_PRIVATE *pms ;
  284. short *sptr ;
  285. int k, bufferlen, readcount = 0, count ;
  286. sf_count_t total = 0 ;
  287. if (! psf->codec_data)
  288. return 0 ;
  289. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  290. sptr = psf->u.sbuf ;
  291. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  292. while (len > 0)
  293. { readcount = (len >= bufferlen) ? bufferlen : len ;
  294. count = msadpcm_read_block (psf, pms, sptr, readcount) ;
  295. for (k = 0 ; k < readcount ; k++)
  296. ptr [total + k] = sptr [k] << 16 ;
  297. total += count ;
  298. len -= readcount ;
  299. if (count != readcount)
  300. break ;
  301. } ;
  302. return total ;
  303. } /* msadpcm_read_i */
  304. static sf_count_t
  305. msadpcm_read_f (SF_PRIVATE *psf, float *ptr, sf_count_t len)
  306. { MSADPCM_PRIVATE *pms ;
  307. short *sptr ;
  308. int k, bufferlen, readcount = 0, count ;
  309. sf_count_t total = 0 ;
  310. float normfact ;
  311. if (! psf->codec_data)
  312. return 0 ;
  313. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  314. normfact = (psf->norm_float == SF_TRUE) ? 1.0 / ((float) 0x8000) : 1.0 ;
  315. sptr = psf->u.sbuf ;
  316. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  317. while (len > 0)
  318. { readcount = (len >= bufferlen) ? bufferlen : len ;
  319. count = msadpcm_read_block (psf, pms, sptr, readcount) ;
  320. for (k = 0 ; k < readcount ; k++)
  321. ptr [total + k] = normfact * (float) (sptr [k]) ;
  322. total += count ;
  323. len -= readcount ;
  324. if (count != readcount)
  325. break ;
  326. } ;
  327. return total ;
  328. } /* msadpcm_read_f */
  329. static sf_count_t
  330. msadpcm_read_d (SF_PRIVATE *psf, double *ptr, sf_count_t len)
  331. { MSADPCM_PRIVATE *pms ;
  332. short *sptr ;
  333. int k, bufferlen, readcount = 0, count ;
  334. sf_count_t total = 0 ;
  335. double  normfact ;
  336. normfact = (psf->norm_double == SF_TRUE) ? 1.0 / ((double) 0x8000) : 1.0 ;
  337. if (! psf->codec_data)
  338. return 0 ;
  339. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  340. sptr = psf->u.sbuf ;
  341. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  342. while (len > 0)
  343. { readcount = (len >= bufferlen) ? bufferlen : len ;
  344. count = msadpcm_read_block (psf, pms, sptr, readcount) ;
  345. for (k = 0 ; k < readcount ; k++)
  346. ptr [total + k] = normfact * (double) (sptr [k]) ;
  347. total += count ;
  348. len -= readcount ;
  349. if (count != readcount)
  350. break ;
  351. } ;
  352. return total ;
  353. } /* msadpcm_read_d */
  354. static sf_count_t
  355. msadpcm_seek (SF_PRIVATE *psf, int mode, sf_count_t offset)
  356. { MSADPCM_PRIVATE *pms ;
  357. int newblock, newsample ;
  358. if (! psf->codec_data)
  359. return 0 ;
  360. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  361. if (psf->datalength < 0 || psf->dataoffset < 0)
  362. { psf->error = SFE_BAD_SEEK ;
  363. return PSF_SEEK_ERROR ;
  364. } ;
  365. if (offset == 0)
  366. { psf_fseek (psf, psf->dataoffset, SEEK_SET) ;
  367. pms->blockcount = 0 ;
  368. msadpcm_decode_block (psf, pms) ;
  369. pms->samplecount = 0 ;
  370. return 0 ;
  371. } ;
  372. if (offset < 0 || offset > pms->blocks * pms->samplesperblock)
  373. { psf->error = SFE_BAD_SEEK ;
  374. return PSF_SEEK_ERROR ;
  375. } ;
  376. newblock = offset / pms->samplesperblock ;
  377. newsample = offset % pms->samplesperblock ;
  378. if (mode == SFM_READ)
  379. { psf_fseek (psf, psf->dataoffset + newblock * pms->blocksize, SEEK_SET) ;
  380. pms->blockcount = newblock ;
  381. msadpcm_decode_block (psf, pms) ;
  382. pms->samplecount = newsample ;
  383. }
  384. else
  385. { /* What to do about write??? */
  386. psf->error = SFE_BAD_SEEK ;
  387. return PSF_SEEK_ERROR ;
  388. } ;
  389. return newblock * pms->samplesperblock + newsample ;
  390. } /* msadpcm_seek */
  391. /*==========================================================================================
  392. ** MS ADPCM Write Functions.
  393. */
  394. void
  395. msadpcm_write_adapt_coeffs (SF_PRIVATE *psf)
  396. { int k ;
  397. for (k = 0 ; k < MSADPCM_ADAPT_COEFF_COUNT ; k++)
  398. psf_binheader_writef (psf, "22", AdaptCoeff1 [k], AdaptCoeff2 [k]) ;
  399. } /* msadpcm_write_adapt_coeffs */
  400. /*==========================================================================================
  401. */
  402. static int
  403. msadpcm_encode_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms)
  404. { unsigned int blockindx ;
  405. unsigned char byte ;
  406. int chan, k, predict, bpred [2], idelta [2], errordelta, newsamp ;
  407. choose_predictor (pms->channels, pms->samples, bpred, idelta) ;
  408. /* Write the block header. */
  409. if (pms->channels == 1)
  410. { pms->block [0] = bpred [0] ;
  411. pms->block [1] = idelta [0] & 0xFF ;
  412. pms->block [2] = idelta [0] >> 8 ;
  413. pms->block [3] = pms->samples [1] & 0xFF ;
  414. pms->block [4] = pms->samples [1] >> 8 ;
  415. pms->block [5] = pms->samples [0] & 0xFF ;
  416. pms->block [6] = pms->samples [0] >> 8 ;
  417. blockindx = 7 ;
  418. byte = 0 ;
  419. /* Encode the samples as 4 bit. */
  420. for (k = 2 ; k < pms->samplesperblock ; k++)
  421. { predict = (pms->samples [k-1] * AdaptCoeff1 [bpred [0]] + pms->samples [k-2] * AdaptCoeff2 [bpred [0]]) >> 8 ;
  422. errordelta = (pms->samples [k] - predict) / idelta [0] ;
  423. if (errordelta < -8)
  424. errordelta = -8 ;
  425. else if (errordelta > 7)
  426. errordelta = 7 ;
  427. newsamp = predict + (idelta [0] * errordelta) ;
  428. if (newsamp > 32767)
  429. newsamp = 32767 ;
  430. else if (newsamp < -32768)
  431. newsamp = -32768 ;
  432. if (errordelta < 0)
  433. errordelta += 0x10 ;
  434. byte = (byte << 4) | (errordelta & 0xF) ;
  435. if (k % 2)
  436. { pms->block [blockindx++] = byte ;
  437. byte = 0 ;
  438. } ;
  439. idelta [0] = (idelta [0] * AdaptationTable [errordelta]) >> 8 ;
  440. if (idelta [0] < 16)
  441. idelta [0] = 16 ;
  442. pms->samples [k] = newsamp ;
  443. } ;
  444. }
  445. else
  446. { /* Stereo file. */
  447. pms->block [0] = bpred [0] ;
  448. pms->block [1] = bpred [1] ;
  449. pms->block [2] = idelta [0] & 0xFF ;
  450. pms->block [3] = idelta [0] >> 8 ;
  451. pms->block [4] = idelta [1] & 0xFF ;
  452. pms->block [5] = idelta [1] >> 8 ;
  453. pms->block [6] = pms->samples [2] & 0xFF ;
  454. pms->block [7] = pms->samples [2] >> 8 ;
  455. pms->block [8] = pms->samples [3] & 0xFF ;
  456. pms->block [9] = pms->samples [3] >> 8 ;
  457. pms->block [10] = pms->samples [0] & 0xFF ;
  458. pms->block [11] = pms->samples [0] >> 8 ;
  459. pms->block [12] = pms->samples [1] & 0xFF ;
  460. pms->block [13] = pms->samples [1] >> 8 ;
  461. blockindx = 14 ;
  462. byte = 0 ;
  463. chan = 1 ;
  464. for (k = 4 ; k < 2 * pms->samplesperblock ; k++)
  465. { chan = k & 1 ;
  466. predict = (pms->samples [k-2] * AdaptCoeff1 [bpred [chan]] + pms->samples [k-4] * AdaptCoeff2 [bpred [chan]]) >> 8 ;
  467. errordelta = (pms->samples [k] - predict) / idelta [chan] ;
  468. if (errordelta < -8)
  469. errordelta = -8 ;
  470. else if (errordelta > 7)
  471. errordelta = 7 ;
  472. newsamp = predict + (idelta [chan] * errordelta) ;
  473. if (newsamp > 32767)
  474. newsamp = 32767 ;
  475. else if (newsamp < -32768)
  476. newsamp = -32768 ;
  477. if (errordelta < 0)
  478. errordelta += 0x10 ;
  479. byte = (byte << 4) | (errordelta & 0xF) ;
  480. if (chan)
  481. { pms->block [blockindx++] = byte ;
  482. byte = 0 ;
  483. } ;
  484. idelta [chan] = (idelta [chan] * AdaptationTable [errordelta]) >> 8 ;
  485. if (idelta [chan] < 16)
  486. idelta [chan] = 16 ;
  487. pms->samples [k] = newsamp ;
  488. } ;
  489. } ;
  490. /* Write the block to disk. */
  491. if ((k = psf_fwrite (pms->block, 1, pms->blocksize, psf)) != pms->blocksize)
  492. psf_log_printf (psf, "*** Warning : short write (%d != %d).n", k, pms->blocksize) ;
  493. memset (pms->samples, 0, pms->samplesperblock * sizeof (short)) ;
  494. pms->blockcount ++ ;
  495. pms->samplecount = 0 ;
  496. return 1 ;
  497. } /* msadpcm_encode_block */
  498. static sf_count_t
  499. msadpcm_write_block (SF_PRIVATE *psf, MSADPCM_PRIVATE *pms, const short *ptr, int len)
  500. { int count, total = 0, indx = 0 ;
  501. while (indx < len)
  502. { count = (pms->samplesperblock - pms->samplecount) * pms->channels ;
  503. if (count > len - indx)
  504. count = len - indx ;
  505. memcpy (&(pms->samples [pms->samplecount * pms->channels]), &(ptr [total]), count * sizeof (short)) ;
  506. indx += count ;
  507. pms->samplecount += count / pms->channels ;
  508. total = indx ;
  509. if (pms->samplecount >= pms->samplesperblock)
  510. msadpcm_encode_block (psf, pms) ;
  511. } ;
  512. return total ;
  513. } /* msadpcm_write_block */
  514. static sf_count_t
  515. msadpcm_write_s (SF_PRIVATE *psf, const short *ptr, sf_count_t len)
  516. { MSADPCM_PRIVATE *pms ;
  517. int writecount, count ;
  518. sf_count_t total = 0 ;
  519. if (! psf->codec_data)
  520. return 0 ;
  521. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  522. while (len > 0)
  523. { writecount = (len > 0x10000000) ? 0x10000000 : (int) len ;
  524. count = msadpcm_write_block (psf, pms, ptr, writecount) ;
  525. total += count ;
  526. len -= count ;
  527. if (count != writecount)
  528. break ;
  529. } ;
  530. return total ;
  531. } /* msadpcm_write_s */
  532. static sf_count_t
  533. msadpcm_write_i (SF_PRIVATE *psf, const int *ptr, sf_count_t len)
  534. { MSADPCM_PRIVATE *pms ;
  535. short *sptr ;
  536. int k, bufferlen, writecount, count ;
  537. sf_count_t total = 0 ;
  538. if (! psf->codec_data)
  539. return 0 ;
  540. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  541. sptr = psf->u.sbuf ;
  542. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  543. while (len > 0)
  544. { writecount = (len >= bufferlen) ? bufferlen : len ;
  545. for (k = 0 ; k < writecount ; k++)
  546. sptr [k] = ptr [total + k] >> 16 ;
  547. count = msadpcm_write_block (psf, pms, sptr, writecount) ;
  548. total += count ;
  549. len -= writecount ;
  550. if (count != writecount)
  551. break ;
  552. } ;
  553. return total ;
  554. } /* msadpcm_write_i */
  555. static sf_count_t
  556. msadpcm_write_f (SF_PRIVATE *psf, const float *ptr, sf_count_t len)
  557. { MSADPCM_PRIVATE *pms ;
  558. short *sptr ;
  559. int k, bufferlen, writecount, count ;
  560. sf_count_t total = 0 ;
  561. float normfact ;
  562. if (! psf->codec_data)
  563. return 0 ;
  564. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  565. normfact = (psf->norm_float == SF_TRUE) ? (1.0 * 0x7FFF) : 1.0 ;
  566. sptr = psf->u.sbuf ;
  567. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  568. while (len > 0)
  569. { writecount = (len >= bufferlen) ? bufferlen : len ;
  570. for (k = 0 ; k < writecount ; k++)
  571. sptr [k] = lrintf (normfact * ptr [total + k]) ;
  572. count = msadpcm_write_block (psf, pms, sptr, writecount) ;
  573. total += count ;
  574. len -= writecount ;
  575. if (count != writecount)
  576. break ;
  577. } ;
  578. return total ;
  579. } /* msadpcm_write_f */
  580. static sf_count_t
  581. msadpcm_write_d (SF_PRIVATE *psf, const double *ptr, sf_count_t len)
  582. { MSADPCM_PRIVATE *pms ;
  583. short *sptr ;
  584. int k, bufferlen, writecount, count ;
  585. sf_count_t total = 0 ;
  586. double  normfact ;
  587. normfact = (psf->norm_double == SF_TRUE) ? (1.0 * 0x7FFF) : 1.0 ;
  588. if (! psf->codec_data)
  589. return 0 ;
  590. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  591. sptr = psf->u.sbuf ;
  592. bufferlen = ARRAY_LEN (psf->u.sbuf) ;
  593. while (len > 0)
  594. { writecount = (len >= bufferlen) ? bufferlen : len ;
  595. for (k = 0 ; k < writecount ; k++)
  596. sptr [k] = lrint (normfact * ptr [total + k]) ;
  597. count = msadpcm_write_block (psf, pms, sptr, writecount) ;
  598. total += count ;
  599. len -= writecount ;
  600. if (count != writecount)
  601. break ;
  602. } ;
  603. return total ;
  604. } /* msadpcm_write_d */
  605. /*========================================================================================
  606. */
  607. static int
  608. msadpcm_close (SF_PRIVATE *psf)
  609. { MSADPCM_PRIVATE *pms ;
  610. pms = (MSADPCM_PRIVATE*) psf->codec_data ;
  611. if (psf->file.mode == SFM_WRITE)
  612. { /*  Now we know static int for certain the length of the file we can
  613. **  re-write the header.
  614. */
  615. if (pms->samplecount && pms->samplecount < pms->samplesperblock)
  616. msadpcm_encode_block (psf, pms) ;
  617. } ;
  618. return 0 ;
  619. } /* msadpcm_close */
  620. /*========================================================================================
  621. ** Static functions.
  622. */
  623. /*----------------------------------------------------------------------------------------
  624. ** Choosing the block predictor.
  625. ** Each block requires a predictor and an idelta for each channel.
  626. ** The predictor is in the range [0..6] which is an indx into the two AdaptCoeff tables.
  627. ** The predictor is chosen by trying all of the possible predictors on a small set of
  628. ** samples at the beginning of the block. The predictor with the smallest average
  629. ** abs (idelta) is chosen as the best predictor for this block.
  630. ** The value of idelta is chosen to to give a 4 bit code value of +/- 4 (approx. half the
  631. ** max. code value). If the average abs (idelta) is zero, the sixth predictor is chosen.
  632. ** If the value of idelta is less then 16 it is set to 16.
  633. **
  634. ** Microsoft uses an IDELTA_COUNT (number of sample pairs used to choose best predictor)
  635. ** value of 3. The best possible results would be obtained by using all the samples to
  636. ** choose the predictor.
  637. */
  638. #define IDELTA_COUNT 3
  639. static void
  640. choose_predictor (unsigned int channels, short *data, int *block_pred, int *idelta)
  641. { unsigned int chan, k, bpred, idelta_sum, best_bpred, best_idelta ;
  642. for (chan = 0 ; chan < channels ; chan++)
  643. { best_bpred = best_idelta = 0 ;
  644. for (bpred = 0 ; bpred < 7 ; bpred++)
  645. { idelta_sum = 0 ;
  646. for (k = 2 ; k < 2 + IDELTA_COUNT ; k++)
  647. idelta_sum += abs (data [k * channels] - ((data [(k - 1) * channels] * AdaptCoeff1 [bpred] + data [(k - 2) * channels] * AdaptCoeff2 [bpred]) >> 8)) ;
  648. idelta_sum /= (4 * IDELTA_COUNT) ;
  649. if (bpred == 0 || idelta_sum < best_idelta)
  650. { best_bpred = bpred ;
  651. best_idelta = idelta_sum ;
  652. } ;
  653. if (! idelta_sum)
  654. { best_bpred = bpred ;
  655. best_idelta = 16 ;
  656. break ;
  657. } ;
  658. } ; /* for bpred ... */
  659. if (best_idelta < 16)
  660. best_idelta = 16 ;
  661. block_pred [chan] = best_bpred ;
  662. idelta [chan] = best_idelta ;
  663. } ;
  664. return ;
  665. } /* choose_predictor */