/*  FITS Label
    Copyright (C) 2003-2017 Jasem Mutlaq <mutlaqja@ikarustech.com>
    Copyright (C) 2016-2017 Robert Lancaster <rlancaste@gmail.com>

    This application 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.
*/

#include <config-kstars.h>

#include <QCursor>
#include <QToolTip>
#include <QDebug>

#include <KMessageBox>
#include <KLocalizedString>

#ifdef HAVE_INDI
#include "basedevice.h"
#include "indi/indilistener.h"
#include "indi/indistd.h"
#include "indi/driverinfo.h"
#endif

#include "Options.h"
#include "kstars.h"
#include "kstarsdata.h"
#include "skymap.h"
#include "auxiliary/ksutils.h"
#include "kspopupmenu.h"
#include "fitslabel.h"
#include "fitsview.h"

#define BASE_OFFSET     50
#define ZOOM_DEFAULT	100.0
#define ZOOM_MIN        10
#define ZOOM_MAX        400
#define ZOOM_LOW_INCR	10
#define ZOOM_HIGH_INCR	50

FITSLabel::FITSLabel(FITSView *view, QWidget *parent) : QLabel(parent)
{
    this->view = view;
}

FITSLabel::~FITSLabel() {}

void FITSLabel::setSize(double w, double h)
{
    width  = w;
    height = h;
    size   = w*h;
}

bool FITSLabel::getMouseButtonDown(){
    return mouseButtonDown;
}

/**
This method was added to make the panning function work.
If the mouse button is released, it resets mouseButtonDown variable and the mouse cursor.
 */
void FITSLabel::mouseReleaseEvent(QMouseEvent *e)
{
    Q_UNUSED(e);
    if(view->getMouseMode()==FITSView::dragMouse){
        mouseButtonDown=false;
        view->updateMouseCursor();
    }
}
/**
I added some things to the top of this method to allow panning and Scope slewing to function.
If you are in the dragMouse mode and the mousebutton is pressed, The method checks the difference
between the location of the last point stored and the current event point to see how the mouse has moved.
Then it moves the scrollbars and thus the view to the right location.
Then it stores the current point so next time it can do it again.
 */
void FITSLabel::mouseMoveEvent(QMouseEvent *e)
{
    float scale=(view->getCurrentZoom() / ZOOM_DEFAULT);

    if(view->getMouseMode()==FITSView::dragMouse&&mouseButtonDown){
        QPoint newPoint=e->globalPos();
        int dx=newPoint.x()-lastMousePoint.x();
        int dy=newPoint.y()-lastMousePoint.y();
        view->horizontalScrollBar()->setValue(view->horizontalScrollBar()->value()-dx);
        view->verticalScrollBar()->setValue(view->verticalScrollBar()->value()-dy);

        lastMousePoint=newPoint;
    }

    double x,y;
    FITSData *view_data = view->getImageData();

    uint8_t *buffer = view_data->getImageBuffer();

    if (buffer == NULL)
        return;

    x = round(e->x() / scale);
    y = round(e->y() / scale);

    x = KSUtils::clamp(x, 1.0, width);
    y = KSUtils::clamp(y, 1.0, height);

    emit newStatus(QString("X:%1 Y:%2").arg( (int)x ).arg( (int)y ), FITS_POSITION);

    // Range is 0 to dim-1 when accessing array
    x -= 1;
    y -= 1;

    QString stringValue;

    switch (view_data->getDataType())
    {
        case TBYTE:
            stringValue = QLocale().toString(buffer[(int) (y * width + x)]);
            break;

        case TSHORT:
            stringValue = QLocale().toString( (reinterpret_cast<int16_t*>(buffer)) [(int) (y * width + x)]);
            break;

        case TUSHORT:
            stringValue = QLocale().toString( (reinterpret_cast<uint16_t*>(buffer)) [(int) (y * width + x)]);
            break;

        case TLONG:
            stringValue = QLocale().toString( (reinterpret_cast<int32_t*>(buffer)) [(int) (y * width + x)]);
            break;

        case TULONG:
            stringValue = QLocale().toString( (reinterpret_cast<uint32_t*>(buffer)) [(int) (y * width + x)]);
            break;

        case TFLOAT:
            stringValue = QLocale().toString( (reinterpret_cast<float*>(buffer)) [(int) (y * width + x)], 'f', 5);
            break;

        case TLONGLONG:
            stringValue = QLocale().toString(static_cast<int>((reinterpret_cast<int64_t*>(buffer)) [(int) (y * width + x)]));
            break;

        case TDOUBLE:
            stringValue = QLocale().toString( (reinterpret_cast<float*>(buffer)) [(int) (y * width + x)], 'f', 5);

        break;

        default:
        break;
    }

    emit newStatus(stringValue, FITS_VALUE);

    if (view_data->hasWCS()&&view->getMouseMode()!=FITSView::selectMouse)
    {
        int index = x + y * width;

        wcs_point * wcs_coord = view_data->getWCSCoord();

        if (wcs_coord)
        {
            if (index > size)
                return;

            ra.setD(wcs_coord[index].ra);
            dec.setD(wcs_coord[index].dec);

            emit newStatus(QString("%1 , %2").arg( ra.toHMSString()).arg(dec.toDMSString()), FITS_WCS);
        }

        bool objFound=false;
        foreach(FITSSkyObject *listObject, view_data->objList){
            if((std::abs(listObject->x()-x)<5/scale)&&(std::abs(listObject->y()-y)<5/scale)){
                QToolTip::showText(e->globalPos(), listObject->skyObject()->name() +"\n"+listObject->skyObject()->longname() , this);
                objFound=true;
                break;
            }
        }
        if(!objFound)
            QToolTip::hideText();
    }

    double HFR = view->getImageData()->getHFR(x, y);

    if (HFR > 0)
        QToolTip::showText(e->globalPos(), i18nc("Half Flux Radius", "HFR: %1", QString::number(HFR, 'g' , 3)), this);

    //setCursor(Qt::CrossCursor);

    e->accept();
}

/**
I added some things to the top of this method to allow panning and Scope slewing to function.
If in dragMouse mode, the Panning function works by storing the cursor position when the mouse was pressed and setting
the mouseButtonDown variable to true.
If in ScopeMouse mode and the mouse is clicked, if there is WCS data and a scope is available, the method will verify that you actually
do want to slew to the WCS coordinates associated with the click location.  If so, it calls the centerTelescope function.
 */

void FITSLabel::mousePressEvent(QMouseEvent *e)
{
    float scale=(view->getCurrentZoom() / ZOOM_DEFAULT);

    if(view->getMouseMode()==FITSView::dragMouse)
    {
        mouseButtonDown=true;
        lastMousePoint=e->globalPos();
        view->updateMouseCursor();
    } else if(e->buttons() & Qt::LeftButton&&view->getMouseMode()==FITSView::scopeMouse)
    {
#ifdef HAVE_INDI
        FITSData *view_data = view->getImageData();
        if (view_data->hasWCS())
        {

            wcs_point * wcs_coord = view_data->getWCSCoord();
            double x,y;
            x = round(e->x() / scale);
            y = round(e->y() / scale);

            x = KSUtils::clamp(x, 1.0, width);
            y = KSUtils::clamp(y, 1.0, height);
            int index = x + y * width;
            if(KMessageBox::Continue==KMessageBox::warningContinueCancel(NULL, "Slewing to Coordinates: \nRA: " + dms(wcs_coord[index].ra).toHMSString() + "\nDec: " + dms(wcs_coord[index].dec).toDMSString(),
                                 i18n("Continue Slew"),  KStandardGuiItem::cont(), KStandardGuiItem::cancel(), "continue_slew_warning")){
                centerTelescope(wcs_coord[index].ra/15.0, wcs_coord[index].dec);
                view->setMouseMode(view->lastMouseMode);
                view->updateScopeButton();
            }
        }
#endif
    }

    double x,y;

    x = round(e->x() / scale);
    y = round(e->y() / scale);

    x = KSUtils::clamp(x, 1.0, width);
    y = KSUtils::clamp(y, 1.0, height);

#ifdef HAVE_INDI
    FITSData *view_data = view->getImageData();

    if (e->buttons() & Qt::RightButton&&view->getMouseMode()!=FITSView::scopeMouse)
    {
        mouseReleaseEvent(e);
        if (view_data->hasWCS())
        {
            foreach(FITSSkyObject *listObject, view_data->objList){
                if((std::abs(listObject->x()-x)<5/scale)&&(std::abs(listObject->y()-y)<5/scale)){
                    SkyObject *object=listObject->skyObject();
                    KSPopupMenu *pmenu;
                    pmenu=new KSPopupMenu();
                    object->initPopupMenu(pmenu);
                    QList<QAction *> actions= pmenu->actions();
                    foreach(QAction *action,actions){
                        if(action->text().left(7)=="Starhop")
                            pmenu->removeAction(action);
                        if(action->text().left(7)=="Angular")
                            pmenu->removeAction(action);
                        if(action->text().left(8)=="Add flag")
                            pmenu->removeAction(action);
                        if(action->text().left(12)=="Attach Label")
                            pmenu->removeAction(action);
                    }
                    pmenu->popup(e->globalPos());
                    KStars::Instance()->map()->setClickedObject(object);
                    break;
                }
            }
        }

        if (fabs(view->markerCrosshair.x()-x) <= 15 && fabs(view->markerCrosshair.y()-y) <= 15)
            emit markerSelected(0, 0);
    }
#endif

    if (e->buttons() & Qt::LeftButton&&view->getMouseMode()==FITSView::selectMouse)
    {
        emit pointSelected(x, y);
    }

}

void FITSLabel::mouseDoubleClickEvent(QMouseEvent *e)
{
    double x,y;

    x = round(e->x() / (view->getCurrentZoom() / ZOOM_DEFAULT));
    y = round(e->y() / (view->getCurrentZoom() / ZOOM_DEFAULT));

    x = KSUtils::clamp(x, 1.0, width);
    y = KSUtils::clamp(y, 1.0, height);

    emit markerSelected(x, y);

    return;
}

void FITSLabel::centerTelescope(double raJ2000, double decJ2000)
{
#ifdef HAVE_INDI

    if (INDIListener::Instance()->size() == 0)
    {
        KMessageBox::sorry(0, i18n("KStars did not find any active telescopes."));
        return;
    }

    foreach(ISD::GDInterface *gd, INDIListener::Instance()->getDevices())
    {
        INDI::BaseDevice *bd = gd->getBaseDevice();

        if (gd->getType() != KSTARS_TELESCOPE)
            continue;

        if (bd == NULL)
            continue;

        if (bd->isConnected() == false)
        {
            KMessageBox::error(0, i18n("Telescope %1 is offline. Please connect and retry again.", gd->getDeviceName()));
            return;
        }

        ISD::GDSetCommand SlewCMD(INDI_SWITCH, "ON_COORD_SET", "TRACK", ISS_ON, this);

        SkyObject selectedObject;

        selectedObject.setRA0(raJ2000);
        selectedObject.setDec0(decJ2000);

        selectedObject.apparentCoord(J2000, KStarsData::Instance()->ut().djd());

        gd->setProperty(&SlewCMD);
        gd->runCommand(INDI_SEND_COORDS, &selectedObject);

        return;
    }

    KMessageBox::sorry(0, i18n("KStars did not find any active telescopes."));

#else

    Q_UNUSED(raJ2000);
    Q_UNUSED(decJ2000);

#endif    
}
