debug_utils.h

Go to the documentation of this file.

 
#ifndef KIDS_DEBUG_UTILS_H
#define KIDS_DEBUG_UTILS_H
 
#include <iomanip>
#include <iostream>
 
#include "kids/fmt.h"
 
#define PRINT_ARRAY(_A, _n1, _n2)                                                           \
    ({                                                                                      \
        std::cout << #_A << " = np.array([\n";                                              \
        int _idxA = 0;                                                                      \
        for (int _i = 0; _i < (_n1); ++_i) {                                                \
            for (int _j = 0; _j < (_n2); ++_j) std::cout << FMT(4) << (_A)[_idxA++] << ","; \
            std::cout << std::endl;                                                         \
        }                                                                                   \
        { std::cout << "])\n"; }                                                            \
    })
 
#define PRINT_ARRAY_8(_A, _n1, _n2)                                                         \
    ({                                                                                      \
        std::cout << #_A << " = np.array([\n";                                              \
        int _idxA = 0;                                                                      \
        for (int _i = 0; _i < (_n1); ++_i) {                                                \
            for (int _j = 0; _j < (_n2); ++_j) std::cout << FMT(8) << (_A)[_idxA++] << ","; \
            std::cout << std::endl;                                                         \
        }                                                                                   \
        { std::cout << "])\n"; }                                                            \
    })
 
 
#endif  // KIDS_DEBUG_UTILS_H