geo/tcExportText.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 tcExportText.cpp
  22  * @brief Exports a set of channels to a file.
  23  */
  24
  25 #include "tcExportText.h"
  26 #include "tcChannel.h"
  27 #include "tcChannelManager.h"
  28 #include "tcGeoImageData.h"
  29 #include "tcPixelData.h"
  30
  31 #include <QVBoxLayout>
  32 #include <QLabel>
  33 #include <QListWidget>
  34 #include <QComboBox>
  35 #include <QPushButton>
  36 #include <QMessageBox>
  37 #include <QFileDialog>
  38 #include <QTextStream>
  39
  40 /*
  41  * Constructors + destructor
  42  */
  43
  44 /// Primary constructor.
  45 tcExportText::tcExportText(const QList<tcGeoImageData*>& imagery, QWidget* parent)
  46 : QWidget(parent)
  47 , m_imagery(imagery)
  48 {
  49   tcChannelManager* channelManager = m_imagery[0]->channelManager();
  50   setWindowTitle(tr("Export as Text"));
  51
  52   QVBoxLayout* layout = new QVBoxLayout(this);
  53
  54   QLabel* labelClassificationChannel = new QLabel(tr("Classification channel"), this);
  55     layout->addWidget(labelClassificationChannel);
  56   m_comboClassificationChannel = new QComboBox(this);
  57     layout->addWidget(m_comboClassificationChannel);
  58     labelClassificationChannel->setBuddy(m_comboClassificationChannel);
  59     for (int i = 0; i < channelManager->numChannels(); ++i)
  60     {
  61       tcChannel* channel = channelManager->channel(i);
  62       m_comboClassificationChannel->addItem(channel->name());
  63     }
  64     connect(m_comboClassificationChannel, SIGNAL(currentIndexChanged(int)),
  65             this,    SLOT(setClassificationChannel(int)));
  66
  67   QLabel* labelFilterChannel = new QLabel(tr("Filter channel"), this);
  68     layout->addWidget(labelFilterChannel);
  69   m_comboFilterChannel = new QComboBox(this);
  70     layout->addWidget(m_comboFilterChannel);
  71     labelFilterChannel->setBuddy(m_comboFilterChannel);
  72     m_comboFilterChannel->addItem(tr("(Export all pixels)"));
  73     for (int i = 0; i < channelManager->numChannels(); ++i)
  74     {
  75       tcChannel* channel = channelManager->channel(i);
  76       m_comboFilterChannel->addItem(channel->name());
  77     }
  78     connect(m_comboFilterChannel, SIGNAL(currentIndexChanged(int)),
  79             this,    SLOT(setFilterChannel(int)));
  80   m_comboFilterType = new QComboBox(this);
  81     layout->addWidget(m_comboFilterType);
  82     m_comboFilterType->addItem(tr("Export dark values"));
  83     m_comboFilterType->addItem(tr("Export light values"));
  84     m_comboFilterType->setEnabled(false);
  85     connect(m_comboFilterType, SIGNAL(currentIndexChanged(int)),
  86             this,    SLOT(setFilterType(int)));
  87
  88   QLabel* labelChannels = new QLabel(tr("Channels to export"), this);
  89     layout->addWidget(labelChannels);
  90   m_listChannels = new QListWidget(this);
  91     layout->addWidget(m_listChannels);
  92     labelChannels->setBuddy(m_listChannels);
  93     for (int i = 0; i < channelManager->numChannels(); ++i)
  94     {
  95       tcChannel* channel = channelManager->channel(i);
  96       m_listChannels->addItem(channel->name());
  97       QListWidgetItem* item = m_listChannels->item(i);
  98       item->setCheckState(Qt::Unchecked);
  99     }
 100
 101   QPushButton* btnSave = new QPushButton(tr("Export"), this);
 102     layout->addWidget(btnSave);
 103     connect(btnSave, SIGNAL(clicked(bool)), this, SLOT(exportToFile()));
 104 }
 105
 106 /// Destructor.
 107 tcExportText::~tcExportText()
 108 {
 109 }
 110
 111 /*
 112  * Public slots
 113  */
 114
 115 /// Export to a file.
 116 void tcExportText::exportToFile()
 117 {
 118   tcChannelManager* channelManager = m_imagery[0]->channelManager();
 119
 120   // Get the data together.
 121   tcChannel* classification = channelManager->channel(m_comboClassificationChannel->currentIndex());
 122   tcChannel* filter = channelManager->channel(m_comboFilterChannel->currentIndex()-1);
 123   int filterType = m_comboFilterType->currentIndex();
 124   QList<tcChannel*> channels;
 125   for (int i = 0; i < m_listChannels->count(); ++i)
 126   {
 127     if (m_listChannels->item(i)->checkState() == Qt::Checked)
 128     {
 129       tcChannel* channel = channelManager->channel(i);
 130       if (0 != channel)
 131       {
 132         channels += channel;
 133       }
 134     }
 135   }
 136   Q_ASSERT(0 != classification);
 137   Q_ASSERT(filterType < 2);
 138
 139   // Check there are enough channels.
 140   if (channels.count() == 0)
 141   {
 142     QMessageBox::warning(this,
 143                          tr("No Channels Ticked."),
 144                          tr("You haven't ticked any channels to export."));
 145     return;
 146   }
 147
 148   // Get the portion of the image
 149   Q_ASSERT(0);
 150   /*
 151   tcGeoImageData::LocalCoord min = m_imagery[0]->minLocal();
 152   tcGeoImageData::LocalCoord range = m_imagery[0]->rangeLocal();
 153   tcGeoImageData::LocalCoord minEffective = m_imagery[0]->minEffectiveLocal();
 154   tcGeoImageData::LocalCoord rangeEffective = m_imagery[0]->rangeEffectiveLocal();
 155   double rx1 = (minEffective.x - min.x) / range.x;
 156   double ry1 = 1.0 - (minEffective.y + rangeEffective.y - min.y) / range.y;
 157   double rx2 = (minEffective.x + rangeEffective.x - min.x) / range.x;
 158   double ry2 = 1.0 - (minEffective.y - min.y) / range.y;
 159   */
 160   double rx1 = 0.0;
 161   double ry1 = 0.0;
 162   double rx2 = 1.0;
 163   double ry2 = 1.0;
 164
 165   double x1,y1,x2,y2;
 166
 167 #define REWRITE_PORTION() \
 168     { x1 = rx1; y1 = ry1; x2 = rx2; y2 = ry2; }
 169
 170   // Get all the pixel data objects
 171   REWRITE_PORTION();
 172   Reference<tcPixelData<float> > classificationData = dynamicCast<tcPixelData<float>*>(classification->portion(&x1,&y1,&x2,&y2));
 173   if (0 == classificationData)
 174   {
 175     QMessageBox::warning(this,
 176                          tr("No Classification Data."),
 177                          tr("No classification pixel data was found in this portion."));
 178     return;
 179   }
 180   Reference<tcAbstractPixelData> filterData;
 181   if (0 != filter)
 182   {
 183     REWRITE_PORTION();
 184     filterData = filter->portion(&x1,&y1,&x2,&y2);
 185     if (0 == filterData)
 186     {
 187       QMessageBox::warning(this,
 188           tr("No Filter Data."),
 189           tr("No filter pixel data was found in this portion."));
 190       return;
 191     }
 192   }
 193   QList<Reference<tcAbstractPixelData> > channelsData;
 194   foreach (tcChannel* channel, channels)
 195   {
 196     REWRITE_PORTION();
 197     channelsData += channel->portion(&x1,&y1,&x2,&y2);
 198     if (0 == channelsData[channelsData.count()-1])
 199     {
 200       QMessageBox::warning(this,
 201           tr("No %1 Data.").arg(channel->name()),
 202           tr("No %1 pixel data was found in this portion.").arg(channel->name()));
 203       return;
 204     }
 205   }
 206
 207   QString filename = QFileDialog::getSaveFileName(this,
 208                         tr("Export channels"),
 209                         QString(),
 210                         tr("ASCII Data Files (*.adf)"));
 211
 212   if (!filename.isEmpty())
 213   {
 214     // Open the file ready to write the data
 215     QFile file(filename);
 216     if (!file.open(QIODevice::WriteOnly | QIODevice::Text))
 217     {
 218       QMessageBox::warning(this,
 219           tr("Open failed."),
 220           tr("Could not open '%1' for writing.").arg(filename));
 221       return;
 222     }
 223
 224     QTextStream out(&file);
 225
 226     // Go through the classification data and sample the other channels
 227     int width = classificationData->width();
 228     int height = classificationData->height();
 229     for (int j = 0; j < height; ++j)
 230     {
 231       for (int i = 0; i < width; ++i)
 232       {
 233         int index = j*height + i;
 234         float x = (float)i/(width-1);
 235         float y = (float)j/(height-1);
 236         // Check filter
 237         if (0 != filterData)
 238         {
 239           float val = filterData->sampleFloat(x,y);
 240           if (filterType == 0)
 241           {
 242             if (val > 0.0f)
 243             {
 244               continue;
 245             }
 246           }
 247           else if (filterType == 1)
 248           {
 249             if (val <= 0.0f)
 250             {
 251               continue;
 252             }
 253           }
 254         }
 255         // Write vales
 256         out << classificationData->buffer()[index];
 257         for (int i = 0; i < channelsData.count(); ++i)
 258         {
 259           out << " " << channelsData[i]->sampleFloat(x, y);
 260         }
 261         out << "\n";
 262       }
 263     }
 264   }
 265 }
 266
 267 /*
 268  * Private slots
 269  */
 270
 271 /// Set the channel to classify the pixels.
 272 void tcExportText::setClassificationChannel(int channel)
 273 {
 274 }
 275
 276 /// Set the channel to filter which pixels to export.
 277 void tcExportText::setFilterChannel(int channel)
 278 {
 279   m_comboFilterType->setEnabled(channel != 0);
 280 }
 281
 282 /// Set the type of filtering.
 283 void tcExportText::setFilterType(int channel)
 284 {
 285 }