test_serial_comm.cpp

#include "pico_comm.h"
#include <string>

// FIXME The length of the read and write buffers should not be fixed. Need to figure out how to
// do that
const DWORD dwReadBuffLength = 10;
const DWORD dwWriteBuffLength = 3;

bool ReadPort(BYTE (*bBuffer)[dwReadBuffLength + 1], SerialComm* SerialManager){
// Read port managed by the SerialManager and store the output to
// the passed buffer.
int iIn;
std::cout << "Press any key to read buffer. Input '-1' to exit: ";
    std::cin >> iIn;
    if (iIn != -1){
        DWORD dwBytesRead;
        (*SerialManager).ReadBytes(bBuffer, dwReadBuffLength, &dwBytesRead);
        if (dwBytesRead != dwReadBuffLength){
            // Haven't read the desired number of bytes. For now, error.
            std::cerr << "Requested " << dwReadBuffLength << " bytes, but got " << dwBytesRead << ".\n";
            return FALSE;
        }

        printf("Bytes read: ");
        for (int i = 0; i < sizeof(*bBuffer); i++){
            printf("%c ", (*bBuffer)[i]);
            printf("%u ", (*bBuffer)[i]);
        }
        printf("\n");
        return TRUE;
    }
    return TRUE;
}

bool WriteToPort(BYTE (*bData)[dwWriteBuffLength + 1], SerialComm* SerialManager){
// Write the contents of the passed byte array to the port managed by
// the SerialManager.

    DWORD dwBytesWritten;
    (*SerialManager).WriteBytes(bData, dwWriteBuffLength, &dwBytesWritten);
    if (dwBytesWritten != dwWriteBuffLength){
        // Havent' written the desired number of bytes.
        std::cerr << "Gave " << dwWriteBuffLength << " bytes to write, but only wrote ";
        std::cerr << dwBytesWritten << ".\n";
        std::cerr << "dwWriteBuffLength: " << dwWriteBuffLength << "\n";
        std::cerr << "dwBytesWritten: " << dwBytesWritten << "\n";
        return FALSE;
    }
    return TRUE;
}

int main(){
    std::string CommPort;
    std::cout << "Insert serial port to use: ";
    std::cin >> CommPort;
    PicoComm SerialManager = PicoComm(CommPort.c_str());
    // PicoComm SerialManager("COM6");

    // Buffers to store byte strings when reading and writing to port.
    BYTE bReadBuffer[dwReadBuffLength + 1] = {0};
    BYTE bWriteBuffer[dwWriteBuffLength + 1];

    do{
        // Choose whether to read or write to the buffer by inputting '1' or '2' respectively.
        std::cout << "To read from port, input '1'. To write to port, input '2'. To exit, input '3': ";
        std::string _Input;
        std::cin >> _Input;
        int iReadOrWrite = (int) _Input[0] - '0';   /* Convert user input to int */

        if ((iReadOrWrite < 0) | (iReadOrWrite > 9)){
            // First char of input can't be converted to an int, it's not a number.
            std::cerr << "Input is not a number.\n";
            continue;
        }

        if(iReadOrWrite == 1){  /* Read from port */
            ReadPort(&bReadBuffer, &SerialManager);
        }
        else if(iReadOrWrite == 2){ /* Write to port */
            std::cout << "Enter " << dwWriteBuffLength << " characters to write: ";
            std::cin >> _Input;
            for (int i=0; i < dwWriteBuffLength; i++){
                // (Somewhat) safely write input to buffer
                if (_Input[i] == '\0')
                {
                    printf("Only entered %i characters.\n", i);
                    continue;
                }
                bWriteBuffer[i] = (BYTE)_Input[i];
            }
            bWriteBuffer[dwWriteBuffLength] = NULL;
            printf("Trying to write array ");
            for (int i = 0; i < dwWriteBuffLength; i++){
                printf("%x::", bWriteBuffer[i]);
            };
            printf("\n");

            WriteToPort(&bWriteBuffer, &SerialManager);
        }
        else if(iReadOrWrite == 3){ /* Exit */
            SerialManager.CloseSerialConnection();
            break;
        }
        else{   /* Invalid input */
            std::cout << "Invalid input.\n";
        }
    } while(true);
}