开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > OpenGL > 查看源码
selectionpopup.cpp
资源名称：celestia-1.6.0.tar.gz　[点击查看]
上传用户：center1979
上传日期：2022-07-26
资源大小：50633k
文件大小：18k
源码类别：

OpenGL

开发平台：

Visual C++

selectionpopup.cpp：源码内容
  1. /***************************************************************************
  2.                           selectionpopup.cpp  -  description
  3.                              -------------------
  4.     begin                : 2003-05-06
  5.     copyright            : (C) 2002 by Christophe Teyssier
  6.     email                : chris@teyssier.org
  7.  ***************************************************************************/
  9. /***************************************************************************
  10.  *                                                                         *
  11.  *   This program is free software; you can redistribute it and/or modify  *
  12.  *   it under the terms of the GNU General Public License as published by  *
  13.  *   the Free Software Foundation; either version 2 of the License, or     *
  14.  *   (at your option) any later version.                                   *
  15.  *                                                                         *
  16.  ***************************************************************************/
  19. #include <libintl.h>
  20. #include <sstream>
  21. #include <fstream>
  22. #include <algorithm>
  24. #include "selectionpopup.h"
  26. #include <qlabel.h>
  28. #include <klocale.h>
  29. #include <kurl.h>
  30. #include <krun.h>
  32. #include "celutil/utf8.h"
  33. #include "celestia/celestiacore.h"
  34. #include "celengine/axisarrow.h"
  35. #include "celengine/planetgrid.h"
  38. SelectionPopup::SelectionPopup(QWidget* parent, CelestiaCore* _appCore, Selection _sel):
  39. KPopupMenu(parent),
  40.  appCore(_appCore),
  41.  sel(_sel),
  42.  baseId(0)
  43. {
  44. }
  46. SelectionPopup::~SelectionPopup() {
  48. }
  50. void SelectionPopup::init()
  51. {
  52.     QPalette pal( palette() );
  53.     pal.setColor( QPalette::Normal, QColorGroup::Background, QColor( "White" ) );
  54.     pal.setColor( QPalette::Normal, QColorGroup::Foreground, QColor( "Black" ) );
  55.     pal.setColor( QPalette::Inactive, QColorGroup::Foreground, QColor( "Black" ) );
  57.     QFont rsFont = font();
  58.     rsFont.setPointSize( rsFont.pointSize() - 2 );
  60.     Simulation *sim = appCore->getSimulation();
  61.     Vec3d v = sel.getPosition(sim->getTime()) - sim->getObserver().getPosition();
  63.     if (sel.body() != NULL)
  64.     {
  65.         insertTitle(QString::fromUtf8(sel.body()->getName(true).c_str()), 0, 0);
  66.         insert(this, sel);
  67.     }
  68.     else if (sel.star() != NULL)
  69.     {
  70.         std::string name = sim->getUniverse()->getStarCatalog()->getStarName(*sel.star());
  72.         double distance = v.length() * 1e-6;
  73.         char buff[50];
  75.         ostringstream o;
  76.         setlocale(LC_NUMERIC, "");
  78.         if (abs(distance) >= astro::AUtoLightYears(1000.0f))
  79.             sprintf(buff, "%'.3f %s", distance, _("ly"));
  80.         else if (abs(distance) >= astro::kilometersToLightYears(10000000.0))
  81.             sprintf(buff, "%'.3f %s", astro::lightYearsToAU(distance), _("au"));
  82.         else if (abs(distance) > astro::kilometersToLightYears(1.0f))
  83.             sprintf(buff, "%'.3f km", astro::lightYearsToKilometers(distance));
  84.         else
  85.             sprintf(buff, "%'.3f m", astro::lightYearsToKilometers(distance) * 1000.0f);
  87.         o << i18n("Distance: ") << QString::fromUtf8(buff) << "n";
  89.         o << i18n("Abs (app) mag: ");
  91.         sprintf(buff, "%'.2f (%.2f)",
  92.                    sel.star()->getAbsoluteMagnitude(),
  93.                    astro::absToAppMag(sel.star()->getAbsoluteMagnitude(),
  94.                                       (float) distance));
  95.         o << QString::fromUtf8(buff) << "n";
  97.         o << i18n("Class: ");
  98.         sprintf(buff, "%s", sel.star()->getSpectralType());
  99.         o << QString::fromUtf8(buff) << "n";
  101.         o << i18n("Surface Temp: ");
  102.         sprintf(buff, "%'.0f K", sel.star()->getTemperature());
  103.         o << QString::fromUtf8(buff) << "n";
  105.         o << i18n("Radius: ");
  106.         sprintf(buff, "%'.2f %s", sel.star()->getRadius() / 696000.0f, _("Rsun"));
  107.         o << QString::fromUtf8(buff);
  109.         setlocale(LC_NUMERIC, "C");
  111.         QLabel *starDetails = new QLabel(QString(o.str().c_str()), this);
  112.         starDetails->setPalette(pal);
  113.         starDetails->setFont(rsFont);
  115.         insertTitle(QString::fromUtf8(ReplaceGreekLetterAbbr(name).c_str()), 0, 0);
  116.         insertItem(starDetails);
  117.         insertSeparator();
  118.         insert(this, sel);
  119.     }
  120.     else if (sel.deepsky() != NULL)
  121.     {
  122.         insertTitle(QString::fromUtf8(sim->getUniverse()->getDSOCatalog()->getDSOName(sel.deepsky()).c_str()), 0);
  123.         insert(this, sel);
  124.     }
  126.     if (sim->getUniverse() != NULL)
  127.     {
  128.         MarkerList* markers = sim->getUniverse()->getMarkers();
  129.         if (markers->size() > 0)
  130.         {
  131.             KPopupMenu *markMenu = new KPopupMenu(this);
  132.             int j=1;
  133.             for (std::vector<Marker>::iterator i = markers->begin(); i < markers->end(); i++)
  134.             {
  135.                 KPopupMenu *objMenu = new KPopupMenu(this);
  136.                 insert(objMenu, (*i).object(), false);
  137.                 markMenu->insertItem(QString::fromUtf8(getSelectionName((*i).object())), objMenu);
  138.                 j++;
  139.             }
  140.             insertItem(i18n("Marked objects"), markMenu);
  141.         }
  142.     }
  143. }
  145. void SelectionPopup::process(int id)
  146. {
  147.     if (id < 0) return;
  149.     Simulation *sim = appCore->getSimulation();
  151.     sel = getSelectionFromId(id);
  152.     int actionId = id - baseId;
  154.     if (actionId == 1) {
  155.         sim->setSelection(sel);
  156.         return;
  157.     }
  158.     if (actionId == 2) {
  159.         sim->setSelection(sel);
  160.         appCore->charEntered('c');
  161.         return;
  162.     }
  163.     if (actionId == 3) {
  164.         sim->setSelection(sel);
  165.         appCore->charEntered('g');
  166.         return;
  167.     }
  168.     if (actionId == 4) {
  169.         sim->setSelection(sel);
  170.         appCore->charEntered('f');
  171.         return;
  172.     }
  173.     if (actionId == 5) {
  174.         sim->setSelection(sel);
  175.         appCore->charEntered('y');
  176.         return;
  177.     }
  178.     if (actionId == 6) {
  179.         QString url;
  180.         if (sel.body() != NULL)
  181.         {
  182.             url = QString(sel.body()->getInfoURL().c_str());
  183.             if (url == "")
  184.             {
  185.                 QString name = QString(sel.body()->getName().c_str()).lower();
  186.                 url = QString("http://www.nineplanets.org/") + name + ".html";
  187.             }
  188.         }
  189.         else if (sel.star() != NULL)
  190.         {
  191.             if (sel.star()->getCatalogNumber() != 0)
  192.                 url = QString("http://simbad.u-strasbg.fr/sim-id.pl?protocol=html&Ident=HIP %1")
  193.                       .arg(sel.star()->getCatalogNumber() & ~0xf0000000);
  194.             else
  195.                 url = QString("http://www.nineplanets.org/sun.html");
  196.         }
  197.         else if (sel.deepsky() != NULL)
  198.         {
  199.             url = QString(sel.deepsky()->getInfoURL().c_str());
  200.             if (url == "")
  201.                 url = QString("http://simbad.u-strasbg.fr/sim-id.pl?protocol=html&Ident=%1")
  202.                       .arg(sim->getUniverse()->getDSOCatalog()->getDSOName(sel.deepsky()).c_str());
  203.         }
  204.         KRun::runURL(url, "text/html");
  205.         return;
  206.     }
  207.     if (actionId == 7)
  208.     {
  209.         if (sim->getUniverse() != NULL)
  210.             sim->getUniverse()->unmarkObject(sel, 1);
  211.         return;
  212.     }
  213.     if (actionId == 8)
  214.     {
  215.         sim->getUniverse()->unmarkAll();
  216.         return;
  217.     }
  218.     if (actionId >= 10 && actionId < 25)
  219.     {
  220.         if (sim->getUniverse() != NULL)
  221.         {
  222.             MarkerRepresentation markerRep((MarkerRepresentation::Symbol) (actionId - 1), 10.0f, Color(0.0f, 1.0f, 0.0f, 0.9f));
  224.             sim->getUniverse()->markObject(sel, markerRep, 1);
  225.         }
  226.         return;
  227.     }
  228.     if (actionId >=25 && actionId < 31 && sel.body() != NULL) 
  229.     {
  231.         switch(actionId)
  232.         {
  233.         case 25:
  234. appCore->toggleReferenceMark("body axes");
  235.             break;
  236.         case 26:
  237. appCore->toggleReferenceMark("frame axes");
  238.             break;
  239.         case 27:
  240. appCore->toggleReferenceMark("sun direction");
  241.             break;
  242.         case 28:
  243. appCore->toggleReferenceMark("velocity vector");
  244.             break;
  245. case 29:
  246. appCore->toggleReferenceMark("planetographic grid");
  247.     break;
  248. case 30:
  249. appCore->toggleReferenceMark("terminator");
  250.     break;
  251.         }
  253.     }
  254.     if (actionId == 31) {
  255.         sim->getActiveObserver()->setDisplayedSurface("");
  256.         return;
  257.     }
  258.     if (actionId > 31) {
  259.         std::vector<std::string>* altSurfaces = sel.body()->getAlternateSurfaceNames();
  260.         if (altSurfaces != NULL && (int) altSurfaces->size() > actionId - 32)
  261.         {
  262.             sim->getActiveObserver()->setDisplayedSurface((*altSurfaces)[actionId - 32]);
  263.         }
  264.     }
  265. }
  267. const char* SelectionPopup::getSelectionName(const Selection& sel) const
  268. {
  269.     if (sel.body() != NULL)
  270.         return sel.body()->getName(true).c_str();
  271.     else if (sel.star() != NULL)
  272.         return ReplaceGreekLetterAbbr(appCore->getSimulation()->getUniverse()->getStarCatalog()->getStarName(*(sel.star()))).c_str();
  273.     else if (sel.deepsky() != NULL)
  274.         return appCore->getSimulation()->getUniverse()->getDSOCatalog()->getDSOName(sel.deepsky()).c_str();
  276.     return "";
  277. }
  279. Selection SelectionPopup::getSelectionFromId(int id)
  280. {
  281.     for(std::vector< std::pair<int, Selection> >::const_iterator i = baseIds.begin() + 1;
  282.         i != baseIds.end();
  283.         ++i) {
  284.         if ((*i).first > id) {
  285.             baseId = (*(i-1)).first;
  286.             return (*(i-1)).second;
  287.         }
  288.     }
  289.     baseId = baseIds.back().first;
  290.     return baseIds.back().second;
  291. }
  293. void SelectionPopup::insert(KPopupMenu *popup, Selection sel, bool showSubObjects)
  294. {
  295.     baseIds.push_back(make_pair(baseId, sel));
  296.     int locBaseId = baseId;
  298.     popup->insertItem(i18n("&Select"), baseId + 1);
  299.     popup->insertItem(i18n("&Center"), baseId + 2);
  300.     popup->insertItem(i18n("&Goto"), baseId + 3);
  301.     popup->insertItem(i18n("&Follow"), baseId + 4);
  302.     if (sel.star() == NULL && sel.deepsky() == NULL)
  303.         popup->insertItem(i18n("S&ynch Orbit"), baseId + 5);
  304.     popup->insertItem(i18n("&Info"), baseId + 6);
  305.     if (baseId == 0)
  306.         popup->insertItem(i18n("Unmark &All"), baseId + 8);
  308.     if (appCore->getSimulation()->getUniverse()->isMarked(sel, 1))
  309.     {
  310.         popup->insertItem(i18n("&Unmark"), baseId + 7);
  311.     }
  312.     else
  313.     {
  314.         KPopupMenu *markMenu = new KPopupMenu(this);
  315.         markMenu->insertItem(i18n("Diamond"), baseId + 10);
  316.         markMenu->insertItem(i18n("Triangle"), baseId + 11);
  317.         markMenu->insertItem(i18n("Square"), baseId + 12);
  318.         markMenu->insertItem(i18n("Filled Square"), baseId + 13);
  319.         markMenu->insertItem(i18n("Plus"), baseId + 14);
  320.         markMenu->insertItem(i18n("X"), baseId + 15);
  321.         markMenu->insertItem(i18n("Left Arrow"), baseId + 16);
  322.         markMenu->insertItem(i18n("Right Arrow"), baseId + 17);
  323.         markMenu->insertItem(i18n("Up Arrow"), baseId + 18);
  324.         markMenu->insertItem(i18n("Down Arrow"), baseId + 19);
  325.         markMenu->insertItem(i18n("Circle"), baseId + 20);
  326.         markMenu->insertItem(i18n("Disk"), baseId + 21);
  327.         popup->insertItem(i18n("&Mark"), markMenu);
  328.     }
  330.     if (sel.body() != NULL) {
  331.         KPopupMenu *refVectorMenu = new KPopupMenu(this);
  332.         refVectorMenu->setCheckable(true);
  333.         popup->insertItem(i18n("&Reference Vectors"), refVectorMenu);
  334.         refVectorMenu->insertItem(i18n("Show Body Axes"), baseId + 25);
  335.         refVectorMenu->setItemChecked(baseId + 25, sel.body()->findReferenceMark("body axes"));
  336.         refVectorMenu->insertItem(i18n("Show Frame Axes"), baseId + 26);
  337.         refVectorMenu->setItemChecked(baseId + 26, sel.body()->findReferenceMark("frame axes"));
  338.         refVectorMenu->insertItem(i18n("Show Sun Direction"), baseId + 27);
  339.         refVectorMenu->setItemChecked(baseId + 27, sel.body()->findReferenceMark("sun direction"));
  340.         refVectorMenu->insertItem(i18n("Show Velocity Vector"), baseId + 28);
  341.         refVectorMenu->setItemChecked(baseId + 28, sel.body()->findReferenceMark("velocity vector"));
  342.         refVectorMenu->insertItem(i18n("Show Planetographic Grid"), baseId + 29);
  343.         refVectorMenu->setItemChecked(baseId + 29, sel.body()->findReferenceMark("planetographic grid"));
  344.         refVectorMenu->insertItem(i18n("Show Terminator"), baseId + 30);
  345.         refVectorMenu->setItemChecked(baseId + 30, sel.body()->findReferenceMark("terminator"));
  346.     }
  348.     baseId += 31;
  350.     if (showSubObjects && sel.body() != NULL)
  351.     {
  352.         std::vector<std::string>* altSurfaces = sel.body()->getAlternateSurfaceNames();
  353.         if (altSurfaces != NULL)
  354.         {
  355.             if (!altSurfaces->empty())
  356.             {
  357.                 KPopupMenu *surfaces = new KPopupMenu(this);
  358.                 surfaces->insertItem(i18n("Normal"), locBaseId + 31);
  359.                 int j=0;
  360.                 for (std::vector<std::string>::const_iterator i = altSurfaces->begin();
  361.                      i < altSurfaces->end(); i++, j++)
  362.                 {
  363.                     surfaces->insertItem(QString((*i).c_str()), locBaseId + 32 + j);
  364.                 }
  365.                 baseId += 7 + j;
  366.                 popup->insertItem(i18n("&Alternate Surfaces"), surfaces);
  367.             }
  368.             delete altSurfaces;
  369.         }
  371.         const PlanetarySystem* satellites = sel.body()->getSatellites();
  372.         if (satellites != NULL && satellites->getSystemSize() != 0)
  373.         {
  374.             insertPlanetaryMenu(popup, getSelectionName(sel), satellites);
  375.         }
  376.     }
  377.     else if (showSubObjects && sel.star() != NULL)
  378.     {
  379.         Simulation *sim = appCore->getSimulation();
  380.         SolarSystemCatalog* solarSystemCatalog = sim->getUniverse()->getSolarSystemCatalog();
  381.         SolarSystemCatalog::iterator iter = solarSystemCatalog->find(sel.star()->getCatalogNumber());
  382.         if (iter != solarSystemCatalog->end())
  383.         {
  384.             SolarSystem* solarSys = iter->second;
  385.             insertPlanetaryMenu(popup, getSelectionName(sel), solarSys->getPlanets());
  386.         }
  387.     }
  388. }
  390. struct BodyComparePredicate
  391. {
  392.     bool operator()(Body* a, Body* b)
  393.     {
  394.         return (a->getName(true).compare(b->getName(true)) < 0);
  395.     }
  396. };
  399. void SelectionPopup::insertPlanetaryMenu(KPopupMenu* popup, const string& parentName, const PlanetarySystem* psys) {
  400.     std::vector<Body*> asteroids;
  401.     std::vector<Body*> comets;
  402.     std::vector<Body*> invisibles;
  403.     std::vector<Body*> moons;
  404.     std::vector<Body*> planets;
  405.     std::vector<Body*> spacecraft;
  406.     std::vector<Body*> smallBodies;
  407.     std::vector<Body*> dwarfPlanets;
  409.     std::vector<std::vector<Body*>* > objects;
  410.     std::vector<std::string> menuNames;
  412.     // Add each vector of PlanetarySystem bodies to a vector to iterate over
  413.     objects.push_back(&asteroids);
  414.     menuNames.push_back(QString::fromUtf8(_("Asteroids")));
  415.     objects.push_back(&comets);
  416.     menuNames.push_back(QString::fromUtf8(_("Comets")));
  417.     objects.push_back(&moons);
  418.     menuNames.push_back(QString::fromUtf8(_("Moons")));
  419.     objects.push_back(&planets);
  420.     menuNames.push_back(QString::fromUtf8(_("Planets")));
  421.     objects.push_back(&spacecraft);
  422.     menuNames.push_back(QString::fromUtf8(_("Spacecraft")));
  423.     objects.push_back(&smallBodies);
  424.     menuNames.push_back(QString::fromUtf8(_("Small Bodies")));
  425.     objects.push_back(&dwarfPlanets);
  426.     menuNames.push_back(QString::fromUtf8(_("Dwarf Planets")));
  428.     for (int i = 0; i < psys->getSystemSize(); i++) {
  429.         Body* body = psys->getBody(i);
  430.         switch(body->getClassification()) {
  431.         case Body::Asteroid:
  432.             asteroids.push_back(body);
  433.             break;
  434.         case Body::Comet:
  435.             comets.push_back(body);
  436.             break;
  437.         case Body::Moon:
  438.             moons.push_back(body);
  439.             break;
  440.         case Body::Planet:
  441.             planets.push_back(body);
  442.             break;
  443.         case Body::Spacecraft:
  444.             spacecraft.push_back(body);
  445.             break;
  446.         case Body::SmallBody:
  447.             smallBodies.push_back(body);
  448.             break;
  449.         case Body::DwarfPlanet:
  450.             dwarfPlanets.push_back(body);
  451.             break;
  452.         case Body::Invisible:
  453.         case Body::Barycenter:
  454.             break;
  455.         }
  456.     }
  458.     // Count how many submenus we need to create
  460.     int numSubMenus = 0;
  461.     int nonEmpty = 0;
  462.     for (std::vector<std::vector<Body*>* >::const_iterator obj = objects.begin();
  463.          obj != objects.end();
  464.          ++obj) {
  465.         if ((*obj)->size() > 0) {
  466.             numSubMenus++;
  467.             nonEmpty = obj - objects.begin();
  468.         }
  469.     }
  471.     if (numSubMenus == 0) return;
  473.     KPopupMenu *submenu = new KPopupMenu(this);
  474.     popup->insertSeparator();
  476.     if (numSubMenus > 1)
  477.         popup->insertItem(i18n("Orbiting Bodies"), submenu);
  478.     else
  479.         popup->insertItem(menuNames[nonEmpty], submenu);
  481.     std::vector<std::string>::const_iterator menuName = menuNames.begin();
  482.     BodyComparePredicate pred;
  484.     for (std::vector<std::vector<Body*>* >::const_iterator obj = objects.begin();
  485.          obj != objects.end();
  486.          ++obj) {
  487.         // Only generate a submenu if the vector is not empty
  488.         if ((*obj)->size() > 0) {
  489.             // Don't create a submenu for a single item
  490.             if ((*obj)->size() == 1) {
  491.                 std::vector<Body*>::const_iterator it = (*obj)->begin();
  492.                 Selection s(*it);
  493.                 KPopupMenu *objMenu = new KPopupMenu(this);
  494.                 insert(objMenu, s);
  495.                 submenu->insertItem(QString::fromUtf8((*it)->getName(true).c_str()), objMenu);
  496.             } else {
  497.                 // Skip sorting if we are dealing with the planets in our own Solar System.
  498.                  if (parentName != "Sol" || *menuName != QString::fromUtf8(_("Planets")))
  499.                      std::sort((*obj)->begin(), (*obj)->end(), pred);
  501.                 if (numSubMenus > 1) {
  502.                     // Add items to submenu
  503.                     KPopupMenu* objMenu = new KPopupMenu(this);
  504.                     for(std::vector<Body*>::const_iterator it = (*obj)->begin(); it != (*obj)->end(); ++it) {
  505.                         Selection s(*it);
  506.                         KPopupMenu *menu = new KPopupMenu(this);
  507.                         insert(menu, s);
  508.                         objMenu->insertItem(QString::fromUtf8((*it)->getName(true).c_str()), menu);
  509.                     }
  510.                     submenu->insertItem(*menuName, objMenu);
  511.                 } else {
  512.                     // Just add items to the popup
  513.                     for(std::vector<Body*>::const_iterator it = (*obj)->begin(); it != (*obj)->end(); ++it) {
  514.                         Selection s(*it);
  515.                         KPopupMenu *menu = new KPopupMenu(this);
  516.                         insert(menu, s);
  517.                         submenu->insertItem(QString::fromUtf8((*it)->getName(true).c_str()), menu);
  518.                     }
  519.                 }
  520.             }
  521.         }
  522.         menuName++;
  523.     }
  524. }