From 79374c8424deb05bd5f4a38480ade696b6d6a27e Mon Sep 17 00:00:00 2001 From: LooseEthics <36419655+LooseEthics@users.noreply.github.com> Date: Thu, 15 Sep 2022 18:28:46 +0200 Subject: [PATCH] opencv test --- Profilometer.cpp | 111 +++++++++++++++++++++++++++++--------- opencv test.cpp | 137 +++++++++++++++++++++++++++++++++++++++++++++++ pixel_delta.py | 22 ++++++-- 3 files changed, 242 insertions(+), 28 deletions(-) create mode 100644 opencv test.cpp diff --git a/Profilometer.cpp b/Profilometer.cpp index 4a93574..4f1711a 100644 --- a/Profilometer.cpp +++ b/Profilometer.cpp @@ -41,11 +41,17 @@ public: // 0 < sigma < pi/2 double dist; // distance between camera and laser - double phi, theta; - // coordinate angles - void fastinit(int h, int w, double a, double b, double s, double d, double p, double t) { - //TODO: overload this + double tan_alpha; + double tan_beta; + double* tan_phi_arr; + double* tan_theta_arr; + double sin_sigma; + double cos_sigma; + double p_h; + + + void fastinit(int h, int w, double a, double b, double s, double d) { // class variable initializer height = h; width = w; @@ -53,8 +59,26 @@ public: beta = b; sigma = s; dist = d; - phi = p; - theta = t; + trig_precalc(); + } + + void trig_precalc() { + tan_alpha = tan(alpha); + tan_beta = tan(beta); + tan_phi_arr = (double*)malloc(sizeof(double) * height); + for (int i = 0; i < height; i++) { + //tan_phi_arr[i] = tan_alpha / height * (2 * i + 1 - height); + tan_phi_arr[i] = tan_alpha / height * (2 * i - height); + } + tan_theta_arr = (double*)malloc(sizeof(double) * width); + for (int i = 0; i < width; i++) { + //tan_theta_arr[i] = -tan_beta / width * (2 * i - 1 - width); + tan_theta_arr[i] = -tan_beta / width * (2 * i - width); + } + sin_sigma = sin(sigma); + cos_sigma = cos(sigma); + p_h = 2 * dist / height * tan_alpha / sin_sigma; + } void print_conf() { @@ -64,15 +88,15 @@ public: std::cout << "Beta: " << beta << "\n"; std::cout << "Sigma: " << sigma << "\n"; std::cout << "Dist: " << dist << "\n"; - std::cout << "Phi: " << phi << "\n"; - std::cout << "Theta: " << theta << "\n"; + //std::cout << "Phi: " << phi << "\n"; + //std::cout << "Theta: " << theta << "\n"; } - + double get_phi(int y) { // phi is angle below camera center line of given y coordinate - above center is negative // returns phi at pixel center, removing the +1 shifts this to low edge // alpha >= phi >= -alpha - phi = atan(tan(alpha)/height * (2 * y + 1 - height)); + double phi = atan(tan_phi_arr[y]); return phi; } @@ -81,40 +105,57 @@ public: // returns theta at pixel center, removing the -1 shifts this to low edge // beta >= theta >= -beta // if viewed from above, positive theta means pixel vector is to the left of center, i.e. x < width/2 - theta = atan(tan(beta) / width * (width - 2 * x - 1)); + double theta = atan(tan_theta_arr[x]); return theta; } - double pixel_delta_approx() { + double sin_sp(int y) { + // sin( sigma + phi) + return sin_sigma + cos_sigma * tan_phi_arr[y] / sqrt(1 + tan_phi_arr[y]); + } + + double cos_sp(int y) { + // cos( sigma + phi) + return cos_sigma + sin_sigma * tan_phi_arr[y] / sqrt(1 + tan_phi_arr[y]); + } + + double pixel_delta_approx(int y) { // returns approximation of change in elevation from y to y+1 // assumes that epsilon=0 - double delta = 2 * dist * tan(alpha) * cos(phi) / (height * sin(sigma + phi) * sin(sigma + phi)); + //double delta = 2 * dist * tan(alpha) * cos(phi) / (height * sin(sigma + phi) * sin(sigma + phi)); + double delta = p_h * sin_sigma / (sin_sigma + cos_sigma * tan_phi_arr[y]); return delta; } - double pixel_delta_exact() { + double pixel_delta_exact(int y) { // returns exact change in elevation from y to y+1 - double tan_eps = 2 * tan(alpha) * cos(phi) * cos(phi) / (height + 2 * tan(alpha) * sin(phi) * cos(phi)); - std::cout << "Tan(eps) at phi=" << phi << ": " << tan_eps << "\n"; - double delta = 2 * dist * tan(alpha) * cos(phi) * (cos(phi) - sin(phi) * tan_eps) / (height * sin(sigma + phi) * (sin(sigma + phi) + cos(sigma + phi) * tan_eps)); + //double tan_eps = 2 * tan(alpha) * cos(phi) * cos(phi) / (height + 2 * tan(alpha) * sin(phi) * cos(phi)); + double tan_phi = tan_phi_arr[y]; + double tan_eps = 2 * tan_alpha / (height * (1 + tan_phi * tan_phi) + 2 * tan_alpha * tan_phi); + std::cout << "Tan(eps) at phi=" << get_phi(y) << ": " << tan_eps << "\n"; + //double delta = 2 * dist * tan(alpha) * cos(phi) * (cos(phi) - sin(phi) * tan_eps) / (height * sin(sigma + phi) * (sin(sigma + phi) + cos(sigma + phi) * tan_eps)); + //double delta = dist * tan_eps / (sin_sp(y) * (sin_sp(y) + tan_eps * cos_sp(y))); + double delta = 2 * dist * tan_alpha * (1 - tan_phi_arr[y] * tan_eps) / (height * sin_sp(y) * (sin_sp(y) + cos_sp(y) * tan_eps)); return delta; } - - double get_elevation() { + double get_elevation(int y) { // elevation above horizontal camera plane // singularity if phi = -sigma // flat projection surface -> x coordinate irrelevant - double e = dist / tan(phi + sigma); + //double e = dist / tan(phi + sigma); + double e = dist * cos_sp(y) / sin_sp(y); return e; } - - double get_horizontal_dist() { + + double get_horizontal_dist(int x, int y) { // get horizontal distance from center line, same sign convention as theta // generally depends on both x and y - double f = dist * tan(theta) / sin(sigma + phi); + //double f = dist * tan(theta) / sin(sigma + phi); + double f = dist * tan_theta_arr[x] / sin_sp(y); return f; } + /* void approx_test_phi(double new_phi) { // print exact and approximate delta and the error at given phi // changes phi @@ -126,7 +167,18 @@ public: std::cout << "Error absolute: " << (de - da) << "\n"; std::cout << "Error relative: " << (de - da)/de << "\n"; } + */ + void approx_test(int y) { + double da = pixel_delta_approx(y); + std::cout << "Delta approx.: " << da << "\n"; + double de = pixel_delta_exact(y); + std::cout << "Delta exact: " << de << "\n"; + std::cout << "Error absolute: " << (de - da) << "\n"; + std::cout << "Error relative: " << (de - da) / de << "\n"; + } + + /* void edge_test() { // run approx_test_phi at bottom edge, phi=0, and top edge // preserves phi @@ -139,14 +191,25 @@ public: approx_test_phi(-1*alpha); phi = old_phi; } + */ + void edge_test() { + // run approx_test_phi at y = {0, height/2, height-1} + std::cout << "\ny = h - 1\n"; + approx_test(height - 1); + std::cout << "\ny = h/2\n"; + approx_test(height/2); + std::cout << "\ny = 0\n"; + approx_test(0); + } }; int main() { PMeter PM; - PM.fastinit(1000, 1000, M_PI / 6, M_PI / 6, M_PI / 3, 5.0, 0.0, 0.0); + // fastinit(int h, int w, double a, double b, double s, double d) + PM.fastinit(3120, 4208, M_PI / 180 * 21.5, M_PI / 180 * 43, M_PI / 180 * 51, 0.05); PM.print_conf(); diff --git a/opencv test.cpp b/opencv test.cpp new file mode 100644 index 0000000..95a7b71 --- /dev/null +++ b/opencv test.cpp @@ -0,0 +1,137 @@ +// opencv test.cpp : This file contains the 'main' function. Program execution begins and ends there. +// + +#include +#include +#include +#include + +using namespace cv; +using namespace std; + +int main_img() +{ + string path = "Resources/test.png"; + Mat img = imread(path); + imshow("Image", img); + waitKey(0); + return 0; +} + +int main_vid() { + string path = "Resources/vid.mp4"; + VideoCapture cap(path); + Mat img; + + while (true){ + cap.read(img); + imshow("Image", img); + waitKey(10); + } + return 0; +} + +int main_cam() { + VideoCapture cap(0); + Mat img; + + while (true) { + cap.read(img); + imshow("Image", img); + waitKey(10); + } + return 0; +} + +int main() { + string path = "Resources/PM_test_1.png"; + Mat img = imread(path); + + + int optimize = 0; + int tag = 1; + // Brightest array setup + int *maxrow_arr = (int *)malloc(sizeof(int) * img.cols); + for (int i = 0; i < img.cols; i++) { + maxrow_arr[i] = -1; + } + + if (optimize) { + //float thrcols[1] = { 0.5 }; + float thrcols[3] = { 0.25, 0.5, 0.75 }; + float thrscale = 0.7; + int thrsum = 0; + int thrcnt = 0; + + // Bruteforce threshold columns + // Get brightness threshold + for (float ratio : thrcols) { + int maxrow = 0; + int maxbr = 0; + int i = (int)(ratio * img.cols); + for (int j = 0; j < img.rows; j++) { + Vec3b& px = img.at(j, i); + if (px[0] + px[1] + px[2] > maxbr) { + maxbr = px[0] + px[1] + px[2]; + maxrow = j; + } + } + maxrow_arr[i] = maxrow; + thrsum += maxbr; + thrcnt += 1; + + if (tag) { + Vec3b& px = img.at(maxrow, i); + px.val[0] = 0; + px.val[1] = 0; + px.val[2] = 255; + } + } + + // Optimization constants + int thr = (int)(thrscale * thrsum / thrcnt); + int voffset = 8; + int hoffset = 8; + + + // Left pass + for (int i = (int)(0.5*img.cols - 1); i >= 0; i--) { + int maxrow = 0; + int maxbr = 0; + //// TODO + } + + // Right pass + //// TODO + + } + else { + for (int i = 0; i < img.cols; i++) { + int maxrow = 0; + int maxbr = 0; + for (int j = 0; j < img.rows; j++) { + Vec3b& px = img.at(j, i); + if (px[0] + px[1] + px[2] > maxbr) { + maxbr = px[0] + px[1] + px[2]; + maxrow = j; + } + } + maxrow_arr[i] = maxrow; + + if (tag) { + Vec3b& px = img.at(maxrow, i); + px.val[0] = 0; + px.val[1] = 0; + px.val[2] = 255; + } + } + } + + for (int i = 0; i < img.cols; i++) { + cout << maxrow_arr[i] << ", "; + } + imshow("Image", img); + + waitKey(0); + return 0; +} diff --git a/pixel_delta.py b/pixel_delta.py index 732839b..08e2dd8 100644 --- a/pixel_delta.py +++ b/pixel_delta.py @@ -7,10 +7,13 @@ import numpy as np # phi is angle between camera center vector and target pixel vector # d is distance between laser plane and camera # h is vertical pixel count -alpha = pi/6 -sigma = pi/3 -d = 5 -h = 1000 +alpha = pi/180*21.5 +sigma = pi/180*51 +r = 0.05 +d = r*tan(sigma)*.65 +#d = r +h = 3120 +phi_r = atan(d/r)-sigma def get_phi(y, h, alpha): # get phi of pixel row y @@ -30,16 +33,27 @@ def crunch(alpha, sigma, phi, d, h): def nprint(phi_text, phi): nums = crunch(alpha, sigma, phi, d, h) print(phi_text) + print(f"Phi: {phi}") print(f"Delta approx.: {nums['da']}") print(f"Tan(eps): {nums['te']}") print(f"Delta exact: {nums['de']}") print(f"Error (absolute): {nums['ea']}") print(f"Error (relative): {nums['er']}") + print(f"Height: {d/tan(sigma+phi)}") print("") +print(f"alpha is {alpha} ({alpha/pi} pi)") +print(f"sigma is {sigma} ({sigma/pi} pi)") +print(f"d at r={r} is {d}") + nprint('Maximal phi', alpha) nprint('Zero phi', 0) +nprint('phi_r', phi_r) nprint('Minimal phi', -alpha) +nprint('y=h/2', get_phi(h/2, h, alpha)) + +#for i in range(0,10): +# nprint(f'phi = {i}/10*alpha', i/10*alpha) print(f"phi at y = 50 is {get_phi(50, h, alpha)}")