File indexing completed on 2024-04-21 14:45:57

 /*
     SPDX-FileCopyrightText: 2017 Jasem Mutlaq <mutlaqja@ikarustech.com>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 
 #include "opships.h"
 
 #include "kstars.h"
 #include "hipsmanager.h"
 #include "Options.h"
 #include "skymap.h"
 #include "auxiliary/ksnotification.h"
 #include "auxiliary/filedownloader.h"
 #include "auxiliary/kspaths.h"
 
 #include <KConfigDialog>
 
 #include <QCheckBox>
 #include <QComboBox>
 #include <QFileDialog>
 #include <QPushButton>
 #include <QStringList>
 
 // Qt version calming
 #include <qtkeepemptyparts.h>
 
 static const QStringList hipsKeys = { "ID", "obs_title", "obs_description", "hips_order", "hips_frame", "hips_tile_width", "hips_tile_format", "hips_service_url", "moc_sky_fraction"};
 
 OpsHIPSDisplay::OpsHIPSDisplay() : QFrame(KStars::Instance())
 {
     setupUi(this);
 }
 
 OpsHIPSCache::OpsHIPSCache() : QFrame(KStars::Instance())
 {
     setupUi(this);
 
     connect(selectDirectoryB, &QPushButton::clicked, this, [this]()
     {
         QString dir = QFileDialog::getExistingDirectory(this, i18nc("@title:window", "HiPS Offline Storage"),
                       kcfg_HIPSOfflinePath->text());
 
         if (dir.isEmpty())
             return;
 
         kcfg_HIPSOfflinePath->setText(dir);
 
         QDir hipsDirectory(dir);
         auto orders = hipsDirectory.entryList(QDir::AllDirs | QDir::NoDotAndDotDot);
         HIPSManager::Instance()->setOfflineLevels(orders);
         HIPSManager::Instance()->setCurrentSource("DSS Colored");
     });
 }
 
 OpsHIPS::OpsHIPS() : QFrame(KStars::Instance())
 {
     setupUi(this);
 
     //Get a pointer to the KConfigDialog
     m_ConfigDialog = KConfigDialog::exists("hipssettings");
 
     QString path = QDir(KSPaths::writableLocation(QStandardPaths::AppLocalDataLocation)).filePath(
                        QLatin1String("hips_previews/"));
     QDir dir;
     dir.mkpath(path);
 
     connect(refreshSourceB, SIGNAL(clicked()), this, SLOT(slotRefresh()));
 
     connect(sourcesList, SIGNAL(itemChanged(QListWidgetItem*)), this, SLOT(slotItemUpdated(QListWidgetItem*)));
     connect(sourcesList, SIGNAL(itemClicked(QListWidgetItem*)), this, SLOT(slotItemClicked(QListWidgetItem*)));
 
     if (sourcesList->count() == 0)
         slotRefresh();
 }
 
 void OpsHIPS::slotRefresh()
 {
     downloadJob = new FileDownloader();
 
     downloadJob->setProgressDialogEnabled(true, i18n("HiPS Update"), i18n("Downloading HiPS sources..."));
 
     QObject::connect(downloadJob, SIGNAL(downloaded()), this, SLOT(downloadReady()));
     QObject::connect(downloadJob, SIGNAL(error(QString)), this, SLOT(downloadError(QString)));
 
     downloadJob->get(
         QUrl("http://alasky.unistra.fr/MocServer/query?hips_service_url=*&dataproduct_type=!catalog&dataproduct_type=!cube&&moc_sky_fraction=1&get=record"));
 }
 
 void OpsHIPS::downloadReady()
 {
     sources.clear();
 
     QTextStream stream(downloadJob->downloadedData());
 
     QStringList hipsTitles;
 
     QMap<QString, QString> oneSource;
     while (stream.atEnd() == false)
     {
         QString line = stream.readLine();
         if (line.isEmpty())
         {
             sources.append(oneSource);
             oneSource.clear();
             continue;
         }
 
         QStringList keyvalue = line.split('=', Qt::KeepEmptyParts);
         QString key   = keyvalue[0].simplified();
         if (hipsKeys.contains(key) == false)
             continue;
         QString value = keyvalue[1].simplified();
         oneSource[key] = value;
         if (key == "obs_title")
             hipsTitles << value;
     }
 
     // Get existing database sources
     QList<QMap<QString, QString>> dbSources;
     KStarsData::Instance()->userdb()->GetAllHIPSSources(dbSources);
 
     // Get existing database titles
     QStringList dbTitles;
     for (QMap<QString, QString> oneSource : dbSources)
         dbTitles << oneSource["obs_title"];
 
     // Add all titles to list widget
     sourcesList->addItems(hipsTitles);
     QListWidgetItem* item = nullptr;
 
     // Make sources checkable and check sources that already exist in the database
     sourcesList->blockSignals(true);
     for(int i = 0; i < sourcesList->count(); ++i)
     {
         item = sourcesList->item(i);
         item->setFlags(item->flags() | Qt::ItemIsUserCheckable);
         item->setCheckState(dbTitles.contains(item->text()) ? Qt::Checked : Qt::Unchecked);
 
         if (item->text() == Options::hIPSSource())
         {
             item->setSelected(true);
             sourcesList->scrollToItem(item);
             slotItemClicked(item);
         }
     }
     sourcesList->blockSignals(false);
 
     // Delete job later
     downloadJob->deleteLater();
 }
 
 void OpsHIPS::downloadError(const QString &errorString)
 {
     KSNotification::error(i18n("Error downloading HiPS sources: %1", errorString));
     downloadJob->deleteLater();
 }
 
 void OpsHIPS::slotItemUpdated(QListWidgetItem *item)
 {
     for(QMap<QString, QString> &oneSource : sources)
     {
         if (oneSource.value("obs_title") == item->text())
         {
             if (item->checkState() == Qt::Checked)
                 KStarsData::Instance()->userdb()->AddHIPSSource(oneSource);
             else
                 KStarsData::Instance()->userdb()->DeleteHIPSSource(oneSource.value("ID"));
             break;
         }
     }
 }
 
 void OpsHIPS::slotItemClicked(QListWidgetItem *item)
 {
     for(QMap<QString, QString> &oneSource : sources)
     {
         if (oneSource.value("obs_title") == item->text())
         {
             sourceDescription->setText(oneSource.value("obs_description"));
             // Get stored preview, if not found, it will be downloaded.
             setPreview(oneSource.value("ID"), oneSource.value("hips_service_url"));
             break;
         }
     }
 }
 
 void OpsHIPS::setPreview(const QString &id, const QString &url)
 {
     uint hash = qHash(id);
     QString previewName = QString("%1.jpg").arg(hash);
 
     QString currentPreviewPath = QDir(KSPaths::locate(QStandardPaths::AppLocalDataLocation,
                                       QLatin1String("hips_previews"))).filePath(previewName);
     if (currentPreviewPath.isEmpty() == false)
     {
         sourceImage->setPixmap(QPixmap(currentPreviewPath));
     }
     else
     {
         currentPreviewPath = QDir(KSPaths::writableLocation(QStandardPaths::AppLocalDataLocation)).filePath(
                                  QLatin1String("hips_previews/") + previewName);
 
         previewJob = new FileDownloader();
         connect(previewJob, SIGNAL(downloaded()), this, SLOT(previewReady()));
 
         previewJob->setDownloadedFileURL(QUrl::fromLocalFile(currentPreviewPath));
 
         previewJob->get(QUrl(url + QLatin1String("/preview.jpg")));
     }
 }
 
 void OpsHIPS::previewReady()
 {
     QString previewFile = previewJob->getDownloadedFileURL().toLocalFile();
     QFileInfo previewInfo(previewFile);
     // If less than 1kb then it's junk
     if (previewInfo.size() < 1024)
     {
         sourceImage->setPixmap(QPixmap(":/images/noimage.png"));
         QFile::remove(previewFile);
     }
     else
         sourceImage->setPixmap(QPixmap(previewFile));
 }