source: vbox/trunk/src/libs/libvorbis-1.3.7/vq/make_residue_books.pl@ 96468

#!/usr/bin/perl

# quick, very dirty little script so that we can put all the
# information for building a residue book set (except the original
# partitioning) in one spec file.

#eg:

# >res0_128_128 interleaved
# haux 44c0_s/resaux_0.vqd res0_96_128aux 0,4,2 9
# :1 res0_128_128_1.vqd, 4, nonseq cull, 0 +- 1
# :2 res0_128_128_2.vqd, 4, nonseq, 0 +- 1(.7) 2
# :3 res0_128_128_3.vqd, 4, nonseq, 0 +- 1(.7) 3 5
# :4 res0_128_128_4.vqd, 2, nonseq, 0 +- 1(.7) 3 5 8 11
# :5 res0_128_128_5.vqd, 1, nonseq, 0 +- 1 3 5 8 11 14 17 20 24 28 31 35 39 


die "Could not open $ARGV[0]: $!" unless open (F,$ARGV[0]);

$goflag=0;
while($line=<F>){

    print "#### $line";
    if($line=~m/^GO/){
        $goflag=1;
        next;
    }

    if($goflag==0){
        if($line=~m/\S+/ && !($line=~m/^\#/) ){
            my $command=$line;
            print ">>> $command";
            die "Couldn't shell command.\n\tcommand:$command\n" 
                if syst($command);
        }
        next;
    }

    # >res0_128_128
    if($line=~m/^>(\S+)\s+(\S*)/){
        # set the output name
        $globalname=$1;
        $interleave=$2;
        next;
    }

    # haux 44c0_s/resaux_0.vqd res0_96_128aux 0,4,2 9
    if($line=~m/^h(.*)/){
        # build a huffman book (no mapping) 
        my($name,$datafile,$bookname,$interval,$range)=split(' ',$1);
 
        # check the desired subdir to see if the data file exists
        if(-e $datafile){
            my $command="cp $datafile $bookname.tmp";
            print ">>> $command\n";
            die "Couldn't access partition data file.\n\tcommand:$command\n" 
                if syst($command);

            my $command="huffbuild $bookname.tmp $interval";
            print ">>> $command\n";
            die "Couldn't build huffbook.\n\tcommand:$command\n" 
                if syst($command);

            my $command="rm $bookname.tmp";
            print ">>> $command\n";
            die "Couldn't remove temporary file.\n\tcommand:$command\n" 
                if syst($command);
        }else{
            my $command="huffbuild $bookname.tmp 0-$range";
            print ">>> $command\n";
            die "Couldn't build huffbook.\n\tcommand:$command\n" 
                if syst($command);

        }
        next;
    }

    # :1 res0_128_128_1.vqd, 4, nonseq, 0 +- 1
    if($line=~m/^:(.*)/){
        my($namedata,$dim,$seqp,$vals)=split(',',$1);
        my($name,$datafile)=split(' ',$namedata);
        # build value list
        my$plusminus="+";
        my$list;
        my$thlist;
        my$count=0;
        foreach my$val (split(' ',$vals)){
            if($val=~/\-?\+?\d+/){
                my$th;

                # got an explicit threshhint?
                if($val=~/([0-9\.]+)\(([^\)]+)/){
                    $val=$1;
                    $th=$2;
                }

                if($plusminus=~/-/){
                    $list.="-$val ";
                    if(defined($th)){
                        $thlist.="," if(defined($thlist));
                        $thlist.="-$th";
                    }
                    $count++;
                }
                if($plusminus=~/\+/){
                    $list.="$val ";
                    if(defined($th)){
                        $thlist.="," if(defined($thlist));
                        $thlist.="$th";
                    }
                    $count++;
                }
            }else{
                $plusminus=$val;
            }
        }
        die "Couldn't open temp file $globalname$name.vql: $!" unless
            open(G,">$globalname$name.vql");
        print G "$count $dim 0 ";
        if($seqp=~/non/){
            print G "0\n$list\n";
        }else{  
            print G "1\n$list\n";
        }
        close(G);

        my $command="latticebuild $globalname$name.vql > $globalname$name.vqh";
        print ">>> $command\n";
        die "Couldn't build latticebook.\n\tcommand:$command\n" 
            if syst($command);

        if(-e $datafile){
        
            if($interleave=~/non/){
                $restune="res1tune";
            }else{
                $restune="res0tune";
            }
            
            if($seqp=~/cull/){
                my $command="$restune $globalname$name.vqh $datafile 1 > temp$$.vqh";
                print ">>> $command\n";
                die "Couldn't tune latticebook.\n\tcommand:$command\n" 
                    if syst($command);
            }else{
                my $command="$restune $globalname$name.vqh $datafile > temp$$.vqh";
                print ">>> $command\n";
                die "Couldn't tune latticebook.\n\tcommand:$command\n" 
                    if syst($command);
            }

            my $command="mv temp$$.vqh $globalname$name.vqh";
            print ">>> $command\n";
            die "Couldn't rename latticebook.\n\tcommand:$command\n" 
                if syst($command);

        }else{
            print "No matching training file; leaving this codebook untrained.\n";
        }

        my $command="rm $globalname$name.vql";
        print ">>> $command\n";
        die "Couldn't remove temp files.\n\tcommand:$command\n" 
            if syst($command);

        next;
    }
}

$command="rm -f temp$$.vqd";
print ">>> $command\n";
die "Couldn't remove temp files.\n\tcommand:$command\n" 
    if syst($command);

sub syst{
    system(@_)/256;
}