#pragma once

#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/STLExtras.h"
#include "llvm/Support/raw_ostream.h"

#include <fstream>
#include <limits>
#include <string>

namespace onnx_mlir {

std::fstream openReportFile(const std::string& name);
std::string formatReportMemory(uint64_t bytes);

struct ReportField {
  std::string label;
  std::string value;
};

void printReportFlatFields(llvm::raw_ostream& os, llvm::ArrayRef<ReportField> fields);
void printReportFieldBlock(llvm::raw_ostream& os, llvm::StringRef title, llvm::ArrayRef<ReportField> fields);
void printReportTotalsBlock(llvm::raw_ostream& os, llvm::ArrayRef<ReportField> fields);
void printReportPerCoreAndTotalFields(llvm::raw_ostream& os,
                                      llvm::ArrayRef<ReportField> perCoreFields,
                                      llvm::ArrayRef<ReportField> totalFields);
void printReportEntrySeparator(llvm::raw_ostream& os, bool hasNextEntry);

template <typename EntryTy>
int32_t getFirstReportCoreId(const EntryTy& entry) {
  if (entry.coreIds.empty())
    return std::numeric_limits<int32_t>::max();
  return entry.coreIds.front();
}

template <typename EntryRange>
void sortReportEntriesByFirstCore(EntryRange& entries) {
  llvm::stable_sort(entries, [](const auto& lhs, const auto& rhs) {
    int32_t lhsFirstCore = getFirstReportCoreId(lhs);
    int32_t rhsFirstCore = getFirstReportCoreId(rhs);
    if (lhsFirstCore != rhsFirstCore)
      return lhsFirstCore < rhsFirstCore;
    return lhs.id < rhs.id;
  });
}

} // namespace onnx_mlir