diff options
Diffstat (limited to 'src/blind-matrix-orthoproject.c')
| -rw-r--r-- | src/blind-matrix-orthoproject.c | 98 |
1 files changed, 52 insertions, 46 deletions
diff --git a/src/blind-matrix-orthoproject.c b/src/blind-matrix-orthoproject.c index 4a9cb3c..5bac543 100644 --- a/src/blind-matrix-orthoproject.c +++ b/src/blind-matrix-orthoproject.c @@ -1,56 +1,13 @@ /* See LICENSE file for copyright and license details. */ +#ifndef TYPE #include "common.h" USAGE("[-c]") static int per_channel = 0; -#define PROCESS(TYPE)\ - do {\ - typedef TYPE pixel_t[4];\ - pixel_t matrix[9];\ - pixel_t buf[2];\ - TYPE x2, y2, norm2;\ - size_t i;\ - \ - for (i = 0; i < stream->n_chan; i++) {\ - matrix[0][i] = 1, matrix[1][i] = 0, matrix[2][i] = 0;\ - matrix[3][i] = 0, matrix[4][i] = 1, matrix[5][i] = 0;\ - matrix[6][i] = 0, matrix[7][i] = 0, matrix[8][i] = 1;\ - }\ - \ - while (eread_frame(stream, buf)) {\ - if (per_channel) {\ - for (i = 0; i < stream->n_chan; i++) {\ - x2 = buf[0][i] * buf[0][i];\ - y2 = buf[1][i] * buf[1][i];\ - norm2 = x2 + y2;\ - matrix[0][i] = x2 / norm2;\ - matrix[4][i] = y2 / norm2;\ - matrix[3][i] = matrix[1][i] = buf[0][i] * buf[1][i] / norm2;\ - }\ - } else {\ - buf[0][1] *= buf[0][3];\ - buf[1][1] *= buf[1][3];\ - x2 = buf[0][1] * buf[0][1];\ - y2 = buf[1][1] * buf[1][1];\ - norm2 = x2 + y2;\ - matrix[0][0] = x2 / norm2;\ - matrix[4][0] = y2 / norm2;\ - matrix[3][0] = matrix[1][0] = buf[0][1] * buf[1][1] / norm2;\ - for (i = 1; i < stream->n_chan; i++) {\ - matrix[0][i] = matrix[0][0];\ - matrix[1][i] = matrix[1][0];\ - matrix[3][i] = matrix[3][0];\ - matrix[4][i] = matrix[4][0];\ - }\ - }\ - ewriteall(STDOUT_FILENO, matrix, sizeof(matrix), "<stdout>");\ - }\ - } while (0) - -static void process_lf(struct stream *stream) { PROCESS(double); } -static void process_f(struct stream *stream) { PROCESS(float); } +#define FILE "blind-matrix-orthoproject.c" +#include "define-functions.h" int main(int argc, char *argv[]) @@ -89,3 +46,52 @@ main(int argc, char *argv[]) process(&stream); return 0; } + +#else + +static void +PROCESS(struct stream *stream) +{ + typedef TYPE pixel_t[4]; + pixel_t matrix[9]; + pixel_t buf[2]; + TYPE x2, y2, norm2; + size_t i; + + for (i = 0; i < stream->n_chan; i++) { + matrix[0][i] = 1, matrix[1][i] = 0, matrix[2][i] = 0; + matrix[3][i] = 0, matrix[4][i] = 1, matrix[5][i] = 0; + matrix[6][i] = 0, matrix[7][i] = 0, matrix[8][i] = 1; + } + + while (eread_frame(stream, buf)) { + if (per_channel) { + for (i = 0; i < stream->n_chan; i++) { + x2 = buf[0][i] * buf[0][i]; + y2 = buf[1][i] * buf[1][i]; + norm2 = x2 + y2; + matrix[0][i] = x2 / norm2; + matrix[4][i] = y2 / norm2; + matrix[3][i] = matrix[1][i] = buf[0][i] * buf[1][i] / norm2; + } + } else { + buf[0][1] *= buf[0][3]; + buf[1][1] *= buf[1][3]; + x2 = buf[0][1] * buf[0][1]; + y2 = buf[1][1] * buf[1][1]; + norm2 = x2 + y2; + matrix[0][0] = x2 / norm2; + matrix[4][0] = y2 / norm2; + matrix[3][0] = matrix[1][0] = buf[0][1] * buf[1][1] / norm2; + for (i = 1; i < stream->n_chan; i++) { + matrix[0][i] = matrix[0][0]; + matrix[1][i] = matrix[1][0]; + matrix[3][i] = matrix[3][0]; + matrix[4][i] = matrix[4][0]; + } + } + ewriteall(STDOUT_FILENO, matrix, sizeof(matrix), "<stdout>"); + } +} + +#endif |
