mapping0.c 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811
  1. /********************************************************************
  2. * *
  3. * THIS FILE IS PART OF THE OggVorbis SOFTWARE CODEC SOURCE CODE. *
  4. * USE, DISTRIBUTION AND REPRODUCTION OF THIS LIBRARY SOURCE IS *
  5. * GOVERNED BY A BSD-STYLE SOURCE LICENSE INCLUDED WITH THIS SOURCE *
  6. * IN 'COPYING'. PLEASE READ THESE TERMS BEFORE DISTRIBUTING. *
  7. * *
  8. * THE OggVorbis SOURCE CODE IS (C) COPYRIGHT 1994-2010 *
  9. * by the Xiph.Org Foundation http://www.xiph.org/ *
  10. * *
  11. ********************************************************************
  12. function: channel mapping 0 implementation
  13. last mod: $Id: mapping0.c 19441 2015-01-21 01:17:41Z xiphmont $
  14. ********************************************************************/
  15. #include <stdlib.h>
  16. #include <stdio.h>
  17. #include <string.h>
  18. #include <math.h>
  19. #include <ogg/ogg.h>
  20. #include "vorbis/codec.h"
  21. #include "codec_internal.h"
  22. #include "codebook.h"
  23. #include "window.h"
  24. #include "registry.h"
  25. #include "psy.h"
  26. #include "misc.h"
  27. /* simplistic, wasteful way of doing this (unique lookup for each
  28. mode/submapping); there should be a central repository for
  29. identical lookups. That will require minor work, so I'm putting it
  30. off as low priority.
  31. Why a lookup for each backend in a given mode? Because the
  32. blocksize is set by the mode, and low backend lookups may require
  33. parameters from other areas of the mode/mapping */
  34. static void mapping0_free_info(vorbis_info_mapping *i){
  35. vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)i;
  36. if(info){
  37. memset(info,0,sizeof(*info));
  38. _ogg_free(info);
  39. }
  40. }
  41. static void mapping0_pack(vorbis_info *vi,vorbis_info_mapping *vm,
  42. oggpack_buffer *opb){
  43. int i;
  44. vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)vm;
  45. /* another 'we meant to do it this way' hack... up to beta 4, we
  46. packed 4 binary zeros here to signify one submapping in use. We
  47. now redefine that to mean four bitflags that indicate use of
  48. deeper features; bit0:submappings, bit1:coupling,
  49. bit2,3:reserved. This is backward compatable with all actual uses
  50. of the beta code. */
  51. if(info->submaps>1){
  52. oggpack_write(opb,1,1);
  53. oggpack_write(opb,info->submaps-1,4);
  54. }else
  55. oggpack_write(opb,0,1);
  56. if(info->coupling_steps>0){
  57. oggpack_write(opb,1,1);
  58. oggpack_write(opb,info->coupling_steps-1,8);
  59. for(i=0;i<info->coupling_steps;i++){
  60. oggpack_write(opb,info->coupling_mag[i],ov_ilog(vi->channels-1));
  61. oggpack_write(opb,info->coupling_ang[i],ov_ilog(vi->channels-1));
  62. }
  63. }else
  64. oggpack_write(opb,0,1);
  65. oggpack_write(opb,0,2); /* 2,3:reserved */
  66. /* we don't write the channel submappings if we only have one... */
  67. if(info->submaps>1){
  68. for(i=0;i<vi->channels;i++)
  69. oggpack_write(opb,info->chmuxlist[i],4);
  70. }
  71. for(i=0;i<info->submaps;i++){
  72. oggpack_write(opb,0,8); /* time submap unused */
  73. oggpack_write(opb,info->floorsubmap[i],8);
  74. oggpack_write(opb,info->residuesubmap[i],8);
  75. }
  76. }
  77. /* also responsible for range checking */
  78. static vorbis_info_mapping *mapping0_unpack(vorbis_info *vi,oggpack_buffer *opb){
  79. int i,b;
  80. vorbis_info_mapping0 *info=_ogg_calloc(1,sizeof(*info));
  81. codec_setup_info *ci=vi->codec_setup;
  82. memset(info,0,sizeof(*info));
  83. if(vi->channels<=0)goto err_out;
  84. b=oggpack_read(opb,1);
  85. if(b<0)goto err_out;
  86. if(b){
  87. info->submaps=oggpack_read(opb,4)+1;
  88. if(info->submaps<=0)goto err_out;
  89. }else
  90. info->submaps=1;
  91. b=oggpack_read(opb,1);
  92. if(b<0)goto err_out;
  93. if(b){
  94. info->coupling_steps=oggpack_read(opb,8)+1;
  95. if(info->coupling_steps<=0)goto err_out;
  96. for(i=0;i<info->coupling_steps;i++){
  97. /* vi->channels > 0 is enforced in the caller */
  98. int testM=info->coupling_mag[i]=
  99. oggpack_read(opb,ov_ilog(vi->channels-1));
  100. int testA=info->coupling_ang[i]=
  101. oggpack_read(opb,ov_ilog(vi->channels-1));
  102. if(testM<0 ||
  103. testA<0 ||
  104. testM==testA ||
  105. testM>=vi->channels ||
  106. testA>=vi->channels) goto err_out;
  107. }
  108. }
  109. if(oggpack_read(opb,2)!=0)goto err_out; /* 2,3:reserved */
  110. if(info->submaps>1){
  111. for(i=0;i<vi->channels;i++){
  112. info->chmuxlist[i]=oggpack_read(opb,4);
  113. if(info->chmuxlist[i]>=info->submaps || info->chmuxlist[i]<0)goto err_out;
  114. }
  115. }
  116. for(i=0;i<info->submaps;i++){
  117. oggpack_read(opb,8); /* time submap unused */
  118. info->floorsubmap[i]=oggpack_read(opb,8);
  119. if(info->floorsubmap[i]>=ci->floors || info->floorsubmap[i]<0)goto err_out;
  120. info->residuesubmap[i]=oggpack_read(opb,8);
  121. if(info->residuesubmap[i]>=ci->residues || info->residuesubmap[i]<0)goto err_out;
  122. }
  123. return info;
  124. err_out:
  125. mapping0_free_info(info);
  126. return(NULL);
  127. }
  128. #include "os.h"
  129. #include "lpc.h"
  130. #include "lsp.h"
  131. #include "envelope.h"
  132. #include "mdct.h"
  133. #include "psy.h"
  134. #include "scales.h"
  135. #if 0
  136. static long seq=0;
  137. static ogg_int64_t total=0;
  138. static float FLOOR1_fromdB_LOOKUP[256]={
  139. 1.0649863e-07F, 1.1341951e-07F, 1.2079015e-07F, 1.2863978e-07F,
  140. 1.3699951e-07F, 1.4590251e-07F, 1.5538408e-07F, 1.6548181e-07F,
  141. 1.7623575e-07F, 1.8768855e-07F, 1.9988561e-07F, 2.128753e-07F,
  142. 2.2670913e-07F, 2.4144197e-07F, 2.5713223e-07F, 2.7384213e-07F,
  143. 2.9163793e-07F, 3.1059021e-07F, 3.3077411e-07F, 3.5226968e-07F,
  144. 3.7516214e-07F, 3.9954229e-07F, 4.2550680e-07F, 4.5315863e-07F,
  145. 4.8260743e-07F, 5.1396998e-07F, 5.4737065e-07F, 5.8294187e-07F,
  146. 6.2082472e-07F, 6.6116941e-07F, 7.0413592e-07F, 7.4989464e-07F,
  147. 7.9862701e-07F, 8.5052630e-07F, 9.0579828e-07F, 9.6466216e-07F,
  148. 1.0273513e-06F, 1.0941144e-06F, 1.1652161e-06F, 1.2409384e-06F,
  149. 1.3215816e-06F, 1.4074654e-06F, 1.4989305e-06F, 1.5963394e-06F,
  150. 1.7000785e-06F, 1.8105592e-06F, 1.9282195e-06F, 2.0535261e-06F,
  151. 2.1869758e-06F, 2.3290978e-06F, 2.4804557e-06F, 2.6416497e-06F,
  152. 2.8133190e-06F, 2.9961443e-06F, 3.1908506e-06F, 3.3982101e-06F,
  153. 3.6190449e-06F, 3.8542308e-06F, 4.1047004e-06F, 4.3714470e-06F,
  154. 4.6555282e-06F, 4.9580707e-06F, 5.2802740e-06F, 5.6234160e-06F,
  155. 5.9888572e-06F, 6.3780469e-06F, 6.7925283e-06F, 7.2339451e-06F,
  156. 7.7040476e-06F, 8.2047000e-06F, 8.7378876e-06F, 9.3057248e-06F,
  157. 9.9104632e-06F, 1.0554501e-05F, 1.1240392e-05F, 1.1970856e-05F,
  158. 1.2748789e-05F, 1.3577278e-05F, 1.4459606e-05F, 1.5399272e-05F,
  159. 1.6400004e-05F, 1.7465768e-05F, 1.8600792e-05F, 1.9809576e-05F,
  160. 2.1096914e-05F, 2.2467911e-05F, 2.3928002e-05F, 2.5482978e-05F,
  161. 2.7139006e-05F, 2.8902651e-05F, 3.0780908e-05F, 3.2781225e-05F,
  162. 3.4911534e-05F, 3.7180282e-05F, 3.9596466e-05F, 4.2169667e-05F,
  163. 4.4910090e-05F, 4.7828601e-05F, 5.0936773e-05F, 5.4246931e-05F,
  164. 5.7772202e-05F, 6.1526565e-05F, 6.5524908e-05F, 6.9783085e-05F,
  165. 7.4317983e-05F, 7.9147585e-05F, 8.4291040e-05F, 8.9768747e-05F,
  166. 9.5602426e-05F, 0.00010181521F, 0.00010843174F, 0.00011547824F,
  167. 0.00012298267F, 0.00013097477F, 0.00013948625F, 0.00014855085F,
  168. 0.00015820453F, 0.00016848555F, 0.00017943469F, 0.00019109536F,
  169. 0.00020351382F, 0.00021673929F, 0.00023082423F, 0.00024582449F,
  170. 0.00026179955F, 0.00027881276F, 0.00029693158F, 0.00031622787F,
  171. 0.00033677814F, 0.00035866388F, 0.00038197188F, 0.00040679456F,
  172. 0.00043323036F, 0.00046138411F, 0.00049136745F, 0.00052329927F,
  173. 0.00055730621F, 0.00059352311F, 0.00063209358F, 0.00067317058F,
  174. 0.00071691700F, 0.00076350630F, 0.00081312324F, 0.00086596457F,
  175. 0.00092223983F, 0.00098217216F, 0.0010459992F, 0.0011139742F,
  176. 0.0011863665F, 0.0012634633F, 0.0013455702F, 0.0014330129F,
  177. 0.0015261382F, 0.0016253153F, 0.0017309374F, 0.0018434235F,
  178. 0.0019632195F, 0.0020908006F, 0.0022266726F, 0.0023713743F,
  179. 0.0025254795F, 0.0026895994F, 0.0028643847F, 0.0030505286F,
  180. 0.0032487691F, 0.0034598925F, 0.0036847358F, 0.0039241906F,
  181. 0.0041792066F, 0.0044507950F, 0.0047400328F, 0.0050480668F,
  182. 0.0053761186F, 0.0057254891F, 0.0060975636F, 0.0064938176F,
  183. 0.0069158225F, 0.0073652516F, 0.0078438871F, 0.0083536271F,
  184. 0.0088964928F, 0.009474637F, 0.010090352F, 0.010746080F,
  185. 0.011444421F, 0.012188144F, 0.012980198F, 0.013823725F,
  186. 0.014722068F, 0.015678791F, 0.016697687F, 0.017782797F,
  187. 0.018938423F, 0.020169149F, 0.021479854F, 0.022875735F,
  188. 0.024362330F, 0.025945531F, 0.027631618F, 0.029427276F,
  189. 0.031339626F, 0.033376252F, 0.035545228F, 0.037855157F,
  190. 0.040315199F, 0.042935108F, 0.045725273F, 0.048696758F,
  191. 0.051861348F, 0.055231591F, 0.058820850F, 0.062643361F,
  192. 0.066714279F, 0.071049749F, 0.075666962F, 0.080584227F,
  193. 0.085821044F, 0.091398179F, 0.097337747F, 0.10366330F,
  194. 0.11039993F, 0.11757434F, 0.12521498F, 0.13335215F,
  195. 0.14201813F, 0.15124727F, 0.16107617F, 0.17154380F,
  196. 0.18269168F, 0.19456402F, 0.20720788F, 0.22067342F,
  197. 0.23501402F, 0.25028656F, 0.26655159F, 0.28387361F,
  198. 0.30232132F, 0.32196786F, 0.34289114F, 0.36517414F,
  199. 0.38890521F, 0.41417847F, 0.44109412F, 0.46975890F,
  200. 0.50028648F, 0.53279791F, 0.56742212F, 0.60429640F,
  201. 0.64356699F, 0.68538959F, 0.72993007F, 0.77736504F,
  202. 0.82788260F, 0.88168307F, 0.9389798F, 1.F,
  203. };
  204. #endif
  205. static int mapping0_forward(vorbis_block *vb){
  206. vorbis_dsp_state *vd=vb->vd;
  207. vorbis_info *vi=vd->vi;
  208. codec_setup_info *ci=vi->codec_setup;
  209. private_state *b=vb->vd->backend_state;
  210. vorbis_block_internal *vbi=(vorbis_block_internal *)vb->internal;
  211. int n=vb->pcmend;
  212. int i,j,k;
  213. int *nonzero = alloca(sizeof(*nonzero)*vi->channels);
  214. float **gmdct = _vorbis_block_alloc(vb,vi->channels*sizeof(*gmdct));
  215. int **iwork = _vorbis_block_alloc(vb,vi->channels*sizeof(*iwork));
  216. int ***floor_posts = _vorbis_block_alloc(vb,vi->channels*sizeof(*floor_posts));
  217. float global_ampmax=vbi->ampmax;
  218. float *local_ampmax=alloca(sizeof(*local_ampmax)*vi->channels);
  219. int blocktype=vbi->blocktype;
  220. int modenumber=vb->W;
  221. vorbis_info_mapping0 *info=ci->map_param[modenumber];
  222. vorbis_look_psy *psy_look=b->psy+blocktype+(vb->W?2:0);
  223. vb->mode=modenumber;
  224. for(i=0;i<vi->channels;i++){
  225. float scale=4.f/n;
  226. float scale_dB;
  227. float *pcm =vb->pcm[i];
  228. float *logfft =pcm;
  229. iwork[i]=_vorbis_block_alloc(vb,n/2*sizeof(**iwork));
  230. gmdct[i]=_vorbis_block_alloc(vb,n/2*sizeof(**gmdct));
  231. scale_dB=todB(&scale) + .345; /* + .345 is a hack; the original
  232. todB estimation used on IEEE 754
  233. compliant machines had a bug that
  234. returned dB values about a third
  235. of a decibel too high. The bug
  236. was harmless because tunings
  237. implicitly took that into
  238. account. However, fixing the bug
  239. in the estimator requires
  240. changing all the tunings as well.
  241. For now, it's easier to sync
  242. things back up here, and
  243. recalibrate the tunings in the
  244. next major model upgrade. */
  245. #if 0
  246. if(vi->channels==2){
  247. if(i==0)
  248. _analysis_output("pcmL",seq,pcm,n,0,0,total-n/2);
  249. else
  250. _analysis_output("pcmR",seq,pcm,n,0,0,total-n/2);
  251. }else{
  252. _analysis_output("pcm",seq,pcm,n,0,0,total-n/2);
  253. }
  254. #endif
  255. /* window the PCM data */
  256. _vorbis_apply_window(pcm,b->window,ci->blocksizes,vb->lW,vb->W,vb->nW);
  257. #if 0
  258. if(vi->channels==2){
  259. if(i==0)
  260. _analysis_output("windowedL",seq,pcm,n,0,0,total-n/2);
  261. else
  262. _analysis_output("windowedR",seq,pcm,n,0,0,total-n/2);
  263. }else{
  264. _analysis_output("windowed",seq,pcm,n,0,0,total-n/2);
  265. }
  266. #endif
  267. /* transform the PCM data */
  268. /* only MDCT right now.... */
  269. mdct_forward(b->transform[vb->W][0],pcm,gmdct[i]);
  270. /* FFT yields more accurate tonal estimation (not phase sensitive) */
  271. drft_forward(&b->fft_look[vb->W],pcm);
  272. logfft[0]=scale_dB+todB(pcm) + .345; /* + .345 is a hack; the
  273. original todB estimation used on
  274. IEEE 754 compliant machines had a
  275. bug that returned dB values about
  276. a third of a decibel too high.
  277. The bug was harmless because
  278. tunings implicitly took that into
  279. account. However, fixing the bug
  280. in the estimator requires
  281. changing all the tunings as well.
  282. For now, it's easier to sync
  283. things back up here, and
  284. recalibrate the tunings in the
  285. next major model upgrade. */
  286. local_ampmax[i]=logfft[0];
  287. for(j=1;j<n-1;j+=2){
  288. float temp=pcm[j]*pcm[j]+pcm[j+1]*pcm[j+1];
  289. temp=logfft[(j+1)>>1]=scale_dB+.5f*todB(&temp) + .345; /* +
  290. .345 is a hack; the original todB
  291. estimation used on IEEE 754
  292. compliant machines had a bug that
  293. returned dB values about a third
  294. of a decibel too high. The bug
  295. was harmless because tunings
  296. implicitly took that into
  297. account. However, fixing the bug
  298. in the estimator requires
  299. changing all the tunings as well.
  300. For now, it's easier to sync
  301. things back up here, and
  302. recalibrate the tunings in the
  303. next major model upgrade. */
  304. if(temp>local_ampmax[i])local_ampmax[i]=temp;
  305. }
  306. if(local_ampmax[i]>0.f)local_ampmax[i]=0.f;
  307. if(local_ampmax[i]>global_ampmax)global_ampmax=local_ampmax[i];
  308. #if 0
  309. if(vi->channels==2){
  310. if(i==0){
  311. _analysis_output("fftL",seq,logfft,n/2,1,0,0);
  312. }else{
  313. _analysis_output("fftR",seq,logfft,n/2,1,0,0);
  314. }
  315. }else{
  316. _analysis_output("fft",seq,logfft,n/2,1,0,0);
  317. }
  318. #endif
  319. }
  320. {
  321. float *noise = _vorbis_block_alloc(vb,n/2*sizeof(*noise));
  322. float *tone = _vorbis_block_alloc(vb,n/2*sizeof(*tone));
  323. for(i=0;i<vi->channels;i++){
  324. /* the encoder setup assumes that all the modes used by any
  325. specific bitrate tweaking use the same floor */
  326. int submap=info->chmuxlist[i];
  327. /* the following makes things clearer to *me* anyway */
  328. float *mdct =gmdct[i];
  329. float *logfft =vb->pcm[i];
  330. float *logmdct =logfft+n/2;
  331. float *logmask =logfft;
  332. vb->mode=modenumber;
  333. floor_posts[i]=_vorbis_block_alloc(vb,PACKETBLOBS*sizeof(**floor_posts));
  334. memset(floor_posts[i],0,sizeof(**floor_posts)*PACKETBLOBS);
  335. for(j=0;j<n/2;j++)
  336. logmdct[j]=todB(mdct+j) + .345; /* + .345 is a hack; the original
  337. todB estimation used on IEEE 754
  338. compliant machines had a bug that
  339. returned dB values about a third
  340. of a decibel too high. The bug
  341. was harmless because tunings
  342. implicitly took that into
  343. account. However, fixing the bug
  344. in the estimator requires
  345. changing all the tunings as well.
  346. For now, it's easier to sync
  347. things back up here, and
  348. recalibrate the tunings in the
  349. next major model upgrade. */
  350. #if 0
  351. if(vi->channels==2){
  352. if(i==0)
  353. _analysis_output("mdctL",seq,logmdct,n/2,1,0,0);
  354. else
  355. _analysis_output("mdctR",seq,logmdct,n/2,1,0,0);
  356. }else{
  357. _analysis_output("mdct",seq,logmdct,n/2,1,0,0);
  358. }
  359. #endif
  360. /* first step; noise masking. Not only does 'noise masking'
  361. give us curves from which we can decide how much resolution
  362. to give noise parts of the spectrum, it also implicitly hands
  363. us a tonality estimate (the larger the value in the
  364. 'noise_depth' vector, the more tonal that area is) */
  365. _vp_noisemask(psy_look,
  366. logmdct,
  367. noise); /* noise does not have by-frequency offset
  368. bias applied yet */
  369. #if 0
  370. if(vi->channels==2){
  371. if(i==0)
  372. _analysis_output("noiseL",seq,noise,n/2,1,0,0);
  373. else
  374. _analysis_output("noiseR",seq,noise,n/2,1,0,0);
  375. }else{
  376. _analysis_output("noise",seq,noise,n/2,1,0,0);
  377. }
  378. #endif
  379. /* second step: 'all the other crap'; all the stuff that isn't
  380. computed/fit for bitrate management goes in the second psy
  381. vector. This includes tone masking, peak limiting and ATH */
  382. _vp_tonemask(psy_look,
  383. logfft,
  384. tone,
  385. global_ampmax,
  386. local_ampmax[i]);
  387. #if 0
  388. if(vi->channels==2){
  389. if(i==0)
  390. _analysis_output("toneL",seq,tone,n/2,1,0,0);
  391. else
  392. _analysis_output("toneR",seq,tone,n/2,1,0,0);
  393. }else{
  394. _analysis_output("tone",seq,tone,n/2,1,0,0);
  395. }
  396. #endif
  397. /* third step; we offset the noise vectors, overlay tone
  398. masking. We then do a floor1-specific line fit. If we're
  399. performing bitrate management, the line fit is performed
  400. multiple times for up/down tweakage on demand. */
  401. #if 0
  402. {
  403. float aotuv[psy_look->n];
  404. #endif
  405. _vp_offset_and_mix(psy_look,
  406. noise,
  407. tone,
  408. 1,
  409. logmask,
  410. mdct,
  411. logmdct);
  412. #if 0
  413. if(vi->channels==2){
  414. if(i==0)
  415. _analysis_output("aotuvM1_L",seq,aotuv,psy_look->n,1,1,0);
  416. else
  417. _analysis_output("aotuvM1_R",seq,aotuv,psy_look->n,1,1,0);
  418. }else{
  419. _analysis_output("aotuvM1",seq,aotuv,psy_look->n,1,1,0);
  420. }
  421. }
  422. #endif
  423. #if 0
  424. if(vi->channels==2){
  425. if(i==0)
  426. _analysis_output("mask1L",seq,logmask,n/2,1,0,0);
  427. else
  428. _analysis_output("mask1R",seq,logmask,n/2,1,0,0);
  429. }else{
  430. _analysis_output("mask1",seq,logmask,n/2,1,0,0);
  431. }
  432. #endif
  433. /* this algorithm is hardwired to floor 1 for now; abort out if
  434. we're *not* floor1. This won't happen unless someone has
  435. broken the encode setup lib. Guard it anyway. */
  436. if(ci->floor_type[info->floorsubmap[submap]]!=1)return(-1);
  437. floor_posts[i][PACKETBLOBS/2]=
  438. floor1_fit(vb,b->flr[info->floorsubmap[submap]],
  439. logmdct,
  440. logmask);
  441. /* are we managing bitrate? If so, perform two more fits for
  442. later rate tweaking (fits represent hi/lo) */
  443. if(vorbis_bitrate_managed(vb) && floor_posts[i][PACKETBLOBS/2]){
  444. /* higher rate by way of lower noise curve */
  445. _vp_offset_and_mix(psy_look,
  446. noise,
  447. tone,
  448. 2,
  449. logmask,
  450. mdct,
  451. logmdct);
  452. #if 0
  453. if(vi->channels==2){
  454. if(i==0)
  455. _analysis_output("mask2L",seq,logmask,n/2,1,0,0);
  456. else
  457. _analysis_output("mask2R",seq,logmask,n/2,1,0,0);
  458. }else{
  459. _analysis_output("mask2",seq,logmask,n/2,1,0,0);
  460. }
  461. #endif
  462. floor_posts[i][PACKETBLOBS-1]=
  463. floor1_fit(vb,b->flr[info->floorsubmap[submap]],
  464. logmdct,
  465. logmask);
  466. /* lower rate by way of higher noise curve */
  467. _vp_offset_and_mix(psy_look,
  468. noise,
  469. tone,
  470. 0,
  471. logmask,
  472. mdct,
  473. logmdct);
  474. #if 0
  475. if(vi->channels==2){
  476. if(i==0)
  477. _analysis_output("mask0L",seq,logmask,n/2,1,0,0);
  478. else
  479. _analysis_output("mask0R",seq,logmask,n/2,1,0,0);
  480. }else{
  481. _analysis_output("mask0",seq,logmask,n/2,1,0,0);
  482. }
  483. #endif
  484. floor_posts[i][0]=
  485. floor1_fit(vb,b->flr[info->floorsubmap[submap]],
  486. logmdct,
  487. logmask);
  488. /* we also interpolate a range of intermediate curves for
  489. intermediate rates */
  490. for(k=1;k<PACKETBLOBS/2;k++)
  491. floor_posts[i][k]=
  492. floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]],
  493. floor_posts[i][0],
  494. floor_posts[i][PACKETBLOBS/2],
  495. k*65536/(PACKETBLOBS/2));
  496. for(k=PACKETBLOBS/2+1;k<PACKETBLOBS-1;k++)
  497. floor_posts[i][k]=
  498. floor1_interpolate_fit(vb,b->flr[info->floorsubmap[submap]],
  499. floor_posts[i][PACKETBLOBS/2],
  500. floor_posts[i][PACKETBLOBS-1],
  501. (k-PACKETBLOBS/2)*65536/(PACKETBLOBS/2));
  502. }
  503. }
  504. }
  505. vbi->ampmax=global_ampmax;
  506. /*
  507. the next phases are performed once for vbr-only and PACKETBLOB
  508. times for bitrate managed modes.
  509. 1) encode actual mode being used
  510. 2) encode the floor for each channel, compute coded mask curve/res
  511. 3) normalize and couple.
  512. 4) encode residue
  513. 5) save packet bytes to the packetblob vector
  514. */
  515. /* iterate over the many masking curve fits we've created */
  516. {
  517. int **couple_bundle=alloca(sizeof(*couple_bundle)*vi->channels);
  518. int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels);
  519. for(k=(vorbis_bitrate_managed(vb)?0:PACKETBLOBS/2);
  520. k<=(vorbis_bitrate_managed(vb)?PACKETBLOBS-1:PACKETBLOBS/2);
  521. k++){
  522. oggpack_buffer *opb=vbi->packetblob[k];
  523. /* start out our new packet blob with packet type and mode */
  524. /* Encode the packet type */
  525. oggpack_write(opb,0,1);
  526. /* Encode the modenumber */
  527. /* Encode frame mode, pre,post windowsize, then dispatch */
  528. oggpack_write(opb,modenumber,b->modebits);
  529. if(vb->W){
  530. oggpack_write(opb,vb->lW,1);
  531. oggpack_write(opb,vb->nW,1);
  532. }
  533. /* encode floor, compute masking curve, sep out residue */
  534. for(i=0;i<vi->channels;i++){
  535. int submap=info->chmuxlist[i];
  536. int *ilogmask=iwork[i];
  537. nonzero[i]=floor1_encode(opb,vb,b->flr[info->floorsubmap[submap]],
  538. floor_posts[i][k],
  539. ilogmask);
  540. #if 0
  541. {
  542. char buf[80];
  543. sprintf(buf,"maskI%c%d",i?'R':'L',k);
  544. float work[n/2];
  545. for(j=0;j<n/2;j++)
  546. work[j]=FLOOR1_fromdB_LOOKUP[iwork[i][j]];
  547. _analysis_output(buf,seq,work,n/2,1,1,0);
  548. }
  549. #endif
  550. }
  551. /* our iteration is now based on masking curve, not prequant and
  552. coupling. Only one prequant/coupling step */
  553. /* quantize/couple */
  554. /* incomplete implementation that assumes the tree is all depth
  555. one, or no tree at all */
  556. _vp_couple_quantize_normalize(k,
  557. &ci->psy_g_param,
  558. psy_look,
  559. info,
  560. gmdct,
  561. iwork,
  562. nonzero,
  563. ci->psy_g_param.sliding_lowpass[vb->W][k],
  564. vi->channels);
  565. #if 0
  566. for(i=0;i<vi->channels;i++){
  567. char buf[80];
  568. sprintf(buf,"res%c%d",i?'R':'L',k);
  569. float work[n/2];
  570. for(j=0;j<n/2;j++)
  571. work[j]=iwork[i][j];
  572. _analysis_output(buf,seq,work,n/2,1,0,0);
  573. }
  574. #endif
  575. /* classify and encode by submap */
  576. for(i=0;i<info->submaps;i++){
  577. int ch_in_bundle=0;
  578. long **classifications;
  579. int resnum=info->residuesubmap[i];
  580. for(j=0;j<vi->channels;j++){
  581. if(info->chmuxlist[j]==i){
  582. zerobundle[ch_in_bundle]=0;
  583. if(nonzero[j])zerobundle[ch_in_bundle]=1;
  584. couple_bundle[ch_in_bundle++]=iwork[j];
  585. }
  586. }
  587. classifications=_residue_P[ci->residue_type[resnum]]->
  588. class(vb,b->residue[resnum],couple_bundle,zerobundle,ch_in_bundle);
  589. ch_in_bundle=0;
  590. for(j=0;j<vi->channels;j++)
  591. if(info->chmuxlist[j]==i)
  592. couple_bundle[ch_in_bundle++]=iwork[j];
  593. _residue_P[ci->residue_type[resnum]]->
  594. forward(opb,vb,b->residue[resnum],
  595. couple_bundle,zerobundle,ch_in_bundle,classifications,i);
  596. }
  597. /* ok, done encoding. Next protopacket. */
  598. }
  599. }
  600. #if 0
  601. seq++;
  602. total+=ci->blocksizes[vb->W]/4+ci->blocksizes[vb->nW]/4;
  603. #endif
  604. return(0);
  605. }
  606. static int mapping0_inverse(vorbis_block *vb,vorbis_info_mapping *l){
  607. vorbis_dsp_state *vd=vb->vd;
  608. vorbis_info *vi=vd->vi;
  609. codec_setup_info *ci=vi->codec_setup;
  610. private_state *b=vd->backend_state;
  611. vorbis_info_mapping0 *info=(vorbis_info_mapping0 *)l;
  612. int i,j;
  613. long n=vb->pcmend=ci->blocksizes[vb->W];
  614. float **pcmbundle=alloca(sizeof(*pcmbundle)*vi->channels);
  615. int *zerobundle=alloca(sizeof(*zerobundle)*vi->channels);
  616. int *nonzero =alloca(sizeof(*nonzero)*vi->channels);
  617. void **floormemo=alloca(sizeof(*floormemo)*vi->channels);
  618. /* recover the spectral envelope; store it in the PCM vector for now */
  619. for(i=0;i<vi->channels;i++){
  620. int submap=info->chmuxlist[i];
  621. floormemo[i]=_floor_P[ci->floor_type[info->floorsubmap[submap]]]->
  622. inverse1(vb,b->flr[info->floorsubmap[submap]]);
  623. if(floormemo[i])
  624. nonzero[i]=1;
  625. else
  626. nonzero[i]=0;
  627. memset(vb->pcm[i],0,sizeof(*vb->pcm[i])*n/2);
  628. }
  629. /* channel coupling can 'dirty' the nonzero listing */
  630. for(i=0;i<info->coupling_steps;i++){
  631. if(nonzero[info->coupling_mag[i]] ||
  632. nonzero[info->coupling_ang[i]]){
  633. nonzero[info->coupling_mag[i]]=1;
  634. nonzero[info->coupling_ang[i]]=1;
  635. }
  636. }
  637. /* recover the residue into our working vectors */
  638. for(i=0;i<info->submaps;i++){
  639. int ch_in_bundle=0;
  640. for(j=0;j<vi->channels;j++){
  641. if(info->chmuxlist[j]==i){
  642. if(nonzero[j])
  643. zerobundle[ch_in_bundle]=1;
  644. else
  645. zerobundle[ch_in_bundle]=0;
  646. pcmbundle[ch_in_bundle++]=vb->pcm[j];
  647. }
  648. }
  649. _residue_P[ci->residue_type[info->residuesubmap[i]]]->
  650. inverse(vb,b->residue[info->residuesubmap[i]],
  651. pcmbundle,zerobundle,ch_in_bundle);
  652. }
  653. /* channel coupling */
  654. for(i=info->coupling_steps-1;i>=0;i--){
  655. float *pcmM=vb->pcm[info->coupling_mag[i]];
  656. float *pcmA=vb->pcm[info->coupling_ang[i]];
  657. for(j=0;j<n/2;j++){
  658. float mag=pcmM[j];
  659. float ang=pcmA[j];
  660. if(mag>0)
  661. if(ang>0){
  662. pcmM[j]=mag;
  663. pcmA[j]=mag-ang;
  664. }else{
  665. pcmA[j]=mag;
  666. pcmM[j]=mag+ang;
  667. }
  668. else
  669. if(ang>0){
  670. pcmM[j]=mag;
  671. pcmA[j]=mag+ang;
  672. }else{
  673. pcmA[j]=mag;
  674. pcmM[j]=mag-ang;
  675. }
  676. }
  677. }
  678. /* compute and apply spectral envelope */
  679. for(i=0;i<vi->channels;i++){
  680. float *pcm=vb->pcm[i];
  681. int submap=info->chmuxlist[i];
  682. _floor_P[ci->floor_type[info->floorsubmap[submap]]]->
  683. inverse2(vb,b->flr[info->floorsubmap[submap]],
  684. floormemo[i],pcm);
  685. }
  686. /* transform the PCM data; takes PCM vector, vb; modifies PCM vector */
  687. /* only MDCT right now.... */
  688. for(i=0;i<vi->channels;i++){
  689. float *pcm=vb->pcm[i];
  690. mdct_backward(b->transform[vb->W][0],pcm,pcm);
  691. }
  692. /* all done! */
  693. return(0);
  694. }
  695. /* export hooks */
  696. const vorbis_func_mapping mapping0_exportbundle={
  697. &mapping0_pack,
  698. &mapping0_unpack,
  699. &mapping0_free_info,
  700. &mapping0_forward,
  701. &mapping0_inverse
  702. };