ecvqtrain.c

spiht for linux this is used to decod and encode vedio i wich all enjoy

/*
 *
 * QccPack: Quantization, compression, and coding utilities
 * Copyright (C) 1997-2009  James E. Fowler
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 *
 */


#include "ecvqtrain.h"

#define USG_STRING "[-i %: %s:initialcodebookfile] [-s %: %d:initialcodebooksize] [-I %: %d:iterations] [-t %: %f:threshold] %f:lambda %s:trainingfile %s:codebookfile"


int initialcodebook_specified = 0;
int codebooksize_specified = 0;
int iterations_specified = 0;
int threshold_specified = 0;

QccVQCodebook Codebook;
int CodebookSize;
int Iterations;
float Threshold;

float Lambda;

QccDataset TrainingData;

QccString FinalCodebookFilename;


int main(int argc, char *argv[])
{
  QccInit(argc, argv);
  
  QccDatasetInitialize(&TrainingData);
  QccVQCodebookInitialize(&Codebook);

  if (QccParseParameters(argc, argv,
                         USG_STRING,
                         &initialcodebook_specified,
                         Codebook.filename,
                         &codebooksize_specified,
                         &CodebookSize,
                         &iterations_specified,
                         &Iterations,
                         &threshold_specified,
                         &Threshold,
                         &Lambda,
                         TrainingData.filename, 
                         FinalCodebookFilename))
    QccErrorExit();
  
  if ((initialcodebook_specified && codebooksize_specified) ||
      (!initialcodebook_specified && !codebooksize_specified))
    {
      QccErrorAddMessage("%s: Either -i or -s must be specified (but not both)\n",
                         argv[0]);
      QccErrorExit();
    }
  
  if ((iterations_specified && threshold_specified) ||
      (!iterations_specified && !threshold_specified))
    {
      QccErrorAddMessage("%s: Either -I or -t must be specified (but not both)\n",
                         argv[0]);
      QccErrorExit();
    }
  
  if (Lambda < 0.0)
    {
      QccErrorAddMessage("%s: Lambda must be nonnegative",
                         argv[0]);
      QccErrorExit();
    }
  
  if (QccDatasetReadWholefile(&TrainingData))
    {
      QccErrorAddMessage("%s: Error calling QccDatasetReadWholefile()",
                         argv[0]);
      QccErrorExit();
    }
  
  if (initialcodebook_specified)
    {
      if (QccVQCodebookRead(&Codebook))
        {
          QccErrorAddMessage("%s: Error reading initial codebook file %s",
                             argv[0], Codebook.filename);
          QccErrorExit();
        }
      if (Codebook.codeword_dimension !=
          TrainingData.vector_dimension)
        {
          QccErrorAddMessage("%s: Vector dimensions of %s and %s must be the same\n",
                             argv[0], Codebook.filename, TrainingData.filename);
          QccErrorExit();
        }
    }
  else
    {
      Codebook.codeword_dimension = TrainingData.vector_dimension;
      Codebook.num_codewords = CodebookSize;
      QccVQCodebookSetIndexLength(&Codebook);
      if (QccVQCodebookCreateRandomCodebook(&Codebook,
                                            TrainingData.max_val,
                                            TrainingData.min_val))
        {
          QccErrorAddMessage("%s: Error making random initial codebook",
                             argv[0]);
          QccErrorExit();
        }
    }
  
  if (iterations_specified)
    {
      if (QccVQEntropyConstrainedVQTraining(&TrainingData,
                                            &Codebook,
                                            (double)Lambda,
                                            Iterations, (double)0.0,
                                            NULL))
        {
          QccErrorAddMessage("%s: Error calling QccVQEntropyConstrainedVQTraining()",
                             argv[0]);
          QccErrorExit();
        }
    }
  else
    {
      if (QccVQEntropyConstrainedVQTraining(&TrainingData,
                                            &Codebook,
                                            (double)Lambda,
                                            (int)0, (double)Threshold,
                                            NULL))
        {
          QccErrorAddMessage("%s: Error calling QccVQEntropyConstrainedVQTraining()",
                             argv[0]);
          QccErrorExit();
        }
    }
  
  QccStringCopy(Codebook.filename, FinalCodebookFilename);
  
  if (QccVQCodebookWrite(&Codebook))
    {
      QccErrorAddMessage("%s: Error writing codebook %s",
                         argv[0], Codebook.filename);
      QccErrorExit();
    }

  QccDatasetFree(&TrainingData);
  QccVQCodebookFree(&Codebook);

  QccExit;
}