tcMainWindow.cpp

   1 /***************************************************************************
   2  *   This file is part of Tecorrec.                                        *
   3  *   Copyright 2008 James Hogan <james@albanarts.com>                      *
   4  *                                                                         *
   5  *   Tecorrec is free software: you can redistribute it and/or modify      *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation, either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   Tecorrec is distributed in the hope that it will be useful,           *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with Tecorrec.  If not, write to the Free Software Foundation,  *
  17  *   Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.   *
  18  ***************************************************************************/
  19
  20 /**
  21  * @file tcMainWindow.h
  22  * @brief Main application window.
  23  */
  24
  25 #include "tcMainWindow.h"
  26 #include "tcViewportWidget.h"
  27 #include "tcColourMapWidget.h"
  28 #include <tcGlobe.h>
  29 #include <tcSrtmCache.h>
  30 #include <tcGeoImageData.h>
  31 #include <tcChannelManager.h>
  32 #include <tcChannel.h>
  33 #include <tcChannelConfigWidget.h>
  34 #include <tcExportText.h>
  35
  36 #include <qwt_plot.h>
  37 #include <qwt_plot_curve.h>
  38
  39 #include <QHBoxLayout>
  40 #include <QWidget>
  41 #include <QMenuBar>
  42 #include <QMenu>
  43 #include <QStatusBar>
  44 #include <QDockWidget>
  45 #include <QLabel>
  46 #include <QComboBox>
  47 #include <QRadioButton>
  48 #include <QSlider>
  49 #include <QPushButton>
  50 #include <QButtonGroup>
  51
  52 /*
  53  * Constructors + destructor
  54  */
  55
  56 /// Default constructor.
  57 tcMainWindow::tcMainWindow()
  58 : QMainWindow()
  59 , m_viewport(new tcViewportWidget(this))
  60 , m_colourMap(new tcColourMapWidget(QStringList()
  61                   << tr("Red") << tr("Green") << tr("Blue") << tr("A") << tr("B") << tr("C"), this))
  62 , m_globe(new tcGlobe(6378.137e3))
  63 , m_exportText(0)
  64 {
  65   m_viewport->setGlobe(m_globe);
  66   setCentralWidget(m_viewport);
  67
  68   {
  69     // Menu bar
  70     QMenuBar* menus = menuBar();
  71     {
  72       QMenu* menuFile = menus->addMenu(tr("File", "The menu"));
  73       {
  74         QMenu* menuFileExport = menuFile->addMenu(tr("Export", "The menu"));
  75         {
  76           QAction* menuFileExportText = menuFileExport->addAction(tr("Export as Text"),
  77               this, SLOT(exportText()));
  78         }
  79       }
  80       QMenu* menuView = menus->addMenu(tr("View", "The menu"));
  81       {
  82         QMenu* menuViewQuality = menuView->addMenu(tr("Elevation Quality", "The menu"));
  83         {
  84           QAction* menuViewQualityAdaptive = menuViewQuality->addAction(tr("Adaptive Resolution"),
  85               m_viewport, SLOT(setQualityAdaptive()));
  86           QAction* menuViewQualityFull = menuViewQuality->addAction(tr("Full Resolution"),
  87               m_viewport, SLOT(setQualityFull()));
  88         }
  89         QMenu* menuViewRender = menuView->addMenu(tr("Render Options", "The menu"));
  90         {
  91           QAction* menuViewRenderNormal = menuViewRender->addAction(tr("Normal", "Polygon mode"),
  92               m_viewport, SLOT(setPolygonModeNormal()));
  93           QAction* menuViewRenderWireframe = menuViewRender->addAction(tr("Wireframe", "Polygon mode"),
  94               m_viewport, SLOT(setPolygonModeWireframe()));
  95         }
  96         QMenu* menuViewRenderColour = menuView->addMenu(tr("Colour Coding", "The menu"));
  97         {
  98           QAction* menuViewRenderColourNone = menuViewRenderColour->addAction(tr("No colour coding"),
  99               m_viewport, SLOT(setColourCodingNone()));
 100           QAction* menuViewRenderColourElevationSampleAlignment = menuViewRenderColour->addAction(tr("Alignment of elevation samples"),
 101               m_viewport, SLOT(setColourCodingElevationSampleAlignment()));
 102         }
 103       }
 104     }
 105   }
 106
 107   {
 108     // Status bar
 109     QStatusBar* statusBar = new QStatusBar(this);
 110     connect(m_viewport, SIGNAL(mouseGeoTextChanged(const QString&)),
 111             statusBar,  SLOT(showMessage(const QString&)));
 112     foreach (tcGeoImageData* imagery, m_globe->imagery())
 113     {
 114       connect(imagery->channelManager(), SIGNAL(progressing(const QString&)),
 115               statusBar, SLOT(showMessage(const QString&)));
 116     }
 117     connect(m_globe->dem(), SIGNAL(progressing(const QString&)),
 118             statusBar, SLOT(showMessage(const QString&)));
 119     setStatusBar(statusBar);
 120   }
 121
 122   QDockWidget* dockElevation;
 123   {
 124     // Docker for elevation data settings
 125     QDockWidget* docker = new QDockWidget(tr("Elevation Data"), this);
 126     dockElevation = docker;
 127       QWidget* dockerWidget = new QWidget(docker);
 128       QVBoxLayout* layout = new QVBoxLayout(dockerWidget);
 129       docker->setWidget(dockerWidget);
 130     addDockWidget(Qt::LeftDockWidgetArea, docker);
 131
 132     // Combo box to select data set
 133     QStringList elevDataSets = tcSrtmCache::dataSets();
 134     QString elevTypeNames[2] = {
 135       tr("Primary DEM"),
 136       tr("Secondary DEM")
 137     };
 138     const char* elevDatasetSlot[2] = { SLOT(setPrimaryElevationDataSet(const QString&)), SLOT(setSecondaryElevationDataSet(const QString&)) };
 139     const char* elevFlatSlot[2] = { SLOT(setPrimaryElevationFlat()), SLOT(setSecondaryElevationFlat()) };
 140     const char* elevRawSlot[2] = { SLOT(setPrimaryElevationRaw()), SLOT(setSecondaryElevationRaw()) };
 141     const char* elevCorrectedSlot[2] = { SLOT(setPrimaryElevationCorrected()), SLOT(setSecondaryElevationCorrected()) };
 142     for (int i = 0; i < 2; ++i)
 143     {
 144       QLabel* elevInterpolationLabel = new QLabel(elevTypeNames[i], dockerWidget);
 145         layout->addWidget(elevInterpolationLabel);
 146
 147       QComboBox* elevDataSet = new QComboBox(dockerWidget);
 148         layout->addWidget(elevDataSet);
 149         elevDataSet->addItem("");
 150         foreach (QString dataSet, elevDataSets)
 151         {
 152           elevDataSet->addItem(dataSet);
 153         }
 154         connect(elevDataSet, SIGNAL(currentIndexChanged(const QString&)), m_viewport, elevDatasetSlot[i]);
 155         if (-1 != elevDataSet->currentIndex())
 156         {
 157           m_viewport->setPrimaryElevationDataSet(elevDataSet->currentText());
 158         }
 159
 160       // Radio buttons
 161       QButtonGroup* elevGroup = new QButtonGroup(dockerWidget);
 162       QRadioButton* elevFlat = new QRadioButton(tr("Don't Show Elevation (Flat)"), dockerWidget);
 163         layout->addWidget(elevFlat);
 164         elevGroup->addButton(elevFlat);
 165         connect(elevFlat, SIGNAL(pressed()), m_viewport, elevFlatSlot[i]);
 166       QRadioButton* elevRaw = new QRadioButton(tr("Raw SRTM Elevation Data"), dockerWidget);
 167         layout->addWidget(elevRaw);
 168         elevGroup->addButton(elevRaw);
 169         connect(elevRaw, SIGNAL(pressed()), m_viewport, elevRawSlot[i]);
 170       QRadioButton* elevCorrected = new QRadioButton(tr("Corrected SRTM Elevation Data"), dockerWidget);
 171         layout->addWidget(elevCorrected);
 172         elevGroup->addButton(elevCorrected);
 173         elevCorrected->setChecked(true);
 174         connect(elevCorrected, SIGNAL(pressed()), m_viewport, elevCorrectedSlot[i]);
 175
 176       // Spacer
 177       layout->addStretch();
 178     }
 179
 180     QLabel* elevInterpolationLabel = new QLabel(tr("Interpolate between DEMs"), dockerWidget);
 181       layout->addWidget(elevInterpolationLabel);
 182     QSlider* elevCorrectionLevel = new QSlider(Qt::Horizontal, dockerWidget);
 183       layout->addWidget(elevCorrectionLevel);
 184       elevCorrectionLevel->setRange(0,100);
 185       elevCorrectionLevel->setValue(100);
 186       connect(elevCorrectionLevel, SIGNAL(valueChanged(int)), m_viewport, SLOT(setElevationInterpolation(int)));
 187
 188     // Spacer
 189     layout->addStretch();
 190   }
 191
 192   {
 193     // Docker for satellite imagery
 194     QDockWidget* docker = new QDockWidget(tr("Imagery Data"), this);
 195       QWidget* dockerWidget = new QWidget(docker);
 196       QVBoxLayout* layout = new QVBoxLayout(dockerWidget);
 197       docker->setWidget(dockerWidget);
 198     tabifyDockWidget(dockElevation, docker);
 199
 200     // Grid of colours
 201     layout->addWidget(m_colourMap);
 202     foreach (tcGeoImageData* imagery, m_globe->imagery())
 203     {
 204       const tcChannelManager* manager = imagery->channelManager();
 205       int numChannels = manager->numChannels();
 206       m_colourMap->addInputGroupSeparator(imagery->name());
 207       for (int i = 0; i < numChannels; ++i)
 208       {
 209         tcChannel* channel = manager->channel(i);
 210         m_colourMap->addInputBand(channel->name(), channel->description());
 211       }
 212     }
 213     connect(m_colourMap, SIGNAL(inputBandClicked(int, int)), this, SLOT(configureChannel(int, int)));
 214     connect(m_colourMap, SIGNAL(inputGroupClicked(int)), m_viewport, SLOT(sunView(int)));
 215     connect(m_colourMap, SIGNAL(inputBandChanged(int, int, int)), m_viewport, SLOT(setColourMapping(int, int, int)));
 216     for (int i = 0; i < 3; ++i)
 217     {
 218       m_colourMap->setInputBand(i, 2-i);
 219     }
 220   }
 221
 222   {
 223     // Docker for processing
 224     QDockWidget* docker = new QDockWidget(tr("Processing"), this);
 225       QWidget* dockerWidget = new QWidget(docker);
 226       QVBoxLayout* layout = new QVBoxLayout(dockerWidget);
 227       docker->setWidget(dockerWidget);
 228     tabifyDockWidget(dockElevation, docker);
 229
 230     {
 231       QwtPlot* plot = new QwtPlot(this);
 232       layout->addWidget(plot);
 233       plot->setTitle("This is a test");
 234       //plot->setAutoLegend(true);
 235       //plot->setLegendPosition(Qwt::bottom);
 236       //plot->setAxisTitle(xBottom, "x");
 237       //plot->setAxisTitle(yLeft, "y");
 238       QwtPlotCurve* cSin = new QwtPlotCurve("y=sin(x)");
 239       cSin->setPen(QPen(Qt::red));
 240       const int points = 500;
 241       double x[points];
 242       double sn[points];
 243       for (int i = 0; i < points; ++i)
 244       {
 245         x[i] = (3.0 * 3.14 / double(points))*double(i);
 246         sn[i] = 2.0*sin(x[i]);
 247       }
 248       cSin->setData(x, sn, points);
 249       cSin->attach(plot);
 250       plot->replot();
 251     }
 252
 253     // Spacer
 254     layout->addStretch();
 255   }
 256 }
 257
 258 /// Destructor.
 259 tcMainWindow::~tcMainWindow()
 260 {
 261   delete m_globe;
 262 }
 263
 264 /*
 265  * Private slots
 266  */
 267
 268 /// Show configuration for a channel.
 269 void tcMainWindow::configureChannel(int channel, int group)
 270 {
 271   tcChannelManager* manager = m_globe->imagery()[group]->channelManager();
 272   tcChannelConfigWidget* configWidget = manager->channel(channel)->configWidget();
 273   if (0 != configWidget)
 274   {
 275     connect(configWidget, SIGNAL(updated()),
 276             m_viewport, SLOT(updateGL()));
 277     connect(configWidget, SIGNAL(newSlice(const tcGeo&, const tcGeo&)),
 278             m_viewport, SLOT(setSlice(const tcGeo&, const tcGeo&)));
 279     connect(configWidget, SIGNAL(texturePointRequested(QObject*, const char*)),
 280             m_viewport, SLOT(texturePointMode(QObject*, const char*)));
 281     configWidget->setWindowFlags(Qt::Tool | Qt::WindowStaysOnTopHint);
 282     configWidget->show();
 283   }
 284 }
 285
 286 /// Export as text.
 287 void tcMainWindow::exportText()
 288 {
 289   if (0 == m_exportText)
 290   {
 291     m_exportText = new tcExportText(m_globe->imagery(), this);
 292   }
 293   m_exportText->setWindowFlags(Qt::Dialog | Qt::WindowStaysOnTopHint);
 294   m_exportText->show();
 295 }