optics / imie

  #ifndef DISTORTIONDIALOG_H
  #define DISTORTIONDIALOG_H
  

  #include <QtWidgets/QDialog>

  #include <QDragEnterEvent>
  #include <QGraphicsScene>
  #include <QGraphicsView>
  #include <QGraphicsPixmapItem>
  #include <qfiledialog.h>
  #include <qinputdialog.h>
  #include "ui_distortiondialog.h"
  #include "globals.h"
  
  extern QGraphicsScene* distortionScene;
  extern QGraphicsView* distortionWindow;
  extern QGraphicsPixmapItem* pixmapItem;
  
  class qtDistortionDialog : public QDialog
  {

      Q_OBJECT

  
  public:

      qtDistortionDialog(QWidget *parent = 0, Qt::WindowFlags flags = 0);

      ~qtDistortionDialog();
      bool updating;

  
  private:

      Ui::DistortionDialogClass ui;

  
  public slots:

      void on_btnComputeDistortionMap_clicked() {

          //get the filename
          QString filename = QFileDialog::getSaveFileName(NULL, "Distortion File");
  
          int steps = ui.spinSteps->value();

  
          //delete the pixmap if one previously existed
          if(pixmapItem != NULL)
              delete(pixmapItem);
  
          //create an image
          QImage image(steps, steps, QImage::Format_RGB888);
  
          QColor color(255, 255, 0);
  

          double minIn, maxIn, minOut, maxOut;
  
          minIn = ui.spinInStart->value();
          maxIn = ui.spinInEnd->value();
          minOut = ui.spinOutStart->value();
          maxOut = ui.spinOutEnd->value();

          //compute the distortion map (2D steps x steps image)

          float* distortionMap = (float*)malloc(sizeof(float) * steps * steps);

          DistortionMap(distortionMap, minIn, maxIn, minOut, maxOut, steps, filename.toUtf8().constData());

  
          //compute the extrema
          float minDistortion = 99999;
          float maxDistortion = distortionMap[0];

          for(int i=1; i<steps*steps; i++) {

              if(distortionMap[i] < minDistortion && distortionMap[i] > 0)
                  minDistortion = distortionMap[i];
              if(distortionMap[i] > maxDistortion)
                  maxDistortion = distortionMap[i];
          }
          cout<<minDistortion<<"----------"<<maxDistortion<<endl;

          FILE* outraw = fopen("out.raw", "wb");
          fwrite(distortionMap, sizeof(float), steps * steps, outraw);
          fclose(outraw);

  
          //compute the distortion map image
          int i, o;
          float intensity;
          float v;
          for(i=0; i<steps; i++)

              for(o=0; o<steps; o++) {

                  v = distortionMap[o * steps + i];
                  intensity = (v - minDistortion) / (maxDistortion - minDistortion) * 255;
  

                  if(intensity < 0) {

                      //cout<<"low intensity"<<endl;

                      intensity = 0;
                  }

  
                  //QColor color(intensity, (float)o / (float)steps * 255, (float)i / (float)steps * 255);
                  QColor color(intensity, intensity, intensity);
                  image.setPixel(o, (steps-1-i), color.rgb());
              }
  
          pixmapItem = new QGraphicsPixmapItem(QPixmap::fromImage(image));
          distortionScene->addItem(pixmapItem);
          distortionWindow->show();
  
  
      }
  };
  
  #endif