diff options
author | Stanislaw Halik <sthalik@misaki.pl> | 2017-03-30 13:35:35 +0200 |
---|---|---|
committer | Stanislaw Halik <sthalik@misaki.pl> | 2017-04-06 04:21:35 +0200 |
commit | e7158125d3c3b63371ddc7dc2d5b7d7865d91e64 (patch) | |
tree | 9ce314487879712916403b0f1dd92c5a96246aeb /pose-widget/glwidget.cpp | |
parent | 8660d0f4e141b32ce306144e95547a40926c62ed (diff) |
pose-widget: rename unit and header
Diffstat (limited to 'pose-widget/glwidget.cpp')
-rw-r--r-- | pose-widget/glwidget.cpp | 352 |
1 files changed, 0 insertions, 352 deletions
diff --git a/pose-widget/glwidget.cpp b/pose-widget/glwidget.cpp deleted file mode 100644 index beddc5e9..00000000 --- a/pose-widget/glwidget.cpp +++ /dev/null @@ -1,352 +0,0 @@ -/* Copyright (c) 2013, 2015 Stanislaw Halik <sthalik@misaki.pl> - * - * Permission to use, copy, modify, and/or distribute this software for any - * purpose with or without fee is hereby granted, provided that the above - * copyright notice and this permission notice appear in all copies. - */ - -#include "glwidget.h" -#include "compat/util.hpp" -#include "compat/timer.hpp" -#include <cmath> -#include <algorithm> -#include <QPainter> -#include <QPaintEvent> - -#include <QDebug> - -using namespace pose_widget_impl; - -pose_transform::pose_transform(QWidget* dst) : - dst(dst), - image(w, h, QImage::Format_ARGB32), - image2(w, h, QImage::Format_ARGB32), - width(w), height(h) -{ - front = QImage(QString(":/images/side1.png")); - back = QImage(QString(":/images/side6.png")); - - image.fill(Qt::transparent); - image2.fill(Qt::transparent); - - rotateBy(0, 0, 0, 0, 0, 0, QSize(w, h)); - - project_quad_texture(); - - start(); -} - -pose_transform::~pose_transform() -{ - requestInterruption(); - wait(); -} - -void pose_widget::paintEvent(QPaintEvent* event) -{ - QPainter p(this); - xform.with_image_lock([&](const QImage& image) { - p.drawImage(event->rect(), image); - }); -} - -void pose_transform::run() -{ - for (;;) - { - if (isInterruptionRequested()) - break; - - static struct cruft - { - void lock() {} - void unlock() {} - } not_a_mutex; - - const cv_status st = cvar.wait_for(not_a_mutex, std::chrono::milliseconds(2000)); - if (st == cv_status::timeout) - continue; - - project_quad_texture(); - } -} - -pose_widget::pose_widget(QWidget* parent) : QWidget(parent), xform(this) -{ -} - -pose_widget::~pose_widget() -{ -} - -void pose_widget::rotateBy(double xAngle, double yAngle, double zAngle, double x, double y, double z) -{ - xform.rotateBy(xAngle, yAngle, zAngle, x, y, z, size()); -} - -void pose_transform::rotateBy(double xAngle, double yAngle, double zAngle, double x, double y, double z, const QSize& size) -{ - using std::sin; - using std::cos; - - static constexpr double d2r = M_PI / 180; - - euler::euler_t euler(-zAngle * d2r, xAngle * d2r, -yAngle * d2r); - euler::rmat r = euler::euler_to_rmat(euler); - - lock_guard l(mtx); - - for (int i = 0; i < 3; i++) - for (int j = 0; j < 3; j++) - rotation(i, j) = num(r(i, j)); - - translation = vec3(x, y, z); - - cvar.notify_one(); -} - -class Triangle { - num dot00, dot01, dot11, invDenom; - vec2 v0, v1, origin; -public: - Triangle(const vec2& p1, const vec2& p2, const vec2& p3); - bool barycentric_coords(const vec2& px, vec2& uv, int& i) const; -}; - -inline vec3 pose_transform::normal(const vec3& p1, const vec3& p2, const vec3& p3) -{ - using std::sqrt; - - vec3 u = p2 - p1; - vec3 v = p3 - p1; - - vec3 tmp = u.cross(v); - - num i = 1/sqrt(tmp.dot(tmp)); - - return tmp * i; -} - -Triangle::Triangle(const vec2& p1, const vec2& p2, const vec2& p3) -{ - using std::fabs; - - origin = p1; - - v0 = vec2(p3 - p1); - v1 = vec2(p2 - p1); - - dot00 = v0.dot(v0); - dot01 = v0.dot(v1); - dot11 = v1.dot(v1); - - const num denom = dot00 * dot11 - dot01 * dot01; - - if (fabs(denom) < num(1e3)) - { - // for perpendicular plane, ensure u and v don't come out right - // this is done here to avoid branching below, in a hot loop - invDenom = 0; - dot00 = dot01 = dot11 = 0; - v0 = v1 = vec2(0, 0); - } - else - invDenom = 1 / denom; -} - -bool Triangle::barycentric_coords(const vec2& px, vec2& uv, int& i) const -{ - i = 0; - const vec2 v2 = px - origin; - const num dot12 = v1.dot(v2); - const num dot02 = v0.dot(v2); - num u = (dot11 * dot02 - dot01 * dot12) * invDenom; - num v = (dot00 * dot12 - dot01 * dot02) * invDenom; - if (!(u >= 0 && v >= 0)) - return false; - if (u + v > 1) - { - i = 1; - - u = 1 - u; - v = 1 - v; - } - uv = vec2(u, v); - return u >= 0 && v >= 0 && u + v <= 1; -} - -void pose_transform::project_quad_texture() -{ - num dir; - vec2 pt[4]; - const int sx = width - 1, sy = height - 1; - vec2 projected[3]; - - lock_guard l(mtx); - - { - const int sx_ = (sx - std::max(0, (sx - sy)/2)) * 5/9; - const int sy_ = (sy - std::max(0, (sy - sx)/2)) * 5/9; - - static constexpr const double c = 85/100.; - - const vec3 dst_corners[] = - { - vec3(-sx_/2. * c, -sy_/2., 0), - vec3(sx_/2. * c, -sy_/2., 0), - vec3(-sx_/2. * c, sy_/2., 0), - vec3(sx_/2. * c, sy_/2., 0.) - }; - - for (int i = 0; i < 4; i++) - pt[i] = project(dst_corners[i]) + vec2(sx/2, sy/2); - - vec3 normal1(0, 0, 1); - vec3 normal2; - { - vec3 foo[3]; - for (int i = 0; i < 3; i++) - { - foo[i] = project2(dst_corners[i]); - projected[i] = project(dst_corners[i]) + vec2(sx/2, sy/2); - } - normal2 = normal(foo[0], foo[1], foo[2]); - } - dir = normal1.dot(normal2); - } - - const QImage& tex = dir < 0 ? back : front; - const int ow = tex.width(), oh = tex.height(); - - vec2 origs[2][3] = - { - { - vec2(0, 0), - vec2(ow-1, 0), - vec2(0, oh-1) - }, - { - vec2(ow-1, oh-1), - vec2(0, oh-1) - vec2(ow-1, oh-1), - vec2(ow-1, 0) - vec2(ow-1, oh-1), - } - }; - - Triangle t(projected[0], projected[1], projected[2]); - - const int orig_pitch = tex.bytesPerLine(); - const int dest_pitch = image.bytesPerLine(); - - const unsigned char* orig = tex.bits(); - unsigned char* dest = image.bits(); - - const int orig_depth = tex.depth() / 8; - const int dest_depth = image.depth() / 8; - - /* image breakage? */ - if (orig_depth != 4) - { - qDebug() << "pose-widget: octopus must be saved as .png with 32-bit depth"; - return; - } - - const vec3 half = rotation * vec3(.5f, .5f, 0); - - for (int y = 1; y < sy; y++) - for (int x = 1; x < sx; x++) - { - vec2 pos(x, y); - vec2 uv; - int i; - if (t.barycentric_coords(pos, uv, i)) - { - const float fx = origs[i][0].x() - + uv.x() * origs[i][2].x() - + uv.y() * origs[i][1].x(); - const float fy = origs[i][0].y() - + uv.x() * origs[i][2].y() - + uv.y() * origs[i][1].y(); - - const int px_ = fx + half.x(); - const int py_ = fy + half.y(); - const int px = fx; - const int py = fy; - const float ax_ = fx - px; - const float ay_ = fy - py; - const float ax = 1 - ax_; - const float ay = 1 - ay_; - - // 0, 0 -- ax, ay - const int orig_pos = py * orig_pitch + px * orig_depth; - const unsigned char r = orig[orig_pos + 2]; - const unsigned char g = orig[orig_pos + 1]; - const unsigned char b = orig[orig_pos + 0]; - - // 1, 1 -- ax_, ay_ - const int orig_pos_ = py_ * orig_pitch + px_ * orig_depth; - const unsigned char r_ = orig[orig_pos_ + 2]; - const unsigned char g_ = orig[orig_pos_ + 1]; - const unsigned char b_ = orig[orig_pos_ + 0]; - - // 1, 0 -- ax_, ay - const int orig_pos__ = py * orig_pitch + px_ * orig_depth; - const unsigned char r__ = orig[orig_pos__ + 2]; - const unsigned char g__ = orig[orig_pos__ + 1]; - const unsigned char b__ = orig[orig_pos__ + 0]; - - // 0, 1 -- ax, ay_ - const int orig_pos___ = py_ * orig_pitch + px * orig_depth; - const unsigned char r___ = orig[orig_pos___ + 2]; - const unsigned char g___ = orig[orig_pos___ + 1]; - const unsigned char b___ = orig[orig_pos___ + 0]; - - const unsigned char a1 = orig[orig_pos + 3]; - const unsigned char a2 = orig[orig_pos_ + 3]; - const unsigned char a3 = orig[orig_pos__ + 3]; - const unsigned char a4 = orig[orig_pos___ + 3]; - - const int pos = y * dest_pitch + x * dest_depth; - - dest[pos + 2] = (r * ax + r__ * ax_) * ay + (r___ * ax + r_ * ax_) * ay_; - dest[pos + 1] = (g * ax + g__ * ax_) * ay + (g___ * ax + g_ * ax_) * ay_; - dest[pos + 0] = (b * ax + b__ * ax_) * ay + (b___ * ax + b_ * ax_) * ay_; - dest[pos + 3] = (a1 * ax + a3 * ax_) * ay + (a4 * ax + a2 * ax_) * ay_; - } - } - - { - lock_guard l2(mtx2); - image2.fill(Qt::transparent); - std::swap(image, image2); - } - - run_in_thread_async(dst->thread(), [this]() { dst->update(); }); -} - -vec2 pose_transform::project(const vec3 &point) -{ - vec3 ret = rotation * point; - num z = std::max<num>(.75f, 1 + translation.z()/-60); - num w = width, h = height; - num x = w * translation.x() / 2 / -40; - if (std::abs(x) > w/2) - x = x > 0 ? w/2 : w/-2; - num y = h * translation.y() / 2 / -40; - if (std::abs(y) > h/2) - y = y > 0 ? h/2 : h/-2; - return vec2(z * (ret.x() + x), z * (ret.y() + y)); -} - -vec3 pose_transform::project2(const vec3 &point) -{ - return rotation * point; -} - - -template<typename F> -inline void pose_transform::with_image_lock(F&& fun) -{ - lock_guard l(mtx2); - - fun(image2); -} |