Decoration.h

/* Copyright 2017 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_PLOT_DECORATION
#define COM_DAFER45_TBTK_PLOT_DECORATION
 
#include <vector>
 
namespace TBTK{
namespace Plot{
 
class Decoration{
public:
    enum class LineStyle {Line, Point};
 
    Decoration();
 
    Decoration(
        const std::vector<unsigned char> &color,
        LineStyle lineStyle,
        unsigned int size = 1
    );
 
    ~Decoration();
 
    void setColor(const std::vector<unsigned char> &color);
 
    const std::vector<unsigned char>& getColor() const;
 
    const LineStyle& getLineStyle() const;
 
    unsigned int getSize() const;
private:
    std::vector<unsigned char> color;
 
    LineStyle lineStyle;
 
    unsigned int size;
};
 
inline Decoration::Decoration(){
    color = {0, 0, 0};
    lineStyle = LineStyle::Point;
    size = 1;
}
 
inline Decoration::Decoration(
    const std::vector<unsigned char> &color,
    LineStyle lineStyle,
    unsigned int size
){
    this->color = color;
    this->lineStyle = lineStyle;
    this->size = size;
}
 
inline Decoration::~Decoration(){
}
 
inline void Decoration::setColor(const std::vector<unsigned char> &color){
    TBTKAssert(
        color.size() == 3,
        "Decoration::setColor()",
        "Color must have three components but '" << color.size() << "'"
        << " components given.",
        ""
    );
    this->color = color;
}
 
inline const std::vector<unsigned char>& Decoration::getColor() const{
    return color;
}
 
inline const Decoration::LineStyle& Decoration::getLineStyle() const{
    return lineStyle;
}
 
inline unsigned int Decoration::getSize() const{
    return size;
}
 
};  //End namespace Plot
};  //End namespace TBTK
 
#endif