#include "loadrunable.h"
#include "imageringlist.h"
#include <libraw/libraw.h>
#include "imageinfo.h"
#include <QFileInfo>
#include <QPainter>
#include <QElapsedTimer>
#include <QDebug>
#include <iostream>
#include <libexif/exif-data.h>
#include <fitsio2.h>
#include "rawimage.h"
#include "starfit.h"

LoadRunable::LoadRunable(const QString &file, Image *receiver, AnalyzeLevel level) :
    m_file(file),
    m_receiver(receiver),
    m_analyzeLevel(level)
{
}

void loadExifEntry(ImageInfoData &info, ExifContent *content, ExifTag tag)
{
    char val[1024];
    ExifEntry *entry = exif_content_get_entry(content, tag);
    if(entry)
    {
        exif_entry_get_value(entry, val, sizeof(val));
        info.append(StringPair(exif_tag_get_title(tag), QString(val)));
    }
}

void drawPeaks(QImage &img, const std::vector<Peak> &peaks)
{
    QPixmap pix = QPixmap::fromImage(img);
    QPainter painter(&pix);
    painter.setPen(Qt::red);
    for(auto peak : peaks)
    {
        painter.drawEllipse(QPoint(peak.x(), peak.y()), 5, 5);
    }
    img = pix.toImage();
}

void drawStars(QImage &img, const std::vector<Star> &stars)
{
    QPixmap pix = QPixmap::fromImage(img);
    QPainter painter(&pix);
    painter.setPen(Qt::red);
    for(auto star : stars)
    {
        painter.drawEllipse(QPointF(star.m_x, star.m_y), star.hw20X(), star.hw20Y());
    }
    img = pix.toImage();
}

void printStarModel(int radius, const std::vector<double> &data, const Star &star)
{
    QString d = "d=[";
    QString m = "m=[";
    for(int y=0; y<radius; y++)
    {
        for(int x=0; x<radius; x++)
        {
            d += QString::number(data[y*radius+x]) + ",";
            m += QString::number(gauss_model(star.m_am, star.m_x, star.m_y, star.m_sx, star.m_sy, x, y)) + ",";
        }
        d += ";";
        m += ";";
    }
    d += "];";
    m += "];";
    //std::cout << star.m_am << " " << star.m_sx << star.m_sy << std::endl;
    std::cout << d.toStdString() << std::endl;
    std::cout << m.toStdString() << std::endl << std::endl;
}

bool loadRAW(QString path, ImageInfoData &info, RawImage **image, QImage *qimage)
{
    if(!image && !qimage)
        return false;

    LibRaw raw;
    raw.open_file(path.toLocal8Bit().data());
    raw.imgdata.params.half_size = true;
    raw.imgdata.params.use_camera_wb = true;
    raw.imgdata.params.user_flip = 0;
    if(raw.unpack())
        return false;

    if(image)
    {
        libraw_rawdata_t rawdata = raw.imgdata.rawdata;
        size_t size = rawdata.sizes.width*rawdata.sizes.height;

        std::vector<uint16_t> out;
        out.resize(size);
        size_t d = 0;
        uint h=rawdata.sizes.top_margin+rawdata.sizes.height;
        uint w=rawdata.sizes.left_margin+rawdata.sizes.width;
        size_t pitch = rawdata.sizes.raw_pitch/sizeof(uint16_t);

        for(size_t i=rawdata.sizes.top_margin;i<h;i++)
        {
            for(size_t o=rawdata.sizes.left_margin;o<w;o++)
            {
                uint16_t p = rawdata.raw_image[i*pitch+o];
                out[d++] = p;
            }
        }
        *image = new RawImage(rawdata.sizes.width, rawdata.sizes.height, RawImage::UINT16);
        memcpy((*image)->data(), &out[0], sizeof(uint16_t)*d);
    }

    if(qimage)
    {
        raw.dcraw_process();
        libraw_processed_image_t *rawImg = raw.dcraw_make_mem_image();
        QImage img(rawImg->width, rawImg->height, QImage::Format_RGB888);
        uint scanLine = rawImg->width*rawImg->colors;

        for(uint i=0; i<rawImg->height; i++)
        {
            memcpy(img.scanLine(i), rawImg->data+(i*scanLine), scanLine);
        }

        QString shutterSpeed = QString::number(raw.imgdata.other.shutter);
        if(raw.imgdata.other.shutter < 1)
        {
            shutterSpeed = QString("1/%1s").arg(1.0f/raw.imgdata.other.shutter);
        }
        info.append(StringPair(QObject::tr("Width"), QString::number(rawImg->width)));
        info.append(StringPair(QObject::tr("Height"), QString::number(rawImg->height)));
        info.append(StringPair(QObject::tr("ISO"), QString::number(raw.imgdata.other.iso_speed)));
        info.append(StringPair(QObject::tr("Shutter speed"), shutterSpeed));
#if LIBRAW_MINOR_VERSION>=19
        info.append(StringPair(QObject::tr("Camera temperature"), QString::number(raw.imgdata.other.CameraTemperature)));
#endif
        raw.dcraw_clear_mem(rawImg);

        *qimage = img;
    }

    return true;
}

bool loadFITS(QString path, ImageInfoData &info, RawImage **image, QImage *qimage)
{
    if(!image && !qimage)
        return false;

    fitsfile *file;
    int status = 0;
    int type;
    fits_open_image(&file, path.toLocal8Bit().data(), READONLY, &status);
    fits_get_hdu_type(file, &type, &status);

    if(type == IMAGE_HDU)
    {
        int imgtype;
        int naxis;
        long naxes[3] = {0};
        fits_get_img_param(file, 3, &imgtype, &naxis, naxes, &status);

        if(naxis >= 2 && naxis <= 3 && status == 0)
        {
            int cvtype;
            int fitstype;
            std::vector<cv::Mat> cvimg;
            long fpixel[3] = {1,1,1};
            switch(imgtype)
            {
            case BYTE_IMG:
                cvtype = CV_8U;
                fitstype = TBYTE;
                break;
            case SHORT_IMG:
                cvtype = CV_16U;
                fitstype = TUSHORT;
                break;
            case FLOAT_IMG:
                cvtype = CV_32F;
                fitstype = TFLOAT;
                break;
            }

            size_t size = naxes[0]*naxes[1];
            size_t w = naxes[0];
            size_t h = naxes[1];

            info.append(StringPair(QObject::tr("Width"), QString::number(naxes[0])));
            info.append(StringPair(QObject::tr("Height"), QString::number(naxes[1])));

            for (int i=1; i==1 || i<=naxes[2]; i++)
            {
                cv::Mat tmp(h, w, cvtype);
                fpixel[2] = i;
                fits_read_pix(file, fitstype, fpixel, size, NULL, tmp.ptr(), NULL, &status);
                cvimg.push_back(tmp);
            }

            if(cvimg.size() == 1)
            {
                *image = new RawImage(cvimg[0]);
            }
            if(cvimg.size() == 3)
            {
                cv::Mat rgb;
                cv::merge(cvimg, rgb);
                *image = new RawImage(rgb);
            }
        }
    }

    int nheader = 0, more;
    fits_get_hdrspace(file, &nheader, &more, &status);
    for(int i=1; i<=nheader; i++)
    {
        char key[FLEN_KEYWORD];

        char val[FLEN_VALUE];
        char comm[FLEN_COMMENT];
        fits_read_keyn(file, i, key, val, comm, &status);
        info.append(StringPair(key, QString(val)));
    }

    fits_close_file(file, &status);
    if(status)
    {
        char err[100];
        fits_get_errstatus(status, err);
        info.append(StringPair(QObject::tr("Error"), QString(err)));
    }

    return true;
}

void LoadRunable::run()
{
    if(!m_receiver->isCurrent())
    {
        return;
    }
    QElapsedTimer timer;
    ImageInfoData info;
    QFileInfo finfo(m_file);
    info.append(StringPair(QObject::tr("Filename"), finfo.fileName()));

    QImage img;
    RawImage *rawImage = nullptr;
    bool raw = false;
    if(m_file.endsWith(".CR2", Qt::CaseInsensitive))
    {
        timer.start();
        loadRAW(m_file, info, &rawImage, nullptr);
        raw = true;
        qDebug() << "LoadRaw" << timer.elapsed();
    }
    else if(m_file.endsWith(".FIT", Qt::CaseInsensitive) || m_file.endsWith(".FITS", Qt::CaseInsensitive))
    {
        loadFITS(m_file, info, &rawImage, nullptr);
    }
    else
    {
        img = QImage(m_file);
        ExifData *exif = exif_data_new_from_file(m_file.toLocal8Bit().constData());
        info.append(StringPair(QObject::tr("Width"), QString::number(img.width())));
        info.append(StringPair(QObject::tr("Height"), QString::number(img.height())));
        if(exif)
        {
            loadExifEntry(info, exif->ifd[EXIF_IFD_EXIF], EXIF_TAG_ISO_SPEED_RATINGS);
            loadExifEntry(info, exif->ifd[EXIF_IFD_EXIF], EXIF_TAG_SHUTTER_SPEED_VALUE);
            exif_data_free(exif);
        }
    }

    if(rawImage && m_analyzeLevel >= Statistics)
    {
        double mean, median, min, max;
        double stdDev;
        timer.start();
        rawImage->imageStats(&mean, &stdDev, &median, &min, &max);
        qDebug() << "image stats" << timer.restart();
        info.append(StringPair(QObject::tr("Mean"), QString::number(mean)));
        info.append(StringPair(QObject::tr("Standart deviation"), QString::number(stdDev)));
        info.append(StringPair(QObject::tr("Median"), QString::number(median)));
        info.append(StringPair(QObject::tr("Minimum"), QString::number(min)));
        info.append(StringPair(QObject::tr("Maximum"), QString::number(max)));

        if(m_analyzeLevel >= Peaks)
        {
            std::vector<Peak> peaks;
            if(raw) {
                rawImage->quarter();
                qDebug() << "quarter" << timer.restart();
            }
            RawImage *medianImage = rawImage->medianFilter();
            qDebug() << "median" << timer.restart();
            int numPeaks = medianImage->findPeaks(median+stdDev*2, 20, peaks);
            delete medianImage;
            qDebug() << "peaks" << timer.restart();
            if(m_analyzeLevel == Peaks)
                drawPeaks(img, peaks);
            qDebug() << "draw peaks" << timer.restart();
            info.append(StringPair(QObject::tr("Peaks"), QString::number(numPeaks)));
            info.append(StringPair(QObject::tr("Peaks draw"), QString::number(peaks.size())));

            if(m_analyzeLevel>= Stars)
            {
                double fwhmX = 0;
                double fwhmY = 0;
                const int radius = 13;
                StarFit starFit(radius);
                std::vector<Star> stars;
                for(uint i=0; i<peaks.size(); i++)
                {
                    Peak p = peaks[i];
                    std::vector<double> r;
                    int x = p.x();
                    int y = p.y();
                    rawImage->rect(x, y, radius, radius, r);
                    Star star = starFit.fitStar(r, false);
                    if(star.valid())
                    {
                        //printStarModel(radius, r, star);
                        star.m_x += x;
                        star.m_y += y;
                        fwhmX += star.fwhmX();
                        fwhmY += star.fwhmY();
                        stars.push_back(star);
                    }
                }
                drawStars(img, stars);
                info.append(StringPair(QObject::tr("FWHM X"), QString::number(fwhmX/stars.size())));
                info.append(StringPair(QObject::tr("FWHM Y"), QString::number(fwhmY/stars.size())));
            }
            qDebug() << "Star fit" << timer.restart();
        }
    }

    QMetaObject::invokeMethod(m_receiver, "imageLoaded", Qt::QueuedConnection, Q_ARG(QImage, img), Q_ARG(void*, rawImage), Q_ARG(ImageInfoData, info));
}