Utilities.hpp

Go to the documentation of this file.

#pragma once
 
#include "slang/ast/Compilation.h"
#include "slang/text/FormatBuffer.h"
#include "slang/text/SourceLocation.h"
#include "slang/text/SourceManager.h"
 
#include <algorithm>
#include <cstddef>
#include <string>
#include <string_view>
#include <vector>
 
#include <fmt/format.h>
 
namespace slang::netlist {
 
struct Utilities {

  static auto locationStr(ast::Compilation const &compilation,
                          SourceLocation location) {
    if (location.buffer() != SourceLocation::NoLocation.buffer()) {
      auto filename = compilation.getSourceManager()->getFileName(location);
      auto line = compilation.getSourceManager()->getLineNumber(location);
      auto column = compilation.getSourceManager()->getColumnNumber(location);
      return fmt::format("{}:{}:{}", filename, line, column);
    }
    return std::string("?");
  }
 
  using Row = std::vector<std::string>;
  using Table = std::vector<Row>;
 
  struct TableFormatConfig {
 
    // Spaces between columns
    size_t padding;
 
    TableFormatConfig() : padding(2) {}
  };

  static auto formatTable(FormatBuffer &buffer, const Row &header,
                          const Table &rows, TableFormatConfig cfg = {}) {
 
    const std::size_t cols = header.size();
 
    // Compute column widths: max of header and all rows for each column.
    std::vector<std::size_t> widths(cols);
 
    for (std::size_t col = 0; col < cols; ++col) {
      std::size_t max_width = header[col].size();
      for (const auto &row : rows) {
        if (col < row.size()) {
          max_width = std::max(max_width, row[col].size());
        }
      }
      widths[col] = max_width;
    }
 
    auto appendRow = [&](const Row &row) {
      for (std::size_t col = 0; col < cols; ++col) {
        std::string_view value;
        if (col < row.size()) {
          value = row[col];
        } else {
          value = "";
        }
 
        // Left-align in a field of `widths[col]`.
        buffer.format("{:<{}}", value, widths[col]);
 
        // Padding between columns (except after the last).
        if (col + 1 < cols && cfg.padding > 0) {
          buffer.format("{: >{}}", "", cfg.padding);
        }
      }
      buffer.format("\n");
    };
 
    appendRow(header);
    for (const auto &row : rows) {
      appendRow(row);
    }
  }
};
 
} // namespace slang::netlist