Marcio T
4 years ago
committed by
GitHub
GitHub
13 changed files with 364 additions and 18 deletions
@ -0,0 +1,268 @@ |
|||
/***********************
|
|||
* bed_mesh_screen.cpp * |
|||
***********************/ |
|||
|
|||
/****************************************************************************
|
|||
* Written By Marcio Teixeira 2020 * |
|||
* * |
|||
* This program is free software: you can redistribute it and/or modify * |
|||
* it under the terms of the GNU General Public License as published by * |
|||
* the Free Software Foundation, either version 3 of the License, or * |
|||
* (at your option) any later version. * |
|||
* * |
|||
* This program is distributed in the hope that it will be useful, * |
|||
* but WITHOUT ANY WARRANTY; without even the implied warranty of * |
|||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * |
|||
* GNU General Public License for more details. * |
|||
* * |
|||
* To view a copy of the GNU General Public License, go to the following * |
|||
* location: <http://www.gnu.org/licenses/>. *
|
|||
****************************************************************************/ |
|||
|
|||
#include "../config.h" |
|||
|
|||
#if ENABLED(TOUCH_UI_FTDI_EVE) && HAS_MESH |
|||
|
|||
#include "screens.h" |
|||
#include "screen_data.h" |
|||
|
|||
using namespace FTDI; |
|||
using namespace Theme; |
|||
using namespace ExtUI; |
|||
|
|||
#ifdef TOUCH_UI_PORTRAIT |
|||
#define GRID_COLS 2 |
|||
#define GRID_ROWS 10 |
|||
|
|||
#define MESH_POS BTN_POS(1, 2), BTN_SIZE(2,5) |
|||
#define Z_LABEL_POS BTN_POS(1, 8), BTN_SIZE(1,1) |
|||
#define Z_VALUE_POS BTN_POS(2, 8), BTN_SIZE(1,1) |
|||
#define WAIT_POS BTN_POS(1, 8), BTN_SIZE(2,1) |
|||
#define BACK_POS BTN_POS(1,10), BTN_SIZE(2,1) |
|||
#else |
|||
#define GRID_COLS 5 |
|||
#define GRID_ROWS 5 |
|||
|
|||
#define MESH_POS BTN_POS(2,1), BTN_SIZE(4,5) |
|||
#define Z_LABEL_POS BTN_POS(1,3), BTN_SIZE(1,1) |
|||
#define Z_VALUE_POS BTN_POS(1,4), BTN_SIZE(2,1) |
|||
#define WAIT_POS BTN_POS(1,3), BTN_SIZE(2,2) |
|||
#define BACK_POS BTN_POS(1,5), BTN_SIZE(2,1) |
|||
#endif |
|||
|
|||
void BedMeshScreen::drawMesh(int16_t x, int16_t y, int16_t w, int16_t h, ExtUI::bed_mesh_t data, uint8_t opts) { |
|||
CommandProcessor cmd; |
|||
|
|||
#define TRANSFORM_2(X,Y,Z) (X), (Y) // No transform
|
|||
#define TRANSFORM_1(X,Y,Z) TRANSFORM_2((X) + (Y) * slant, (Y) - (Z), 0) // Perspective
|
|||
#define TRANSFORM(X,Y,Z) TRANSFORM_1(float(X)/(cols-1) - 0.5, float(Y)/(rows-1) - 0.5, (Z)) // Normalize
|
|||
|
|||
constexpr uint8_t rows = GRID_MAX_POINTS_Y; |
|||
constexpr uint8_t cols = GRID_MAX_POINTS_X; |
|||
const float slant = 0.5; |
|||
const float bounds_min[] = {TRANSFORM(0 ,0 ,0)}; |
|||
const float bounds_max[] = {TRANSFORM(cols,rows,0)}; |
|||
const float scale_x = float(w)/(bounds_max[0] - bounds_min[0]); |
|||
const float scale_y = float(h)/(bounds_max[1] - bounds_min[1]); |
|||
const float center_x = x + w/2; |
|||
const float center_y = y + h/2; |
|||
|
|||
float val_mean = 0; |
|||
float val_max = -INFINITY; |
|||
float val_min = INFINITY; |
|||
uint8_t val_cnt = 0; |
|||
|
|||
if (opts & USE_AUTOSCALE) { |
|||
// Compute the mean
|
|||
for (uint8_t y = 0; y < rows; y++) { |
|||
for (uint8_t x = 0; x < cols; x++) { |
|||
const float val = data[x][y]; |
|||
if (!isnan(val)) { |
|||
val_mean += val; |
|||
val_max = max(val_max, val); |
|||
val_min = min(val_min, val); |
|||
val_cnt++; |
|||
} |
|||
} |
|||
} |
|||
if (val_cnt) { |
|||
val_mean /= val_cnt; |
|||
val_min -= val_mean; |
|||
val_max -= val_mean; |
|||
} else { |
|||
val_mean = 0; |
|||
val_min = 0; |
|||
val_max = 0; |
|||
} |
|||
} |
|||
|
|||
const float scale_z = ((val_max == val_min) ? 1 : 1/(val_max - val_min)) * 0.1; |
|||
|
|||
#undef TRANSFORM_2 |
|||
#define TRANSFORM_2(X,Y,Z) center_x + (X) * scale_x, center_y + (Y) * scale_y // Scale and position
|
|||
#define VALUE(X,Y) ((data && ISVAL(X,Y)) ? data[X][Y] : 0) |
|||
#define ISVAL(X,Y) (data ? !isnan(data[X][Y]) : true) |
|||
#define HEIGHT(X,Y) (VALUE(X,Y) * scale_z) |
|||
|
|||
uint16_t basePointSize = min(scale_x,scale_y) / max(cols,rows); |
|||
|
|||
cmd.cmd(SAVE_CONTEXT()) |
|||
.cmd(VERTEX_FORMAT(0)) |
|||
.cmd(TAG_MASK(false)) |
|||
.cmd(SAVE_CONTEXT()); |
|||
|
|||
for (uint8_t y = 0; y < rows; y++) { |
|||
for (uint8_t x = 0; x < cols; x++) { |
|||
if (ISVAL(x,y)) { |
|||
const bool hasLeftSegment = x < cols - 1 && ISVAL(x+1,y); |
|||
const bool hasRightSegment = y < rows - 1 && ISVAL(x,y+1); |
|||
if (hasLeftSegment || hasRightSegment) { |
|||
cmd.cmd(BEGIN(LINE_STRIP)); |
|||
if (hasLeftSegment) cmd.cmd(VERTEX2F(TRANSFORM(x + 1, y , HEIGHT(x + 1, y )))); |
|||
cmd.cmd( VERTEX2F(TRANSFORM(x , y , HEIGHT(x , y )))); |
|||
if (hasRightSegment) cmd.cmd(VERTEX2F(TRANSFORM(x , y + 1, HEIGHT(x , y + 1)))); |
|||
} |
|||
} |
|||
} |
|||
|
|||
if (opts & USE_POINTS) { |
|||
cmd.cmd(POINT_SIZE(basePointSize * 2)); |
|||
cmd.cmd(BEGIN(POINTS)); |
|||
for (uint8_t x = 0; x < cols; x++) { |
|||
if (ISVAL(x,y)) { |
|||
if (opts & USE_COLORS) { |
|||
const float val_dev = VALUE(x, y) - val_mean; |
|||
const uint8_t neg_byte = sq(val_dev) / sq(val_dev < 0 ? val_min : val_max) * 0xFF; |
|||
const uint8_t pos_byte = 255 - neg_byte; |
|||
cmd.cmd(COLOR_RGB(pos_byte, pos_byte, 0xFF)); |
|||
} |
|||
cmd.cmd(VERTEX2F(TRANSFORM(x, y, HEIGHT(x, y)))); |
|||
} |
|||
} |
|||
if (opts & USE_COLORS) { |
|||
cmd.cmd(RESTORE_CONTEXT()) |
|||
.cmd(SAVE_CONTEXT()); |
|||
} |
|||
} |
|||
} |
|||
cmd.cmd(RESTORE_CONTEXT()) |
|||
.cmd(TAG_MASK(true)); |
|||
|
|||
if (opts & USE_TAGS) { |
|||
cmd.cmd(COLOR_MASK(false, false, false, false)) |
|||
.cmd(POINT_SIZE(basePointSize * 10)) |
|||
.cmd(BEGIN(POINTS)); |
|||
for (uint8_t y = 0; y < rows; y++) { |
|||
for (uint8_t x = 0; x < cols; x++) { |
|||
const uint8_t tag = pointToTag(x, y); |
|||
cmd.tag(tag).cmd(VERTEX2F(TRANSFORM(x, y, HEIGHT(x, y)))); |
|||
} |
|||
} |
|||
cmd.cmd(COLOR_MASK(true, true, true, true)); |
|||
} |
|||
|
|||
if (opts & USE_HIGHLIGHT) { |
|||
const uint8_t tag = screen_data.BedMeshScreen.highlightedTag; |
|||
uint8_t x, y; |
|||
tagToPoint(tag, x, y); |
|||
cmd.cmd(COLOR_A(128)) |
|||
.cmd(POINT_SIZE(basePointSize * 6)) |
|||
.cmd(BEGIN(POINTS)) |
|||
.tag(tag).cmd(VERTEX2F(TRANSFORM(x, y, HEIGHT(x, y)))); |
|||
} |
|||
cmd.cmd(END()); |
|||
cmd.cmd(RESTORE_CONTEXT()); |
|||
} |
|||
|
|||
uint8_t BedMeshScreen::pointToTag(uint8_t x, uint8_t y) { |
|||
return y * (GRID_MAX_POINTS_X) + x + 10; |
|||
} |
|||
|
|||
void BedMeshScreen::tagToPoint(uint8_t tag, uint8_t &x, uint8_t &y) { |
|||
x = (tag - 10) % (GRID_MAX_POINTS_X); |
|||
y = (tag - 10) / (GRID_MAX_POINTS_X); |
|||
} |
|||
|
|||
void BedMeshScreen::onEntry() { |
|||
screen_data.BedMeshScreen.highlightedTag = 0; |
|||
screen_data.BedMeshScreen.count = 0; |
|||
BaseScreen::onEntry(); |
|||
} |
|||
|
|||
float BedMeshScreen::getHightlightedValue() { |
|||
if (screen_data.BedMeshScreen.highlightedTag) { |
|||
xy_uint8_t pt; |
|||
tagToPoint(screen_data.BedMeshScreen.highlightedTag, pt.x, pt.y); |
|||
return ExtUI::getMeshPoint(pt); |
|||
} |
|||
return NAN; |
|||
} |
|||
|
|||
void BedMeshScreen::drawHighlightedPointValue() { |
|||
char str[16]; |
|||
const float val = getHightlightedValue(); |
|||
const bool isGood = !isnan(val); |
|||
if (isGood) |
|||
dtostrf(val, 5, 3, str); |
|||
else |
|||
strcpy_P(str, PSTR("-")); |
|||
|
|||
CommandProcessor cmd; |
|||
cmd.font(Theme::font_medium) |
|||
.text(Z_LABEL_POS, GET_TEXT_F(MSG_MESH_EDIT_Z)) |
|||
.text(Z_VALUE_POS, str) |
|||
.colors(action_btn) |
|||
.tag(1).button( BACK_POS, GET_TEXT_F(MSG_BACK)) |
|||
.tag(0); |
|||
} |
|||
|
|||
void BedMeshScreen::onRedraw(draw_mode_t what) { |
|||
if (what & BACKGROUND) { |
|||
CommandProcessor cmd; |
|||
cmd.cmd(CLEAR_COLOR_RGB(bg_color)) |
|||
.cmd(CLEAR(true,true,true)); |
|||
|
|||
// Draw the shadow and tags
|
|||
cmd.cmd(COLOR_RGB(0x444444)); |
|||
BedMeshScreen::drawMesh(MESH_POS, nullptr, USE_POINTS | USE_TAGS); |
|||
cmd.cmd(COLOR_RGB(bg_text_enabled)); |
|||
} |
|||
|
|||
if (what & FOREGROUND) { |
|||
const bool levelingFinished = screen_data.BedMeshScreen.count >= GRID_MAX_POINTS; |
|||
if (levelingFinished) drawHighlightedPointValue(); |
|||
|
|||
BedMeshScreen::drawMesh(MESH_POS, ExtUI::getMeshArray(), |
|||
USE_POINTS | USE_HIGHLIGHT | USE_AUTOSCALE | (levelingFinished ? USE_COLORS : 0)); |
|||
} |
|||
} |
|||
|
|||
bool BedMeshScreen::onTouchStart(uint8_t tag) { |
|||
screen_data.BedMeshScreen.highlightedTag = tag; |
|||
return true; |
|||
} |
|||
|
|||
bool BedMeshScreen::onTouchEnd(uint8_t tag) { |
|||
switch(tag) { |
|||
case 1: |
|||
GOTO_PREVIOUS(); |
|||
return true; |
|||
default: |
|||
return false; |
|||
} |
|||
} |
|||
|
|||
void BedMeshScreen::onMeshUpdate(const int8_t, const int8_t, const float) { |
|||
if (AT_SCREEN(BedMeshScreen)) |
|||
onRefresh(); |
|||
} |
|||
|
|||
void BedMeshScreen::onMeshUpdate(const int8_t x, const int8_t y, const ExtUI::probe_state_t state) { |
|||
if (state == ExtUI::PROBE_FINISH) { |
|||
screen_data.BedMeshScreen.highlightedTag = pointToTag(x, y); |
|||
screen_data.BedMeshScreen.count++; |
|||
} |
|||
BedMeshScreen::onMeshUpdate(x, y, 0); |
|||
} |
|||
|
|||
#endif // TOUCH_UI_FTDI_EVE
|
|||
Loading…
Reference in new issue