X-Git-Url: http://git.kpe.io/?a=blobdiff_plain;f=src%2Fviews.cpp;h=72ed48dd8a4f2fd3abfb758aee970ac39adc8d8c;hb=e69a48c43eb11e0a6ffcf7f5edf28a1f10672d12;hp=7c47aa6415671807810ac4274111fdee5f3d90de;hpb=c6cda8844a491b71759e5dd5edba830d0b809cfd;p=ctsim.git

diff --git a/src/views.cpp b/src/views.cpp
index 7c47aa6..72ed48d 100644
--- a/src/views.cpp
+++ b/src/views.cpp
@@ -9,7 +9,7 @@
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: views.cpp,v 1.52 2001/01/06 15:33:15 kevin Exp $
+**  $Id: views.cpp,v 1.56 2001/01/08 02:02:18 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
@@ -1049,15 +1049,26 @@ ImageFileView::OnPlotRow (wxCommandEvent& event)
   
   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) {
     double* pX = new double [nx];
-    double* pY = new double [nx];
+    double* pYReal = new double [nx];
+    double *pYImag = NULL;
+    double *pYMag = NULL;
+    if (rIF.isComplex()) {
+      pYImag = new double [nx];
+      pYMag = new double [nx];
+    }
     for (int i = 0; i < nx; i++) {
       pX[i] = i;
-      pY[i] = v[i][yCursor];
+      pYReal[i] = v[i][yCursor];
+      if (rIF.isComplex()) {
+        pYImag[i] = vImag[i][yCursor];
+        pYMag[i] = ::sqrt (v[i][yCursor] * v[i][yCursor] + vImag[i][yCursor] * vImag[i][yCursor]);
+      }
     }
     PlotFileDocument* pPlotDoc = dynamic_cast<PlotFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.plt", wxDOC_SILENT));
     if (! pPlotDoc) {
@@ -1074,12 +1085,37 @@ ImageFileView::OnPlotRow (wxCommandEvent& event)
       rPlotFile.addEzsetCommand ("lxfrac 0");
       rPlotFile.addEzsetCommand ("box");
       rPlotFile.addEzsetCommand ("grid");
-      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.addEzsetCommand ("curve 1");
+      rPlotFile.addEzsetCommand ("color 1");
+      if (rIF.isComplex()) {
+        rPlotFile.addEzsetCommand ("dash 1");
+        rPlotFile.addEzsetCommand ("curve 2");
+        rPlotFile.addEzsetCommand ("color 4");
+        rPlotFile.addEzsetCommand ("dash 3");
+        rPlotFile.addEzsetCommand ("curve 3");
+        rPlotFile.addEzsetCommand ("color 0");
+        rPlotFile.addEzsetCommand ("solid");
+        rPlotFile.setCurveSize (4, nx);
+      } else
+        rPlotFile.setCurveSize (2, nx);
       rPlotFile.addColumn (0, pX);
-      rPlotFile.addColumn (1, pY);
+      rPlotFile.addColumn (1, pYReal); 
+      if (rIF.isComplex()) {
+        rPlotFile.addColumn (2, pYImag);
+        rPlotFile.addColumn (3, pYMag);
+      }
+      for (unsigned int iL = 0; iL < rIF.nLabels(); iL++)
+        rPlotFile.addDescription (rIF.labelGet(iL).getLabelString().c_str());
+      os << " Plot of " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+      *theApp->getLog() << os.str().c_str() << "\n";
+      rPlotFile.addDescription (os.str().c_str());
     }
     delete pX;
-    delete pY;
+    delete pYReal;
+    if (rIF.isComplex()) {
+      delete pYImag;
+      delete pYMag;
+    }
     if (theApp->getSetModifyNewDocs())
       pPlotDoc->Modify(true);
     pPlotDoc->UpdateAllViews();
@@ -1097,15 +1133,26 @@ ImageFileView::OnPlotCol (wxCommandEvent& event)
   
   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 && xCursor < nx) {
     double* pX = new double [ny];
-    double* pY = new double [ny];
+    double* pYReal = new double [ny];
+    double* pYImag = NULL;
+    double* pYMag = NULL;
+    if (rIF.isComplex()) {
+      pYImag = new double [ny];
+      pYMag = new double [ny];
+    }
     for (int i = 0; i < ny; i++) {
       pX[i] = i;
-      pY[i] = v[xCursor][i];
+      pYReal[i] = v[xCursor][i];
+      if (rIF.isComplex()) {
+        pYImag[i] = vImag[xCursor][i];
+        pYMag[i] = ::sqrt (v[xCursor][i] * v[xCursor][i] + vImag[xCursor][i] * vImag[xCursor][i]);
+      }
     }
     PlotFileDocument* pPlotDoc = dynamic_cast<PlotFileDocument*>(theApp->getDocManager()->CreateDocument("untitled.plt", wxDOC_SILENT));
     if (! pPlotDoc) {
@@ -1122,12 +1169,37 @@ ImageFileView::OnPlotCol (wxCommandEvent& event)
       rPlotFile.addEzsetCommand ("lxfrac 0");
       rPlotFile.addEzsetCommand ("box");
       rPlotFile.addEzsetCommand ("grid");
-      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.addEzsetCommand ("curve 1");
+      rPlotFile.addEzsetCommand ("color 1");
+      if (rIF.isComplex()) {
+        rPlotFile.addEzsetCommand ("dash 1");
+        rPlotFile.addEzsetCommand ("curve 2");
+        rPlotFile.addEzsetCommand ("color 4");
+        rPlotFile.addEzsetCommand ("dash 3");
+        rPlotFile.addEzsetCommand ("curve 3");
+        rPlotFile.addEzsetCommand ("color 0");
+        rPlotFile.addEzsetCommand ("solid");
+        rPlotFile.setCurveSize (4, ny);
+      } else
+        rPlotFile.setCurveSize (2, ny);
       rPlotFile.addColumn (0, pX);
-      rPlotFile.addColumn (1, pY);
+      rPlotFile.addColumn (1, pYReal); 
+      if (rIF.isComplex()) {
+        rPlotFile.addColumn (2, pYImag);
+        rPlotFile.addColumn (3, pYMag);
+      }
+      for (unsigned int iL = 0; iL < rIF.nLabels(); iL++)
+        rPlotFile.addDescription (rIF.labelGet(iL).getLabelString().c_str());
+      os << " Plot of " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+      *theApp->getLog() << os.str().c_str() << "\n";
+      rPlotFile.addDescription (os.str().c_str());
     }
     delete pX;
-    delete pY;
+    delete pYReal;
+    if (rIF.isComplex()) {
+      delete pYImag;
+      delete pYMag;
+    }
     if (theApp->getSetModifyNewDocs())
       pPlotDoc->Modify(true);
     pPlotDoc->UpdateAllViews();
@@ -1195,19 +1267,25 @@ ImageFileView::OnPlotFFTRow (wxCommandEvent& event)
       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 ("dash 1");
+        rPlotFile.addEzsetCommand ("curve 2");
+        rPlotFile.addEzsetCommand ("color 4");
+        rPlotFile.addEzsetCommand ("dash 3");
+        rPlotFile.addEzsetCommand ("curve 3");
+        rPlotFile.addEzsetCommand ("color 0");
+        rPlotFile.addEzsetCommand ("solid");
+       rPlotFile.addEzsetCommand ("box");
       rPlotFile.addEzsetCommand ("grid");
-      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.setCurveSize (4, nx);
       rPlotFile.addColumn (0, pX);
       rPlotFile.addColumn (1, pYReal);
       rPlotFile.addColumn (2, pYImag);
       rPlotFile.addColumn (3, pYMag);
+      for (int iL = 0; iL < rIF.nLabels(); iL++)
+        rPlotFile.addDescription (rIF.labelGet(iL).getLabelString().c_str());
+      os << " FFT Plot of " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+      *theApp->getLog() << os.str().c_str() << "\n";
+      rPlotFile.addDescription (os.str().c_str());
     }
     delete pX;
     delete pYReal;
@@ -1232,16 +1310,46 @@ ImageFileView::OnPlotFFTCol (wxCommandEvent& event)
   
   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 && xCursor < nx) {
+    fftw_complex* pcIn = new fftw_complex [ny];
+    double *pdTemp = new double [ny];
+
+    int i;
+    for (i = 0; i < ny; i++)
+      pdTemp[i] = v[xCursor][i];
+    Fourier::shuffleNaturalToFourierOrder (pdTemp, ny);
+    for (i = 0; i < ny; i++) 
+      pcIn[i].re = pdTemp[i];
+
+    for (i = 0; i < ny; i++) {
+      if (rIF.isComplex())
+        pdTemp[i] = vImag[xCursor][i];
+      else
+      pdTemp[i] = 0;
+    }
+    Fourier::shuffleNaturalToFourierOrder (pdTemp, ny);
+    for (i = 0; i < ny; i++)
+      pcIn[i].im = pdTemp[i];
+
+    fftw_plan plan = fftw_create_plan (ny, FFTW_BACKWARD, FFTW_IN_PLACE);
+    fftw_one (plan, pcIn, NULL);
+    fftw_destroy_plan (plan);
+
     double* pX = new double [ny];
-    double* pY = new double [ny];
-    for (int i = 0; i < ny; i++) {
+    double* pYReal = new double [ny];
+    double* pYImag = new double [ny];
+    double* pYMag = new double [ny];
+    for (i = 0; i < ny; i++) {
       pX[i] = i;
-      pY[i] = v[xCursor][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);
     }
+
     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");
@@ -1252,17 +1360,38 @@ ImageFileView::OnPlotFFTCol (wxCommandEvent& event)
       std::string title("title ");
       title += os.str();
       rPlotFile.addEzsetCommand (title.c_str());
-      rPlotFile.addEzsetCommand ("xlabel Row");
+      rPlotFile.addEzsetCommand ("xlabel Column");
       rPlotFile.addEzsetCommand ("ylabel Pixel Value");
       rPlotFile.addEzsetCommand ("lxfrac 0");
-      rPlotFile.addEzsetCommand ("box");
+      rPlotFile.addEzsetCommand ("curve 1");
+      rPlotFile.addEzsetCommand ("color 1");
+       rPlotFile.addEzsetCommand ("dash 1");
+        rPlotFile.addEzsetCommand ("curve 2");
+        rPlotFile.addEzsetCommand ("color 4");
+        rPlotFile.addEzsetCommand ("dash 3");
+        rPlotFile.addEzsetCommand ("curve 3");
+        rPlotFile.addEzsetCommand ("color 0");
+        rPlotFile.addEzsetCommand ("solid");
+       rPlotFile.addEzsetCommand ("box");
       rPlotFile.addEzsetCommand ("grid");
-      rPlotFile.setCurveSize (2, nx);
+      rPlotFile.setCurveSize (4, ny);
       rPlotFile.addColumn (0, pX);
-      rPlotFile.addColumn (1, pY);
+      rPlotFile.addColumn (1, pYReal);
+      rPlotFile.addColumn (2, pYImag);
+      rPlotFile.addColumn (3, pYMag);
+      for (int iL = 0; iL < rIF.nLabels(); iL++)
+        rPlotFile.addDescription (rIF.labelGet(iL).getLabelString().c_str());
+      os << " FFT Plot of " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+      *theApp->getLog() << os.str().c_str() << "\n";
+      rPlotFile.addDescription (os.str().c_str());
     }
     delete pX;
-    delete pY;
+    delete pYReal;
+    delete pYImag;
+    delete pYMag;
+    delete pdTemp;
+    delete [] pcIn;
+
     if (theApp->getSetModifyNewDocs())
       pPlotDoc->Modify(true);
     pPlotDoc->UpdateAllViews();
@@ -1326,10 +1455,28 @@ ImageFileView::OnCompareCol (wxCommandEvent& event)
         rPlotFile.addEzsetCommand ("dash 5");
         rPlotFile.addEzsetCommand ("box");
         rPlotFile.addEzsetCommand ("grid");
-        rPlotFile.setCurveSize (3, nx);
+        rPlotFile.setCurveSize (3, ny);
         rPlotFile.addColumn (0, pX);
         rPlotFile.addColumn (1, pY1);
         rPlotFile.addColumn (2, pY2);
+
+        unsigned int iL;
+        for (iL = 0; iL < rIF.nLabels(); iL++) {
+          std::string s = GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+          s += ": ";
+          s += rIF.labelGet(iL).getLabelString();
+          rPlotFile.addDescription (s.c_str());
+        }
+        for (iL = 0; iL < rIF.nLabels(); iL++) {
+          std::string s = pCompareDoc->GetFirstView()->GetFrame()->GetTitle().c_str();
+          s += ": ";
+          s += rCompareIF.labelGet(iL).getLabelString();
+          rPlotFile.addDescription (s.c_str());
+        }
+        os << " Between " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str() << " and "
+          << pCompareDoc->GetFirstView()->GetFrame()->GetTitle().c_str();
+        *theApp->getLog() << os.str().c_str() << "\n";
+        rPlotFile.addDescription (os.str().c_str());
       }
       delete pX;
       delete pY1;
@@ -1399,10 +1546,27 @@ ImageFileView::OnCompareRow (wxCommandEvent& event)
         rPlotFile.addEzsetCommand ("dash 5");
         rPlotFile.addEzsetCommand ("box");
         rPlotFile.addEzsetCommand ("grid");
-        rPlotFile.setCurveSize (3, ny);
+        rPlotFile.setCurveSize (3, nx);
         rPlotFile.addColumn (0, pX);
         rPlotFile.addColumn (1, pY1);
         rPlotFile.addColumn (2, pY2);
+        unsigned int iL;
+        for (iL = 0; iL < rIF.nLabels(); iL++) {
+          std::string s = GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str();
+          s += ": ";
+          s += rIF.labelGet(iL).getLabelString();
+          rPlotFile.addDescription (s.c_str());
+        }
+        for (iL = 0; iL < rIF.nLabels(); iL++) {
+          std::string s = pCompareDoc->GetFirstView()->GetFrame()->GetTitle().c_str();
+          s += ": ";
+          s += rCompareIF.labelGet(iL).getLabelString();
+          rPlotFile.addDescription (s.c_str());
+        }
+        os << " Between " << GetDocument()->GetFirstView()->GetFrame()->GetTitle().c_str() << " and "
+          << pCompareDoc->GetFirstView()->GetFrame()->GetTitle().c_str();
+        *theApp->getLog() << os.str().c_str() << "\n";
+        rPlotFile.addDescription (os.str().c_str());
       }
       delete pX;
       delete pY1;
@@ -1458,9 +1622,18 @@ ImageFileView::OnPlotHistogram (wxCommandEvent& event)
     rPlotFile.addEzsetCommand ("ylabel Count");
     rPlotFile.addEzsetCommand ("box");
     rPlotFile.addEzsetCommand ("grid");
-    rPlotFile.setCurveSize (2, nx);
+    rPlotFile.setCurveSize (2, NUMBER_HISTOGRAM_BINS);
     rPlotFile.addColumn (0, pX);
     rPlotFile.addColumn (1, pY);
+    for (unsigned int iL = 0; iL < rIF.nLabels(); iL++) {
+      std::string s = GetDocument()->GetFirstView()->GetFrame()->GetTitle();
+      s += ": ";
+      s += rIF.labelGet(iL).getLabelString();
+      rPlotFile.addDescription (s.c_str());
+    }
+    os << " Plot of " << GetDocument()->GetFirstView()->GetFrame()->GetTitle();
+    *theApp->getLog() << os.str().c_str() << "\n";
+    rPlotFile.addDescription (os.str().c_str());
     delete pX;
     delete pY;
     if (theApp->getSetModifyNewDocs())
@@ -2251,7 +2424,7 @@ PlotFileView::OnProperties (wxCommandEvent& event)
   const PlotFile& rPlot = GetDocument()->getPlotFile();
   std::ostringstream os;
   os << "Columns: " << rPlot.getNumColumns() << ", Records: " << rPlot.getNumRecords() << "\n";
-  rPlot.printHeaders (os);
+  rPlot.printHeadersBrief (os);
   *theApp->getLog() << os.str().c_str();
   wxMessageDialog dialogMsg (m_frame, os.str().c_str(), "Plot File Properties", wxOK | wxICON_INFORMATION);
   dialogMsg.ShowModal();