package net.sourceforge.jaad.aac.error;
import net.sourceforge.jaad.aac.AACException;
import net.sourceforge.jaad.aac.huffman.HCB;
import net.sourceforge.jaad.aac.syntax.SyntaxConstants;
import net.sourceforge.jaad.aac.syntax.IBitStream;
import net.sourceforge.jaad.aac.syntax.ICSInfo;
import net.sourceforge.jaad.aac.syntax.ICStream;
/**
* This class is part of JAAD ( jaadec.sourceforge.net ) that is distributed
* under the Public Domain license. Code changes provided by the JCodec project
* are distributed under FreeBSD license.
*
* Huffman Codeword Reordering
* Decodes spectral data for ICStreams if error resilience is used for
* section data.
*
* @author in-somnia
*/
//TODO: needs decodeSpectralDataER() in BitStream
public class HCR implements SyntaxConstants {
private static class Codeword {
int cb, decoded, sp_offset;
BitsBuffer bits;
private void fill(int sp, int cb) {
sp_offset = sp;
this.cb = cb;
decoded = 0;
bits = new BitsBuffer();
}
}
private static final int NUM_CB = 6;
private static final int NUM_CB_ER = 22;
private static final int MAX_CB = 32;
private static final int VCB11_FIRST = 16;
private static final int VCB11_LAST = 31;
private static final int[] PRE_SORT_CB_STD = {11, 9, 7, 5, 3, 1};
private static final int[] PRE_SORT_CB_ER = {11, 31, 30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 9, 7, 5, 3, 1};
private static final int[] MAX_CW_LEN = {0, 11, 9, 20, 16, 13, 11, 14, 12, 17, 14, 49,
0, 0, 0, 0, 14, 17, 21, 21, 25, 25, 29, 29, 29, 29, 33, 33, 33, 37, 37, 41};
private static boolean isGoodCB(int cb, int sectCB) {
boolean b = false;
if((sectCB>HCB.ZERO_HCB&§CB<=HCB.ESCAPE_HCB)||(sectCB>=VCB11_FIRST&§CB<=VCB11_LAST)) {
if(cb<HCB.ESCAPE_HCB) b = ((sectCB==cb)||(sectCB==cb+1));
else b = (sectCB==cb);
}
return b;
}
//sectionDataResilience = hDecoder->aacSectionDataResilienceFlag
public static void decodeReorderedSpectralData(ICStream ics, IBitStream _in, short[] spectralData, boolean sectionDataResilience) throws AACException {
final ICSInfo info = ics.getInfo();
final int windowGroupCount = info.getWindowGroupCount();
final int maxSFB = info.getMaxSFB();
final int[] swbOffsets = info.getSWBOffsets();
final int swbOffsetMax = info.getSWBOffsetMax();
//TODO:
//final SectionData sectData = ics.getSectionData();
final int[][] sectStart = new int[0][0]; //sectData.getSectStart();
final int[][] sectEnd = new int[0][0]; //sectData.getSectEnd();
final int[] numSec = new int[0]; //sectData.getNumSec();
final int[][] sectCB = new int[0][0]; //sectData.getSectCB();
final int[][] sectSFBOffsets = new int[0][0]; //info.getSectSFBOffsets();
//check parameter
final int spDataLen = ics.getReorderedSpectralDataLength();
if(spDataLen==0) return;
final int longestLen = ics.getLongestCodewordLength();
if(longestLen==0||longestLen>=spDataLen) throw new AACException("length of longest HCR codeword out of range");
//create spOffsets
final int[] spOffsets = new int[8];
final int shortFrameLen = spectralData.length/8;
spOffsets[0] = 0;
int g;
for(g = 1; g<windowGroupCount; g++) {
spOffsets[g] = spOffsets[g-1]+shortFrameLen*info.getWindowGroupLength(g-1);
}
final Codeword[] codeword = new Codeword[512];
final BitsBuffer[] segment = new BitsBuffer[512];
int lastCB;
int[] preSortCB;
if(sectionDataResilience) {
preSortCB = PRE_SORT_CB_ER;
lastCB = NUM_CB_ER;
}
else {
preSortCB = PRE_SORT_CB_STD;
lastCB = NUM_CB;
}
int PCWs_done = 0;
int segmentsCount = 0;
int numberOfCodewords = 0;
int bitsread = 0;
int sfb, w_idx, i, thisCB, thisSectCB, cws;
//step 1: decode PCW's (set 0), and stuff data in easier-to-use format
for(int sortloop = 0; sortloop<lastCB; sortloop++) {
//select codebook to process this pass
thisCB = preSortCB[sortloop];
for(sfb = 0; sfb<maxSFB; sfb++) {
for(w_idx = 0; 4*w_idx<(Math.min(swbOffsets[sfb+1], swbOffsetMax)-swbOffsets[sfb]); w_idx++) {
for(g = 0; g<windowGroupCount; g++) {
for(i = 0; i<numSec[g]; i++) {
if((sectStart[g][i]<=sfb)&&(sectEnd[g][i]>sfb)) {
/* check whether codebook used here is the one we want to process */
thisSectCB = sectCB[g][i];
if(isGoodCB(thisCB, thisSectCB)) {
//precalculation
int sect_sfb_size = sectSFBOffsets[g][sfb+1]-sectSFBOffsets[g][sfb];
int inc = (thisSectCB<HCB.FIRST_PAIR_HCB) ? 4 : 2;
int group_cws_count = (4*info.getWindowGroupLength(g))/inc;
int segwidth = Math.min(MAX_CW_LEN[thisSectCB], longestLen);
//read codewords until end of sfb or end of window group
for(cws = 0; (cws<group_cws_count)&&((cws+w_idx*group_cws_count)<sect_sfb_size); cws++) {
int sp = spOffsets[g]+sectSFBOffsets[g][sfb]+inc*(cws+w_idx*group_cws_count);
//read and decode PCW
if(PCWs_done==0) {
//read in normal segments
if(bitsread+segwidth<=spDataLen) {
segment[segmentsCount].readSegment(segwidth, _in);
bitsread += segwidth;
//Huffman.decodeSpectralDataER(segment[segmentsCount], thisSectCB, spectralData, sp);
//keep leftover bits
segment[segmentsCount].rewindReverse();
segmentsCount++;
}
else {
//remaining after last segment
if(bitsread<spDataLen) {
int additional_bits = spDataLen-bitsread;
segment[segmentsCount].readSegment(additional_bits, _in);
segment[segmentsCount].len += segment[segmentsCount-1].len;
segment[segmentsCount].rewindReverse();
if(segment[segmentsCount-1].len>32) {
segment[segmentsCount-1].bufb = segment[segmentsCount].bufb
+segment[segmentsCount-1].showBits(segment[segmentsCount-1].len-32);
segment[segmentsCount-1].bufa = segment[segmentsCount].bufa
+segment[segmentsCount-1].showBits(32);
}
else {
segment[segmentsCount-1].bufa = segment[segmentsCount].bufa
+segment[segmentsCount-1].showBits(segment[segmentsCount-1].len);
segment[segmentsCount-1].bufb = segment[segmentsCount].bufb;
}
segment[segmentsCount-1].len += additional_bits;
}
bitsread = spDataLen;
PCWs_done = 1;
codeword[0].fill(sp, thisSectCB);
}
}
else {
codeword[numberOfCodewords-segmentsCount].fill(sp, thisSectCB);
}
numberOfCodewords++;
}
}
}
}
}
}
}
}
if(segmentsCount==0) throw new AACException("no segments _in HCR");
final int numberOfSets = numberOfCodewords/segmentsCount;
//step 2: decode nonPCWs
int trial, codewordBase, segmentID, codewordID;
for(int set = 1; set<=numberOfSets; set++) {
for(trial = 0; trial<segmentsCount; trial++) {
for(codewordBase = 0; codewordBase<segmentsCount; codewordBase++) {
segmentID = (trial+codewordBase)%segmentsCount;
codewordID = codewordBase+set*segmentsCount-segmentsCount;
//data up
if(codewordID>=numberOfCodewords-segmentsCount) break;
if((codeword[codewordID].decoded==0)&&(segment[segmentID].len>0)) {
if(codeword[codewordID].bits.len!=0) segment[segmentID].concatBits(codeword[codewordID].bits);
int tmplen = segment[segmentID].len;
/*int ret = Huffman.decodeSpectralDataER(segment[segmentID], codeword[codewordID].cb,
spectralData, codeword[codewordID].sp_offset);
if(ret>=0) codeword[codewordID].decoded = 1;
else {
codeword[codewordID].bits = segment[segmentID];
codeword[codewordID].bits.len = tmplen;
}*/
}
}
}
for(i = 0; i<segmentsCount; i++) {
segment[i].rewindReverse();
}
}
}
}