printer.h

Go to the documentation of this file.

#pragma once

#include <atomic>
#include <iostream>
#include <mutex>
#include <thread>

#include <glog/logging.h>

#include "tc/core/rtc.h" // for Duration

namespace tc {
namespace autotune {

class Printer {
 public:
  Printer(
      size_t generation,
      size_t total,
      const std::atomic_size_t& currentCompilationJob,
      const std::atomic_size_t& numEvaluations);
  ~Printer();

  void record(Duration runtime);
  void stop();

  void printAll();

 private:
  void printLoop();

  size_t generation_;
  std::vector<Duration> runtimes_;
  mutable std::mutex runtimesMtx_;

  std::atomic_bool stopPrinting_{false};
  std::thread printerThread_;

  const size_t total_;
  const std::atomic_size_t& currentCompilationJob_;
  const std::atomic_size_t& numEvaluations_;
};

} // namespace autotune
} // namespace tc