r355: Polar conversions of projections
[ctsim.git] / src / views.cpp
index aba89cc3198de6c70e40e6b0c0ff058d65ff1fae..7c47aa6415671807810ac4274111fdee5f3d90de 100644 (file)
@@ -9,7 +9,7 @@
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: views.cpp,v 1.51 2001/01/04 21:28:41 kevin Exp $
+**  $Id: views.cpp,v 1.52 2001/01/06 15:33:15 kevin Exp $
 **
 **  This program is free software; you can redistribute it and/or modify
 **  it under the terms of the GNU General Public License (version 2) as
@@ -188,14 +188,22 @@ EVT_MENU(IFMENU_IMAGE_SUBTRACT, ImageFileView::OnSubtract)
 EVT_MENU(IFMENU_IMAGE_MULTIPLY, ImageFileView::OnMultiply)
 EVT_MENU(IFMENU_IMAGE_DIVIDE, ImageFileView::OnDivide)
 EVT_MENU(IFMENU_IMAGE_SCALESIZE, ImageFileView::OnScaleSize)
-#ifdef HAVE_FFTW
+#ifdef HAVE_FFT
 EVT_MENU(IFMENU_FILTER_FFT, ImageFileView::OnFFT)
 EVT_MENU(IFMENU_FILTER_IFFT, ImageFileView::OnIFFT)
+EVT_MENU(IFMENU_FILTER_FFT_ROWS, ImageFileView::OnFFTRows)
+EVT_MENU(IFMENU_FILTER_IFFT_ROWS, ImageFileView::OnIFFTRows)
+EVT_MENU(IFMENU_FILTER_FFT_COLS, ImageFileView::OnFFTCols)
+EVT_MENU(IFMENU_FILTER_IFFT_COLS, ImageFileView::OnIFFTCols)
 #endif
 EVT_MENU(IFMENU_FILTER_MAGNITUDE, ImageFileView::OnMagnitude)
 EVT_MENU(IFMENU_FILTER_PHASE, ImageFileView::OnPhase)
 EVT_MENU(IFMENU_PLOT_ROW, ImageFileView::OnPlotRow)
 EVT_MENU(IFMENU_PLOT_COL, ImageFileView::OnPlotCol)
+#ifdef HAVE_FFT
+EVT_MENU(IFMENU_PLOT_FFT_ROW, ImageFileView::OnPlotFFTRow)
+EVT_MENU(IFMENU_PLOT_FFT_COL, ImageFileView::OnPlotFFTCol)
+#endif
 EVT_MENU(IFMENU_PLOT_HISTOGRAM, ImageFileView::OnPlotHistogram)
 END_EVENT_TABLE()
 
@@ -577,7 +585,6 @@ void
 ImageFileView::OnFFT (wxCommandEvent& event)
 {
   ImageFile& rIF = GetDocument()->getImageFile();
-  wxProgressDialog dlgProgress (wxString("FFT"), wxString("FFT Progress"), 1, m_frame, wxPD_APP_MODAL);
   rIF.fft (rIF);
   rIF.labelAdd ("FFT Image");
   m_bMinSpecified = false;
@@ -591,7 +598,6 @@ void
 ImageFileView::OnIFFT (wxCommandEvent& event)
 {
   ImageFile& rIF = GetDocument()->getImageFile();
-  wxProgressDialog dlgProgress (wxString("IFFT"), wxString("IFFT Progress"), 1, m_frame, wxPD_APP_MODAL);
   rIF.ifft (rIF);
   rIF.labelAdd ("IFFT Image");
   m_bMinSpecified = false;
@@ -600,6 +606,58 @@ ImageFileView::OnIFFT (wxCommandEvent& event)
     GetDocument()->Modify(TRUE);
   GetDocument()->UpdateAllViews(this);
 }
+
+void
+ImageFileView::OnFFTRows (wxCommandEvent& event)
+{
+  ImageFile& rIF = GetDocument()->getImageFile();
+  rIF.fftRows (rIF);
+  rIF.labelAdd ("FFT Rows");
+  m_bMinSpecified = false;
+  m_bMaxSpecified = false;
+  if (theApp->getSetModifyNewDocs())
+    GetDocument()->Modify(TRUE);
+  GetDocument()->UpdateAllViews(this);
+}
+
+void
+ImageFileView::OnIFFTRows (wxCommandEvent& event)
+{
+  ImageFile& rIF = GetDocument()->getImageFile();
+  rIF.ifftRows (rIF);
+  rIF.labelAdd ("IFFT Rows");
+  m_bMinSpecified = false;
+  m_bMaxSpecified = false;
+  if (theApp->getSetModifyNewDocs())
+    GetDocument()->Modify(TRUE);
+  GetDocument()->UpdateAllViews(this);
+}
+
+void
+ImageFileView::OnFFTCols (wxCommandEvent& event)
+{
+  ImageFile& rIF = GetDocument()->getImageFile();
+  rIF.fftCols (rIF);
+  rIF.labelAdd ("FFT Columns");
+  m_bMinSpecified = false;
+  m_bMaxSpecified = false;
+  if (theApp->getSetModifyNewDocs())
+    GetDocument()->Modify(TRUE);
+  GetDocument()->UpdateAllViews(this);
+}
+
+void
+ImageFileView::OnIFFTCols (wxCommandEvent& event)
+{
+  ImageFile& rIF = GetDocument()->getImageFile();
+  rIF.ifftCols (rIF);
+  rIF.labelAdd ("IFFT Columns");
+  m_bMinSpecified = false;
+  m_bMaxSpecified = false;
+  if (theApp->getSetModifyNewDocs())
+    GetDocument()->Modify(TRUE);
+  GetDocument()->UpdateAllViews(this);
+}
 #endif
 
 void
@@ -706,7 +764,11 @@ ImageFileView::CreateCanvas (wxView *view, wxFrame *parent)
 wxFrame*
 ImageFileView::CreateChildFrame(wxDocument *doc, wxView *view)
 {
+#if CTSIM_MDI
+  wxMDIChildFrame *subframe = new wxMDIChildFrame(theApp->getMainFrame(), -1, "ImageFile Frame", wxPoint(-1, -1), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
+#else
   wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "ImageFile Frame", wxPoint(-1, -1), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
+#endif
   
   wxMenu *file_menu = new wxMenu;
   
@@ -739,8 +801,12 @@ ImageFileView::CreateChildFrame(wxDocument *doc, wxView *view)
   filter_menu->Append (IFMENU_FILTER_EXP, "&Exp");
   filter_menu->AppendSeparator();
 #ifdef HAVE_FFT
-  filter_menu->Append (IFMENU_FILTER_FFT, "&FFT");
-  filter_menu->Append (IFMENU_FILTER_IFFT, "&IFFT");
+  filter_menu->Append (IFMENU_FILTER_FFT, "2D &FFT");
+  filter_menu->Append (IFMENU_FILTER_IFFT, "2D &IFFT");
+  filter_menu->Append (IFMENU_FILTER_FFT_ROWS, "FFT Rows");
+  filter_menu->Append (IFMENU_FILTER_IFFT_ROWS, "IFFT Rows");
+  filter_menu->Append (IFMENU_FILTER_FFT_COLS, "FFT Columns");
+  filter_menu->Append (IFMENU_FILTER_IFFT_COLS, "IFFT Columns");
   filter_menu->Append (IFMENU_FILTER_FOURIER, "F&ourier");
   filter_menu->Append (IFMENU_FILTER_INVERSE_FOURIER, "Inverse Fo&urier");
 #else
@@ -765,6 +831,9 @@ ImageFileView::CreateChildFrame(wxDocument *doc, wxView *view)
   analyze_menu->Append (IFMENU_PLOT_COL, "Plot &Column");
   analyze_menu->Append (IFMENU_PLOT_HISTOGRAM, "Plot &Histogram");
   analyze_menu->AppendSeparator();
+  analyze_menu->Append (IFMENU_PLOT_FFT_ROW, "Plot FFT Row");
+  analyze_menu->Append (IFMENU_PLOT_FFT_COL, "Plot FFT Column");
+  analyze_menu->AppendSeparator();
   analyze_menu->Append (IFMENU_COMPARE_IMAGES, "Compare &Images...");
   analyze_menu->Append (IFMENU_COMPARE_ROW, "Compare &Row");
   analyze_menu->Append (IFMENU_COMPARE_COL, "Compare &Column");
@@ -876,7 +945,7 @@ ImageFileView::OnClose (bool deleteWindow)
   if (!GetDocument()->Close())
     return false;
   
-  m_canvas->Clear();
+  // m_canvas->Clear();
   m_canvas->m_pView = NULL;
   m_canvas = NULL;
   wxString s(theApp->GetAppName());
@@ -958,6 +1027,7 @@ ImageFileView::OnScaleSize (wxCommandEvent& event)
     }
     ImageFile& rScaledIF = pScaledDoc->getImageFile();
     rScaledIF.setArraySize (iNewNX, iNewNY);
+    rScaledIF.labelsCopy (rIF);
     rScaledIF.labelAdd (os.str().c_str());
     rIF.scaleImage (rScaledIF);
     *theApp->getLog() << os.str().c_str() << "\n";
@@ -1064,6 +1134,142 @@ ImageFileView::OnPlotCol (wxCommandEvent& event)
   }
 }
 
+#ifdef HAVE_FFT
+void
+ImageFileView::OnPlotFFTRow (wxCommandEvent& event)
+{
+  int xCursor, yCursor;
+  if (! m_canvas->GetCurrentCursor (xCursor, yCursor)) {
+    wxMessageBox ("No row selected. Please use left mouse button on image to select column","Error");
+    return;
+  }
+  
+  const ImageFile& rIF = dynamic_cast<ImageFileDocument*>(GetDocument())->getImageFile();
+  ImageFileArrayConst v = rIF.getArray();
+  ImageFileArrayConst vImag = rIF.getImaginaryArray();
+  int nx = rIF.nx();
+  int ny = rIF.ny();
+  
+  if (v != NULL && yCursor < ny) {
+    fftw_complex* pcIn = new fftw_complex [nx];
+
+    int i;
+    for (i = 0; i < nx; i++) {
+      pcIn[i].re = v[i][yCursor];
+      if (rIF.isComplex())
+        pcIn[i].im = vImag[i][yCursor];
+      else
+        pcIn[i].im = 0;
+    }
+
+    fftw_plan plan = fftw_create_plan (nx, FFTW_FORWARD, FFTW_IN_PLACE);
+    fftw_one (plan, pcIn, NULL);
+    fftw_destroy_plan (plan);
+
+    double* pX = new double [nx];
+    double* pYReal = new double [nx];
+    double* pYImag = new double [nx];
+    double* pYMag = new double [nx];
+    for (i = 0; i < nx; i++) {
+      pX[i] = i;
+      pYReal[i] = pcIn[i].re;
+      pYImag[i] = pcIn[i].im;
+      pYMag[i] = ::sqrt (pcIn[i].re * pcIn[i].re + pcIn[i].im * pcIn[i].im);
+    }
+    Fourier::shuffleFourierToNaturalOrder (pYReal, nx);
+    Fourier::shuffleFourierToNaturalOrder (pYImag, nx);
+    Fourier::shuffleFourierToNaturalOrder (pYMag, nx);
+
+    PlotFileDocument* pPlotDoc = dynamic_cast<PlotFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.plt", wxDOC_SILENT));
+    if (! pPlotDoc) {
+      sys_error (ERR_SEVERE, "Internal error: unable to create Plot file");
+    } else {
+      PlotFile& rPlotFile = pPlotDoc->getPlotFile();
+      std::ostringstream os;
+      os << "Row " << yCursor;
+      std::string title("title ");
+      title += os.str();
+      rPlotFile.addEzsetCommand (title.c_str());
+      rPlotFile.addEzsetCommand ("xlabel Column");
+      rPlotFile.addEzsetCommand ("ylabel Pixel Value");
+      rPlotFile.addEzsetCommand ("lxfrac 0");
+      rPlotFile.addEzsetCommand ("curve 1");
+      rPlotFile.addEzsetCommand ("color 1");
+      rPlotFile.addEzsetCommand ("curve 2");
+      rPlotFile.addEzsetCommand ("color 2");
+      rPlotFile.addEzsetCommand ("dash 1");
+      rPlotFile.addEzsetCommand ("curve 3");
+      rPlotFile.addEzsetCommand ("dash 2");
+      rPlotFile.addEzsetCommand ("color 3");
+      rPlotFile.addEzsetCommand ("box");
+      rPlotFile.addEzsetCommand ("grid");
+      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.addColumn (0, pX);
+      rPlotFile.addColumn (1, pYReal);
+      rPlotFile.addColumn (2, pYImag);
+      rPlotFile.addColumn (3, pYMag);
+    }
+    delete pX;
+    delete pYReal;
+    delete pYImag;
+    delete pYMag;
+    delete [] pcIn;
+
+    if (theApp->getSetModifyNewDocs())
+      pPlotDoc->Modify(true);
+    pPlotDoc->UpdateAllViews();
+  }
+}
+
+void
+ImageFileView::OnPlotFFTCol (wxCommandEvent& event)
+{
+  int xCursor, yCursor;
+  if (! m_canvas->GetCurrentCursor (xCursor, yCursor)) {
+    wxMessageBox ("No column selected. Please use left mouse button on image to select column","Error");
+    return;
+  }
+  
+  const ImageFile& rIF = dynamic_cast<ImageFileDocument*>(GetDocument())->getImageFile();
+  ImageFileArrayConst v = rIF.getArray();
+  int nx = rIF.nx();
+  int ny = rIF.ny();
+  
+  if (v != NULL && xCursor < nx) {
+    double* pX = new double [ny];
+    double* pY = new double [ny];
+    for (int i = 0; i < ny; i++) {
+      pX[i] = i;
+      pY[i] = v[xCursor][i];
+    }
+    PlotFileDocument* pPlotDoc = dynamic_cast<PlotFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.plt", wxDOC_SILENT));
+    if (! pPlotDoc) {
+      sys_error (ERR_SEVERE, "Internal error: unable to create Plot file");
+    } else {
+      PlotFile& rPlotFile = pPlotDoc->getPlotFile();
+      std::ostringstream os;
+      os << "Column " << xCursor;
+      std::string title("title ");
+      title += os.str();
+      rPlotFile.addEzsetCommand (title.c_str());
+      rPlotFile.addEzsetCommand ("xlabel Row");
+      rPlotFile.addEzsetCommand ("ylabel Pixel Value");
+      rPlotFile.addEzsetCommand ("lxfrac 0");
+      rPlotFile.addEzsetCommand ("box");
+      rPlotFile.addEzsetCommand ("grid");
+      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.addColumn (0, pX);
+      rPlotFile.addColumn (1, pY);
+    }
+    delete pX;
+    delete pY;
+    if (theApp->getSetModifyNewDocs())
+      pPlotDoc->Modify(true);
+    pPlotDoc->UpdateAllViews();
+  }
+}
+#endif
+
 void
 ImageFileView::OnCompareCol (wxCommandEvent& event)
 {
@@ -1301,6 +1507,10 @@ PhantomView::PhantomView(void)
   m_dDefaultFieldOfView = 1;
   m_iDefaultGeometry = Scanner::GEOMETRY_PARALLEL;
   m_iDefaultTrace = Trace::TRACE_NONE;
+
+  m_iDefaultRasterNX = 256;
+  m_iDefaultRasterNY = 256;
+  m_iDefaultRasterNSamples = 2;
 }
 
 PhantomView::~PhantomView(void)
@@ -1416,15 +1626,15 @@ PhantomView::OnProjections (wxCommandEvent& event)
 void
 PhantomView::OnRasterize (wxCommandEvent& event)
 {
-  DialogGetRasterParameters dialogRaster (m_frame, 256, 256, 1);
+  DialogGetRasterParameters dialogRaster (m_frame, m_iDefaultRasterNX, m_iDefaultRasterNY, m_iDefaultRasterNSamples);
   int retVal = dialogRaster.ShowModal();
   if (retVal == wxID_OK) {
-    int xSize = dialogRaster.getXSize();
-    int ySize = dialogRaster.getYSize();
-    int nSamples = dialogRaster.getNSamples();
-    if (nSamples < 1)
-      nSamples = 1;
-    if (xSize > 0 && ySize > 0) {
+    m_iDefaultRasterNX = dialogRaster.getXSize();
+    m_iDefaultRasterNY  = dialogRaster.getYSize();
+    m_iDefaultRasterNSamples = dialogRaster.getNSamples();
+    if (m_iDefaultRasterNSamples < 1)
+      m_iDefaultRasterNSamples = 1;
+    if (m_iDefaultRasterNX > 0 && m_iDefaultRasterNY > 0) {
       const Phantom& rPhantom = GetDocument()->getPhantom();
       ImageFileDocument* pRasterDoc = dynamic_cast<ImageFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.if", wxDOC_SILENT));
       if (! pRasterDoc) {
@@ -1433,11 +1643,11 @@ PhantomView::OnRasterize (wxCommandEvent& event)
       }
       ImageFile& imageFile = pRasterDoc->getImageFile();
       
-      imageFile.setArraySize (xSize, ySize);
+      imageFile.setArraySize (m_iDefaultRasterNX, m_iDefaultRasterNX);
       wxProgressDialog dlgProgress (wxString("Rasterize"), wxString("Rasterization Progress"), imageFile.nx() + 1, m_frame, wxPD_CAN_ABORT);
       Timer timer;
       for (unsigned int i = 0; i < imageFile.nx(); i++) {
-        rPhantom.convertToImagefile (imageFile, nSamples, Trace::TRACE_NONE, i, 1, true);
+        rPhantom.convertToImagefile (imageFile, m_iDefaultRasterNSamples, Trace::TRACE_NONE, i, 1, true);
         if (! dlgProgress.Update(i+1)) {
           pRasterDoc->DeleteAllViews();
           return;
@@ -1447,7 +1657,8 @@ PhantomView::OnRasterize (wxCommandEvent& event)
         pRasterDoc->Modify(true);
       pRasterDoc->UpdateAllViews(this);
       std::ostringstream os;
-      os << "Rasterize Phantom " << rPhantom.name() << ": XSize=" << xSize << ", YSize=" << ySize << ", nSamples=" << nSamples;
+      os << "Rasterize Phantom " << rPhantom.name() << ": XSize=" << m_iDefaultRasterNX << ", YSize=" 
+        << m_iDefaultRasterNY << ", nSamples=" << m_iDefaultRasterNSamples;
       *theApp->getLog() << os.str().c_str() << "\n";
       imageFile.labelAdd (os.str().c_str(), timer.timerEnd());
       ImageFileView* rasterView = dynamic_cast<ImageFileView*>(pRasterDoc->GetFirstView());
@@ -1479,8 +1690,12 @@ PhantomView::CreateCanvas (wxView *view, wxFrame *parent)
 wxFrame*
 PhantomView::CreateChildFrame(wxDocument *doc, wxView *view)
 {
+#if CTSIM_MDI
+  wxMDIChildFrame *subframe = new wxMDIChildFrame(theApp->getMainFrame(), -1, "Phantom Frame", wxPoint(10, 10), wxSize(256, 256), wxDEFAULT_FRAME_STYLE);
+#else
   wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "Phantom Frame", wxPoint(10, 10), wxSize(256, 256), wxDEFAULT_FRAME_STYLE);
-  
+#endif
+
   wxMenu *file_menu = new wxMenu;
   
   file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
@@ -1556,17 +1771,18 @@ PhantomView::OnClose (bool deleteWindow)
   if (!GetDocument()->Close())
     return false;
   
-  m_canvas->Clear();
+//  m_canvas->Clear();
   m_canvas->m_pView = NULL;
   m_canvas = NULL;
-  wxString s(wxTheApp->GetAppName());
-  if (m_frame)
-    m_frame->SetTitle(s);
+//  wxString s(wxTheApp->GetAppName());
+//  if (m_frame)
+//    m_frame->SetTitle(s);
   SetFrame(NULL);
   
   Activate(false);
   
   if (deleteWindow) {
+
     delete m_frame;
     return true;
   }
@@ -1834,8 +2050,12 @@ ProjectionFileView::CreateCanvas (wxView *view, wxFrame *parent)
 wxFrame*
 ProjectionFileView::CreateChildFrame(wxDocument *doc, wxView *view)
 {
+#ifdef CTSIM_MDI
+  wxMDIChildFrame *subframe = new wxMDIChildFrame (theApp->getMainFrame(), -1, "Projection Frame", wxPoint(10, 10), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
+#else
   wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "Projection Frame", wxPoint(10, 10), wxSize(0, 0), wxDEFAULT_FRAME_STYLE);
-  
+#endif
+
   wxMenu *file_menu = new wxMenu;
   
   file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
@@ -1967,7 +2187,7 @@ ProjectionFileView::OnClose (bool deleteWindow)
   if (!GetDocument()->Close())
     return false;
   
-  m_canvas->Clear();
+  // m_canvas->Clear();
   m_canvas->m_pView = NULL;
   m_canvas = NULL;
   wxString s(wxTheApp->GetAppName());
@@ -2118,8 +2338,12 @@ PlotFileView::CreateCanvas (wxView *view, wxFrame *parent)
 wxFrame*
 PlotFileView::CreateChildFrame(wxDocument *doc, wxView *view)
 {
+#ifdef CTSIM_MDI
+  wxMDIChildFrame *subframe = new wxMDIChildFrame (theApp->getMainFrame(), -1, "Plot Frame", wxPoint(10, 10), wxSize(500, 300), wxDEFAULT_FRAME_STYLE);
+#else
   wxDocChildFrame *subframe = new wxDocChildFrame(doc, view, theApp->getMainFrame(), -1, "Plot Frame", wxPoint(10, 10), wxSize(500, 300), wxDEFAULT_FRAME_STYLE);
-  
+#endif
+
   wxMenu *file_menu = new wxMenu;
   
   file_menu->Append(MAINMENU_FILE_CREATE_PHANTOM, "Cr&eate Phantom...");
@@ -2259,7 +2483,7 @@ PlotFileView::OnClose (bool deleteWindow)
   if (!GetDocument()->Close())
     return false;
   
-  m_canvas->Clear();
+  // m_canvas->Clear();
   m_canvas->m_pView = NULL;
   m_canvas = NULL;
   wxString s(wxTheApp->GetAppName());