// Copyright © 2019-2023
// 
// 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.

#pragma once

#include "cache_sim.h"

namespace vortex {

class CacheCluster : public SimObject<CacheCluster> {
public:
    std::vector<std::vector<SimPort<MemReq>>> CoreReqPorts;
    std::vector<std::vector<SimPort<MemRsp>>> CoreRspPorts;
    SimPort<MemReq> MemReqPort;
    SimPort<MemRsp> MemRspPort;

    CacheCluster(const SimContext& ctx, 
                 const char* name, 
                 uint32_t num_units, 
                 uint32_t num_caches, 
                 uint32_t num_requests,
                 const CacheSim::Config& config) 
        : SimObject(ctx, name)        
        , CoreReqPorts(num_units, std::vector<SimPort<MemReq>>(num_requests, this))
        , CoreRspPorts(num_units, std::vector<SimPort<MemRsp>>(num_requests, this))
        , MemReqPort(this)
        , MemRspPort(this)
        , caches_(MAX(num_caches, 0x1)) {

        CacheSim::Config config2(config);
        if (0 == num_caches) {
            num_caches = 1;
            config2.bypass = true;
        }

        char sname[100];
        
        std::vector<MemSwitch::Ptr> unit_arbs(num_units);
        for (uint32_t u = 0; u < num_units; ++u) {
            snprintf(sname, 100, "%s-unit-arb-%d", name, u);
            unit_arbs.at(u) = MemSwitch::Create(sname, ArbiterType::RoundRobin, num_requests, config.num_inputs);
            for (uint32_t i = 0; i < num_requests; ++i) {
                this->CoreReqPorts.at(u).at(i).bind(&unit_arbs.at(u)->ReqIn.at(i));
                unit_arbs.at(u)->RspIn.at(i).bind(&this->CoreRspPorts.at(u).at(i));
            }
        }

        std::vector<MemSwitch::Ptr> mem_arbs(config.num_inputs);
        for (uint32_t i = 0; i < config.num_inputs; ++i) {
            snprintf(sname, 100, "%s-mem-arb-%d", name, i);
            mem_arbs.at(i) = MemSwitch::Create(sname, ArbiterType::RoundRobin, num_units, num_caches);
            for (uint32_t u = 0; u < num_units; ++u) {              
                unit_arbs.at(u)->ReqOut.at(i).bind(&mem_arbs.at(i)->ReqIn.at(u));
                mem_arbs.at(i)->RspIn.at(u).bind(&unit_arbs.at(u)->RspOut.at(i));
            }            
        }

        snprintf(sname, 100, "%s-cache-arb", name);
        auto cache_arb = MemSwitch::Create(sname, ArbiterType::RoundRobin, num_caches, 1);

        for (uint32_t i = 0; i < num_caches; ++i) {
            snprintf(sname, 100, "%s-cache%d", name, i);
            caches_.at(i) = CacheSim::Create(sname, config2);

            for (uint32_t j = 0; j < config.num_inputs; ++j) {
                mem_arbs.at(j)->ReqOut.at(i).bind(&caches_.at(i)->CoreReqPorts.at(j));
                caches_.at(i)->CoreRspPorts.at(j).bind(&mem_arbs.at(j)->RspOut.at(i));
            }

            caches_.at(i)->MemReqPort.bind(&cache_arb->ReqIn.at(i));
            cache_arb->RspIn.at(i).bind(&caches_.at(i)->MemRspPort);
        }

        cache_arb->ReqOut.at(0).bind(&this->MemReqPort);
        this->MemRspPort.bind(&cache_arb->RspOut.at(0));
    }

    ~CacheCluster() {}

    void reset() {}
    
    void tick() {}

    CacheSim::PerfStats perf_stats() const {
        CacheSim::PerfStats perf;
        for (auto cache : caches_) {
            perf += cache->perf_stats();
        }   
        return perf;
    }
    
private:
    std::vector<CacheSim::Ptr> caches_;
};

}