/*
 * Copyright (c) 2009 Sean C. Rhea (srhea@srhea.net)
 *               2011 Mark Liversedge (liversedge@gmail.com)
 *
 * 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 "HistogramWindow.h"
#include "MainWindow.h"
#include "PowerHist.h"
#include "RideFile.h"
#include "RideFileCache.h"
#include "RideItem.h"
#include "Settings.h"
#include <QtGui>
#include <assert.h>

#include "Zones.h"
#include "HrZones.h"

HistogramWindow::HistogramWindow(MainWindow *mainWindow, bool rangemode) : GcWindow(mainWindow), mainWindow(mainWindow), stale(true), source(NULL), rangemode(rangemode), useCustom(false), useToToday(false)
{
    setInstanceName("Histogram Window");

    QWidget *c = new QWidget;
    QFormLayout *cl = new QFormLayout(c);
    cl->setFieldGrowthPolicy(QFormLayout::ExpandingFieldsGrow);
    setControls(c);


    // Main layout
    QGridLayout *mainLayout = new QGridLayout(this);
    mainLayout->setContentsMargins(2,2,2,2);

    //
    // reveal controls widget
    //

    // reveal widget
    revealBackground = new QWidget(this);
    revealBackground->setStyleSheet("background-color: rgba(100%, 100%, 100%, 100%)");
    revealControls = new QWidget(this);
    revealControls->setFixedHeight(50);
    revealControls->setStyleSheet("background-color: rgba(100%, 100%, 100%, 0%)");
    revealControls->setSizePolicy(QSizePolicy::Expanding, QSizePolicy::Minimum);

    revealAnim = new QPropertyAnimation(revealControls, "pos");
    revealAnim->setDuration(200);
    revealAnim->setEasingCurve(QEasingCurve(QEasingCurve::InSine));
    revealAnim->setKeyValueAt(0,QPoint(2,-20));
    revealAnim->setKeyValueAt(0.5,QPoint(2,15));
    revealAnim->setKeyValueAt(1,QPoint(2,20));

    revealBgAnim = new QPropertyAnimation(revealBackground, "geometry");
    revealBgAnim->setEasingCurve(QEasingCurve(QEasingCurve::InSine));
    revealBgAnim->setDuration(200);

    groupAnim = new QParallelAnimationGroup();
    groupAnim->addAnimation(revealAnim);
    groupAnim->addAnimation(revealBgAnim);

    // reveal controls
    rWidth = new QLabel(tr("Bin Width"), revealControls);
    rBinEdit = new QLineEdit(revealControls);
    rBinEdit->setFixedWidth(30);
    rBinSlider = new QSlider(Qt::Horizontal, revealControls);
    rBinSlider->setTickPosition(QSlider::TicksBelow);
    rBinSlider->setTickInterval(10);
    rBinSlider->setMinimum(1);
    rBinSlider->setMaximum(100);
    rShade = new QCheckBox(tr("Shade zones"), revealControls);
    rZones = new QCheckBox(tr("Show in zones"), revealControls);

    // layout reveal controls
    QHBoxLayout *r = new QHBoxLayout;
    r->setContentsMargins(0,0,0,0);
    r->addStretch();
    r->addWidget(rWidth);
    r->addWidget(rBinEdit);
    r->addWidget(rBinSlider);
    QVBoxLayout *v = new QVBoxLayout;
    v->addWidget(rShade);
    v->addWidget(rZones);
    r->addSpacing(20);
    r->addLayout(v);
    r->addStretch();
    revealControls->setLayout(r);

    // hide them initially
    revealControls->hide();

    // plot
    QVBoxLayout *vlayout = new QVBoxLayout;
    vlayout->setSpacing(10);
    powerHist = new PowerHist(mainWindow);
    vlayout->addWidget(powerHist);

    mainLayout->addLayout(vlayout,0,0);
    mainLayout->addWidget(revealBackground,0,0, Qt::AlignTop);
    mainLayout->addWidget(revealControls,0,0, Qt::AlignTop);
    revealBackground->raise();
    revealControls->raise();
    setLayout(mainLayout);

#ifdef GC_HAVE_LUCENE
    // search filter box
    isFiltered = false;
    searchBox = new SearchFilterBox(this, mainWindow);
    connect(searchBox, SIGNAL(searchClear()), this, SLOT(clearFilter()));
    connect(searchBox, SIGNAL(searchResults(QStringList)), this, SLOT(setFilter(QStringList)));
    cl->addRow(new QLabel(tr("Filter")), searchBox);
#endif

    // spacing if we have a range
    if (rangemode) cl->addWidget(new QLabel(""));

    // date selection
    dateSetting = new DateSettingsEdit(this);
    if (rangemode) cl->addRow(new QLabel(tr("Date Range")), dateSetting);

    // data series
    cl->addWidget(new QLabel("")); // spacing
    seriesCombo = new QComboBox();
    addSeries();
    cl->addRow(new QLabel(tr("Data Series")), seriesCombo);

    // bin width
    QHBoxLayout *binWidthLayout = new QHBoxLayout;
    QLabel *binWidthLabel = new QLabel(tr("Bin width"), this);
    binWidthLineEdit = new QLineEdit(this);
    binWidthLineEdit->setFixedWidth(30);

    binWidthLayout->addWidget(binWidthLineEdit);
    binWidthSlider = new QSlider(Qt::Horizontal);
    binWidthSlider->setTickPosition(QSlider::TicksBelow);
    binWidthSlider->setTickInterval(1);
    binWidthSlider->setMinimum(1);
    binWidthSlider->setMaximum(100);
    binWidthLayout->addWidget(binWidthSlider);
    cl->addRow(binWidthLabel, binWidthLayout);

    showSumY = new QComboBox();
    showSumY->addItem(tr("Absolute Time"));
    showSumY->addItem(tr("Percentage Time"));

    cl->addWidget(new QLabel("")); // spacing
    cl->addRow(new QLabel("Show"), showSumY);

    showLnY = new QCheckBox;
    showLnY->setText(tr("Log Y"));
    cl->addRow(new QLabel(""), showLnY);

    showZeroes = new QCheckBox;
    showZeroes->setText(tr("With zeros"));
    cl->addRow(new QLabel(""), showZeroes);

    shadeZones = new QCheckBox;
    shadeZones->setText(tr("Shade zones"));
    cl->addRow(new QLabel(""), shadeZones);

    showInZones = new QCheckBox;
    showInZones->setText(tr("Show in zones"));
    cl->addRow(new QLabel(""), showInZones);


    // sort out default values
    setHistTextValidator();
    showLnY->setChecked(powerHist->islnY());
    showZeroes->setChecked(powerHist->withZeros());
    shadeZones->setChecked(powerHist->shade);
    binWidthSlider->setValue(powerHist->binWidth());
    rBinSlider->setValue(powerHist->binWidth());
    rShade->setChecked(powerHist->shade);
    setHistBinWidthText();

    // set the defaults etc
    updateChart();

    // the bin slider/input update each other
    // only the input box triggers an update to the chart
    connect(binWidthSlider, SIGNAL(valueChanged(int)), this, SLOT(setBinWidthFromSlider()));
    connect(binWidthLineEdit, SIGNAL(editingFinished()), this, SLOT(setBinWidthFromLineEdit()));
    connect(rBinSlider, SIGNAL(valueChanged(int)), this, SLOT(setrBinWidthFromSlider()));
    connect(rBinEdit, SIGNAL(editingFinished()), this, SLOT(setrBinWidthFromLineEdit()));
    connect(rZones, SIGNAL(stateChanged(int)), this, SLOT(setZoned(int)));
    connect(rShade, SIGNAL(stateChanged(int)), this, SLOT(setShade(int)));

    // when season changes we need to retrieve data from the cache then update the chart
    if (rangemode) {
        connect(this, SIGNAL(dateRangeChanged(DateRange)), this, SLOT(dateRangeChanged(DateRange)));
    } else {
        dateSetting->hide();
        connect(this, SIGNAL(rideItemChanged(RideItem*)), this, SLOT(rideSelected()));
        connect(mainWindow, SIGNAL(intervalSelected()), this, SLOT(intervalSelected()));
    }

    // if any of the controls change we pass the chart everything
    connect(showLnY, SIGNAL(stateChanged(int)), this, SLOT(updateChart()));
    connect(showZeroes, SIGNAL(stateChanged(int)), this, SLOT(updateChart()));
    connect(seriesCombo, SIGNAL(currentIndexChanged(int)), this, SLOT(updateChart()));
    connect(showInZones, SIGNAL(stateChanged(int)), this, SLOT(setZoned(int)));
    connect(showInZones, SIGNAL(stateChanged(int)), this, SLOT(updateChart()));
    connect(shadeZones, SIGNAL(stateChanged(int)), this, SLOT(setShade(int)));
    connect(shadeZones, SIGNAL(stateChanged(int)), this, SLOT(updateChart()));
    connect(showSumY, SIGNAL(currentIndexChanged(int)), this, SLOT(updateChart()));

    connect(mainWindow, SIGNAL(zonesChanged()), this, SLOT(zonesChanged()));
    connect(mainWindow, SIGNAL(configChanged()), powerHist, SLOT(configChanged()));

    connect(mainWindow, SIGNAL(rideAdded(RideItem*)), this, SLOT(rideAddorRemove(RideItem*)));
    connect(mainWindow, SIGNAL(rideDeleted(RideItem*)), this, SLOT(rideAddorRemove(RideItem*)));
}

void
HistogramWindow::rideSelected()
{
    if (!amVisible()) return;

    RideItem *ride = myRideItem;

    if (!ride || (rangemode && !stale)) return;

    if (rangemode) {
        // get range that applies to this ride
        powerRange = mainWindow->zones()->whichRange(ride->dateTime.date());
        hrRange = mainWindow->hrZones()->whichRange(ride->dateTime.date());
    }

    // update
    updateChart();
}

void
HistogramWindow::rideAddorRemove(RideItem *)
{
    stale = true;
}

void
HistogramWindow::intervalSelected()
{
    if (!amVisible()) return;

    RideItem *ride = myRideItem;

    // null? or not plotting current ride, ignore signal
    if (!ride || rangemode) return;

    // update
    interval = true;
    updateChart();
}

void
HistogramWindow::zonesChanged()
{
    if (!amVisible()) return;

    powerHist->refreshZoneLabels();
    powerHist->replot();
}

void
HistogramWindow::useCustomRange(DateRange range)
{
    // plot using the supplied range
    useCustom = true;
    useToToday = false;
    custom = range;
    dateRangeChanged(custom);
}

void
HistogramWindow::useStandardRange()
{
    useToToday= useCustom = false;
    dateRangeChanged(myDateRange);
}

void
HistogramWindow::useThruToday()
{
    // plot using the supplied range
    useCustom = false;
    useToToday = true;
    custom = myDateRange;
    if (custom.to > QDate::currentDate()) custom.to = QDate::currentDate();
    dateRangeChanged(custom);
}

void HistogramWindow::dateRangeChanged(DateRange dateRange)
{
    // has it changed?
    if (dateRange.from != cfrom || dateRange.to != cto) 
        stale = true;

    if (!amVisible() || !stale) return;

    updateChart();
}

void HistogramWindow::addSeries()
{
    // setup series list
    seriesList << RideFile::watts
               << RideFile::wattsKg
               << RideFile::hr
               << RideFile::kph
               << RideFile::cad
               << RideFile::nm;

    foreach (RideFile::SeriesType x, seriesList) 
        seriesCombo->addItem(RideFile::seriesName(x), static_cast<int>(x));
}

void
HistogramWindow::setBinWidthFromSlider()
{
    if (powerHist->binWidth() != binWidthSlider->value()) {
        //RideFile::SeriesType series = static_cast<RideFile::SeriesType>(seriesCombo->itemData(seriesCombo->currentIndex()).toInt());
        powerHist->setBinWidth(binWidthSlider->value());
        setHistBinWidthText();
        rBinEdit->setText(binWidthLineEdit->text());
        rBinSlider->setValue(binWidthSlider->value());
        updateChart();
    }
}

void
HistogramWindow::setrBinWidthFromSlider()
{
    if (powerHist->binWidth() != rBinSlider->value()) {
        powerHist->setBinWidth(rBinSlider->value());
        setHistBinWidthText();
        rBinEdit->setText(binWidthLineEdit->text());
        updateChart();
    }
}

void
HistogramWindow::setHistBinWidthText()
{
    binWidthLineEdit->setText(QString("%1").arg(powerHist->getBinWidthRealUnits(), 0, 'g', 3));
    rBinEdit->setText(QString("%1").arg(powerHist->getBinWidthRealUnits(), 0, 'g', 3));

}

void
HistogramWindow::setHistTextValidator()
{
    double delta = powerHist->getDelta();
    int digits = powerHist->getDigits();

    QValidator *validator;
    if (digits == 0) {

        validator = new QIntValidator(binWidthSlider->minimum() * delta,
                                      binWidthSlider->maximum() * delta,
                                      binWidthLineEdit);
    } else {

        validator = new QDoubleValidator(binWidthSlider->minimum() * delta,
                                         binWidthSlider->maximum() * delta,
                                         digits,
                                         binWidthLineEdit);
    }
    binWidthLineEdit->setValidator(validator);
    rBinEdit->setValidator(validator);
}

void
HistogramWindow::setBinWidthFromLineEdit()
{
    double value = binWidthLineEdit->text().toDouble();
    if (value != powerHist->binWidth()) {
        binWidthSlider->setValue(powerHist->setBinWidthRealUnits(value));
        rBinSlider->setValue(powerHist->setBinWidthRealUnits(value));
        setHistBinWidthText();

        updateChart();
    }
}

void
HistogramWindow::setrBinWidthFromLineEdit()
{
    double value = rBinEdit->text().toDouble();
    if (value != powerHist->binWidth()) {
        rBinSlider->setValue(powerHist->setBinWidthRealUnits(value));
        binWidthSlider->setValue(powerHist->setBinWidthRealUnits(value));
        updateChart();
    }
}

void
HistogramWindow::setZoned(int x)
{
    rZones->setCheckState((Qt::CheckState)x);
    showInZones->setCheckState((Qt::CheckState)x);
    
}

void
HistogramWindow::setShade(int x)
{
    rShade->setCheckState((Qt::CheckState)x);
    shadeZones->setCheckState((Qt::CheckState)x);
}

void
HistogramWindow::resizeEvent(QResizeEvent *)
{
    revealBgAnim->setKeyValueAt(0, QRect(2, contentsMargins().top(), width()-4, 0));
    revealBgAnim->setKeyValueAt(0.5, QRect(2, contentsMargins().top(), width()-4, 45));
    revealBgAnim->setKeyValueAt(1, QRect(2, contentsMargins().top(), width()-4, 50));
}

void
HistogramWindow::updateChart()
{
    // refresh the ridefile cache if it is stale
    if (stale) {

        RideFileCache *old = source;

        if (rangemode) {

            DateRange use;
            if (useCustom) {

                use = custom;

            } else if (useToToday) {

                use = myDateRange;
                QDate today = QDate::currentDate();
                if (use.to > today) use.to = today;

            } else {

                use = myDateRange;
            }

#ifdef GC_HAVE_LUCENE
            source = new RideFileCache(mainWindow, use.from, use.to, isFiltered, files);
#else
            source = new RideFileCache(mainWindow, use.from, use.to);
#endif
            cfrom = use.from;
            cto = use.to;
            stale = false;

            if (old) delete old; // guarantee source pointer changes
        }
        stale = false; // well we tried
    }

    // set data
    if (rangemode && source) {
        powerHist->setData(source);
    } else 
        powerHist->setData(myRideItem, interval); // intervals selected forces data to
                                                  // be recomputed since interval selection
                                                  // has changed.


    // and now the controls
    powerHist->setShading(shadeZones->isChecked() ? true : false);
    powerHist->setZoned(showInZones->isChecked() ? true : false);
    powerHist->setlnY(showLnY->isChecked() ? true : false);
    powerHist->setWithZeros(showZeroes->isChecked() ? true : false);
    powerHist->setSumY(showSumY->currentIndex()== 0 ? true : false);

    // and which series to plot
    powerHist->setSeries(static_cast<RideFile::SeriesType>(seriesCombo->itemData(seriesCombo->currentIndex()).toInt()));

    // Correct binWidth if not valid for the selected series
    if (binWidthLineEdit->text().toDouble()<powerHist->getDelta())
        binWidthSlider->setValue(powerHist->getDelta());
    else
        powerHist->setBinWidth(binWidthLineEdit->text().toDouble());

    // now go plot yourself
    //powerHist->setAxisTitle(int axis, QString label);
    powerHist->recalc(interval); // interval changed? force recalc
    powerHist->replot();

    interval = false;// we force a recalc whem called coz intervals
                     // have been selected. The recalc routine in
                     // powerhist optimises out, but doesn't keep track
                     // of interval selection -- simplifies the setters
                     // and getters, so worth this 'hack'.
}

#ifdef GC_HAVE_LUCENE
void 
HistogramWindow::clearFilter()
{
    isFiltered = false;
    files.clear();
    stale = true;
    updateChart();
}

void
HistogramWindow::setFilter(QStringList list)
{
    isFiltered = true;
    files = list;
    stale = true;
    updateChart();
}
#endif