optics / imie

  #ifndef DISTORTIONDIALOG_H
  #define DISTORTIONDIALOG_H
  
  #include <QtGui/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::WFlags flags = 0);
      ~qtDistortionDialog();
      bool updating;

  
  private:

      Ui::DistortionDialogClass ui;

  
  public slots:

      void on_btnComputeDistortionMap_clicked() {

  
          int steps = 100;
  
          //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);
  
  
          //compute the distortion map
          float* distortionMap = (float*)malloc(sizeof(float) * steps * steps);
          DistortionMap(distortionMap, steps);
  
          //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;
  
          //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