-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathcore.c
80 lines (66 loc) · 1.78 KB
/
core.c
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <assert.h>
#include <math.h>
#include <inttypes.h>
#include <unistd.h>
typedef struct {
uint8_t* ptr;
int64_t width;
int64_t height;
} image;
// this function will be provided at runtime
int64_t function(int64_t x, int64_t y, image inputs[], size_t num_inputs);
// only for storing values in output
uint8_t* at(uint8_t* image, int64_t width, int64_t x, int64_t y) {
return &image[y*width + x];
}
// core functions
int64_t core_isqrt(int64_t input) {
return (int64_t) sqrt((double)input);
}
// declare function, then define it inline to encourage inlining
int64_t core_input_at(int64_t x, int64_t y,
image inputs[], size_t num_inputs, size_t input);
inline int64_t core_input_at(int64_t x, int64_t y,
image inputs[], size_t num_inputs, size_t input)
{
(void)num_inputs;
if (x >= inputs[input].width || x < 0) {
return 0;
}
if (y >= inputs[input].height || y < 0) {
return 0;
}
image i = inputs[input];
return (int64_t) i.ptr[y * i.width + x];
}
// "main"
void jitfunction(
int64_t width,
int64_t height,
uint8_t* output_ptr,
uint8_t** input_ptrs,
uint64_t num_inputs)
{
image output;
output.ptr = output_ptr;
output.width = width;
output.height = height;
image inputs[num_inputs];
for (size_t i = 0; i < num_inputs; i++) {
inputs[i].ptr = input_ptrs[i];
inputs[i].width = width;
inputs[i].height = height;
}
for (int x = 0; x < output.width; x++) {
for (int y = 0; y < output.height; y++) {
int64_t res = function(x, y, inputs, num_inputs);
// clamp the pixel
// if (res >= 255) res = 255;
// if (res <= 0) res = 0;
*at(output.ptr, output.width, x, y) = (uint8_t) res;
}
}
}