EventSet.hpp

Go to the documentation of this file.

// <StatisticSet> -*- C++ -*-
 
 
#pragma once
 
#include <iostream>
#include <array>
 
#include "sparta/utils/SpartaException.hpp"
#include "sparta/utils/SpartaAssert.hpp"
#include "sparta/events/EventNode.hpp"
 
namespace sparta
{
    class EventSet : public TreeNode
    {
    public:
 
        static constexpr char NODE_NAME[] = "events";
 
        EventSet(TreeNode* parent) :
            TreeNode(NODE_NAME,
                     TreeNode::GROUP_NAME_BUILTIN,
                     TreeNode::GROUP_IDX_NONE,
                     "Event Set")
        {
            if(parent){
                setExpectedParent_(parent);
                parent->addChild(this);
            }
        }
 
        ~EventSet() {}
 
        // Overload of TreeNode::stringize
        virtual std::string stringize(bool pretty=false) const override {
            (void) pretty;
            std::stringstream ss;
            std::array<EventsVector, NUM_SCHEDULING_PHASES>::size_type event_cnt = 0;
            for(const auto & ea : events_) {
                event_cnt += ea.size();
            }
            ss << '<' << getLocation() << ' ' << event_cnt << " events>";
            return ss.str();
        }
 
    public:
        typedef std::vector<EventNode *> EventsVector;
 
        EventsVector & getEvents(sparta::SchedulingPhase phase) {
            return events_[static_cast<uint32_t>(phase)];
        }
 
    private:
 
        virtual void onAddingChild_(TreeNode* child) override {
            if(isFinalized()){
                throw SpartaException("Cannot add a child event once a EventSet is finalized. "
                                    "Error with: ")
                    << getLocation();
            }
 
            EventNode* event_node = dynamic_cast<EventNode*>(child);
            if(nullptr != event_node){
                // Add event to events_ list for tracking.
                events_[static_cast<uint32_t>(event_node->getSchedulingPhase())].push_back(event_node);
                return;
            }
 
            throw SpartaException("Cannot add TreeNode child ")
                << child->getName() << " to EventSet " << getLocation()
                << " because the child is not a CounterBase or Event";
        }
 
        std::array<EventsVector, NUM_SCHEDULING_PHASES> events_;
    };
}