test_audiophash_main.cpp

来自「pHash is an implementation of various pe」· C++ 代码 · 共 103 行

#include <cstdio>
#include <getopt.h>
#include "audiophash.h"

int main(int argc, char** argv){

   struct option long_option[] = 
   {
       {"help", 0, NULL, 'h'},
       {"file1", 1, 0, 'f'},
       {"file2", 1, 0, 'g'},
       {"threshold", 1, 0, 't'},
       {"block_size",1, NULL, 'b'},
       {NULL, 0, NULL, 0},
   };
   char *file1, *file2;
   float threshold = 0.30;
   int block_size = 256;
   int morehelp= 0;
   while(1){
       int c;
       if ((c = getopt_long(argc, argv, "f:g:hb:t:",long_option,NULL)) < 0)
	   break;
       switch (c){
       case 'h' :
	   morehelp++;
	   break;
       case 'f':
	   file1 = optarg;
	   break;
       case 'g':
	   file2 = optarg;
	   break;
       case 't':
	   threshold = atof(optarg);
	   break;
       case 'b':
	   block_size = atoi(optarg);
       }
   }
	   
   printf("file1: %s\n",file1);
   printf("file2: %s\n",file2);
   printf("threshold: %f\n",threshold);
   printf("blocksize: %d\n",block_size);


    int sr = 8000;
    int channels = 1;
 
    int N = 0;
    float *buf = ph_readaudio(file1,sr,channels, N); 

    if (!buf){
	fprintf(stderr," cannot read file %s, no such file\n", file1);
	return -1;
    }
    int N2 = 0;
    float *buf2 = ph_readaudio(file2,sr,channels, N2);

    if (!buf2){
	fprintf(stderr," cannot read file %s, no such file", file2);
	return -1;
    }
 
    int nb_frames = 0;
    uint32_t *hash  = ph_audiohash(buf,N,sr,nb_frames);

    if (!hash){
	fprintf(stderr,"unable to calculate hash for %s\n",file1);
	return -1;
    }

    int nb_frames2 = 0;
    uint32_t *hash2 = ph_audiohash(buf2,N2,sr,nb_frames2);

    if (!hash2){
	fprintf(stderr,"unable to calculate hash for %s\n",file2);
	return -1;
    }

    int Nc = 0;
    double maxC = 0;
    double *C = ph_audio_distance_ber(hash, nb_frames, hash2, nb_frames2, threshold, block_size, Nc); 

    printf("Nc = %d, ",Nc);

    for (int i = 0;i < Nc; i++){
        if (C[i] > maxC)
	    maxC = C[i];
    }
    printf("cs = %f\n",maxC);

  
    free(buf);
    free(buf2);
    free(hash);
    free(hash2);
    free(C);

    return 0;
}