aggregateTree.h

//
// Copyright 2018 Pixar
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
 
#ifndef PXR_BASE_TRACE_AGGREGATE_TREE_H
#define PXR_BASE_TRACE_AGGREGATE_TREE_H
 
#include "pxr/pxr.h"
 
#include "pxr/base/trace/api.h"
#include "pxr/base/trace/aggregateNode.h"
 
PXR_NAMESPACE_OPEN_SCOPE
 
class TraceCollection;
 
TF_DECLARE_WEAK_AND_REF_PTRS(TraceAggregateTree);
TF_DECLARE_WEAK_AND_REF_PTRS(TraceEventTree);
 
 
class TraceAggregateTree : public TfRefBase, public TfWeakBase {
public:
    using This = TraceAggregateTree;
    using ThisPtr = TraceAggregateTreePtr;
    using ThisRefPtr = TraceAggregateTreeRefPtr;
 
    using TimeStamp = TraceEvent::TimeStamp;
    using EventTimes = std::map<TfToken, TimeStamp>;
    using CounterMap = TfHashMap<TfToken, double, TfToken::HashFunctor>;
 
    static ThisRefPtr New() {
        return TfCreateRefPtr(new This());
    }
 
    TraceAggregateNodePtr GetRoot() { return _root; }
 
    const EventTimes& GetEventTimes() const { return _eventTimes; }
 
    const CounterMap& GetCounters() const { return _counters; }
 
    TRACE_API int GetCounterIndex(const TfToken &key) const;
 
    TRACE_API bool AddCounter(const TfToken &key, int index, double totalValue);
 
    TRACE_API void Clear();
 
    TRACE_API void Append(
        const TraceEventTreeRefPtr& eventTree,
        const TraceCollection& collection);
 
private:
    TRACE_API TraceAggregateTree();
 
    using _CounterIndexMap =TfHashMap<TfToken, int, TfToken::HashFunctor>;
 
    TraceAggregateNodeRefPtr _root;
    EventTimes _eventTimes;
    CounterMap _counters;
    _CounterIndexMap _counterIndexMap;
    int _counterIndex;
 
    friend class Trace_AggregateTreeBuilder;
};
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif // PXR_BASE_TRACE_AGGREGATE_TREE_H