spook.cpp

// A example to illustrate the usage of skeletonide.
// We load an in image and call the pipeline in a loop.
// TODO Check for completion flags

#include "skeletonide.h"
#include "halide_target_check.h"
#include "halide_image_io.h"
#include <stdio.h>
#include <chrono>

#define ITERS 100

using namespace Halide;
using namespace Halide::Tools;

int main(int argc, char **argv){
    Target target = find_gpu_target();
    if (!target.has_gpu_feature()) {
        return -1;
    }
    Runtime::Buffer<uint8_t> input = Tools::load_and_convert_image("test/images/tiled_horse.png");
    Runtime::Buffer<uint8_t> output(input.width(), input.height());
    auto orig_input = input;
    auto t1 = std::chrono::high_resolution_clock::now();
    for(int j=0; j<ITERS; j++){
        for(int i=0; i<29; i++){ // TODO remove this hardcoded limit.
            skel(input, output);
            input = output;
        }
        input = orig_input;
    }
    auto t2 = std::chrono::high_resolution_clock::now();
    auto dur = std::chrono::duration_cast<std::chrono::milliseconds>(t2 - t1).count();
    printf("Mean time to skeletonize: %fms\n", dur /(float)ITERS);
    Tools::convert_and_save_image(output, "spook_out.png");
    printf("Image saved\n");
    return 0;
}