PerfNode.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html

#pragma once

#include "MooseTypes.h"

#include <map>

class PerfNode
{
public:
  PerfNode(const PerfID id) : _id(id), _total_time(0), _num_calls(0) {}

  PerfID id() const { return _id; }

  void setStartTime(const std::chrono::time_point<std::chrono::steady_clock> time)
  {
    _start_time = time;
  }

  void addTime(const std::chrono::time_point<std::chrono::steady_clock> time)
  {
    _total_time += time - _start_time;
  }

  void incrementNumCalls() { _num_calls++; }

  void addTime(const std::chrono::steady_clock::duration time) { _total_time += time; }

  PerfNode * getChild(const PerfID id)
  {
    // RHS insertion on purpose
    auto & child_node = _children[id];

    if (!child_node)
      child_node.reset(new PerfNode(id));

    return child_node.get();
  }

  const std::map<PerfID, std::unique_ptr<PerfNode>> & children() const { return _children; }

  std::chrono::steady_clock::duration selfTime() const;

  std::chrono::steady_clock::duration totalTime() const;

  std::chrono::steady_clock::duration childrenTime() const;

  unsigned long int numCalls() { return _num_calls; }

protected:
  PerfID _id;

  std::chrono::time_point<std::chrono::steady_clock> _start_time;

  std::chrono::steady_clock::duration _total_time;

  long unsigned int _num_calls;

  std::map<PerfID, std::unique_ptr<PerfNode>> _children;
};