profiler.h

/* Copyright 2016 Carnegie Mellon University, NVIDIA Corporation
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include "scanner/util/util.h"

#include <atomic>
#include <fstream>
#include <map>
#include <string>
#include <vector>

namespace storehouse {
class WriteFile;
}

namespace scanner {

enum ProfilerLevel {
  Debug = 0,
  Info = 1,
  Important = 2
};

extern ProfilerLevel PROFILER_LEVEL;

class Profiler {
 public:
  Profiler(timepoint_t base_time);

  Profiler(const Profiler& other);

  void add_interval(const std::string& key, timepoint_t start, timepoint_t end, ProfilerLevel level=ProfilerLevel::Info);

  void increment(const std::string& key, int64_t value);

  void reset(timepoint_t base_time);

  struct TaskRecord {
    std::string key;
    int64_t start;
    int64_t end;
  };

  const std::vector<TaskRecord>& get_records() const;

  const std::map<std::string, int64_t>& get_counters() const;

 protected:
  void spin_lock();
  void unlock();

  timepoint_t base_time_;
  std::atomic_flag lock_;
  std::vector<TaskRecord> records_;
  std::map<std::string, int64_t> counters_;
};

class ProfileBlock {
  public:
  ProfileBlock(Profiler* profiler, std::string label);
  ~ProfileBlock();
protected:
  Profiler* profiler_;
  std::string label_;
  timepoint_t start_;
};

void write_profiler_to_file(storehouse::WriteFile* file, int64_t node,
                            std::string type_name, std::string tag,
                            int64_t worker_num, const Profiler& profiler);

}  // namespace scanner

#include "scanner/util/profiler.inl"