StratBar.cpp

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) OpenGeoSys Community (opengeosys.org)
// SPDX-License-Identifier: BSD-3-Clause
 
#include "StratBar.h"
 
#include <QPainter>
 
StratBar::StratBar(GeoLib::StationBorehole* station,
                   std::map<std::string, DataHolderLib::Color>* stratColors,
                   QGraphicsItem* parent)
    : QGraphicsItem(parent), _station(station)
{
    if (stratColors)
    {
        _stratColors = *stratColors;
    }
}
 
QRectF StratBar::boundingRect() const
{
    return QRectF(0, 0, BARWIDTH + 10, totalLogHeight());
}
 
void StratBar::paint(QPainter* painter,
                     const QStyleOptionGraphicsItem* option,
                     QWidget* widget)
{
    Q_UNUSED(option)
    Q_UNUSED(widget)
 
    double top = 0;
    double height = 0;
 
    QPen pen(Qt::black, 1, Qt::SolidLine, Qt::SquareCap, Qt::RoundJoin);
    pen.setCosmetic(true);
    painter->setPen(pen);
    // painter->drawRect(_bar);
 
    // pen.setWidth(1);
    std::vector<GeoLib::Point*> profile = _station->getProfile();
    std::vector<std::string> soilNames = _station->getSoilNames();
    std::size_t nLayers = profile.size();
 
    painter->drawLine(0, 0, BARWIDTH + 5, 0);
 
    for (std::size_t i = 1; i < nLayers; i++)
    {
        top += height;
        height = logHeight(((*(profile[i - 1]))[2] - (*(profile[i]))[2]));
        QRectF layer(0, top, BARWIDTH, height);
        DataHolderLib::Color const& c(
            DataHolderLib::getColor(soilNames[i], _stratColors));
        QBrush brush(QColor(static_cast<int>(c[0]),
                            static_cast<int>(c[1]),
                            static_cast<int>(c[2]),
                            127),
                     Qt::SolidPattern);
        painter->setBrush(brush);
 
        painter->drawRect(layer);
        painter->drawLine(0,
                          static_cast<int>(layer.bottom()),
                          BARWIDTH + 5,
                          static_cast<int>(layer.bottom()));
        // painter->drawText(BARWIDTH+10, layer.bottom(),
        // QString::number((*(profile[i]))[2]));
    }
}
 
double StratBar::totalLogHeight() const
{
    double height = 0;
    std::vector<GeoLib::Point*> profile = _station->getProfile();
 
    for (std::size_t i = 1; i < profile.size(); i++)
    {
        height += (log((*(profile[i - 1]))[2] - (*(profile[i]))[2] + 1) * 100);
    }
 
    return height;
}