1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
// Test whether min[] and extent[] of an ImageParam are correctly passed into
// the filter.
#include "Halide.h"
#include <stdio.h>
using namespace Halide;
int main(int argc, char **argv) {
Var x("x");
Func f("f");
ImageParam in(Int(32), 1, "in");
Expr factor("factor");
// Multiply by -10 or +10 for coordinates that fall outside the input
// image.
factor = select(x < in.left(), -10,
select(x > in.right(), 10, 1));
f(x) = factor * x;
// Create input and output buffers. The input pixels are never accessed,
// but we initialize them anyway.
Buffer<int> input(5);
Buffer<int> out(10);
input.fill(0);
out.fill(0);
// Change coordinate origin of input and output buffer so that they are
// aligned as follows:
// input |------|
// out |-----------------|
const int INOFF = 4;
const int OUTOFF = 1;
input.set_min(INOFF);
out.set_min(OUTOFF);
in.set(input);
f.realize(out);
// Check correctness of result
int expected[] = {-10, -20, -30, 4, 5, 6, 7, 8, 90, 100};
for (int i = 0; i < out.width(); i++) {
if (out(i + OUTOFF) != expected[i]) {
printf("Unexpected output: %d != %d\n", out(i + OUTOFF), expected[i]);
return 1;
}
}
printf("Success!\n");
return 0;
}
|
