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

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

    Some code fragments were adapted from Peter Kirchgessner's FITS plugin
    SPDX-FileCopyrightText: Peter Kirchgessner <http://members.aol.com/pkirchg>
*/

#pragma once

#include "config-kstars.h"

// From StellarSolver
#ifdef HAVE_STELLARSOLVER
#include <structuredefinitions.h>
#else
#include "structuredefinitions.h"
#endif

#include "kstarsdatetime.h"
#include "bayerparameters.h"
#include "skybackground.h"
#include "fitscommon.h"
#include "fitsstardetector.h"
#include "auxiliary/imagemask.h"

#ifdef WIN32
// This header must be included before fitsio.h to avoid compiler errors with Visual Studio
#include <windows.h>
#endif

#include <fitsio.h>

#include <QFuture>
#include <QFutureWatcher>
#include <QObject>
#include <QRect>
#include <QVariant>
#include <QTemporaryFile>
#include <QNetworkReply>
#include <QTimer>
#include <QQueue>
#include <QMutex>

#ifndef KSTARS_LITE
#include <kxmlguiwindow.h>
#include <wcs.h>
#endif

#include "fitsskyobject.h"
#include "fitsdirwatcher.h"
#if !defined (KSTARS_LITE)
#include "fitsstack.h"
#endif

class QProgressDialog;

class SkyPoint;
class FITSHistogramData;
class Edge;

class FITSData : public QObject
{
        Q_OBJECT

        // Name of FITS file
        Q_PROPERTY(QString filename READ filename)
        // Extension
        Q_PROPERTY(QString extension READ extension)
        // Size of file in bytes
        Q_PROPERTY(qint64 size READ size)
        // Width in pixels
        Q_PROPERTY(quint16 width READ width)
        // Height in pixels
        Q_PROPERTY(quint16 height READ height)
        // FITS MODE --> Normal, Focus, Guide..etc
        Q_PROPERTY(FITSMode mode MEMBER m_Mode)
        // 1 channel (grayscale) or 3 channels (RGB)
        Q_PROPERTY(quint8 channels READ channels)
        // Bits per pixel
        Q_PROPERTY(quint8 bpp READ bpp)
        // Does FITS have WSC header?
        Q_PROPERTY(bool hasWCS READ hasWCS)
        // Does FITS have bayer data?
        Q_PROPERTY(bool hasDebayer READ hasDebayer)

    public:
        explicit FITSData(FITSMode fitsMode = FITS_NORMAL);
        explicit FITSData(const QSharedPointer<FITSData> &other);
        ~FITSData() override;

        /**
         * @brief mutex locks the image buffer during update
         * @return mutex.
         */
        QMutex* mutex() const
        {
            return &m_DataMutex;
        };

        /** Object to hold FITS Header records */
        struct Record
        {
            Record() = default;
            Record(QString k, QString v, QString c) : key(k), value(v), comment(c) {}
            QString key;      /** FITS Header Key */
            QVariant value;   /** FITS Header Value */
            QString comment;  /** FITS Header Comment, if any */
        };

        typedef enum
        {
            Idle,
            Busy,
            Success,
            Failure
        } WCSState;

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Read and Write file/buffer Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /**
         * @brief loadFITS Loading FITS file asynchronously.
         * @param inFilename Path to FITS file (or compressed fits.gz)
         * @return A QFuture that can be watched until the async operation is complete.
         */
        QFuture<bool> loadFromFile(const QString &inFilename);

        /**
         * @brief loadFITSFromMemory Loading FITS from memory buffer.
         * @param buffer The memory buffer containing the fits data.
         * @return bool indicating success or failure.
         */
        bool loadFromBuffer(const QByteArray &buffer);

        /**
         * @brief parseSolution Parse the WCS solution information from the header into the given struct.
         * @param solution Solution structure to fill out.
         * @param stack tells the function to use the stack header rather than "normal" header
         * @return True if parsing successful, false otherwise.
         */
        bool parseSolution(FITSImage::Solution &solution, const bool stack = false) const;

        /* Save FITS or JPG/PNG*/
        bool saveImage(const QString &newFilename);

        // Access functions
        void clearImageBuffers();
        void setImageBuffer(uint8_t *buffer);
        uint8_t const *getImageBuffer() const;
        uint8_t *getWritableImageBuffer();

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Statistics Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Calculate stats
        void calculateStats(bool refresh = false, bool roi = false);
        void saveStatistics(FITSImage::Statistic &other);
        void restoreStatistics(FITSImage::Statistic &other);
        FITSImage::Statistic const &getStatistics() const
        {
            return m_Statistics;
        }

        uint16_t width(bool roi = false) const
        {
            return roi ?  m_ROIStatistics.width : m_Statistics.width;
        }
        uint16_t height(bool roi = false) const
        {
            return roi ?  m_ROIStatistics.height : m_Statistics.height;
        }
        int64_t size(bool roi = false) const
        {
            return roi ?  m_ROIStatistics.size : m_Statistics.size;
        }
        int channels() const
        {
            return m_Statistics.channels;
        }
        uint32_t samplesPerChannel(bool roi = false) const
        {
            return roi ?  m_ROIStatistics.samples_per_channel : m_Statistics.samples_per_channel;
        }
        uint32_t dataType() const
        {
            return m_Statistics.dataType;
        }
        double getMin(uint8_t channel = 0, bool roi = false) const
        {
            return roi ?  m_ROIStatistics.min[channel] : m_Statistics.min[channel];
        }
        double getMax(uint8_t channel = 0, bool roi = false) const
        {
            return roi ?  m_ROIStatistics.max[channel] : m_Statistics.max[channel];

        }
        void setMinMax(double newMin, double newMax, uint8_t channel = 0);
        void getMinMax(double *min, double *max, uint8_t channel = 0) const
        {
            *min = m_Statistics.min[channel];
            *max = m_Statistics.max[channel];
        }
        void setStdDev(double value, uint8_t channel = 0)
        {
            m_Statistics.stddev[channel] = value;
        }
        double getStdDev(uint8_t channel = 0, bool roi = false ) const
        {
            return roi ? m_ROIStatistics.stddev[channel] :  m_Statistics.stddev[channel];
        }
        double getAverageStdDev(bool roi = false) const;
        void setMean(double value, uint8_t channel = 0)
        {
            m_Statistics.mean[channel] = value;
        }
        double getMean(uint8_t channel = 0, bool roi = false) const
        {
            return roi ? m_ROIStatistics.mean[channel] :  m_Statistics.mean[channel];
        }
        // for single channel, just return the mean for channel zero
        // for color, return the average
        double getAverageMean(bool roi = false) const;
        void setMedian(double val, uint8_t channel = 0)
        {
            m_Statistics.median[channel] = val;
        }
        // for single channel, just return the median for channel zero
        // for color, return the average
        double getAverageMedian(bool roi = false) const;
        double getMedian(uint8_t channel = 0, bool roi = false) const
        {
            return roi ? m_ROIStatistics.median[channel] :  m_Statistics.median[channel];
        }

        int getBytesPerPixel() const
        {
            return m_Statistics.bytesPerPixel;
        }
        void setSNR(double val)
        {
            m_Statistics.SNR = val;
        }
        double getSNR() const
        {
            return m_Statistics.SNR;
        }
        uint32_t bpp() const
        {
            switch(m_Statistics.dataType)
            {
                case TBYTE:
                    return 8;
                case TSHORT:
                case TUSHORT:
                    return 16;
                case TLONG:
                case TULONG:
                case TFLOAT:
                    return 32;
                case TLONGLONG:
                case TDOUBLE:
                    return 64;
                default:
                    return 8;
            }
        }
        double getADU() const;

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// FITS Header Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // FITS Record
        bool getRecordValue(const QString &key, QVariant &value, const bool stack = false) const;
        void updateRecordValue(const QString &key, QVariant value, const QString &comment, const bool stack = false);
        const QList<Record> &getRecords() const
        {
            return m_HeaderRecords;
        }

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Star Search & HFR Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Star Detection - Native KStars implementation
        void setStarAlgorithm(StarAlgorithm algorithm)
        {
            starAlgorithm = algorithm;
        }
        int getDetectedStars() const
        {
            return starCenters.count();
        }
        bool areStarsSearched() const
        {
            return starsSearched;
        }
        void appendStar(Edge *newCenter)
        {
            starCenters.append(newCenter);
        }
        const QList<Edge *> &getStarCenters() const
        {
            return starCenters;
        }
        QList<Edge *> getStarCentersInSubFrame(QRect subFrame) const;

        void setStarCenters(const QList<Edge*> &centers)
        {
            qDeleteAll(starCenters);
            starCenters = centers;
        }
        QFuture<bool> findStars(StarAlgorithm algorithm = ALGORITHM_CENTROID, const QRect &trackingBox = QRect());

        void setSkyBackground(const SkyBackground &bg)
        {
            m_SkyBackground = bg;
        }
        const SkyBackground &getSkyBackground() const
        {
            return m_SkyBackground;
        }
        const QVariantMap &getSourceExtractorSettings() const
        {
            return m_SourceExtractorSettings;
        }
        void setSourceExtractorSettings(const QVariantMap &settings)
        {
            m_SourceExtractorSettings = settings;
        }
        // Use SEP (Sextractor Library) to find stars
        template <typename T>
        void getFloatBuffer(float *buffer, int x, int y, int w, int h) const;
        //int findSEPStars(QList<Edge*> &, const int8_t &boundary = int8_t()) const;

        // filter all stars that are visible through the given mask
        int filterStars(QSharedPointer<ImageMask> mask);

        // Half Flux Radius
        const Edge &getSelectedHFRStar() const
        {
            return m_SelectedHFRStar;
        }

        // Calculates the median star eccentricity.
        double getEccentricity();

        double getHFR(HFRType type = HFR_AVERAGE);
        double getHFR(int x, int y, double scale = 1.0);

        // Request an abort of star detection
        void abortStarDetection();

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Date & Time (WCS) Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////

        const KStarsDateTime &getDateTime() const
        {
            return m_DateTime;
        }

        // Set the time, for testing (doesn't set header field)
        void setDateTime(const KStarsDateTime &t)
        {
            m_DateTime = t;
        }

        const QPoint getRoiCenter() const
        {
            return roiCenter;
        }

        void setRoiCenter(QPoint c)
        {
            roiCenter = c;
        }

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// World Coordinate System (WCS) Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Check if a particular point exists within the image
        bool contains(const QPointF &point) const;
        // Does image have valid WCS?
        bool hasWCS()
        {
            return HasWCS;
        }
        // Load WCS data
        bool loadWCS();
        // Get WCS State
        WCSState getWCSState() const
        {
            return m_WCSState;
        }

        /**
             * @brief wcsToPixel Given J2000 (RA0,DE0) coordinates. Find in the image the corresponding pixel coordinates.
             * @param wcsCoord Coordinates of target
             * @param wcsPixelPoint Return XY FITS coordinates
             * @param wcsImagePoint Return XY Image coordinates
             * @return True if conversion is successful, false otherwise.
             */
        bool wcsToPixel(const SkyPoint &wcsCoord, QPointF &wcsPixelPoint, QPointF &wcsImagePoint);

        /**
             * @brief pixelToWCS Convert Pixel coordinates to J2000 world coordinates
             * @param wcsPixelPoint Pixel coordinates in XY Image space.
             * @param wcsCoord Store back WCS world coordinate in wcsCoord
             * @return True if successful, false otherwise.
             */
        bool pixelToWCS(const QPointF &wcsPixelPoint, SkyPoint &wcsCoord);

        /**
             * @brief injectWCS Add WCS keywords
             * @param orientation Solver orientation, degrees E of N.
             * @param ra J2000 Right Ascension
             * @param dec J2000 Declination
             * @param pixscale Pixel scale in arcsecs per pixel
             * @param eastToTheRight if true, then when the image is rotated so that north is up, then east would be to the right on the image.
             */
        void injectWCS(double orientation, double ra, double dec, double pixscale, bool eastToTheRight);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Debayering Functions
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////

        // Debayer
        bool hasDebayer()
        {
            return HasDebayer;
        }

        /**
         * @brief debayer the 1-channel data to 3-channel RGB using the default debayer pattern detected in the FITS header.
         * @param reload If true, it will read the image again from disk before performing debayering. This is necessary to attempt
         * subsequent debayering processes on an already debayered image.
         */
        bool debayer(bool reload = false);
        bool debayer_8bit();
        bool debayer_16bit();
        void getBayerParams(BayerParameters *param);
        void setBayerParams(BayerParameters *param);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Public Histogram Functions
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////

        void resetHistogram()
        {
            m_HistogramConstructed = false;
        }
        double getHistogramBinWidth(int channel = 0)
        {
            return m_HistogramBinWidth[channel];
        }

        // Returns a vector with the counts (y-axis values) for the histogram.
        const QVector<uint32_t> &getCumulativeFrequency(uint8_t channel = 0) const
        {
            return m_CumulativeFrequency[channel];
        }
        // Returns a vector with the values (x-axis values) for the histogram.
        // The value returned is the low end of the histogram interval.
        // The high end is this intensity plus the value returned by getHistogramBinWidth().
        const QVector<double> &getHistogramIntensity(uint8_t channel = 0) const
        {
            return m_HistogramIntensity[channel];
        }
        const QVector<double> &getHistogramFrequency(uint8_t channel = 0) const
        {
            return m_HistogramFrequency[channel];
        }
        int getHistogramBinCount() const
        {
            return m_HistogramBinCount;
        }

        // Returns the histogram bin for the pixel at location x,y in the given channel.
        int32_t histogramBin(int x, int y, int channel) const;

        /**
         * @brief getJMIndex Overall contrast of the image used in find centeroid algorithm. i.e. is the image diffuse?
         * @return Value of JMIndex
         */
        double getJMIndex() const
        {
            return m_JMIndex;
        }

        bool isHistogramConstructed() const
        {
            return m_HistogramConstructed;
        }
        void constructHistogram();

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Filters and Rotations Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Filter
        void applyFilter(FITSScale type, uint8_t *image = nullptr, QVector<double> *targetMin = nullptr,
                         QVector<double> *targetMax = nullptr);

        // Rotation counter. We keep count to rotate WCS keywords on save
        int getRotCounter() const;
        void setRotCounter(int value);

        // Filename
        const QString &filename() const
        {
            return m_Filename;
        }
        void setFilename(QString &filename)
        {
            m_Filename = filename;
        }
        const QString &compressedFilename() const
        {
            return m_compressedFilename;
        }
        bool isCompressed() const
        {
            return m_isCompressed;
        }
        // Extension
        const QString &extension() const
        {
            return m_Extension;
        }
        void setExtension(const QString &extension)
        {
            m_Extension = extension;
        }

        // Horizontal flip counter. We keep count to rotate WCS keywords on save
        int getFlipHCounter() const;
        void setFlipHCounter(int value);

        // Horizontal flip counter. We keep count to rotate WCS keywords on save
        int getFlipVCounter() const;
        void setFlipVCounter(int value);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Object Search Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
#ifndef KSTARS_LITE
        /**
         * @brief search the current image for objects, either on the Skymap or a catalog
         */
        bool searchObjects();
        /**
         * @brief search the current image for catalog objects
         */
        bool searchCatObjects();
        /**
         * @brief get the search center point
         * @return search center point
         */
        QPoint &catROIPt()
        {
            return m_CatROIPt;
        }
        /**
         * @brief get the search circle radius
         * @return search circle radius
         */
        int catROIRadius() const
        {
            return m_CatROIRadius;
        }

        bool findWCSBounds(double &minRA, double &maxRA, double &minDec, double &maxDec);
#endif
        /**
         * @brief set the search center circle with center point and radius
         * @param circle center
         * @param circle radius
         */
        void setCatSearchROI(const QPoint searchCenter, const int radius);
        /**
         * @brief get the sky map objects
         * @return QList of objects
         */
        const QList<FITSSkyObject *> &getSkyObjects() const
        {
            return m_SkyObjects;
        }
        /**
         * @brief get the catalog objects
         * @return QList of catalog objects
         */
        const QList<CatObject> &getCatObjects() const
        {
            return m_CatObjects;
        }
        /**
         * @brief get the catalog object filters
         * @return QList of catalog object filters
         */
        const QStringList &getCatObjectsFilters() const
        {
            return m_CatObjectsFilters;
        }
        /**
         * @brief set the catalog object filters
         * @param QList of filters
         */
        void setCatObjectsFilters(const QStringList filters);
        /**
         * @brief get current status of catalog query
         * @return whether catalog query in progress or not
         */
        const bool &getCatQueryInProgress() const
        {
            return m_CatQueryInProgress;
        }
        /**
         * @brief highlight (and optionally lowlight) a particular catalog item
         * @param highlight item
         * @param lowlight item
         */
        bool highlightCatObject(const int hilite, const int lolite);
        /**
         * @brief applies current filters to object type list showing / hiding items as appropriate
         */
        void filterCatObjects();
        /**
         * @brief returns Object Type Label for the passed in Object Type Code
         * @param object type code
         * @return object type label
         */
        QString getCatObjectLabel(const QString code) const;

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Image Conversion Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Create autostretch image from FITS File
        static QImage FITSToImage(const QString &filename);

        /**
         * @brief ImageToFITS Convert an image file with supported format to a FITS file.
         * @param filename full path to filename without extension
         * @param format file extension. Supported formats are whatever supported by Qt (e.g. PNG, JPG,..etc)
         * @param output Output temporary file path. The created file is generated by the function and store in output.
         * @return True if conversion is successful, false otherwise.
         */
        static bool ImageToFITS(const QString &filename, const QString &format, QString &output);

        QString getLastError() const;

        static bool readableFilename(const QString &filename);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Live Stacking Functions
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////

        /**
         * @brief Get the live stack object pointer
         * @return Live Stack pointer
         */
#if !defined (KSTARS_LITE)
        const QMap<LiveStackChannel, QSharedPointer<FITSStack>> &stacks() const
        {
            return m_Stacks;
        }
#endif
        /**
         * @brief Get the live stacking image buffer
         * @return Live Stacking image buffer
         */
        uint8_t const *getStackImageBuffer() const
        {
            return m_StackImageBuffer;
        }
        /**
         * @brief Get the live stacking image statistics
         * @return Live Stacking image statistics
         */
        FITSImage::Statistic const &getStackStatistics() const
        {
            return m_StackStatistics.stats;
        }

        /**
         * @brief Initialise FITSData for a stack.
         * @param inDirectory Inital directory path
         * @return success.
         */

        /**
         * @brief Load and stack directory of FITS files.
         * @param inDirectory List of paths to directory of FITS files
         * @param params are stacking parameters
         * @return success (or not)
         */
        bool loadStack(const QStringList &inDirectory, const LiveStackData &params);

        /**
         * @brief User request to cancel stacking operation
         */
        void cancelStack();

        /**
         * @brief Load stack from buffer
         * @return A QFuture that can be watched until the async operation is complete.
         */
        QFuture<bool> loadStackBuffer();

        /**
         * @brief Solver results are in so take the next action
         * @param whether the solver timed out or not
         * @param success status
         * @param median hfr
         * @param number of stars
         */
        void solverDone(const bool timedOut, const bool success, const double hfr, const int numStars);

        /**
         * @brief Process new subs into an existing stack
         */
        void incrementalStack();

        /**
         * @brief Load WCS for a FITS file sub loaded during Live Stacking
         * @return success
         */
        bool stackLoadWCS();

        /**
         * @brief Create WCS params for a stack sub if they don't already exist. Indi subs
         * normally have WCS params set (if indi params set) but subs from other sources don't
         * always have these set, e.g. Seestar
         */
        void stackCreateWCSParams();

        /**
         * @brief injectStackWCS to inject a plate solved solution to WCS
         * @param orientation Solver orientation, degrees E of N.
         * @param ra J2000 Right Ascension
         * @param dec J2000 Declination
         * @param pixscale Pixel scale in arcsecs per pixe
         * @param eastToTheRight if true, then when the image is rotated so that north is up, then east would be to the right on the image.
         */
        void injectStackWCS(double orientation, double ra, double dec, double pixscale, bool eastToTheRight);

        /**
         * @brief setStackSubSolution saves the last plate solve sub solution for use with the next sub solution
         * @param ra J2000 Right Ascension
         * @param dec J2000 Declination
         * @param pixscale Pixel scale in arcsecs per pixel
         * @param indexUsed is the index file used
         * @param healpixUsed is the index file used
         */
        void setStackSubSolution(const double ra, const double dec, const double pixscale, const int index, const int healpix);

#if !defined (KSTARS_LITE)
        /**
         * @brief convertMatToFITS converts the passed in cv::Mat image to FITS
         * @param image to convert
         * @return success (or not)
         */
        bool convertMatToFITS(const cv::Mat &image);

        /**
         * @brief calcStackSNR is a utility function to calculate the SNR. Used by fitsdata and fitsstack
         * @param image
         * @return SNR
         */
        static double calcStackSNR(const cv::Mat &image);
#endif

        /**
         * @brief prepareStackBuffer musters stacks from different channels into a single displayable image
         * @return success (or not)
         */
        bool prepareStackBuffer();

        /**
         * @brief prepareStackBufferAsync runs prepareStackBuffer in the background
         */
        void prepareStackBufferAsync();

        /**
         * @brief isStackedImageEmpty determines if there is a displayable image or not
         * @return empty = true
         */
        bool isStackedImageEmpty() const
        {
            return !m_StackedBuffer || m_StackedBuffer->isEmpty();
        }

        /**
         * @brief getStackSNR returns the SNR of the stacked image
         * @return SNR
         */
        const double &getStackSNR() const
        {
            return m_StackSNR;
        }

        /**
         * @brief getLiveStackData returns the current live stack parameters, including outputDirectory
         * @return const reference to LiveStackData
         */
        const LiveStackData &getLiveStackData() const
        {
            return m_LiveStackData;
        }

        /**
         * @brief getStackSavedFrameCount returns how many stacked frames have been saved to outputDirectory
         * @return frame count
         */
        int getStackSavedFrameCount() const
        {
            return m_StackSavedFrameCount;
        }

        /**
         * @brief incrementStackSavedFrameCount increments the saved frame counter after a successful save
         */
        void incrementStackSavedFrameCount()
        {
            ++m_StackSavedFrameCount;
        }

        /**
         * @brief redo Post Processing on the existing stack (user requsted)
         * @param ppParams are the current user requested post processing params
         */
        void redoPostProcessStack(const LiveStackPPData &ppParams);

        /**
         * @brief Get the Live Stack Object, e.g. M51
         * @return object
         */
        LiveStackMetadata const &getLiveStackMetadata() const
        {
            return m_LiveStackMetadata;
        }

    Q_SIGNALS:
        void converted(QImage);

        /**
         * @brief histogramReady Sends signal when histogram construction is complete.
         */
        void histogramReady();

        /**
         * @brief dataChanged Send signal when undelying raw data buffer data changed.
         */
        void dataChanged();

        /**
         * @brief headerChanged Send signal when undelying header data changed.
         */
        void headerChanged();

        /**
         * @brief loadingCatalogData Send signal when starting an external catalog data query
         */
        void loadingCatalogData();
        /**
         * @brief loadedCatalogData Send signal when finished an external catalog data query
         */
        void loadedCatalogData();
        /**
         * @brief catalogQueryFailed Send signal when an external catalog data query fails
         * @param failure text
         */
        void catalogQueryFailed(const QString text);

        /**
         * @brief Signal FITSView then FITSTab to plate solve the current image sub
         */
        void plateSolveSub(const double ra, const double dec, const double pixScale, const int index,
                           const int healpix, const LiveStackFrameWeighting &weighting);

        /**
         * @brief Signal FITSView->FITSTab->FITSViewer that a stacking process is in operation
         */
        void stackInProgress();

        /**
         * @brief Signal FITSView then FITSTab that an align master sub has been chosen
         */
        void alignMasterChosen(const QString &alignMaster);

        /**
         * @brief Signal FITSView the stack is ready to load
         * @param whether the stacking operation has been cancelled
         */
        void stackReady(const bool cancelled = false);

        /**
         * @brief update FITSTab on progress
         * @param ok whether sub being processed was successful or not
         * @param sub just processed
         * @param total number of subs
         */
        void stackUpdateStats(const bool ok, const int sub, const int total, const double meanSNR, const double minSNR,
                              const double maxSNR);

        /**
         * @brief Initialise the Stack Monitor
         * @param timestamp
         * @param list of subframes
         */
        void initStackMon(QDateTime timestamp, const QVector<LiveStackFile> &subs);

        /**
         * @brief Add 1 or more new subs to the Stack Monitor
         * @param timestamp
         * @param list of subframes to add to the monitor
         */
        void addStackMon(QDateTime timestamp, const QVector<LiveStackFile> &subs);

        /**
         * @brief Update an existing sub in the Stack Monitor
         * @param Vector of subs being updated
         * @param vector of stats that contains update data fields
         */
        void updateStackMon(const QVector<LiveStackFile> &subs, const QVector<LiveStackStageInfo> &infos);

    public Q_SLOTS:
        void makeRoiBuffer(QRect roi);

        /**
         * @brief Called when a new file is added to the watched stack directory
         * @param File added to directory
         */
#if !defined (KSTARS_LITE)
        void newStackSubs(QDateTime timestamp, const QVector<LiveStackFile> &newFile);
#endif

    private:
        void loadCommon(const QString &inFilename);
        void releaseMemFileBuffer();
        void releaseStackMemFileBuffer();
        /**
         * @brief privateLoad Load an image (FITS, RAW, or images supported by Qt like jpeg, png).
         * @param Buffer pointer to image data. If buffer is emtpy, read from disk (m_Filename).
         * @return true if successfully loaded, false otherwise.
         */
        bool privateLoad(const QByteArray &buffer);

        // Load Qt-supported images.
        bool loadCanonicalImage(const QByteArray &buffer);
        // Load FITS images.
        bool loadFITSImage(const QByteArray &buffer, const bool isCompressed = false);
        // Load XISF images.
        bool loadXISFImage(const QByteArray &buffer);
        // Save XISF images.
        bool saveXISFImage(const QString &newFilename);
        // Load RAW images.
        bool loadRAWImage(const QByteArray &buffer);

        void rotWCSFITS(int angle, int mirror);
        void calculateMinMax(bool refresh = false, bool roi = false);
        void calculateMedian(bool refresh = false, bool roi = false);

        /**
         * @brief Default bayer parameters based on user configurable parameters (dc1394 / openCV, etc)
         * @param stack is true for live stacker subs, false for regular images
         * @return Configured BayerParameters
         */
        BayerParameters setupBayerParams(const bool stack = false);
        /**
         * @brief Setup bayer parameters based on input parameters
         * @param engine to set
         * @param pattern to set
         * @param algo to set
         * @param offsetX - to set
         * @param offsetY - to set
         * @return Configured BayerParameters
         */
        BayerParameters setupBayerParams(const DebayerEngine engine, const QString pattern, const QVariant algo,
                                         const int offsetX, const int offsetY);
        bool checkDebayerFITS();
        bool checkDebayerXISF(const QString pattern);
        void readWCSKeys();

        /**
             * @brief Update header info used by WCS based on solution info
             * @param orientation Solver orientation, degrees E of N.
             * @param ra J2000 Right Ascension
             * @param dec J2000 Declination
             * @param pixscale Pixel scale in arcsecs per pixel
             * @param eastToTheRight if true, then when the image is rotated so that north is up, then east would be to the right on the image.
             */
        void updateWCSHeaderData(const double orientation, const double ra, const double dec, const double pixscale,
                                 const bool eastToTheRight, const bool stack = false);

        /**
             * @brief Setup WCS parameters for non-FITS files so plate solved solutions can be used with catalog functionality.
             * @param stack whether to use stack header or normal header
             */
        void setupWCSParams(const bool stack = false);

        // Record last FITS error
        void recordLastError(int errorCode);
        void logOOMError(uint32_t requiredMemory = 0);

        // FITS Record
        bool parseHeader(const bool stack = false);
        //int getFITSRecord(QString &recordList, int &nkeys);

        // Templated functions
        template <typename T>
        bool debayer();

        template <typename T>
        bool rotFITS(int rotate, int mirror);

        // Apply Filter
        template <typename T>
        void applyFilter(FITSScale type, uint8_t *targetImage, QVector<double> * min = nullptr, QVector<double> * max = nullptr);

        template <typename T>
        void calculateMinMax(bool roi = false);
        template <typename T>
        void calculateMedian(bool roi = false);

        template <typename T>
        QPair<T, T> getParitionMinMax(uint32_t start, uint32_t stride, bool roi);

        /* Calculate the Gaussian blur matrix and apply it to the image using the convolution filter */
        QVector<double> createGaussianKernel(int size, double sigma);
        template <typename T>
        void convolutionFilter(const QVector<double> &kernel, int kernelSize);
        template <typename T>
        void gaussianBlur(int kernelSize, double sigma);

        /* Calculate running average & standard deviation */
        template <typename T>
        void calculateStdDev( bool roi = false );

        template <typename T>
        void convertToQImage(double dataMin, double dataMax, double scale, double zero, QImage &image);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Private Histogram Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        template <typename T>  void constructHistogramInternal();
        template <typename T> int32_t histogramBinInternal(T value, int channel) const;
        template <typename T> int32_t histogramBinInternal(int x, int y, int channel) const;

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Private Catalog Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /**
         * @brief search the current image for Skymap objects
         */
        bool searchSkyMapObjects();
        /**
         * @brief search the current image for skymap objects
         * @param startPoint the top left point on the sky map
         * @param endPoint the bottom right point on the sky map
         */
        bool findObjectsInImage(SkyPoint startPoint, SkyPoint endPoint);
        /**
         * @brief search Simbad for catalog objects in the current image
         * @param searchCenter is the center point for the search
         * @param search circle radius
         */
        bool findSimbadObjectsInImage(SkyPoint searchCenter, double radius);
        /**
         * @brief whether the Object Type passed in should be displayed
         * @param object type
         * @return whether the passed in object type should be displayed
         */
        bool getCatObjectFilter(const QString type) const;
        /**
         * @brief called when a Simbad query response has arrived
         * @param reply
         */
        void simbadResponseReady(QNetworkReply *reply);
        /**
         * @brief called when a Simbad query times out
         */
        void catTimeout();
        /**
         * @brief add catalog object to the array of objects
         * @param name of object
         * @param type of object
         * @param coord of object
         * @param dist of object from search point
         * @param magnitude of object (if available)
         * @param size of object (if available)
         * @return whether the add was successful
         */
        bool addCatObject(const int num, const QString name, const QString type, const QString coord, const double dist,
                          const double magnitude, const QString sizeStr);

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Private Live Stacking Functions.
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /**
         * @brief Setup which channels (mono, colour, etc.) are required for this stack
         * @param dirs is the array of directories to process
         * @param darks is the array of master darks to process
         * @param flats is the array of master flats to process
         * @return success (or not)
         */
        bool initStackChannels(const QStringList &dirs, const QVector<QString> &darks, const QVector<QString> &flats);

        /**
         * @brief Process alignment master file (load and plate solve)
         * @param alignMaster
         */
        void processAlignMaster(const QString &alignMaster);

        /**
         * @brief Process master files for stacking
         */
        void processMasters();

        /**
         * @brief Async callback when a stack FITS file load has completed
         */
        void stackFITSLoaded();

        /**
         * @brief Load a sub for Live Stacking
         * @param filename to open
         * @return success
         */
        bool stackLoadImage(QString filename);

        /**
         * @brief Load a FITS sub for Live Stacking
         * @param filename to open
         * @param isCompressed
         * @return success
         */
        bool stackLoadFITSImage(QString filename, const bool isCompressed = false);

        /**
         * @brief Load a XISF sub for Live Stacking
         * @param filename to open
         * @return success
         */
        bool stackLoadXISFImage(QString filename);

        /**
         * @brief stackCheckDebayerFITS checks whether a FITS stack sub needs to be debayered
         *        this routine doesn't affect the object variables used by checkDebayer
         * @param if debayering is successful, bayerParams contains the appropriate params
         * @return success
         */
        bool stackCheckDebayerFITS(BayerParameters &bayerParams);

        /**
         * @brief stackCheckDebayerXISF checks whether a XISF stack sub needs to be debayered
         *        this routine doesn't affect the object variables used by checkDebayer
         * @param pattern is the bayer pattern used
         * @param if debayering is successful, bayerParams contains the appropriate params
         * @return success
         */
        bool stackCheckDebayerXISF(const QString pattern, BayerParameters &bayerParams);

        /**
         * @brief stackDebayer debayers a stack sub
         * @param if debayering is successful, debayerParams contains the appropriate params
         * @return success
         */
        template <typename T>
        bool stackDebayer(BayerParameters &bayerParams);

        /**
         * @brief debayerCV debayers an image or stack sub using openCV
         * @param debayerParams
         * @param stack is true if input is a live stacker sub
         * @return success
         */
        template <typename T>
        bool debayerCV(BayerParameters &bayerParams, const bool stack = false);

        /**
         * @brief Verify that the debayer params are consistent for openCV processing
         * @param bayerParams are the input parameters
         * @param openCVParams is the output parameter structure
         * @return success
         */
        bool verifyCVDebayerParams(const BayerParameters &bayerParams, OpenCVParams &openCVParams);

        /**
         * @brief Verify that the debayer params are consistent for dc1394 processing
         * @param bayerParams are the input parameters
         * @param dc1394Params is the output parameter structure
         * @return success
         */
        bool verifyDC1394DebayerParams(const BayerParameters &bayerParams, DC1394Params &dc1394Params);

        /**
         * @brief Work out the next stacking action and start it off
         */
        void nextStackAction();

        /**
         * @brief Process the next sub
         * @param sub to process
         * @return success
         */
        bool processNextSub(LiveStackFile &sub);

        /**
         * @brief Callback to handle an asynchronous stacking operation completion
         */
        void stackProcessDone();

        /**
         * @brief Setup WCS for the image stack based on the master sub WCS
         */
        void stackSetupWCS();

        /**
         * @brief Manage the user cancel stack request within FITSData
         */
        void checkCancelStack();

#if !defined (KSTARS_LITE)
        /**
         * @brief Return the channel for the passed in stack
         * @param stack
         * @return channel
         */
        LiveStackChannel channelForStack(const QSharedPointer<FITSStack> &stack) const;

        /**
         * @brief Perform a Linear fit on the passed in channels (ref=g)
         * @param r - red channel balanced to g
         * @param g - green channel (reference)
         * @param b - blue channel balanced to g
         */
        void linearFit(cv::Mat &r, cv::Mat &g, cv::Mat &b);

        /**
         * @brief Perform a LRGB combination from the passed in L, R, G, B channels
         * @param lum - lum channel
         * @param r - red channel
         * @param g - green channel
         * @param b - blue channel
         * @return combined 3-channel image
         */
        cv::Mat combineLRGB(const cv::Mat &lum, cv::Mat &r, cv::Mat &g, cv::Mat &b);

        /**
         * @brief Initialise m_LiveStackMetadata structure
         * @param target - name of the target
         * @param exposure - sub exposure in sec
         */
        void initLiveStackMetadata(const QString target, const double exposure);

        /**
         * @brief Update m_LiveStackMetadata structure after each stack operation
         */
        void updateLiveStackMetadata();
#endif

        /// Mutex used by LiveStacker to protect updating data buffer
        mutable QMutex m_DataMutex;
        /// Pointer to CFITSIO FITS file struct
        fitsfile *fptr { nullptr };
        /// Generic data image buffer
        uint8_t *m_ImageBuffer { nullptr };
        /// Above buffer size in bytes
        uint32_t m_ImageBufferSize { 0 };
        /// Image Buffer if Selection is to be done
        uint8_t *m_ImageRoiBuffer { nullptr };
        /// Above buffer size in bytes
        uint32_t m_ImageRoiBufferSize { 0 };
        /// Is this a temporary file or one loaded from disk?
        bool m_isTemporary { false };
        /// is this file compress (.fits.fz)?
        bool m_isCompressed { false };
        /// Did we search for stars yet?
        bool starsSearched { false };
        ///Star Selection Algorithm
        StarAlgorithm starAlgorithm { ALGORITHM_GRADIENT };
        /// Do we have WCS keywords in this FITS data?
        bool HasWCS { false };        /// Do we have WCS keywords in this FITS data?
        /// Is the image debayarable?
        bool HasDebayer { false };
        /// Buffer to hold fpack uncompressed data
        uint8_t *m_PackBuffer {nullptr};
        size_t m_PackBufferSize {0};
        void *m_MemFileBuffer {nullptr};
        size_t m_MemFileBufferSize {0};
        bool m_MemFileBufferOwned {false};

        /// Our very own file name
        QString m_Filename, m_compressedFilename, m_Extension;
        /// FITS Mode (Normal, WCS, Guide, Focus..etc)
        FITSMode m_Mode;
        // FITS Observed UTC date time
        KStarsDateTime m_DateTime;

        /// How many times the image was rotated? Useful for WCS keywords rotation on save.
        int rotCounter { 0 };
        /// How many times the image was flipped horizontally?
        int flipHCounter { 0 };
        /// How many times the image was flipped vertically?
        int flipVCounter { 0 };

        /// WCS Struct
        struct wcsprm *m_WCSHandle
        {
            nullptr
        };
        /// Number of coordinate representations found.
        int m_nwcs {0};
        WCSState m_WCSState { Idle };
        /// All the stars we detected, if any.
        QList < Edge * > starCenters;
        QList < Edge * > localStarCenters;
        /// The biggest fattest star in the image.
        Edge m_SelectedHFRStar;

        /// Bayer parameters
        BayerParameters debayerParams;
        QTemporaryFile m_TemporaryDataFile;

        /// Data type of fits pixel in the image. Used when saving FITS again.
        /// There is bit depth and also data type. They're not the same.
        /// 16bit can be either SHORT_IMG or USHORT_IMG, so m_FITSBITPIX specifies which is
        int m_FITSBITPIX {USHORT_IMG};
        FITSImage::Statistic m_Statistics;
        FITSImage::Statistic m_ROIStatistics;

        // A list of header records
        QList < Record > m_HeaderRecords;

        QList < FITSSkyObject * > m_SkyObjects;
        QList < CatObject > m_CatObjects;
        QStringList m_CatObjectsFilters;
        QString m_CatObjQuery;
        bool m_ObjectsSearched {false};
        bool m_CatObjectsSearched {false};
        void resetCatObjectsSearched()
        {
            m_CatObjectsSearched = false;
        }

        FITSImage::Solution m_PlateSolveSolution { 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, FITSImage::POSITIVE, 0.0, 0.0 };

        QString m_LastError;

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Histogram Variables
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        QVector < QVector < uint32_t>> m_CumulativeFrequency;
        QVector < QVector < double>> m_HistogramIntensity;
        QVector < QVector < double>> m_HistogramFrequency;
        QVector < double > m_HistogramBinWidth;
        uint16_t m_HistogramBinCount { 0 };
        double m_JMIndex { 1 };
        bool m_HistogramConstructed { false };

        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        /// Star Detector
        ////////////////////////////////////////////////////////////////////////////////////////
        ////////////////////////////////////////////////////////////////////////////////////////
        // Sky Background
        SkyBackground m_SkyBackground;
        // Detector Settings
        QVariantMap m_SourceExtractorSettings;
        QFuture < bool > m_StarFindFuture;
        QScopedPointer < FITSStarDetector, QScopedPointerDeleteLater > m_StarDetector;

        // Cached values for hfr and eccentricity computations
        double cacheHFR { -1 };
        HFRType cacheHFRType { HFR_AVERAGE };
        double cacheEccentricity { -1 };
        QPoint roiCenter;

        // Catalog Objects
        QSharedPointer < QNetworkAccessManager > m_NetworkAccessManager;
        QTimer m_CatQueryTimer;
        bool m_CatQueryInProgress { false };
        bool m_CatUpdateTable { false };
        QPoint m_CatROIPt { -1, -1 };
        int m_CatROIRadius { -1 };

        // Live Stacking
#if !defined (KSTARS_LITE)
        // Each base channel has its own FITSStack
        QMap < LiveStackChannel, QSharedPointer < FITSStack>> m_Stacks;
        QSharedPointer < FITSStack > m_CurrentStack;
#endif
        QVector < LiveStackFile > m_StackSubs;
        int m_StackSubPos { -1 };
        bool m_StackMultiC { false };
        QMap < LiveStackChannel, QString > m_StackChannelMap;
        QMap < LiveStackChannel, QString > m_DarkChannelMap;
        QMap < LiveStackChannel, QString > m_FlatChannelMap;

        // StackChannelDir associates a LiveStackChannel with its input directory, and optionally links channels that share the
        // same source directory, e.g. if Red and Green both have the same dir, baseChannel for Green will be Red.
        struct StackChannelDir
        {
            LiveStackChannel channel;
            QString dir;
            LiveStackChannel baseChannel;

            StackChannelDir(LiveStackChannel c, const QString & d, LiveStackChannel base)
                : channel(c), dir(d), baseChannel(base) {}
        };
        QVector < StackChannelDir > m_StackChannelDirs;
        // Helper functions for StackChannelDir
        void insertChannel(const LiveStackChannel &c, const QString dir);
        QVector < LiveStackChannel > getBaseChannels();
        LiveStackChannel getBaseChannelForChannel(LiveStackChannel channel);
        QVector < QString > getUniqueStackDirs();
        void getChannelInfoforSub(const QString &sub, LiveStackChannel &base, QVector < LiveStackChannel > &channels);

        QSharedPointer < QByteArray > m_StackedBuffer { nullptr };
        double m_StackSNR { 0.0 };

        LiveStackData m_LiveStackData;
        QSharedPointer < FITSDirWatcher > m_StackDirWatcher;
        QQueue < LiveStackFile > m_StackQ;
        bool m_AlignMasterChosen { false };
        bool m_AlignMasterProcessed { false };
        LiveStackMetadata m_LiveStackMetadata;
        QMap < LiveStackChannel, bool > m_DarksLoadedMap;
        QMap < LiveStackChannel, bool > m_FlatsLoadedMap;
        uint8_t *m_StackImageBuffer { nullptr };
        uint32_t m_StackImageBufferSize { 0 };
        typedef struct
        {
            FITSImage::Statistic stats;
            int cvType;
        } StackStatistics;
        StackStatistics m_StackStatistics;
        struct wcsprm *m_StackWCSHandle
        {
            nullptr
        };
        int m_Stacknwcs { 0 };
        QSharedPointer < wcsprm > m_StackAlignMasterWCS;
        fitsfile *m_Stackfptr { nullptr };
        void *m_StackMemFileBuffer {nullptr};
        size_t m_StackMemFileBufferSize {0};
        bool m_StackMemFileBufferOwned {false};
        QList < Record > m_StackHeaderRecords;
        QFutureWatcher < bool > m_StackWatcher;
        QFutureWatcher < bool > m_StackFITSWatcher;
        QFuture < void > m_StackPrepareFuture;
        typedef enum
        {
            stackFITSNone,
            stackFITSAlignMaster,
            stackFITSDark,
            stackFITSFlat,
            stackFITSSub
        } StackFITSAsyncType;
        StackFITSAsyncType m_StackFITSAsync { stackFITSNone };
        bool m_CancelRequest { false }; // Overall control variable for cancelling a stack operation
        bool m_StackWatcherCancel { false }; // Control variable to cancel background thread stack operation
        bool m_StackFITSWatcherCancel { false }; // Control variable to cancel background thread FITS Load operation
        double m_StackSubRa { 0.0 };
        double m_StackSubDec { 0.0 };
        double m_StackSubPixscale { 0.0 };
        int m_StackSubIndex { 0 };
        int m_StackSubHealpix { 0 };
        int m_StackSavedFrameCount { 0 };  // Counter for saved stacked frames (EkosLive)
};
