print_utils.hpp

#pragma once
 
#include <iomanip>
#include <iostream>
#include <string>
 
namespace itk::itkpix::endec::codec {
 
template <bool enable>
struct print {
  void tag(uint64_t value) {
    if constexpr (!enable) return;
    std::cout << BLUE;
    bits(value, 8);
    std::cout << "(" << value << ")" << RESET;
  }
  void internal_tag(uint64_t value) {
    if constexpr (!enable) return;
    std::cout << BLUE;
    bits(value, 11);
    std::cout << "(" << value << ")" << RESET;
  }
 
  void ccol(uint64_t value) {
    if constexpr (!enable) return;
    std::cout << YELLOW;
    bits(value, 6);
    std::cout << "(" << value << ")" << RESET;
  }
  void qrow(uint64_t value) {
    if constexpr (!enable) return;
    std::cout << GREEN;
    bits(value, 8);
    std::cout << "(" << value << ")" << RESET;
  }
  void hitmap(uint64_t value, size_t length, uint64_t decoded) {
    if constexpr (!enable) return;
    std::cout << RED;
    bits(value, length);
    std::cout << "(";
    bits(decoded, 16);
    std::cout << ")";
    std::cout << RESET;
  }
  void tot(uint64_t value, size_t length) {
    if constexpr (!enable) return;
    std::cout << ORANGE;
    bits(value, length*4);
    std::cout << "(";
    for (int i = 0; i < length; i++) {
      std::cout << std::setw(2) << (value >> (i * 4) & 0xf);
    }
    std::cout << ")" << RESET;
  }
  void ctrl(uint64_t value) {
    if constexpr (!enable) return;
    std::cout << CYAN;
    bits(value >> 1, 1);
    std::cout << MAGENTA;
    bits(value & 1, 1);
    std::cout << RESET;
  }
  void bits(uint64_t value, size_t length) {
    if constexpr (!enable) return;
    for (int i = length - 1; i >= 0; --i) {
      std::cout << ((value >> i) & 1);
    }
  }
 
 private:
  static inline const std::string RED = "\033[31m";
  static inline const std::string GREEN = "\033[32m";
  static inline const std::string YELLOW = "\033[33m";
  static inline const std::string BLUE = "\033[34m";
  static inline const std::string MAGENTA = "\033[35m";
  static inline const std::string CYAN = "\033[36m";
  static inline const std::string RESET = "\033[0m";
  const std::string ORANGE = "\033[38;5;214m";
};
 
}  // itk::itkpix::endec::codec