/*  YUView - YUV player with advanced analytics toolset
*   Copyright (C) 2015  Institut für Nachrichtentechnik
*                       RWTH Aachen University, GERMANY
*
*   YUView 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.
*
*   YUView 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 YUView.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef STATISTICSMODEL_H
#define STATISTICSMODEL_H

#include <assert.h>
#include <string>
#include "displayobject.h"
#include <tikzfile.h>
#include "statisticsextensions.h"

#include <QMap>
#include <QHash>
#include <QFuture>

typedef QList<StatisticsItem> StatisticsItemList;
typedef QVector<StatisticsType> StatisticsTypeList;

class StatisticsObject : public DisplayObject
{
public:
    StatisticsObject(const QString& srcFileName, QObject* parent = 0);
    ~StatisticsObject();

    QString path() {return p_srcFilePath;}
    QString createdtime() {return p_createdTime;}
    QString modifiedtime() {return p_modifiedTime;}

    void loadImage(int idx);

    void saveFrame(int idx, QRect widget, QRect image);

    ValuePairList getValuesAt(int x, int y);

    void setInternalScaleFactor(int internalScaleFactor);

    StatisticsType* getStatisticsType(int typeID);

    void setStatisticsTypeList(StatisticsTypeList typeList);
    StatisticsTypeList getStatisticsTypeList() { return p_statsTypeList; }
    TikZFile* getTikzFile(){return p_tikzFile;}

    int numFrames() { return p_numberFrames; }
    int nrBytes() { return p_numBytes; }
    QString status() { return p_status; }

private:
    //! Scan the header: What types are saved in this file?
    void readHeaderFromFile();
    //! Parser the whole file and get the positions where a new POC/type starts. Save this position in p_pocTypeStartList.
    //! This is performed in the background
    void readFrameAndTypePositionsFromFile();
    //! Load the statistics with frameIdx/type from file and put it into the cache.
    //! If the statistics file is in an interleaved format (types are mixed within one POC) this function also parses
    //! types which were not requested by the given 'type'.
    void readStatisticsFromFile(int frameIdx, int type);

    //! Get statistics. Try cache first, or load (using readStatisticsFromFile())
    StatisticsItemList getStatistics(int frameIdx, int type);

    void drawStatisticsImage(int frameIdx);
    void drawStatisticsImage(StatisticsItemList statsList, StatisticsType statsType);

    void tikZStatisticsImage(int frameIdx, QRect widget, QRect image);
    StatisticsTikzDrawItemList tikZStatisticsImage(StatisticsItemList statsList, StatisticsType statsType);

    //! Error while parsing. Set the error message that will be returned by status(). Also set p_numberFrames to 0, clear p_pocStartList.
    void setErrorState(QString sError);

    QStringList parseCSVLine(QString line, char delimiter);

    QHash< int,QHash< int,StatisticsItemList > > p_statsCache; // 2D map of type StatisticsItemList with indexing: [POC][statsTypeID]
    StatisticsTypeList p_statsTypeList;

    QFuture<void> p_backgroundParserFuture;
    bool p_cancelBackgroundParser;

    QMap<int,QMap<int,qint64> > p_pocTypeStartList;

    QString p_srcFilePath;
    QString p_createdTime;
    QString p_modifiedTime;
    int     p_numBytes;

    int p_zoomFactor;

    int p_numberFrames;
    QString p_status; // Contains the status as text
    // Set if the file is sorted by POC and the types are 'random' within this POC (true)
    // or if the file is sorted by typeID and the POC is 'random'
    bool bFileSortedByPOC;

    // part of the image, which is seen in display widget relative to scaled image
    QRect p_displayedImage;
    // points of the most top left and botoomright statistics element
    // it is used to crop the frame picture according to statistics grid
    QPoint p_tlPoint, p_brPoint;

    TikZFile * p_tikzFile;
};

#endif // STATISTICSMODEL_H