PlotParameters.h

Go to the documentation of this file.

/* Copyright 2019 Kristofer Björnson
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef COM_DAFER45_TBTK_VISUALIZATION_MAT_PLOT_LIB_PLOT_PARAMETERS
#define COM_DAFER45_TBTK_VISUALIZATION_MAT_PLOT_LIB_PLOT_PARAMETERS

#include "TBTK/Visualization/MatPlotLib/matplotlibcpp.h"

#include <string>
#include <utility>

namespace TBTK{
namespace Visualization{
namespace MatPlotLib{

class PlotParameters{
public:
    PlotParameters();

    void setTitle(const std::string &title, bool overwrite = true);

    void setLabelX(const std::string &labelX, bool overwrite = true);

    void setLabelY(const std::string &labelY, bool overwrite = true);

    void setBoundsX(double minX, double maxX, bool overwrite = true);

    void setBoundsY(double minY, double maxY, bool overwrite = true);

    void flush() const;

    void clear();
private:
    std::pair<bool, std::string> title;
    std::pair<bool, std::string> labelX;
    std::pair<bool, std::string> labelY;
    std::pair<bool, double> minX;
    std::pair<bool, double> maxX;
    std::pair<bool, double> minY;
    std::pair<bool, double> maxY;
};

inline PlotParameters::PlotParameters(
) :
    title(false, ""),
    labelX(false, ""),
    labelY(false, ""),
    minX(false, 0),
    maxX(false, 1),
    minY(false, 0),
    maxY(false, 1)
{
}

inline void PlotParameters::setTitle(const std::string &title, bool overwrite){
    if(overwrite || !this->title.first)
        this->title = {true, title};
}

inline void PlotParameters::setLabelX(
    const std::string &labelX,
    bool overwrite
){
    if(overwrite || !this->labelX.first)
        this->labelX = {true, labelX};
}

inline void PlotParameters::setLabelY(
    const std::string &labelY,
    bool overwrite
){
    if(overwrite || !this->labelY.first)
        this->labelY = {true, labelY};
}

inline void PlotParameters::setBoundsX(
    double minX,
    double maxX,
    bool overwrite
){
    if(overwrite || !this->minX.first)
        this->minX = {true, minX};
    if(overwrite || !this->maxX.first)
        this->maxX = {true, maxX};
}

inline void PlotParameters::setBoundsY(
    double minY,
    double maxY,
    bool overwrite
){
    if(overwrite || !this->minY.first)
        this->minY = {true, minY};
    if(overwrite || !this->maxY.first)
        this->maxY = {true, maxY};
}

inline void PlotParameters::flush() const{
    if(title.first)
        matplotlibcpp::title(title.second);
    if(labelX.first)
        matplotlibcpp::xlabel(labelX.second);
    if(labelY.first)
        matplotlibcpp::ylabel(labelY.second);
    if(minX.first)
        matplotlibcpp::xlim(minX.second, maxX.second);
    if(minY.first)
        matplotlibcpp::ylim(minY.second, maxY.second);
}

inline void PlotParameters::clear(){
    title = {false, ""};
    labelX = {false, ""};
    labelY = {false, ""};
    minX = {false, 0};
    maxX = {false, 1};
    minY = {false, 0};
    maxY = {false, 1};
}

};  //End namespace MatPlotLib
};  //End namespace Visualization
};  //End namespace TBTK

#endif