/*
 * Copyright (c) 2020, Technische Universität Kaiserslautern
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 * 1. Redistributions of source code must retain the above copyright notice,
 *    this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * 3. Neither the name of the copyright holder nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * Author: Lukas Steiner
 */

#include "BufferCounterBankwise.h"

#include "DRAMSys/common/dramExtensions.h"

using namespace tlm;

BufferCounterBankwise::BufferCounterBankwise(unsigned requestBufferSize, unsigned numberOfBanks)
    : requestBufferSize(requestBufferSize)
{
    numRequestsOnBank = std::vector<unsigned>(numberOfBanks, 0);
}

bool BufferCounterBankwise::hasBufferSpace() const
{
    return (numRequestsOnBank[lastBankID] < requestBufferSize);
}

void BufferCounterBankwise::storeRequest(const tlm_generic_payload& trans)
{
    lastBankID = ControllerExtension::getBank(trans).ID();
    numRequestsOnBank[lastBankID]++;
    if (trans.is_read())
        numReadRequests++;
    else
        numWriteRequests++;
}

void BufferCounterBankwise::removeRequest(const tlm_generic_payload& trans)
{
    numRequestsOnBank[ControllerExtension::getBank(trans).ID()]--;
    if (trans.is_read())
        numReadRequests--;
    else
        numWriteRequests--;
}

const std::vector<unsigned>& BufferCounterBankwise::getBufferDepth() const
{
    return numRequestsOnBank;
}

unsigned BufferCounterBankwise::getNumReadRequests() const
{
    return numReadRequests;
}

unsigned BufferCounterBankwise::getNumWriteRequests() const
{
    return numWriteRequests;
}