ascii_plots.hpp

Go to the documentation of this file.

#ifndef ASCII_PLOTS_HPP
#define ASCII_PLOTS_HPP

#include "rz_bounds.hpp"

template<typename F>
void ascii_contour_plot(const RZBounds& bounds, F func){
    int nx = 80;
    int ny = 40;
    double dr = bounds.get_r_width()/nx;
    double dz = bounds.get_z_width()/ny;
    printf("\n");
    std::vector<int> prevrow(nx+1,-1);
    for (int j=ny;j>=0;j--){
        printf("\n");
        int prevnum = -1;
        for (int i=0;i<=nx;i++){
            double r = bounds.min_r + i*dr;
            double z = bounds.min_z + j*dz;

            // Get value of function
            double val = func(r, z);

            // Assume the user has normalized the function to (0.0,10.0)
            int num = val;

            num += 48; // Shift so val=0.0 results in char 0

            // Allow values outside from 33 (!) to 125 (})
            num = std::max(33,std::min(num,125));

            char mychar = num;
            if (prevnum!=num || prevrow[i]!=num) printf("%c",mychar);
            else printf(" ");
            prevnum = num;
            prevrow[i] = num;
        }
    }
    printf("\n");
    printf("xlims: [%1.1e, %1.1e]; ylims: [%1.1e, %1.1e]\n", bounds.min_r, bounds.max_r, bounds.min_z, bounds.max_z);
    printf("\n");
}

#endif