Gloria Wang | 37fe158 | 2010-03-12 14:53:20 -0800 | [diff] [blame] | 1 | /************************************************************************ |
Gloria Wang | 2da723a | 2010-03-18 15:56:16 -0700 | [diff] [blame^] | 2 | * Copyright (C) 2002-2009, Xiph.org Foundation |
| 3 | * Copyright (C) 2010, Robin Watts for Pinknoise Productions Ltd |
Gloria Wang | 37fe158 | 2010-03-12 14:53:20 -0800 | [diff] [blame] | 4 | * All rights reserved. |
| 5 | * |
| 6 | * Redistribution and use in source and binary forms, with or without |
Gloria Wang | 2da723a | 2010-03-18 15:56:16 -0700 | [diff] [blame^] | 7 | * modification, are permitted provided that the following conditions |
| 8 | * are met: |
Gloria Wang | 37fe158 | 2010-03-12 14:53:20 -0800 | [diff] [blame] | 9 | * |
| 10 | * * Redistributions of source code must retain the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer. |
| 12 | * * Redistributions in binary form must reproduce the above |
| 13 | * copyright notice, this list of conditions and the following disclaimer |
| 14 | * in the documentation and/or other materials provided with the |
| 15 | * distribution. |
Gloria Wang | 2da723a | 2010-03-18 15:56:16 -0700 | [diff] [blame^] | 16 | * * Neither the names of the Xiph.org Foundation nor Pinknoise |
| 17 | * Productions Ltd nor the names of its contributors may be used to |
| 18 | * endorse or promote products derived from this software without |
| 19 | * specific prior written permission. |
Gloria Wang | 37fe158 | 2010-03-12 14:53:20 -0800 | [diff] [blame] | 20 | * |
| 21 | * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| 22 | * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| 23 | * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| 24 | * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| 25 | * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| 26 | * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| 27 | * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| 28 | * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| 29 | * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| 30 | * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| 31 | * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 32 | ************************************************************************ |
Gloria Wang | 7913073 | 2010-02-08 14:41:04 -0800 | [diff] [blame] | 33 | |
| 34 | function: channel mapping 0 implementation |
| 35 | |
Gloria Wang | 37fe158 | 2010-03-12 14:53:20 -0800 | [diff] [blame] | 36 | ************************************************************************/ |
Gloria Wang | 7913073 | 2010-02-08 14:41:04 -0800 | [diff] [blame] | 37 | |
| 38 | #include <stdlib.h> |
| 39 | #include <stdio.h> |
| 40 | #include <string.h> |
| 41 | #include <math.h> |
| 42 | #include "ogg.h" |
| 43 | #include "os.h" |
| 44 | #include "ivorbiscodec.h" |
| 45 | #include "mdct.h" |
| 46 | #include "codec_internal.h" |
| 47 | #include "codebook.h" |
| 48 | #include "misc.h" |
| 49 | |
| 50 | void mapping_clear_info(vorbis_info_mapping *info){ |
| 51 | if(info){ |
| 52 | if(info->chmuxlist)_ogg_free(info->chmuxlist); |
| 53 | if(info->submaplist)_ogg_free(info->submaplist); |
| 54 | if(info->coupling)_ogg_free(info->coupling); |
| 55 | memset(info,0,sizeof(*info)); |
| 56 | } |
| 57 | } |
| 58 | |
| 59 | static int ilog(unsigned int v){ |
| 60 | int ret=0; |
| 61 | if(v)--v; |
| 62 | while(v){ |
| 63 | ret++; |
| 64 | v>>=1; |
| 65 | } |
| 66 | return(ret); |
| 67 | } |
| 68 | |
| 69 | /* also responsible for range checking */ |
| 70 | int mapping_info_unpack(vorbis_info_mapping *info,vorbis_info *vi, |
| 71 | oggpack_buffer *opb){ |
| 72 | int i; |
| 73 | codec_setup_info *ci=(codec_setup_info *)vi->codec_setup; |
| 74 | memset(info,0,sizeof(*info)); |
| 75 | |
| 76 | if(oggpack_read(opb,1)) |
| 77 | info->submaps=oggpack_read(opb,4)+1; |
| 78 | else |
| 79 | info->submaps=1; |
| 80 | |
| 81 | if(oggpack_read(opb,1)){ |
| 82 | info->coupling_steps=oggpack_read(opb,8)+1; |
| 83 | info->coupling= |
| 84 | _ogg_malloc(info->coupling_steps*sizeof(*info->coupling)); |
| 85 | |
| 86 | for(i=0;i<info->coupling_steps;i++){ |
| 87 | int testM=info->coupling[i].mag=(unsigned char)(oggpack_read(opb,ilog(vi->channels))); |
| 88 | int testA=info->coupling[i].ang=(unsigned char)(oggpack_read(opb,ilog(vi->channels))); |
| 89 | |
| 90 | if(testM<0 || |
| 91 | testA<0 || |
| 92 | testM==testA || |
| 93 | testM>=vi->channels || |
| 94 | testA>=vi->channels) goto err_out; |
| 95 | } |
| 96 | |
| 97 | } |
| 98 | |
| 99 | if(oggpack_read(opb,2)>0)goto err_out; /* 2,3:reserved */ |
| 100 | |
| 101 | if(info->submaps>1){ |
| 102 | info->chmuxlist=_ogg_malloc(sizeof(*info->chmuxlist)*vi->channels); |
| 103 | for(i=0;i<vi->channels;i++){ |
| 104 | info->chmuxlist[i]=(unsigned char)(oggpack_read(opb,4)); |
| 105 | if(info->chmuxlist[i]>=info->submaps)goto err_out; |
| 106 | } |
| 107 | } |
| 108 | |
| 109 | info->submaplist=_ogg_malloc(sizeof(*info->submaplist)*info->submaps); |
| 110 | for(i=0;i<info->submaps;i++){ |
| 111 | int temp=oggpack_read(opb,8); |
| 112 | info->submaplist[i].floor=(char)oggpack_read(opb,8); |
| 113 | if(info->submaplist[i].floor>=ci->floors)goto err_out; |
| 114 | info->submaplist[i].residue=(char)oggpack_read(opb,8); |
| 115 | if(info->submaplist[i].residue>=ci->residues)goto err_out; |
| 116 | } |
| 117 | |
| 118 | return 0; |
| 119 | |
| 120 | err_out: |
| 121 | mapping_clear_info(info); |
| 122 | return -1; |
| 123 | } |
| 124 | |
| 125 | int mapping_inverse(vorbis_dsp_state *vd,vorbis_info_mapping *info){ |
| 126 | vorbis_info *vi=vd->vi; |
| 127 | codec_setup_info *ci=(codec_setup_info *)vi->codec_setup; |
| 128 | |
| 129 | int i,j; |
| 130 | long n=ci->blocksizes[vd->W]; |
| 131 | |
| 132 | ogg_int32_t **pcmbundle= |
| 133 | alloca(sizeof(*pcmbundle)*vi->channels); |
| 134 | int *zerobundle= |
| 135 | alloca(sizeof(*zerobundle)*vi->channels); |
| 136 | int *nonzero= |
| 137 | alloca(sizeof(*nonzero)*vi->channels); |
| 138 | ogg_int32_t **floormemo= |
| 139 | alloca(sizeof(*floormemo)*vi->channels); |
| 140 | |
| 141 | /* recover the spectral envelope; store it in the PCM vector for now */ |
| 142 | for(i=0;i<vi->channels;i++){ |
| 143 | int submap=0; |
| 144 | int floorno; |
| 145 | |
| 146 | if(info->submaps>1) |
| 147 | submap=info->chmuxlist[i]; |
| 148 | floorno=info->submaplist[submap].floor; |
| 149 | |
| 150 | if(ci->floor_type[floorno]){ |
| 151 | /* floor 1 */ |
| 152 | floormemo[i]=alloca(sizeof(*floormemo[i])* |
| 153 | floor1_memosize(ci->floor_param[floorno])); |
| 154 | floormemo[i]=floor1_inverse1(vd,ci->floor_param[floorno],floormemo[i]); |
| 155 | }else{ |
| 156 | /* floor 0 */ |
| 157 | floormemo[i]=alloca(sizeof(*floormemo[i])* |
| 158 | floor0_memosize(ci->floor_param[floorno])); |
| 159 | floormemo[i]=floor0_inverse1(vd,ci->floor_param[floorno],floormemo[i]); |
| 160 | } |
| 161 | |
| 162 | if(floormemo[i]) |
| 163 | nonzero[i]=1; |
| 164 | else |
| 165 | nonzero[i]=0; |
| 166 | memset(vd->work[i],0,sizeof(*vd->work[i])*n/2); |
| 167 | } |
| 168 | |
| 169 | /* channel coupling can 'dirty' the nonzero listing */ |
| 170 | for(i=0;i<info->coupling_steps;i++){ |
| 171 | if(nonzero[info->coupling[i].mag] || |
| 172 | nonzero[info->coupling[i].ang]){ |
| 173 | nonzero[info->coupling[i].mag]=1; |
| 174 | nonzero[info->coupling[i].ang]=1; |
| 175 | } |
| 176 | } |
| 177 | |
| 178 | /* recover the residue into our working vectors */ |
| 179 | for(i=0;i<info->submaps;i++){ |
| 180 | int ch_in_bundle=0; |
| 181 | for(j=0;j<vi->channels;j++){ |
| 182 | if(!info->chmuxlist || info->chmuxlist[j]==i){ |
| 183 | if(nonzero[j]) |
| 184 | zerobundle[ch_in_bundle]=1; |
| 185 | else |
| 186 | zerobundle[ch_in_bundle]=0; |
| 187 | pcmbundle[ch_in_bundle++]=vd->work[j]; |
| 188 | } |
| 189 | } |
| 190 | |
| 191 | res_inverse(vd,ci->residue_param+info->submaplist[i].residue, |
| 192 | pcmbundle,zerobundle,ch_in_bundle); |
| 193 | } |
| 194 | |
| 195 | //for(j=0;j<vi->channels;j++) |
| 196 | //_analysis_output("coupled",seq+j,vb->pcm[j],-8,n/2,0,0); |
| 197 | |
| 198 | /* channel coupling */ |
| 199 | for(i=info->coupling_steps-1;i>=0;i--){ |
| 200 | ogg_int32_t *pcmM=vd->work[info->coupling[i].mag]; |
| 201 | ogg_int32_t *pcmA=vd->work[info->coupling[i].ang]; |
| 202 | |
| 203 | for(j=0;j<n/2;j++){ |
| 204 | ogg_int32_t mag=pcmM[j]; |
| 205 | ogg_int32_t ang=pcmA[j]; |
| 206 | |
| 207 | if(mag>0) |
| 208 | if(ang>0){ |
| 209 | pcmM[j]=mag; |
| 210 | pcmA[j]=mag-ang; |
| 211 | }else{ |
| 212 | pcmA[j]=mag; |
| 213 | pcmM[j]=mag+ang; |
| 214 | } |
| 215 | else |
| 216 | if(ang>0){ |
| 217 | pcmM[j]=mag; |
| 218 | pcmA[j]=mag+ang; |
| 219 | }else{ |
| 220 | pcmA[j]=mag; |
| 221 | pcmM[j]=mag-ang; |
| 222 | } |
| 223 | } |
| 224 | } |
| 225 | |
| 226 | //for(j=0;j<vi->channels;j++) |
| 227 | //_analysis_output("residue",seq+j,vb->pcm[j],-8,n/2,0,0); |
| 228 | |
| 229 | /* compute and apply spectral envelope */ |
| 230 | for(i=0;i<vi->channels;i++){ |
| 231 | ogg_int32_t *pcm=vd->work[i]; |
| 232 | int submap=0; |
| 233 | int floorno; |
| 234 | |
| 235 | if(info->submaps>1) |
| 236 | submap=info->chmuxlist[i]; |
| 237 | floorno=info->submaplist[submap].floor; |
| 238 | |
| 239 | if(ci->floor_type[floorno]){ |
| 240 | /* floor 1 */ |
| 241 | floor1_inverse2(vd,ci->floor_param[floorno],floormemo[i],pcm); |
| 242 | }else{ |
| 243 | /* floor 0 */ |
| 244 | floor0_inverse2(vd,ci->floor_param[floorno],floormemo[i],pcm); |
| 245 | } |
| 246 | } |
| 247 | |
| 248 | //for(j=0;j<vi->channels;j++) |
| 249 | //_analysis_output("mdct",seq+j,vb->pcm[j],-24,n/2,0,1); |
| 250 | |
| 251 | /* transform the PCM data; takes PCM vector, vb; modifies PCM vector */ |
| 252 | /* only MDCT right now.... */ |
| 253 | for(i=0;i<vi->channels;i++) |
| 254 | mdct_backward(n,vd->work[i]); |
| 255 | |
| 256 | //for(j=0;j<vi->channels;j++) |
| 257 | //_analysis_output("imdct",seq+j,vb->pcm[j],-24,n,0,0); |
| 258 | |
| 259 | /* all done! */ |
| 260 | return(0); |
| 261 | } |