#include "Halide.h"

using namespace Halide;

static int sz = 64;
static int numPrints = 17;

int main(int argc, char** argv) {
    Func brighter("brighter");
    Var x("x"), y("y");

    Buffer<uint8_t> input(sz, sz);
    for (int j = 0; j < numPrints; j++) {
        for (int i = 0; i < numPrints; i++) {
            input(i, j) = i + j;
            std::cout << (int)input(i, j) << " ";
        }
        std::cout << std::endl;
    }
    std::cout << std::endl;

    brighter(x, y) = input(x, y) * 2;

    brighter.bound(x, 0, sz).bound(y, 0, sz);

    Target target = get_host_target();
    std::cout << target << std::endl;

    brighter.print_loop_nest();

    Buffer<uint8_t> output(sz, sz);
    try {
        brighter.realize(output, target);
    } catch(Halide::InternalError& ex) {
        std::cout << ex.what() << std::endl;
    }

    for (int j = 0; j < numPrints; j++) {
        for (int i = 0; i < numPrints; i++) {
            std::cout << (int)output(i, j) << " ";
        }
        std::cout << std::endl;
    }
    std::cout << std::endl;
}