summaryrefslogtreecommitdiff
path: root/imagealgos.h
blob: 12dc8cf1fcdb4a31d2d0699ace25c29c0f5ff5b9 (plain)
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
#pragma once

#include <array>
#include <cstddef>
#include <cstdio>

#include <libcamera/pixel_format.h>

#include <QLineF>
#include <QList>

constexpr size_t img_width = 1280;
constexpr size_t img_height = 800;
constexpr uint32_t patternSize = 16;

struct Counters
{
    uint32_t timestampUs { 0 };
    uint32_t measurementCounter { 0 };
    int32_t encoderPosition { 0 };
};

struct Image
{
    int width;
    int height;
    uint16_t data[img_height][img_width];
    uint16_t rotated_cw[img_width][img_height];
    size_t dataSize;
    unsigned int stride;
    libcamera::PixelFormat pixelFormat;
    // float pixels[img_width];
    Counters counters {};
    // unsigned int measurementCounter;
    // uint32_t timestampUs;
    // int32_t encoderPosition;
};

struct Pixels
{
    Counters counters {};
    std::array<float, img_width> pixels { 0.f };

    Pixels& operator+=(const Pixels& other);
    Pixels& operator/=(const float divider);
};


size_t pgm_save(Image *img, FILE *outfile, bool really_save = false);

void unpack_10bit(uint8_t const *src, Image const &image, uint16_t *dest);
void unpack_16bit(uint8_t const *src, Image const &image, uint16_t *dest);

template <typename T>
T median3(const T& a, const T& b, const T& c) {
    using namespace std;
    return max(min(a,b), min(max(a,b),c));
}

void rotate(Image & image);
Pixels process_columns(Image & image);

QList<QLineF> pixelsToLines(const Pixels& rawProfile);
QList<QLineF> pointsToLines(const QList<QPointF>& points);