aboutsummaryrefslogtreecommitdiff
path: root/algo/debug.h
blob: d63f896dc178af444699a2f064ae0786efecc1c2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <bits/stdc++.h>
using namespace std;

template <typename k, typename v>
ostream &operator<<(ostream &os, const pair<k, v> &p);
template <typename t> ostream &operator<<(ostream &os, const vector<t> &vec);
template <typename k, typename v>
ostream &operator<<(ostream &os, const map<k, v> &m);
template <typename k, typename v>
ostream &operator<<(ostream &os, const unordered_map<k, v> &m);
template <typename t> ostream &operator<<(ostream &os, const set<t> &s);
template <typename t>
ostream &operator<<(ostream &os, const unordered_set<t> &s);

template <typename k, typename v>
ostream &operator<<(ostream &os, const pair<k, v> &p) {
  os << "(" << p.first << ", " << p.second << ")";
  return os;
}

template <typename t> ostream &operator<<(ostream &os, const vector<t> &vec) {
  os << "[";
  for (size_t i = 0; i < vec.size(); ++i) {
    if (i > 0)
      os << ", ";
    os << vec[i];
  }
  os << "]";
  return os;
}

template <typename t>
ostream &operator<<(ostream &os, const vector<vector<t>> &vec) {
  os << "[\n";
  for (size_t i = 0; i < vec.size(); ++i) {
    os << "  " << vec[i];
    if (i > 0)
      os << ", ";
    os << '\n';
  }
  os << "]";
  return os;
}

template <typename k, typename v>
ostream &operator<<(ostream &os, const map<k, v> &m) {
  os << "{";
  for (const auto &p : m) {
    os << p.first << ": " << p.second << ", ";
  }
  os << "}";
  return os;
}

template <typename k, typename v>
ostream &operator<<(ostream &os, const unordered_map<k, v> &m) {
  os << "{";
  for (const auto &p : m) {
    os << p.first << ": " << p.second << ", ";
  }
  os << "}";
  return os;
}

template <typename t> ostream &operator<<(ostream &os, const set<t> &s) {
  int i = 0;
  os << "{";
  for (const auto &e : s) {
    if (i > 0)
      os << ", ";
    os << e;
    i++;
  }
  os << "}";
  return os;
}

template <typename t>
ostream &operator<<(ostream &os, const unordered_set<t> &s) {
  int i = 0;
  os << "{";
  for (const auto &e : s) {
    if (i > 0)
      os << ", ";
    os << e;
    i++;
  }
  os << "}";
  return os;
}

template <typename... Ts>
void debug_out(const std::string_view names, const Ts &...ts) {
  size_t start = 0;

  const auto print_single = [&names](size_t start, const auto &value) {
    const auto end = names.find(',', start + 1);
    std::cerr << names.substr(start, end - start) << " = " << value << '\n';

    return end + 2;
  };

  ((start = print_single(start, ts)), ...);
}

#define dbg(...) debug_out(#__VA_ARGS__, __VA_ARGS__)