uv-k1-k5v3-firmware-custom/App/ui/aircopy.c

/* Copyright 2023 Dual Tachyon
 * https://github.com/DualTachyon
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *     Unless required by applicable law or agreed to in writing, software
 *     distributed under the License is distributed on an "AS IS" BASIS,
 *     WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *     See the License for the specific language governing permissions and
 *     limitations under the License.
 */

#ifdef ENABLE_AIRCOPY

#include <string.h>

#include "app/aircopy.h"
#include "driver/st7565.h"
#include "external/printf/printf.h"
#include "misc.h"
#include "radio.h"
#include "ui/aircopy.h"
#include "ui/helper.h"
#include "ui/inputbox.h"

static void set_bit(uint8_t* array, int bit_index) {
    array[bit_index / 8] |= (1 << (bit_index % 8));
}

static int get_bit(uint8_t* array, int bit_index) {
    return (array[bit_index / 8] >> (bit_index % 8)) & 1;
}

void UI_DisplayAircopy(void)
{
    char String[16] = { 0 };
    char *pPrintStr = { 0 };
    uint16_t percent;

    UI_DisplayClear();

    if (gAircopyState == AIRCOPY_READY) {
        pPrintStr = "AIR COPY(RDY)";
    } else if (gAircopyState == AIRCOPY_TRANSFER) {
        pPrintStr = "AIR COPY";
    } else {
        pPrintStr = "AIR COPY(CMP)";
        gAircopyState = AIRCOPY_READY;
    }

    UI_PrintString(pPrintStr, 2, 127, 0, 8);

    if (gInputBoxIndex == 0) {
        uint32_t frequency = gRxVfo->freq_config_RX.Frequency;
        sprintf(String, "%3u.%05u", frequency / 100000, frequency % 100000);
        // show the remaining 2 small frequency digits
        UI_PrintStringSmallNormal(String + 7, 97, 0, 3);
        String[7] = 0;
        // show the main large frequency digits
        UI_DisplayFrequency(String, 16, 2, false);
    } else {
        const char *ascii = INPUTBOX_GetAscii();
        sprintf(String, "%.3s.%.3s", ascii, ascii + 3);
        UI_DisplayFrequency(String, 16, 2, false);
    }

    memset(String, 0, sizeof(String));

    // Get the current map and calculate percentage based on its total blocks
    const AIRCOPY_TransferMap_t *currentMap = AIRCOPY_GetCurrentMap();

    uint16_t doneBlocks = gAirCopyBlockNumber + gErrorsDuringAirCopy;

    if (doneBlocks > currentMap->total_blocks)
        doneBlocks = currentMap->total_blocks;

    percent = (doneBlocks * 10000) / currentMap->total_blocks;

    if (gAirCopyIsSendMode == 0) {
        sprintf(String, "RCV:%02u.%02u%% E:%d", percent / 100, percent % 100, gErrorsDuringAirCopy);
    } else if (gAirCopyIsSendMode == 1) {
        sprintf(String, "SND:%02u.%02u%%", percent / 100, percent % 100);
    }

    // Draw gauge
    if(gAircopyStep != 0)
    {
        UI_PrintString(String, 2, 127, 5, 8);

        gFrameBuffer[4][1] = 0x3c;
        gFrameBuffer[4][2] = 0x42;

        for(uint8_t i = 1; i <= AIRCOPY_BAR_WIDTH + 2; i++)
        {
            gFrameBuffer[4][2 + i] = 0x81;
        }

        gFrameBuffer[4][125] = 0x42;
        gFrameBuffer[4][126] = 0x3c;
    }
    
    // Draw memory selection
    if(gAircopyState == AIRCOPY_READY)
    {
        doneBlocks = 0;

        memset(gFrameBuffer[5], 0, 128);
        memset(gFrameBuffer[6], 0, 128);

        if(gAircopyCurrentMapIndex < AIRCOPY_NUM_BANKS) {   
            sprintf(String, "MEM %03u - %03u%", (gAircopyCurrentMapIndex * 128) + 1, (gAircopyCurrentMapIndex + 1) * 128);
        }
        else
        {
            sprintf(String, "Settings");            
        }
        UI_PrintString(String, 2, 127, 5, 8);
    }

    if (doneBlocks > 0)
    {
        // Track CRC errors per real block index
        if (gErrorsDuringAirCopy != lErrorsDuringAirCopy)
        {
            // Mark the last processed block as faulty
            set_bit(crc, doneBlocks - 1);
            lErrorsDuringAirCopy = gErrorsDuringAirCopy;
        }

        const AIRCOPY_TransferMap_t *currentMap = AIRCOPY_GetCurrentMap();

        uint16_t total = currentMap->total_blocks;
        uint16_t done  = gAirCopyBlockNumber + gErrorsDuringAirCopy;
        
        if (done > total) done = total;

        for (uint8_t col = 0; col < AIRCOPY_BAR_WIDTH; col++)
        {
            /* Map column [0..BAR_WIDTH-1] to block [0..total-1] */
            uint16_t b = (uint16_t)((col * (uint32_t)total) / AIRCOPY_BAR_WIDTH);

            bool processed = (b < done);
            bool error     = processed && get_bit(crc, b);

            if (!processed)
                gFrameBuffer[4][col + 4] = 0x81;   // not yet processed
            else if (error)
                gFrameBuffer[4][col + 4] = 0x81;   // error gap (intentional hole)
            else
                gFrameBuffer[4][col + 4] = 0xBD;   // ok filled
        }

    }

    ST7565_BlitFullScreen();
}

#endif