/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2015-     Equinor ASA
//  Copyright (C) 2015-     Ceetron Solutions AS
//
//  ResInsight 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.
//
//  ResInsight 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 at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#pragma once

#include "RiuPlotCurveSymbol.h"

#include <QColor>
#include <QImage>
#include <QPen>
#include <QString>

class QPainter;
class QPointF;
class QRect;

namespace QtCharts
{
class QScatterSeries;
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
class RiuQtChartsPlotCurveSymbol : public RiuPlotCurveSymbol
{
public:
    RiuQtChartsPlotCurveSymbol( RiuPlotCurveSymbol::PointSymbolEnum riuStyle,
                                const QString&                      label = QString(),
                                LabelPosition labelPosition               = RiuPlotCurveSymbol::LabelAboveSymbol,
                                int           labelFontSizePt             = 8 );

    void renderSymbolLabel( QPainter* painter, const QPointF& position, const QString& label ) const;

    void setSize( int width, int height ) override;

    void setColor( const QColor& color ) override;

    void setPen( const QPen& pen ) override;

    void setPixmap( const QPixmap& pixmap ) override;

    QRect boundingRect() const override;

    void applyToScatterSeries( QtCharts::QScatterSeries* series ) const;

    QImage image() const;

private:
    QImage createTriangleImage( RiuPlotCurveSymbol::PointSymbolEnum symbolStyle ) const;
    QImage createRectImage() const;
    QImage createStar1Image() const;
    QImage createStar2Image() const;
    QImage createHexagonImage() const;
    QImage createEllipseImage() const;
    QImage createCrossImage() const;
    QImage createXCrossImage() const;
    QImage createDiamondImage() const;

    void setImageBrush( QtCharts::QScatterSeries* series, const QImage& image ) const;

    QColor m_color;
    QPen   m_pen;
    int    m_size;
};