/*
 * Copyright (c) 1999-2013 Mark D. Hill and David A. Wood
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * 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;
 * neither the name of the copyright holders 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
 * OWNER 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.
 */

/*
   This file has been modified by Kevin Moore and Dan Nussbaum of the
   Scalable Systems Research Group at Sun Microsystems Laboratories
   (http://research.sun.com/scalable/) to support the Adaptive
   Transactional Memory Test Platform (ATMTP).

   Please send email to atmtp-interest@sun.com with feedback, questions, or
   to request future announcements about ATMTP.

   ----------------------------------------------------------------------

   File modification date: 2008-02-23

   ----------------------------------------------------------------------
*/

#include "mem/ruby/profiler/Profiler.hh"

#include <sys/types.h>
#include <unistd.h>

#include <algorithm>
#include <fstream>

#include "base/stl_helpers.hh"
#include "base/str.hh"
#include "config/build_gpu.hh"
#include "mem/ruby/network/Network.hh"
#include "mem/ruby/profiler/AddressProfiler.hh"
#include "mem/ruby/protocol/MachineType.hh"
#include "mem/ruby/protocol/RubyRequest.hh"

/**
 * the profiler uses GPUCoalescer code even
 * though the GPUCoalescer is not built for
 * all ISAs, which can lead to run/link time
 * errors. here we guard the coalescer code
 * with ifdefs as there is no easy way to
 * refactor this code without removing
 * GPUCoalescer stats from the profiler.
 *
 * eventually we should use probe points
 * here, but until then these ifdefs will
 * serve.
 */
#if BUILD_GPU
#include "mem/ruby/system/GPUCoalescer.hh"

#endif

#include "mem/ruby/system/Sequencer.hh"

using gem5::stl_helpers::operator<<;

Profiler::Profiler(const RubySystemParams &p, RubySystem *rs)
    : m_ruby_system(rs), m_hot_lines(p.hot_lines),
      m_all_instructions(p.all_instructions),
      m_num_vnets(p.number_of_virtual_networks),
      rubyProfilerStats(rs, this)
{
    m_address_profiler_ptr = new AddressProfiler(p.num_of_sequencers, this);
    m_address_profiler_ptr->setHotLines(m_hot_lines);
    m_address_profiler_ptr->setAllInstructions(m_all_instructions);

    if (m_all_instructions) {
        m_inst_profiler_ptr = new AddressProfiler(p.num_of_sequencers, this);
        m_inst_profiler_ptr->setHotLines(m_hot_lines);
        m_inst_profiler_ptr->setAllInstructions(m_all_instructions);
    }
}

Profiler::~Profiler()
{
}

Profiler::
ProfilerStats::ProfilerStats(Stats::Group *parent, Profiler *profiler)
    : Stats::Group(parent),
      perRequestTypeStats(parent),
      perMachineTypeStats(parent),
      perRequestTypeMachineTypeStats(parent),
      ADD_STAT(delayHistogram, "delay histogram for all message"),
      ADD_STAT(m_outstandReqHistSeqr, ""),
      ADD_STAT(m_outstandReqHistCoalsr, ""),
      ADD_STAT(m_latencyHistSeqr, ""),
      ADD_STAT(m_latencyHistCoalsr, ""),
      ADD_STAT(m_hitLatencyHistSeqr, ""),
      ADD_STAT(m_missLatencyHistSeqr, ""),
      ADD_STAT(m_missLatencyHistCoalsr, "")
{
    delayHistogram
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    for (int i = 0; i < profiler->m_num_vnets; i++) {
        delayVCHistogram.push_back(new Stats::Histogram(this));
        delayVCHistogram[i]
            ->init(10)
            .name(csprintf("delayVCHist.vnet_%i", i))
            .desc(csprintf("delay histogram for vnet_%i", i))
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);
    }

    m_outstandReqHistSeqr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_outstandReqHistCoalsr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_latencyHistSeqr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_latencyHistCoalsr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_hitLatencyHistSeqr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_missLatencyHistSeqr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);

    m_missLatencyHistCoalsr
        .init(10)
        .flags(Stats::nozero | Stats::pdf | Stats::oneline);
}

Profiler::ProfilerStats::
PerRequestTypeStats::PerRequestTypeStats(Stats::Group *parent)
    : Stats::Group(parent, "RequestType")
{
    for (int i = 0; i < RubyRequestType_NUM; i++) {
        m_typeLatencyHistSeqr.push_back(new Stats::Histogram(this));
        m_typeLatencyHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.latency_hist_seqr", RubyRequestType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_typeLatencyHistCoalsr.push_back(new Stats::Histogram(this));
        m_typeLatencyHistCoalsr[i]
            ->init(10)
            .name(csprintf("%s.latency_hist_coalsr", RubyRequestType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_hitTypeLatencyHistSeqr.push_back(new Stats::Histogram(this));
        m_hitTypeLatencyHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.hit_latency_hist_seqr", RubyRequestType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_missTypeLatencyHistSeqr.push_back(new Stats::Histogram(this));
        m_missTypeLatencyHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.miss_latency_hist_seqr", RubyRequestType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_missTypeLatencyHistCoalsr.push_back(new Stats::Histogram(this));
        m_missTypeLatencyHistCoalsr[i]
            ->init(10)
            .name(csprintf("%s.miss_latency_hist_coalsr", RubyRequestType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);
    }
}

Profiler::ProfilerStats::
PerMachineTypeStats::PerMachineTypeStats(Stats::Group *parent)
    : Stats::Group(parent, "MachineType")
{
    for (int i = 0; i < MachineType_NUM; i++) {
        m_hitMachLatencyHistSeqr.push_back(new Stats::Histogram(this));
        m_hitMachLatencyHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.hit_mach_latency_hist_seqr", MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_missMachLatencyHistSeqr.push_back(new Stats::Histogram(this));
        m_missMachLatencyHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.miss_mach_latency_hist_seqr", MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_missMachLatencyHistCoalsr.push_back(new Stats::Histogram(this));
        m_missMachLatencyHistCoalsr[i]
            ->init(10)
            .name(csprintf("%s.miss_mach_latency_hist_coalsr",
                           MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_IssueToInitialDelayHistSeqr.push_back(new Stats::Histogram(this));
        m_IssueToInitialDelayHistSeqr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_seqr.issue_to_initial_request",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_IssueToInitialDelayHistCoalsr.push_back(new Stats::Histogram(this));
        m_IssueToInitialDelayHistCoalsr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_coalsr.issue_to_initial_request",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_InitialToForwardDelayHistSeqr.push_back(new Stats::Histogram(this));
        m_InitialToForwardDelayHistSeqr[i]
            ->init(10)
            .name(csprintf("%s.miss_latency_hist_seqr.initial_to_forward",
                           MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_InitialToForwardDelayHistCoalsr
            .push_back(new Stats::Histogram(this));
        m_InitialToForwardDelayHistCoalsr[i]
            ->init(10)
            .name(csprintf("%s.miss_latency_hist_coalsr.initial_to_forward",
                           MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_ForwardToFirstResponseDelayHistSeqr
            .push_back(new Stats::Histogram(this));

        m_ForwardToFirstResponseDelayHistSeqr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_seqr.forward_to_first_response",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_ForwardToFirstResponseDelayHistCoalsr
            .push_back(new Stats::Histogram(this));
        m_ForwardToFirstResponseDelayHistCoalsr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_coalsr.forward_to_first_response",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_FirstResponseToCompletionDelayHistSeqr
            .push_back(new Stats::Histogram(this));
        m_FirstResponseToCompletionDelayHistSeqr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_seqr.first_response_to_completion",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_FirstResponseToCompletionDelayHistCoalsr
            .push_back(new Stats::Histogram(this));
        m_FirstResponseToCompletionDelayHistCoalsr[i]
            ->init(10)
            .name(csprintf(
                "%s.miss_latency_hist_coalsr.first_response_to_completion",
                MachineType(i)))
            .desc("")
            .flags(Stats::nozero | Stats::pdf | Stats::oneline);

        m_IncompleteTimesSeqr.push_back(new Stats::Scalar(this));
        m_IncompleteTimesSeqr[i]
            ->name(csprintf("%s.incomplete_times_seqr", MachineType(i)))
            .desc("")
            .flags(Stats::nozero);
    }
}

Profiler::ProfilerStats::
PerRequestTypeMachineTypeStats::
PerRequestTypeMachineTypeStats(Stats::Group *parent)
    : Stats::Group(parent, "RequestTypeMachineType")
{
    for (int i = 0; i < RubyRequestType_NUM; i++) {
        m_hitTypeMachLatencyHistSeqr
            .push_back(std::vector<Stats::Histogram *>());
        m_missTypeMachLatencyHistSeqr
            .push_back(std::vector<Stats::Histogram *>());
        m_missTypeMachLatencyHistCoalsr
            .push_back(std::vector<Stats::Histogram *>());

        for (int j = 0; j < MachineType_NUM; j++) {
            m_hitTypeMachLatencyHistSeqr[i]
                .push_back(new Stats::Histogram(this));
            m_hitTypeMachLatencyHistSeqr[i][j]
                ->init(10)
                .name(csprintf("%s.%s.hit_type_mach_latency_hist_seqr",
                               RubyRequestType(i), MachineType(j)))
                .desc("")
                .flags(Stats::nozero | Stats::pdf | Stats::oneline);

            m_missTypeMachLatencyHistSeqr[i]
                .push_back(new Stats::Histogram(this));
            m_missTypeMachLatencyHistSeqr[i][j]
                ->init(10)
                .name(csprintf("%s.%s.miss_type_mach_latency_hist_seqr",
                               RubyRequestType(i), MachineType(j)))
                .desc("")
                .flags(Stats::nozero | Stats::pdf | Stats::oneline);

            m_missTypeMachLatencyHistCoalsr[i]
                .push_back(new Stats::Histogram(this));
            m_missTypeMachLatencyHistCoalsr[i][j]
                ->init(10)
                .name(csprintf("%s.%s.miss_type_mach_latency_hist_coalsr",
                               RubyRequestType(i), MachineType(j)))
                .desc("")
                .flags(Stats::nozero | Stats::pdf | Stats::oneline);
        }
    }
}

void
Profiler::collateStats()
{
    if (!m_all_instructions) {
        m_address_profiler_ptr->collateStats();
    }

    if (m_all_instructions) {
        m_inst_profiler_ptr->collateStats();
    }

    for (uint32_t i = 0; i < MachineType_NUM; i++) {
        for (std::map<uint32_t, AbstractController*>::iterator it =
                  m_ruby_system->m_abstract_controls[i].begin();
             it != m_ruby_system->m_abstract_controls[i].end(); ++it) {

            AbstractController *ctr = (*it).second;
            rubyProfilerStats.delayHistogram.add(ctr->getDelayHist());

            for (uint32_t i = 0; i < m_num_vnets; i++) {
                rubyProfilerStats.
                    delayVCHistogram[i]->add(ctr->getDelayVCHist(i));
            }
        }
    }

    for (uint32_t i = 0; i < MachineType_NUM; i++) {
        for (std::map<uint32_t, AbstractController*>::iterator it =
                m_ruby_system->m_abstract_controls[i].begin();
                it != m_ruby_system->m_abstract_controls[i].end(); ++it) {

            AbstractController *ctr = (*it).second;
            Sequencer *seq = ctr->getCPUSequencer();
            if (seq != NULL) {
                rubyProfilerStats.
                    m_outstandReqHistSeqr.add(seq->getOutstandReqHist());
            }
#if BUILD_GPU
            GPUCoalescer *coal = ctr->getGPUCoalescer();
            if (coal != NULL) {
                rubyProfilerStats.
                    m_outstandReqHistCoalsr.add(coal->getOutstandReqHist());
            }
#endif
        }
    }

    for (uint32_t i = 0; i < MachineType_NUM; i++) {
        for (std::map<uint32_t, AbstractController*>::iterator it =
                m_ruby_system->m_abstract_controls[i].begin();
                it != m_ruby_system->m_abstract_controls[i].end(); ++it) {

            AbstractController *ctr = (*it).second;
            Sequencer *seq = ctr->getCPUSequencer();
            if (seq != NULL) {
                // add all the latencies
                rubyProfilerStats.
                        m_latencyHistSeqr.add(seq->getLatencyHist());
                rubyProfilerStats.
                        m_hitLatencyHistSeqr.add(seq->getHitLatencyHist());
                rubyProfilerStats.
                        m_missLatencyHistSeqr.add(seq->getMissLatencyHist());

                // add the per request type latencies
                for (uint32_t j = 0; j < RubyRequestType_NUM; ++j) {
                    rubyProfilerStats
                        .perRequestTypeStats
                        .m_typeLatencyHistSeqr[j]
                        ->add(seq->getTypeLatencyHist(j));
                    rubyProfilerStats
                        .perRequestTypeStats
                        .m_hitTypeLatencyHistSeqr[j]
                        ->add(seq->getHitTypeLatencyHist(j));
                    rubyProfilerStats
                        .perRequestTypeStats
                        .m_missTypeLatencyHistSeqr[j]
                        ->add(seq->getMissTypeLatencyHist(j));
                }

                // add the per machine type miss latencies
                for (uint32_t j = 0; j < MachineType_NUM; ++j) {
                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_hitMachLatencyHistSeqr[j]
                        ->add(seq->getHitMachLatencyHist(j));
                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_missMachLatencyHistSeqr[j]
                        ->add(seq->getMissMachLatencyHist(j));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_IssueToInitialDelayHistSeqr[j]
                        ->add(seq->getIssueToInitialDelayHist(MachineType(j)));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_InitialToForwardDelayHistSeqr[j]
                        ->add(seq
                            ->getInitialToForwardDelayHist(MachineType(j)));
                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_ForwardToFirstResponseDelayHistSeqr[j]
                        ->add(seq
                            ->getForwardRequestToFirstResponseHist(
                                MachineType(j)));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_FirstResponseToCompletionDelayHistSeqr[j]
                        ->add(seq
                            ->getFirstResponseToCompletionDelayHist(
                                MachineType(j)));

                    *(rubyProfilerStats
                        .perMachineTypeStats
                        .m_IncompleteTimesSeqr[j]) +=
                            seq->getIncompleteTimes(MachineType(j));
                }

                // add the per (request, machine) type miss latencies
                for (uint32_t j = 0; j < RubyRequestType_NUM; j++) {
                    for (uint32_t k = 0; k < MachineType_NUM; k++) {
                        rubyProfilerStats
                            .perRequestTypeMachineTypeStats
                            .m_hitTypeMachLatencyHistSeqr[j][k]->add(
                                seq->getHitTypeMachLatencyHist(j,k));
                        rubyProfilerStats
                            .perRequestTypeMachineTypeStats
                            .m_missTypeMachLatencyHistSeqr[j][k]->add(
                                seq->getMissTypeMachLatencyHist(j,k));
                    }
                }
            }
#if BUILD_GPU
            GPUCoalescer *coal = ctr->getGPUCoalescer();
            if (coal != NULL) {
                // add all the latencies
                rubyProfilerStats.
                    m_latencyHistCoalsr.add(coal->getLatencyHist());
                rubyProfilerStats.
                    m_missLatencyHistCoalsr.add(coal->getMissLatencyHist());

                // add the per request type latencies
                for (uint32_t j = 0; j < RubyRequestType_NUM; ++j) {
                    rubyProfilerStats
                        .perRequestTypeStats
                        .m_typeLatencyHistCoalsr[j]
                        ->add(coal->getTypeLatencyHist(j));
                    rubyProfilerStats
                        .perRequestTypeStats
                        .m_missTypeLatencyHistCoalsr[j]
                        ->add(coal->getMissTypeLatencyHist(j));
                }

                // add the per machine type miss latencies
                for (uint32_t j = 0; j < MachineType_NUM; ++j) {
                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_missMachLatencyHistCoalsr[j]
                        ->add(coal->getMissMachLatencyHist(j));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_IssueToInitialDelayHistCoalsr[j]
                        ->add(coal->getIssueToInitialDelayHist(
                            MachineType(j)));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_InitialToForwardDelayHistCoalsr[j]
                        ->add(coal->getInitialToForwardDelayHist(
                            MachineType(j)));
                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_ForwardToFirstResponseDelayHistCoalsr[j]
                        ->add(coal->getForwardRequestToFirstResponseHist(
                            MachineType(j)));

                    rubyProfilerStats
                        .perMachineTypeStats
                        .m_FirstResponseToCompletionDelayHistCoalsr[j]
                        ->add(coal->getFirstResponseToCompletionDelayHist(
                            MachineType(j)));
                }

                // add the per (request, machine) type miss latencies
                for (uint32_t j = 0; j < RubyRequestType_NUM; j++) {
                    for (uint32_t k = 0; k < MachineType_NUM; k++) {
                        rubyProfilerStats
                            .perRequestTypeMachineTypeStats
                            .m_missTypeMachLatencyHistCoalsr[j][k]
                            ->add(coal->getMissTypeMachLatencyHist(j,k));
                    }
                }
            }
#endif
        }
    }
}

void
Profiler::addAddressTraceSample(const RubyRequest& msg, NodeID id)
{
    if (msg.getType() != RubyRequestType_IFETCH) {
        // Note: The following line should be commented out if you
        // want to use the special profiling that is part of the GS320
        // protocol

        // NOTE: Unless PROFILE_HOT_LINES is enabled, nothing will be
        // profiled by the AddressProfiler
        m_address_profiler_ptr->
            addTraceSample(msg.getLineAddress(), msg.getProgramCounter(),
                           msg.getType(), msg.getAccessMode(), id, false);
    }
}