Plot.hpp

/*
 * Copyright (C) Tobias Löw (tobi.loew@protonmail.ch)
 *
 * This file is part of Scatter
 *
 * Scatter is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Scatter is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Scatter.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#pragma once
 
#include <functional>
#include <memory>
#include <scatter/Figure.hpp>
#include <scatter/PlotOptions.hpp>
#include <string>
 
namespace scatter
{
    // forward declarations
    class Renderer;
    class Transform;
    class Figure;
    class Point;
    class Ellipse;
    class Arrow;
    class PlotBase;
    class Axis;
    class Legend;
 
    class Plot : public std::enable_shared_from_this<Plot>
    {
      public:
        Plot() = delete;
 
        virtual ~Plot();
 
        bool save(const std::string &file, const FigureOptions &options = FigureOptions());
 
        PlotOptions &options();
 
        template <class T, class... Args>
        typename T::Options *add(Args &&...args)
        {
            plots_.push_back(std::make_unique<T>(args...));
 
            return dynamic_cast<T *>(plots_.back().get())->options();
        }
 
      private:
        friend void Figure::save(const std::string &);
 
        Plot(const std::string &title, const PlotOptions &options);
 
        void render(Renderer &renderer, const Point &origin, const double &width, const double &height);
 
      private:
        std::vector<std::unique_ptr<PlotBase>> plots_;
 
        std::unique_ptr<Axis> axis_;
 
        std::unique_ptr<Legend> legend_;
 
        PlotOptions options_;
 
      public:
        typedef std::shared_ptr<Plot> Ptr;
 
        static Plot::Ptr create(const PlotOptions &options = PlotOptions());
 
        static Plot::Ptr create(const std::string &title, const PlotOptions &options = PlotOptions());
    };
 
}  // namespace scatter