/*
    SPDX-FileCopyrightText: 2016 Jasem Mutlaq <mutlaqja@ikarustech.com>.

    Based on lin_guider

    SPDX-License-Identifier: GPL-2.0-or-later
*/

#pragma once

#include "matr.h"
#include "indi/indicommon.h"
#include "../guideinterface.h"
#include "guidelog.h"
#include "calibration.h"
#include "calibrationprocess.h"
#include "gmath.h"
#include "ekos_guide_debug.h"
#include <QFile>
#include <QPointer>
#include <QQueue>
#include <QTime>

#include <memory>
#include <random>

class QVector3D;
class FITSData;
class cgmath;
class GuideView;
class Edge;

namespace Ekos
{
class InternalGuider : public GuideInterface
{
        Q_OBJECT

    public:
        InternalGuider();

        bool Connect() override
        {
            state = GUIDE_IDLE;
            return true;
        }
        bool Disconnect() override
        {
            return true;
        }
        bool isConnected() override
        {
            return true;
        }

        bool calibrate() override;
        bool guide() override;
        bool abort() override;
        bool suspend() override;
        bool resume() override;

        bool dither(double pixels) override;
        bool ditherXY(double x, double y);

        bool clearCalibration() override;
        bool restoreCalibration();

        bool reacquire() override;

        bool setFrameParams(uint16_t x, uint16_t y, uint16_t w, uint16_t h, uint16_t binX, uint16_t binY) override;
        bool setGuiderParams(double ccdPixelSizeX, double ccdPixelSizeY, double mountAperture,
                             double mountFocalLength) override;

        // Set Star Position
        void setStarPosition(QVector3D &starCenter) override;

        // Select algorithm
        void setStarDetectionAlgorithm(int index);

        // Reticle Parameters
        bool getReticleParameters(double *x, double *y);

        // Guide Square Box Size
        void setGuideBoxSize(uint32_t value)
        {
            guideBoxSize = value;
        }

        // Guide View
        void setGuideView(const QSharedPointer<GuideView> &guideView);
        // Image Data
        void setImageData(const QSharedPointer<FITSData> &data);

        /**
         * @brief setStreamingMode Enable or disable streaming guide mode.
         *        When enabled, the guider does NOT emit frameCaptureRequested after sending pulses,
         *        because new frames arrive automatically from the camera's continuous stream.
         * @param enabled True for streaming mode, false for single-capture mode.
         */
        void setStreamingMode(bool enabled)
        {
            m_StreamingMode = enabled;
        }

        bool isStreamingMode() const
        {
            return m_StreamingMode;
        }

        bool start();

        bool isGuiding(void) const;
        void setInterface(void);

        void setSubFramed(bool enable)
        {
            m_isSubFramed = enable;
        }

        bool useSubFrame();

        const Calibration &getCalibration() const;

        // Select a guide star automatically
        bool selectAutoStar();
        bool selectAutoStarSEPMultistar();
        bool SEPMultiStarEnabled();

        // Manual Dither
        bool processManualDithering();

        void updateGPGParameters();
        void resetGPG() override;
        void resetDarkGuiding();
        void setExposureTime();
        void setDarkGuideTimerInterval();
        void setTimer(std::unique_ptr<QTimer> &timer, Seconds seconds);

    public Q_SLOTS:
        void setDECSwap(bool enable);

        // AI Guider accessor for EkosLive
        MountSpecificGuider *getAIGuider()
        {
            return pmath ? pmath->getAIGuider() : nullptr;
        }
        const MountSpecificGuider *getAIGuider() const
        {
            return pmath ? pmath->getAIGuider() : nullptr;
        }

    protected Q_SLOTS:
        void trackingStarSelected(int x, int y);
        void setDitherSettled();
        void darkGuide();

    Q_SIGNALS:
        void newMultiPulse(GuideDirection ra_dir, int ra_msecs, GuideDirection dec_dir, int dec_msecs,
                           CaptureAfterPulses followWithCapture);
        void newSinglePulse(GuideDirection dir, int msecs, CaptureAfterPulses followWithCapture);
        //void newStarPosition(QVector3D, bool);
        void DESwapChanged(bool enable);
        // Forwarded from cgmath: AI feed-forward lifecycle (state is AIGuideState cast to int).
        void newAIState(int state, double confidence);
    private:
        // Guiding
        bool processGuiding();
        void startDarkGuiding();
        bool abortDither();
        bool onePulseDither(double pixels);
        void startDitherSettleTimer(int ms);
        void disableDitherSettleTimer();


        void reset();

        // Logging
        void fillGuideInfo(GuideLog::GuideInfo *info);

        std::unique_ptr<cgmath> pmath;
        QSharedPointer<GuideView> m_GuideFrame;
        QSharedPointer<FITSData> m_ImageData;
        bool m_isStarted { false };
        bool m_isSubFramed { false };
        bool m_isFirstFrame { false };
        int m_starLostCounter { 0 };

        // When true, the guide camera is streaming continuously.
        // In this mode processGuiding() must NOT emit frameCaptureRequested after sending pulses,
        // because the next frame will arrive on its own from the stream pipeline.
        bool m_StreamingMode { false };

        QFile logFile;
        uint32_t guideBoxSize { 32 };

        GuiderUtils::Vector m_DitherTargetPosition;
        uint8_t m_DitherRetries {0};

        QElapsedTimer reacquireTimer;
        int m_highRMSCounter {0};

        GuiderUtils::Matrix ROT_Z;
        Ekos::GuideState rememberState { GUIDE_IDLE };

        // Progressive Manual Dither
        QQueue<GuiderUtils::Vector> m_ProgressiveDither;

        // Dark guiding timer
        std::unique_ptr<QTimer> m_darkGuideTimer;
        std::unique_ptr<QTimer> m_captureTimer;
        std::unique_ptr<QTimer> m_ditherSettleTimer;
        std::pair<Seconds, Seconds> calculateGPGTimeStep();
        bool isInferencePeriodFinished();


        // How many high RMS pulses before we stop
        static const uint8_t MAX_RMS_THRESHOLD = 10;
        // How many lost stars before we stop
        static const uint8_t MAX_LOST_STAR_THRESHOLD = 5;

        // Maximum pulse time limit for immediate capture. Any pulses longer that this
        // will be delayed until pulse is over
        static const uint16_t MAX_IMMEDIATE_CAPTURE = 250;
        // When to start capture before pulse delay is over
        static const uint16_t PROPAGATION_DELAY = 100;

        // How many 'random' pixels can we move before we have to force direction reversal?
        static const uint8_t MAX_DITHER_TRAVEL = 15;

        // This keeps track of the distance dithering has moved. It's used to make sure
        // the "random walk" dither doesn't wander too far away.
        QVector3D m_DitherOrigin;

        GuideLog guideLog;

        void iterateCalibration();
        std::unique_ptr<CalibrationProcess> calibrationProcess;
        double calibrationStartX = 0;
        double calibrationStartY = 0;

        // Random numbers for dithering
        std::random_device m_randomDevice;
        std::default_random_engine m_generator;
        std::uniform_real_distribution<double> m_randomAngle;

        bool isPoorGuiding(const cproc_out_params *out);
        void emitAxisPulse(const cproc_out_params *out);
};
}
