TYHistoWidget.cpp

Go to the documentation of this file.

/*
 * Copyright (C) <2012> <EDF-R&D> <FRANCE>
 * This program 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 2 of the License, or
 * (at your option) any later version.
 * This program 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 for more details.
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
 
#include "Tympan/models/business/acoustic/TYSpectre.h"
#include "TYHistoWidget.h"
 
#include <math.h>
#include "TYSpectreWidget.h"
#include <QPainter>
#include "Tympan/models/business/OLocalizator.h"
#include "TYPrintDialog.h"
#include <QtPrintSupport/QPrinter>
#include <QtPrintSupport/QPrintDialog>
 
#define TR(id) OLocalizator::getString("TYSpectreWidget", (id))
 
using namespace Qt;
 
TYHistoWidget::TYHistoWidget(TYSpectreWidget* pSpectreWidget, bool dbType, QWidget* _pParent /*=NULL*/)
    : QWidget(_pParent)
{
    _pSpectreWidget = pSpectreWidget;
    _dbType = dbType;
}
 
TYHistoWidget::~TYHistoWidget() {}
 
void TYHistoWidget::print()
{
    TYPrintDialog* pDialog = new TYPrintDialog(this);
    pDialog->_groupBoxProjet->setEnabled(false);
    pDialog->_groupBoxCalcul->setEnabled(false);
    QPrinter* printer = new QPrinter();
    if (pDialog->exec() == QDialog::Accepted)
    {
        QPrintDialog dialog(printer, this);
        if (dialog.exec())
        {
            QPainter paint(printer);
            QPixmap tmpPix;
 
            int mid = int(printer->width() / 2) -
                      int(paint.fontMetrics().horizontalAdvance(pDialog->_lineEditTete->text()) / 2);
            int midX = int(printer->width() / 2);
            int midY = int(printer->height() / 2);
            paint.drawText(mid, 20, pDialog->_lineEditTete->text());
            mid = int(printer->width() / 2) -
                  int(paint.fontMetrics().horizontalAdvance(pDialog->_lineEditPied->text()) / 2);
            paint.drawText(mid, printer->height() - 20, pDialog->_lineEditPied->text());
            paint.setFont(QFont("Helvetica", 15, QFont::Bold));
            mid = int(printer->width() / 2) -
                  int(paint.fontMetrics().horizontalAdvance(pDialog->_lineEditTitre->text()) / 2);
            paint.drawText(mid, printer->height() * 14 / 15, pDialog->_lineEditTitre->text());
            paint.setFont(QFont());
 
            double Wmetric = printer->width() * 9 / 10;
            double Hmetric = printer->height() * 8 / 9;
            double x = width();
            double y = height();
            double w = Wmetric;
            double h = Wmetric * (y / x);
 
            if (h > Hmetric)
            {
                h = Hmetric;
                w = Hmetric * (x / y);
            }
            paint.translate(midX - (w / 2), midY - (h / 2));
            paintInto(&paint, w, h);
        }
    }
    delete printer;
}
 
void TYHistoWidget::paintEvent(QPaintEvent* event)
{
    int w = 0, h = 0;
    QPainter* p = new QPainter(this);
    w = width();
    h = height();
 
    paintInto(p, w, h);
 
    delete p;
}
 
void TYHistoWidget::paintInto(QPainter* painter, int w, int h)
{
    LPTYSpectre pSpectre = new TYSpectre();
    _pSpectreWidget->spectre(pSpectre);
    TYTabFreq tabFreq = TYSpectre::getTabFreqNorm(pSpectre->getForm());
 
    size_t nbFreq = tabFreq.size();
 
    // Initialisation des parametres
    double width = double(w);
    double height = double(h);
    double value[50], hauteur = NAN, L = width * 2 / 3, hzero = NAN;
    double hmax = 0, largeur = NAN, x = NAN;
    size_t i = 0;
    int val = -1;
    double hmin = 0;
    QString num;
    double textPos = NAN;
    double midFreq = double(nbFreq) / 2;
    bool valid = false;
 
    for (i = 0; i < nbFreq; i++)
    {
        value[i] = pSpectre->getTabValReel()[i];
 
        if (value[i] != TY_SPECTRE_DEFAULT_VALUE)
        {
            valid = true;
        }
 
        if (value[i] >= 0)
        {
            if (value[i] > hmax)
            {
                hmax = value[i];
            }
        }
        else if (value[i] < hmin)
        {
            hmin = value[i];
        }
    }
 
    double echelle = height * 7 / 10 / (-hmin + hmax);
    hzero = height * 8.5 / 10 + hmin * echelle;
 
    if ((hmin == 0) && (hmax == 0))
    {
        echelle = 1;
        hzero = height / 2;
    }
 
    L = width / 2;
    x = width * 1 / 12;
    largeur = L / (nbFreq + 1);
    val = -1;
 
    // Histogramme
    painter->setPen(blue);
 
    if (valid)
    {
        double valMax = 0.0;
 
        for (i = 0; i < nbFreq; i++)
        {
            hauteur = value[i] * echelle;
            painter->fillRect((int)x, (int)(hzero - hauteur), (int)largeur, (int)hauteur,
                              QColor(100, 255, 110));
            painter->drawRect((int)x, (int)hzero - hauteur, (int)largeur, (int)hauteur);
 
            if (val != ROUND(value[i]))
            {
                val = ROUND(value[i]);
                valMax = MAX(val, valMax);
            }
 
            if ((i == int(midFreq)) || (i == (nbFreq - 1)))
            {
                if (hauteur < 0)
                {
                    textPos = hzero - 5;
                }
                else
                {
                    textPos = hzero + 17;
                }
                painter->drawLine((int)x, (int)(hzero + 5), (int)x, (int)(hzero - 5));
                painter->drawText((int)(x - 10), (int)textPos, num.setNum(tabFreq[i]));
            }
 
            x += largeur;
        }
 
        // Val Max
        painter->setPen(QPen(blue, 2));
        painter->drawLine((int)(width * 1 / 12 - 5), (int)(hzero - valMax * echelle),
                          (int)(width * 1 / 12 + 5), (int)(hzero - valMax * echelle));
        painter->drawText((int)(width * 1 / 12 - 25), (int)(hzero - valMax * echelle), num.setNum(valMax));
        // Val Max / 2
        painter->drawLine((int)(width * 1 / 12 - 5), (int)(hzero - (valMax / 2) * echelle),
                          (int)(width * 1 / 12 + 5), (int)(hzero - (valMax / 2) * echelle));
        painter->drawText((int)(width * 1 / 12 - 25), (int)(hzero - (valMax / 2) * echelle),
                          num.setNum((valMax / 2)));
 
        painter->setPen(QPen(black, 2));
        painter->drawLine((int)(width * 1 / 12 - 2), (int)hzero, (int)(width * 11 / 12), (int)hzero);
        painter->drawLine((int)(width * 1 / 12), (int)(height * 18 / 20), (int)(width * 1 / 12),
                          (int)(hzero - hmax * echelle - 10));
        painter->drawText((int)(width * 1 / 12), (int)(hzero - hmax * echelle - 20), "dB");
        painter->drawText((int)(x + 2 * largeur), (int)(height * 18 / 20), TR("id_abscisse"));
 
        x += width / 3 - 4 * largeur;
 
        // Reglage de l'echelle pour dBLin
        double value = pSpectre->valGlobDBLin();
        if (value < 0)
        {
            hmin = value;
            hmax = 0;
        }
        else
        {
            hmax = value;
            hmin = 0;
        }
        echelle = height * 7 / 10 / (-hmin + hmax);
        hzero = height * 8.5 / 10 + hmin * echelle;
        hauteur = value * echelle;
 
        // Construction d'un nouvel axe pour dBLin
        painter->drawLine((int)x, (int)(height * 18 / 20), (int)x, (int)(hzero - hmax * echelle - 10));
        painter->drawLine((int)(x - 5), (int)(hzero - value * echelle), (int)(x + 5),
                          (int)(hzero - value * echelle));
        painter->drawText((int)(x - 15), (int)(hzero - hmax * echelle - 20), num.setNum(value, 'f', 2));
 
        // Dessin de dbLin
        painter->fillRect((int)x, (int)(hzero - hauteur), (int)largeur, (int)hauteur, QColor(100, 255, 110));
        painter->setPen(blue);
        painter->drawRect((int)x, (int)(hzero - hauteur), (int)largeur, (int)hauteur);
        painter->setPen(black);
        painter->drawText((int)(x - 20), (int)(height * 19 / 20), TR("id_spectre_histo_dbZ"));
 
        x += 3 * largeur;
 
        // Reglage de l'echelle pour dBA
        value = pSpectre->valGlobDBA();
        if (value < 0)
        {
            hmin = value;
            hmax = 0;
        }
        else
        {
            hmax = value;
            hmin = 0;
        }
        echelle = height * 7 / 10 / (-hmin + hmax);
        hzero = height * 8.5 / 10 + hmin * echelle;
        hauteur = value * echelle;
 
        painter->setPen(QPen(black, 2));
        // Construction d'un nouvel axe pour dBA
        painter->drawLine((int)x, (int)(height * 18 / 20), (int)x, (int)(hzero - hmax * echelle - 10));
        painter->drawLine((int)(x - 5), (int)(hzero - value * echelle), (int)(x + 5),
                          (int)(hzero - value * echelle));
        painter->drawText((int)x, (int)(hzero - hmax * echelle - 20), num.setNum(value, 'f', 2));
 
        // Dessin de dBA
        painter->fillRect((int)x, (int)(hzero - hauteur), (int)largeur, (int)hauteur, QColor(100, 255, 110));
        painter->setPen(blue);
        painter->drawRect((int)x, (int)(hzero - hauteur), (int)largeur, (int)hauteur);
        painter->setPen(black);
        painter->drawText(x, height * 19 / 20, TR("id_spectre_histo_dbA"));
    }
    else
    {
        painter->setPen(QPen(red, 3));
        painter->drawText((int)(width / 10), (int)(height / 10), TR("id_invalid_spectre"));
    }
}