plot_2D.hpp

Go to the documentation of this file.

//====================================================
//     File data
//====================================================
//====================================================
//     Preprocessor settings
//====================================================
#pragma once
#ifndef OSMANIP_GRAPHICS_PLOT2DCANVAS_HPP
#define OSMANIP_GRAPHICS_PLOT2DCANVAS_HPP
 
//====================================================
//     Headers
//====================================================
 
// My headers
#include <osmanip/graphics/canvas.hpp>
 
// STD headers
#include <cmath>
#include <cstdint>
#include <functional>
#include <string>
 
namespace osm {
 
    //====================================================
    //     Plot2DCanvas
    //====================================================
    class Plot2DCanvas : public Canvas {
        public:
 
            // Constructors
            explicit Plot2DCanvas(uint32_t w, uint32_t h);
 
            // Setters
            void setOffset(float xOff, float yOff);
            void setScale(float xScale, float yScale);
 
            // Getters
            float getOffsetX() const;
            float getOffsetY() const;
            float getScaleX() const;
            float getScaleY() const;
 
            // Draw
            template <typename Y, typename X>
            inline void draw(std::function<Y(X)> function, char c, std::string_view feat = "") {  //
 
                // Find correct boundaries for the loop
                uint32_t x_min = std::max(0, static_cast<int>(std::floor((0 - offset_x_) / scale_x_)));
                uint32_t x_max = std::min(width_, static_cast<uint32_t>(std::ceil((width_ - offset_x_) / scale_x_)));
 
                // Loop over the computed boundaries and draw
                for (uint32_t x = x_min; x < x_max; x++) {
                    float real_x = offset_x_ + x * scale_x_;
                    Y real_y = function(real_x);
                    uint32_t y = static_cast<uint32_t>((real_y - offset_y_) / scale_y_);
                    if (y > 0 && y < height_) put(x, y, c, feat);
                }
            }
 
        private:
 
            // Members
            float offset_x_, offset_y_, scale_x_, scale_y_;
    };
}  // namespace osm
 
#endif