mapping0.c 28 KB

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