UsbTxStream.cpp

//------------------------------------------------------------------------------
//This file is part of milkLib. 
//See license.txt for the full license governing this code.
//------------------------------------------------------------------------------

#include "UsbTxStream.hpp"
#include "UsbDevice.hpp"
#include "UsbMacros.hpp"
#include "RtosInclude.hpp"

UsbTxStream::UsbTxStream(uint8_t ep, size_t size) :
        mBuffer(size) {
    mMaxTimeout = Rtos::TICK_INFINITE;
    mEp = ep;
    UsbDevice::registerInCallback(ep, this);
}

UsbTxStream::~UsbTxStream() {
}

int UsbTxStream::write(const void* buffer, int len, Rtos::TICK timeout) {
    if (timeout > mMaxTimeout) {
        timeout = mMaxTimeout;
    }
    uint8_t* src = (uint8_t*) buffer;
    int left = len;
    RtosTimer timer;
    timer.set(timeout);
    do {
        int written = mBuffer.writeChunk(src, left, 10);
        UsbDevice::notifyDataInReadyForEp(mEp);
        src += written;
        left -= written;
        if (left <= 0) {
            break;
        }
    } while (left > 0 && !timer.isEllapsed());
    return (len - left);
}

void UsbTxStream::onDataInFree(uint8_t ep, uint16_t len) {
    uint8_t buffer[len];
    len = mBuffer.read(buffer, len, 0);
    if (len > 0) {
        UsbDevice::writeBuffer(ep, buffer, len);
    }
}