File indexing completed on 2024-04-21 14:46:14

 /*
     SPDX-FileCopyrightText: 2016 Artem Fedoskin <afedoskin3@gmail.com>
 
     SPDX-License-Identifier: GPL-2.0-or-later
 */
 
 #include "finddialoglite.h"
 
 #include "catalogentrydata.h"
 #include "deepskyobject.h"
 #include "kstarsdata.h"
 #include "kstarslite.h"
 #include "skymaplite.h"
 #include "skyobjectlistmodel.h"
 #include "skycomponents/skymapcomposite.h"
 #include "skyobjects/skyobject.h"
 #include "tools/nameresolver.h"
 
 #include <QQmlContext>
 #include <QSortFilterProxyModel>
 #include <QTimer>
 
 FindDialogLite::FindDialogLite()
 {
     m_filterModel.append(i18n("Any"));
     m_filterModel.append(i18n("Stars"));
     m_filterModel.append(i18n("Solar System"));
     m_filterModel.append(i18n("Open Clusters"));
     m_filterModel.append(i18n("Globular Clusters"));
     m_filterModel.append(i18n("Gaseous Nebulae"));
     m_filterModel.append(i18n("Planetary Nebulae"));
     m_filterModel.append(i18n("Galaxies"));
     m_filterModel.append(i18n("Comets"));
     m_filterModel.append(i18n("Asteroids"));
     m_filterModel.append(i18n("Constellations"));
     m_filterModel.append(i18n("Supernovae"));
     m_filterModel.append(i18n("Satellites"));
     emit filterModelChanged();
 
     fModel      = new SkyObjectListModel(this);
     m_sortModel = new QSortFilterProxyModel(this);
     m_sortModel->setFilterCaseSensitivity(Qt::CaseInsensitive);
     m_sortModel->setSourceModel(fModel);
     m_sortModel->setSortRole(Qt::DisplayRole);
     m_sortModel->setFilterRole(Qt::DisplayRole);
     m_sortModel->setDynamicSortFilter(true);
     KStarsLite::Instance()->qmlEngine()->rootContext()->setContextProperty("SortModel", m_sortModel);
     m_sortModel->sort(0);
 }
 
 FindDialogLite::~FindDialogLite()
 {
 }
 
 void FindDialogLite::filterByType(uint typeIndex)
 {
     KStarsData *data = KStarsData::Instance();
     m_typeIndex      = typeIndex;
 
     switch (typeIndex)
     {
         case 0: // All object types
         {
             QVector<QPair<QString, const SkyObject *>> allObjects;
             foreach (int type, data->skyComposite()->objectLists().keys())
             {
                 allObjects.append(data->skyComposite()->objectLists(SkyObject::TYPE(type)));
             }
             fModel->setSkyObjectsList(allObjects);
             break;
         }
         case 1: //Stars
         {
             QVector<QPair<QString, const SkyObject *>> starObjects;
             starObjects.append(data->skyComposite()->objectLists(SkyObject::STAR));
             starObjects.append(data->skyComposite()->objectLists(SkyObject::CATALOG_STAR));
             fModel->setSkyObjectsList(starObjects);
             break;
         }
         case 2: //Solar system
         {
             QVector<QPair<QString, const SkyObject *>> ssObjects;
             ssObjects.append(data->skyComposite()->objectLists(SkyObject::PLANET));
             ssObjects.append(data->skyComposite()->objectLists(SkyObject::COMET));
             ssObjects.append(data->skyComposite()->objectLists(SkyObject::ASTEROID));
             ssObjects.append(data->skyComposite()->objectLists(SkyObject::MOON));
 
             fModel->setSkyObjectsList(ssObjects);
             break;
         }
         case 3: //Open Clusters
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::OPEN_CLUSTER));
             break;
         case 4: //Globular Clusters
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::GLOBULAR_CLUSTER));
             break;
         case 5: //Gaseous nebulae
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::GASEOUS_NEBULA));
             break;
         case 6: //Planetary nebula
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::PLANETARY_NEBULA));
             break;
         case 7: //Galaxies
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::GALAXY));
             break;
         case 8: //Comets
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::COMET));
             break;
         case 9: //Asteroids
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::ASTEROID));
             break;
         case 10: //Constellations
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::CONSTELLATION));
             break;
         case 11: //Supernovae
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::SUPERNOVA));
             break;
         case 12: //Satellites
             fModel->setSkyObjectsList(data->skyComposite()->objectLists(SkyObject::SATELLITE));
             break;
     }
 }
 
 void FindDialogLite::filterList(QString searchQuery)
 {
     m_searchQuery = searchQuery;
     setIsResolveEnabled(!isInList(searchQuery));
     QString SearchText = processSearchText(searchQuery);
     m_sortModel->setFilterFixedString(SearchText);
     listFiltered = true;
 }
 
 void FindDialogLite::selectObject(int index)
 {
     QVariant sObj     = m_sortModel->data(m_sortModel->index(index, 0), SkyObjectListModel::SkyObjectRole);
     SkyObject *skyObj = (SkyObject *)sObj.value<void *>();
     SkyMapLite::Instance()->slotSelectObject(skyObj);
 }
 
 // Process the search box text to replace equivalent names like "m93" with "m 93"
 QString FindDialogLite::processSearchText(QString text)
 {
     QRegExp re;
     QString searchtext = text;
 
     re.setCaseSensitivity(Qt::CaseInsensitive);
 
     // If it is an NGC/IC/M catalog number, as in "M 76" or "NGC 5139", check for absence of the space
     re.setPattern("^(m|ngc|ic)\\s*\\d*$");
     if (text.contains(re))
     {
         re.setPattern("\\s*(\\d+)");
         searchtext.replace(re, " \\1");
         re.setPattern("\\s*$");
         searchtext.remove(re);
         re.setPattern("^\\s*");
         searchtext.remove(re);
     }
 
     // TODO after KDE 4.1 release:
     // If it is a IAU standard three letter abbreviation for a constellation, then go to that constellation
     // Check for genetive names of stars. Example: alp CMa must go to alpha Canis Majoris
 
     return searchtext;
 }
 
 bool FindDialogLite::isInList(QString searchQuery)
 {
     if (searchQuery.isEmpty())
         return false;
 
     int size         = m_sortModel->rowCount(m_sortModel->index(0, 0));
     QString stripped = searchQuery.remove(' ');
 
     for (int i = 0; i < size; ++i)
     {
         QString s = m_sortModel->data(m_sortModel->index(i, 0), Qt::DisplayRole).toString();
 
         if (s == searchQuery || s.remove(' ') == stripped)
             return true;
     }
     return false;
 }
 
 void FindDialogLite::resolveInInternet(QString searchQuery)
 {
     emit notifyMessage(i18n("Not Implemented.", searchQuery));
 }
 
 void FindDialogLite::setIsResolveEnabled(bool isResolveEnabled)
 {
     if (m_isResolveEnabled != isResolveEnabled)
     {
         m_isResolveEnabled = isResolveEnabled;
         emit isResolveEnabledChanged(isResolveEnabled);
     }
 }