Marcus Chang / AsyncSerial

source/AsyncSerial.cpp

Committer:
marcuschang
Date:
2015-04-02
Revision:
7:5ba3a01e13c4
Parent:
6:9d48f2197243
Child:
8:de7aaaf557ba

File content as of revision 7:5ba3a01e13c4:

#include "AsyncSerial/AsyncSerial.h"

#include "mbed.h"

AsyncSerial::AsyncSerial(PinName tx, PinName rx, PinName rts, PinName cts)
    :   SerialBase(tx, rx),

        sendBuffer(NULL),
        sendLength(0),
        sendIndex(0),

        sendDoneHandler(NULL),
        sendObject(NULL),
        sendMember(),
        sendDoneObject(NULL),

        receiveDoneHandler(NULL),
        receiveObject(NULL),
        receiveMember(),
        receiveDoneObject(NULL),

        waitDoneHandler(NULL),
        waitObject(NULL),
        waitMember(),
        waitDoneObject(NULL),

        isReceiving(false),
        receiveBuffer(NULL),
        receiveMaxLength(0),
        receiveIndex(0),

        insideCondition(false),
        conditionStartBuffer(NULL),
        conditionStartLength(0),
        conditionEndBuffer(NULL),
        conditionEndLength(0),
        conditionIndex(0),
        timeout()
{
    SerialBase::attach<AsyncSerial>(this, &AsyncSerial::getReady, SerialBase::RxIrq);
    SerialBase::attach<AsyncSerial>(this, &AsyncSerial::putDone, SerialBase::TxIrq);
}

void AsyncSerial::putDone()
{
    if (sendLength > 0)
    {
        sendIndex++;

        if (sendIndex < sendLength)
        {
            SerialBase::_base_putc(sendBuffer[sendIndex]);
        }
        else
        {
            sendLength = 0;

            if (sendDoneHandler)
            {
                sendDoneHandler(sendBuffer, sendLength);
                sendDoneHandler = NULL;
            }
            else if (sendObject)
            {
                sendDoneObject(sendObject, sendMember, sendBuffer, sendLength);
                sendObject = NULL;
            }
        }
    }
}

void AsyncSerial::getReady()
{    
    if (isReceiving)
    {
        uint8_t input = SerialBase::_base_getc();

        DEBUG("%c", input);

        if (insideCondition)
        {
            if (receiveBuffer != NULL)
            {
                receiveBuffer[receiveIndex] = input;
                receiveIndex++;

                if (receiveIndex == receiveMaxLength)
                {
                    timeout.detach();
                    getDone(AsyncSerial::RECEIVE_FULL);
                }
            }

            if (conditionEndBuffer != NULL)
            {
                if (input == conditionEndBuffer[conditionIndex])
                {
                    conditionIndex++;

                    if (conditionIndex == conditionEndLength)
                    {
                        timeout.detach();
                        getDone(AsyncSerial::RECEIVE_FOUND);
                    }
                }
                else
                {
                    conditionIndex = 0;
                }
            }
        }
        else
        {
            if (conditionStartBuffer != NULL)
            {
                if (input == conditionStartBuffer[conditionIndex])
                {
                    conditionIndex++;

                    if (conditionIndex == conditionStartLength)
                    {
                        insideCondition = true;
                    }
                }
                else
                {
                    conditionIndex = 0;
                }
            }
        }
    }
}

void AsyncSerial::getDone(uint8_t status)
{
    isReceiving = false;

    DEBUG("getDone: %X %X %X\r\n", this, waitDoneHandler, waitObject);

    if ((receiveBuffer == 0) && (conditionStartBuffer == 0))
    {
        if (waitDoneHandler)
        {
            waitDoneHandler(status);
            waitDoneHandler = NULL;
        }
        else if (waitObject)
        {
            waitDoneObject(waitObject, waitMember, status);
            waitObject = NULL;
        }
    }
    else
    {
        if (receiveDoneHandler)
        {
            receiveDoneHandler(receiveBuffer, receiveIndex, status);
            receiveDoneHandler = NULL;
        }
        else if (receiveObject)
        {
            receiveDoneObject(receiveObject, receiveMember, receiveBuffer, receiveIndex, status);
            receiveObject = NULL;
        }
    }
}

void AsyncSerial::send(send_done_t handler, const char* buffer, uint16_t length)
{
    if (handler && buffer && length)
    {
        sendDoneHandler = handler;

        sendObject = NULL;

        send(buffer, length);
    }
}

void AsyncSerial::send(const char* buffer, uint16_t length)
{
    sendBuffer = buffer;
    sendLength = length;

    sendIndex = 0;
    SerialBase::_base_putc(sendBuffer[sendIndex]);

    DEBUG("send: %X\r\n", waitObject);
}

void AsyncSerial::receive(receive_done_t handler,
                          uint8_t* buffer, uint16_t maxLength,
                          const char* conditionStartBuffer, uint16_t conditionStartLength,
                          const char* conditionEndBuffer, uint16_t conditionEndLength,
                          uint32_t timeoutMilli)
{
    if (handler)
    {
        receiveDoneHandler = handler;

        receiveObject = NULL;

        receive(buffer, maxLength,
                conditionStartBuffer, conditionStartLength,
                conditionEndBuffer, conditionEndLength,
                timeoutMilli);
    }
}

void AsyncSerial::receive(uint8_t* buffer, uint16_t maxLength,
                          const char* _conditionStartBuffer, uint16_t _conditionStartLength,
                          const char* _conditionEndBuffer, uint16_t _conditionEndLength,
                          uint32_t timeoutMilli)
{    
    receiveBuffer = buffer;
    receiveMaxLength = maxLength;
    receiveIndex = 0;

    conditionStartBuffer = _conditionStartBuffer;
    conditionStartLength = _conditionStartLength;
    conditionEndBuffer = _conditionEndBuffer;
    conditionEndLength = _conditionEndLength;
    conditionIndex = 0;

    if ((_conditionStartBuffer != NULL) && (_conditionStartLength != 0))
    {
        insideCondition = false;
    }
    else
    {
        insideCondition = true;
    }

    timeout.attach_us<AsyncSerial>(this, &AsyncSerial::receiveTimeout, timeoutMilli * 1000);
    isReceiving = true;

    DEBUG("receive: %X\r\n", waitObject);
}

void AsyncSerial::wait(wait_done_t handler,
                       const char* conditionEndBuffer, uint16_t conditionEndLength,
                       uint32_t timeoutMilli)
{
    if (handler)
    {
        waitDoneHandler = handler;

        waitObject = NULL;

        receive(NULL, 0,
                NULL, 0,
                conditionEndBuffer, conditionEndLength,
                timeoutMilli);
    }
}

void AsyncSerial::receiveTimeout()
{
    DEBUG("timeout\r\n");
    
    getDone(AsyncSerial::RECEIVE_TIMEOUT);
}

int AsyncSerial::getc()
{
    return SerialBase::_base_getc();
}

int AsyncSerial::putc(int c)
{
    return SerialBase::_base_putc(c);
}