main.cu

#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include "includes/bmp_util.h"
#include "includes/kernel.cuh"

int main(int argc, char *argv[]) {
  // Just an example here - you are free to modify them
  int I_width, I_height, T_width, T_height;
  float *I, *T;
  int x1, y1, x2, y2;

  // set the file location of I, T, and Output

  if (argc != 4) {
    printf("Usage: template-matching original.bmp template.bmp out.bmp\n");
    exit(0);
  }

  I = ReadBMP(argv[1], &I_width, &I_height);
  T = ReadBMP(argv[2], &T_width, &T_height);

  if (I == 0 || T == 0) {
    exit(1);
  }

  if (I_width < T_width || I_height < T_height) {
    fprintf(stderr, "Error: The template is larger than the picture\n");
    exit(EXIT_FAILURE);
  }

  int x, y;

  GetMatch(I, T, I_width, I_height, T_width, T_height, &x, &y);
  x1 = x;
  x2 = x + T_width - 1;
  y1 = y;
  y2 = y + T_height - 1;

  MarkAndSave(argv[1], x1, y1, x2, y2, argv[3]);

  printf("Result is put in: %s\n", argv[3]);

  free(I);
  free(T);
  return 0;
}