Tactility/Tactility/Source/app/touchcalibration/TouchCalibration.cpp

#include <Tactility/Tactility.h>

#include <Tactility/app/touchcalibration/TouchCalibration.h>

#include <Tactility/Logger.h>
#include <Tactility/settings/TouchCalibrationSettings.h>

#include <algorithm>
#include <lvgl.h>

namespace tt::app::touchcalibration {

static const auto LOGGER = Logger("TouchCalibration");

extern const AppManifest manifest;

LaunchId start() {
    return app::start(manifest.appId);
}

class TouchCalibrationApp final : public App {

    static constexpr int32_t TARGET_MARGIN = 24;

    struct Sample {
        uint16_t x;
        uint16_t y;
    };

    Sample samples[4] = {};
    uint8_t sampleCount = 0;

    lv_obj_t* root = nullptr;
    lv_obj_t* target = nullptr;
    lv_obj_t* titleLabel = nullptr;
    lv_obj_t* hintLabel = nullptr;

    static void onPress(lv_event_t* event) {
        auto* self = static_cast<TouchCalibrationApp*>(lv_event_get_user_data(event));
        if (self != nullptr) {
            self->onPressInternal(event);
        }
    }

    static lv_point_t getTargetPoint(uint8_t index, lv_coord_t width, lv_coord_t height) {
        switch (index) {
            case 0:
                return {.x = TARGET_MARGIN, .y = TARGET_MARGIN};
            case 1:
                return {.x = width - TARGET_MARGIN, .y = TARGET_MARGIN};
            case 2:
                return {.x = width - TARGET_MARGIN, .y = height - TARGET_MARGIN};
            default:
                return {.x = TARGET_MARGIN, .y = height - TARGET_MARGIN};
        }
    }

    void updateUi() {
        if (target == nullptr || root == nullptr || titleLabel == nullptr || hintLabel == nullptr) {
            return;
        }

        const auto width = lv_obj_get_content_width(root);
        const auto height = lv_obj_get_content_height(root);

        if (sampleCount < 4) {
            const auto point = getTargetPoint(sampleCount, width, height);
            lv_obj_set_pos(target, point.x - 14, point.y - 14);
            lv_label_set_text(titleLabel, "Touchscreen Calibration");
            lv_label_set_text_fmt(hintLabel, "Tap target %u/4", static_cast<unsigned>(sampleCount + 1));
        }
    }

    void finishCalibration() {
        constexpr int32_t MIN_RANGE = 20;

        const int32_t xMin = (static_cast<int32_t>(samples[0].x) + static_cast<int32_t>(samples[3].x)) / 2;
        const int32_t xMax = (static_cast<int32_t>(samples[1].x) + static_cast<int32_t>(samples[2].x)) / 2;
        const int32_t yMin = (static_cast<int32_t>(samples[0].y) + static_cast<int32_t>(samples[1].y)) / 2;
        const int32_t yMax = (static_cast<int32_t>(samples[2].y) + static_cast<int32_t>(samples[3].y)) / 2;

        settings::touch::TouchCalibrationSettings settings = settings::touch::getDefault();
        settings.enabled = true;
        settings.xMin = xMin;
        settings.xMax = xMax;
        settings.yMin = yMin;
        settings.yMax = yMax;

        if ((xMax - xMin) < MIN_RANGE || (yMax - yMin) < MIN_RANGE || !settings::touch::isValid(settings)) {
            lv_label_set_text(titleLabel, "Calibration Failed");
            lv_label_set_text(hintLabel, "Range invalid. Tap to close.");
            lv_obj_add_flag(target, LV_OBJ_FLAG_HIDDEN);
            setResult(Result::Error);
            return;
        }

        if (!settings::touch::save(settings)) {
            lv_label_set_text(titleLabel, "Calibration Failed");
            lv_label_set_text(hintLabel, "Unable to save settings. Tap to close.");
            lv_obj_add_flag(target, LV_OBJ_FLAG_HIDDEN);
            setResult(Result::Error);
            return;
        }

        LOGGER.info("Saved calibration x=[{}, {}] y=[{}, {}]", xMin, xMax, yMin, yMax);
        lv_label_set_text(titleLabel, "Calibration Complete");
        lv_label_set_text(hintLabel, "Touch anywhere to continue.");
        lv_obj_add_flag(target, LV_OBJ_FLAG_HIDDEN);
        setResult(Result::Ok);
    }

    void onPressInternal(lv_event_t* event) {
        auto* indev = lv_event_get_indev(event);
        if (indev == nullptr) {
            return;
        }

        lv_point_t point = {0, 0};
        lv_indev_get_point(indev, &point);

        if (sampleCount < 4) {
            samples[sampleCount] = {
                .x = static_cast<uint16_t>(std::max(static_cast<lv_coord_t>(0), point.x)),
                .y = static_cast<uint16_t>(std::max(static_cast<lv_coord_t>(0), point.y)),
            };
            sampleCount++;

            if (sampleCount < 4) {
                updateUi();
            } else {
                finishCalibration();
            }
            return;
        }

        stop(manifest.appId);
    }

public:

    void onCreate(AppContext& app) override {
        (void)app;
        settings::touch::setRuntimeCalibrationEnabled(false);
        settings::touch::invalidateCache();
    }

    void onDestroy(AppContext& app) override {
        (void)app;
        settings::touch::setRuntimeCalibrationEnabled(true);
        settings::touch::invalidateCache();
    }

    void onShow(AppContext& app, lv_obj_t* parent) override {
        (void)app;

        lv_obj_set_style_bg_color(parent, lv_color_black(), LV_STATE_DEFAULT);
        lv_obj_set_style_bg_opa(parent, LV_OPA_COVER, LV_STATE_DEFAULT);
        lv_obj_set_style_border_width(parent, 0, LV_STATE_DEFAULT);
        lv_obj_set_style_radius(parent, 0, LV_STATE_DEFAULT);

        root = lv_obj_create(parent);
        lv_obj_set_size(root, LV_PCT(100), LV_PCT(100));
        lv_obj_set_style_bg_opa(root, LV_OPA_TRANSP, LV_STATE_DEFAULT);
        lv_obj_set_style_border_width(root, 0, LV_STATE_DEFAULT);
        lv_obj_set_style_pad_all(root, 0, LV_STATE_DEFAULT);

        titleLabel = lv_label_create(root);
        lv_obj_align(titleLabel, LV_ALIGN_TOP_MID, 0, 14);
        lv_obj_set_style_text_color(titleLabel, lv_color_white(), LV_STATE_DEFAULT);
        lv_label_set_text(titleLabel, "Touchscreen Calibration");

        hintLabel = lv_label_create(root);
        lv_obj_align(hintLabel, LV_ALIGN_BOTTOM_MID, 0, -14);
        lv_obj_set_style_text_color(hintLabel, lv_color_white(), LV_STATE_DEFAULT);
        lv_label_set_text(hintLabel, "Tap target 1/4");

        target = lv_button_create(root);
        lv_obj_set_size(target, 28, 28);
        lv_obj_set_style_radius(target, LV_RADIUS_CIRCLE, LV_STATE_DEFAULT);
        lv_obj_set_style_bg_color(target, lv_palette_main(LV_PALETTE_RED), LV_STATE_DEFAULT);
        // Ensure root receives all presses for sampling.
        lv_obj_remove_flag(target, LV_OBJ_FLAG_CLICKABLE);

        auto* targetLabel = lv_label_create(target);
        lv_label_set_text(targetLabel, "+");
        lv_obj_center(targetLabel);

        lv_obj_add_flag(root, LV_OBJ_FLAG_CLICKABLE);
        lv_obj_add_event_cb(root, onPress, LV_EVENT_PRESSED, this);

        updateUi();
    }
};

extern const AppManifest manifest = {
    .appId = "TouchCalibration",
    .appName = "Touch Calibration",
    .appCategory = Category::System,
    .appFlags = AppManifest::Flags::HideStatusBar | AppManifest::Flags::Hidden,
    .createApp = create<TouchCalibrationApp>
};

} // namespace tt::app::touchcalibration