ResInsight/ApplicationCode/UserInterface/RiuMohrsCirclePlot.h

/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2018     Statoil ASA
//
//  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 "qwt_plot.h"
#include "qwt_plot_item.h"

#include "cafTensor3.h"

#include <array>

class QWidget;
class QwtPlotCurve;
class QwtPlotRescaler;
class QwtPlotTextLabel;
class QwtRoundScaleDraw;
class RimGeoMechView;
class RiuSelectionItem;

//==================================================================================================
//
//
//
//==================================================================================================
class RiuMohrsCirclePlot : public QwtPlot
{
    Q_OBJECT

public:
    RiuMohrsCirclePlot(QWidget* parent);
    ~RiuMohrsCirclePlot();

    void setPrincipals(double p1, double p2, double p3);
    void setPrincipalsAndRedrawPlot(double p1, double p2, double p3);

    void updateOnSelectionChanged(const RiuSelectionItem* selectionItem);
    void clearPlot();

private:
    struct MohrCircle
    {
        MohrCircle(double radius, double centerX)
            : radius(radius), centerX(centerX) {}
        MohrCircle() {};
        double radius;
        double centerX;
    };

private:
    virtual QSize sizeHint() const override;
    virtual QSize minimumSizeHint() const override;

    void redrawCircles();
    void deleteCircles();

    void redrawEnvelope();
    void deleteEnvelope();

    void addInfoLabel();
    void deleteInfoLabel();

    void queryDataAndUpdatePlot(RimGeoMechView* geoMechView, size_t gridIndex, size_t cellIndex);
    
    void setDefaults();
    void createMohrCircles();

    void setFrictionAngle(double frictionAngle);
    void setCohesion(double cohesion);
    void setFactorOfSafety(double fos);

    void updateTransparentCurveOnPrincipals();

    void replotAndScaleAxis();

    static bool isValidPrincipals(double p1, double p2, double p3);
    
    float calculateFOS(const caf::Ten3f& tensor);

private:
    double m_principal1;
    double m_principal2;
    double m_principal3;

    std::array<MohrCircle, 3> m_mohrCircles;

    std::vector<QwtPlotItem*> m_circlePlotItems;
    
    QwtPlotCurve* m_envolopePlotItem;
    QwtPlotCurve* m_transparentCurve;

    double m_frictionAngle;
    double m_cohesion;
    double m_factorOfSafety;

    QwtPlotTextLabel* m_infoTextItem;

    QwtPlotRescaler* m_rescaler;
};