D:/work/code_by/CbmrEngine/example/cbmrmain.cpp

查看本檔案說明文件.

#include <stdio.h>
#include <string.h>
#include "InputStream.h"
#include "WaveFile.h"
#include "pitchlib.h"
#include "SongDb.h"

int main(int argc, char *argv[])
{
  TFileInputStream *IndexStream, *DbStream;
  TCipherInputStream *CDbStream;
  TSongDb SongDb;
  TWaveFile Wave;
  float *fpitch;
  int *pitch, pitchSize;
  int i, from, to;

  if (argc < 2)
  {
    printf("Usage: cbmr <wav file> [head|any]\n");
    return 1;
  }

  printf("> Loading wav file...\n");
  if (!Wave.Load(argv[1]))
  {
    printf("Load wav file %s failed.\n", argv[1]);
    return 3;
  }

  printf("> Loading database...\n");
  IndexStream = new TFileInputStream();
  IndexStream->Open("demo.txt");
//  IndexStream->Open("20020618.txt");

  DbStream = new TFileInputStream();
  DbStream->Open("demo.db");
//  DbStream->Open("20020618.db");
  if (!SongDb.Read(IndexStream, DbStream, false))
  {
    printf("Load database failed.\n");
    return 2;
  }
  delete DbStream;
  delete IndexStream;

  SongDb.SetOption(COMPARE_FROM_HEAD);
  if (argc > 2)
    if (strcmp(argv[2], "any") == 0)
      SongDb.SetOption(COMPARE_FROM_ANYWHERE);

  Wave.SetFormat(11025, 8);

  printf("> Feature extracting...\n");
  fpitch = NULL;
  pitchSize = waveToPitch(Wave.wave, Wave.waveSize, 11025, 16, fpitch, NULL);
  limitPitch(fpitch, pitchSize, 34.0f, 80.0f);
  smoothPitch(fpitch, pitchSize, 7);
  trimPitch(fpitch, pitchSize, from, to);
  pitchSize = to - from + 1;

  pitch = new int[pitchSize];
  for (i = 0; i < pitchSize; i++)
    pitch[i] = fpitch[from + i] * 10;
  delete[] fpitch;

  printf("> Searching...\n");
  SongDb.Compare(pitch, pitchSize);
  delete[] pitch;

  for (i = 0; i < 10; i++)
    printf("%s %.2f\n", SongDb.GetTopSong(i)->Fields->Values("songName"), SongDb.GetTopSong(i)->score);

  return 0;
}

---