1 /*****************************************************************************
5 ** Purpose: Imagefile classes
6 ** Programmer: Kevin Rosenberg
7 ** Date Started: June 2000
9 ** This is part of the CTSim program
10 ** Copyright (C) 1983-2000 Kevin Rosenberg
12 ** $Id: imagefile.cpp,v 1.36 2001/01/12 16:41:56 kevin Exp $
14 ** This program is free software; you can redistribute it and/or modify
15 ** it under the terms of the GNU General Public License (version 2) as
16 ** published by the Free Software Foundation.
18 ** This program is distributed in the hope that it will be useful,
19 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
20 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 ** GNU General Public License for more details.
23 ** You should have received a copy of the GNU General Public License
24 ** along with this program; if not, write to the Free Software
25 ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 ******************************************************************************/
30 const int ImageFile::FORMAT_INVALID = -1;
31 const int ImageFile::FORMAT_PGM = 0;
32 const int ImageFile::FORMAT_PGMASCII = 1;
34 const int ImageFile::FORMAT_PNG = 2;
35 const int ImageFile::FORMAT_PNG16 = 3;
38 const char* ImageFile::s_aszFormatName[] =
48 const char* ImageFile::s_aszFormatTitle[] =
56 const int ImageFile::s_iFormatCount = sizeof(s_aszFormatName) / sizeof(const char*);
60 F32Image::F32Image (int nx, int ny, int dataType)
61 : Array2dFile (nx, ny, sizeof(kfloat32), Array2dFile::PIXEL_FLOAT32, dataType)
65 F32Image::F32Image (void)
68 setPixelFormat (Array2dFile::PIXEL_FLOAT32);
69 setPixelSize (sizeof(kfloat32));
70 setDataType (Array2dFile::DATA_TYPE_REAL);
73 F64Image::F64Image (int nx, int ny, int dataType)
74 : Array2dFile (nx, ny, sizeof(kfloat64), Array2dFile::PIXEL_FLOAT64, dataType)
78 F64Image::F64Image (void)
81 setPixelFormat (PIXEL_FLOAT64);
82 setPixelSize (sizeof(kfloat64));
83 setDataType (Array2dFile::DATA_TYPE_REAL);
87 ImageFile::getCenterCoordinates (unsigned int& iXCenter, unsigned int& iYCenter)
92 iXCenter = (m_nx - 1) / 2;
97 iYCenter = (m_ny - 1) / 2;
102 ImageFile::filterResponse (const char* const domainName, double bw, const char* const filterName, double filt_param, double dInputScale, double dOutputScale)
104 ImageFileArray v = getArray();
105 SignalFilter filter (filterName, domainName, bw, filt_param);
107 unsigned int iXCenter, iYCenter;
108 getCenterCoordinates (iXCenter, iYCenter);
110 for (unsigned int ix = 0; ix < m_nx; ix++)
111 for (unsigned int iy = 0; iy < m_ny; iy++) {
112 long lD2 = ((ix - iXCenter) * (ix - iXCenter)) + ((iy - iYCenter) * (iy - iYCenter));
113 double r = ::sqrt (static_cast<double>(lD2)) * dInputScale;
114 v[ix][iy] = filter.response (r) * dOutputScale;
119 ImageFile::display (void) const
123 getMinMax (pmin, pmax);
125 return (displayScaling (1, pmin, pmax));
129 ImageFile::displayScaling (const int scale, const ImageFileValue pmin, const ImageFileValue pmax) const
133 ImageFileArrayConst v = getArray();
134 if (v == NULL || nx == 0 || ny == 0)
138 int* pPens = new int [nx * ny * scale * scale ];
140 double view_scale = 255 / (pmax - pmin);
141 int id_X11 = g2_open_X11 (nx * scale, ny * scale);
143 for (int i = 0; i < 256; i++) {
144 double cval = i / 255.;
145 grayscale[i] = g2_ink (id_X11, cval, cval, cval);
148 for (int iy = ny - 1; iy >= 0; iy--) {
149 int iRowPos = ((ny - 1 - iy) * scale) * (nx * scale);
150 for (int ix = 0; ix < nx; ix++) {
151 int cval = static_cast<int>((v[ix][iy] - pmin) * view_scale);
156 for (int sy = 0; sy < scale; sy++)
157 for (int sx = 0; sx < scale; sx++)
158 pPens[iRowPos+(sy * nx * scale)+(sx + (ix * scale))] = grayscale[cval];
162 g2_image (id_X11, 0., 0., nx * scale, ny * scale, pPens);
173 // ImageFile::comparativeStatistics Calculate comparative stats
176 // d Normalized root mean squared distance measure
177 // r Normalized mean absolute distance measure
178 // e Worst case distance measure
181 // G.T. Herman, Image Reconstruction From Projections, 1980
184 ImageFile::comparativeStatistics (const ImageFile& imComp, double& d, double& r, double& e) const
186 if (imComp.nx() != m_nx && imComp.ny() != m_ny) {
187 sys_error (ERR_WARNING, "Image sizes differ [ImageFile::comparativeStatistics]");
190 ImageFileArrayConst v = getArray();
191 if (v == NULL || m_nx == 0 || m_ny == 0)
194 ImageFileArrayConst vComp = imComp.getArray();
197 for (unsigned int ix = 0; ix < m_nx; ix++) {
198 for (unsigned int iy = 0; iy < m_ny; iy++) {
202 myMean /= (m_nx * m_ny);
204 double sqErrorSum = 0.;
205 double absErrorSum = 0.;
206 double sqDiffFromMeanSum = 0.;
207 double absValueSum = 0.;
208 for (unsigned int ix2 = 0; ix2 < m_nx; ix2++) {
209 for (unsigned int iy = 0; iy < m_ny; iy++) {
210 double diff = v[ix2][iy] - vComp[ix2][iy];
211 sqErrorSum += diff * diff;
212 absErrorSum += fabs(diff);
213 double diffFromMean = v[ix2][iy] - myMean;
214 sqDiffFromMeanSum += diffFromMean * diffFromMean;
215 absValueSum += fabs(v[ix2][iy]);
219 d = ::sqrt (sqErrorSum / sqDiffFromMeanSum);
220 r = absErrorSum / absValueSum;
225 for (int ix3 = 0; ix3 < hx; ix3++) {
226 for (int iy = 0; iy < hy; iy++) {
227 double avgPixel = 0.25 * (v[2*ix3][2*iy] + v[2*ix3+1][2*iy] + v[2*ix3][2*iy+1] + v[2*ix3+1][2*iy+1]);
228 double avgPixelComp = 0.25 * (vComp[2*ix3][2*iy] + vComp[2*ix3+1][2*iy] + vComp[2*ix3][2*iy+1] + vComp[2*ix3+1][2*iy+1]);
229 double error = fabs (avgPixel - avgPixelComp);
242 ImageFile::printComparativeStatistics (const ImageFile& imComp, std::ostream& os) const
246 if (comparativeStatistics (imComp, d, r, e)) {
247 os << " Normalized root mean squared distance (d): " << d << std::endl;
248 os << " Normalized mean absolute distance (r): " << r << std::endl;
249 os << "Worst case distance (2x2 pixel average) (e): " << e << std::endl;
257 ImageFile::printStatistics (std::ostream& os) const
259 double min, max, mean, mode, median, stddev;
261 statistics (min, max, mean, mode, median, stddev);
263 os << "Real Component Statistics" << std::endl;
265 os << " min: " << min << std::endl;
266 os << " max: " << max << std::endl;
267 os << " mean: " << mean << std::endl;
268 os << " mode: " << mode << std::endl;
269 os << "median: " << median << std::endl;
270 os << "stddev: " << stddev << std::endl;
273 statistics (getImaginaryArray(), min, max, mean, mode, median, stddev);
274 os << std::endl << "Imaginary Component Statistics" << std::endl;
275 os << " min: " << min << std::endl;
276 os << " max: " << max << std::endl;
277 os << " mean: " << mean << std::endl;
278 os << " mode: " << mode << std::endl;
279 os << "median: " << median << std::endl;
280 os << "stddev: " << stddev << std::endl;
286 ImageFile::statistics (double& min, double& max, double& mean, double& mode, double& median, double& stddev) const
288 ImageFileArrayConst v = getArray();
289 statistics (v, min, max, mean, mode, median, stddev);
294 ImageFile::statistics (ImageFileArrayConst v, double& min, double& max, double& mean, double& mode, double& median, double& stddev) const
299 if (v == NULL || nx == 0 || ny == 0)
302 std::vector<double> vecImage;
304 vecImage.resize (nx * ny);
305 for (int ix = 0; ix < nx; ix++) {
306 for (int iy = 0; iy < ny; iy++)
307 vecImage[iVec++] = v[ix][iy];
310 vectorNumericStatistics (vecImage, nx * ny, min, max, mean, mode, median, stddev);
314 ImageFile::getMinMax (double& min, double& max) const
318 ImageFileArrayConst v = getArray();
320 if (v == NULL || nx == 0 || ny == 0)
325 for (int ix = 0; ix < nx; ix++) {
326 for (int iy = 0; iy < ny; iy++) {
336 ImageFile::convertRealToComplex ()
338 if (dataType() != Array2dFile::DATA_TYPE_REAL)
341 if (! reallocRealToComplex())
344 ImageFileArray vImag = getImaginaryArray();
345 for (unsigned int ix = 0; ix < m_nx; ix++) {
346 ImageFileColumn vCol = vImag[ix];
347 for (unsigned int iy = 0; iy < m_ny; iy++)
355 ImageFile::convertComplexToReal ()
357 if (dataType() != Array2dFile::DATA_TYPE_COMPLEX)
360 ImageFileArray vReal = getArray();
361 ImageFileArray vImag = getImaginaryArray();
362 for (unsigned int ix = 0; ix < m_nx; ix++) {
363 ImageFileColumn vRealCol = vReal[ix];
364 ImageFileColumn vImagCol = vImag[ix];
365 for (unsigned int iy = 0; iy < m_ny; iy++) {
366 CTSimComplex c (*vRealCol, *vImagCol);
367 *vRealCol++ = std::abs (c);
372 return reallocComplexToReal();
376 ImageFile::subtractImages (const ImageFile& rRHS, ImageFile& result) const
378 if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) {
379 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]");
383 if (isComplex() || rRHS.isComplex() && ! result.isComplex())
384 result.convertRealToComplex();
386 ImageFileArrayConst vLHS = getArray();
387 ImageFileArrayConst vLHSImag = getImaginaryArray();
388 ImageFileArrayConst vRHS = rRHS.getArray();
389 ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray();
390 ImageFileArray vResult = result.getArray();
391 ImageFileArray vResultImag = result.getImaginaryArray();
393 for (unsigned int ix = 0; ix < m_nx; ix++) {
394 for (unsigned int iy = 0; iy < m_ny; iy++) {
395 vResult[ix][iy] = vLHS[ix][iy] - vRHS[ix][iy];
396 if (result.isComplex()) {
397 vResultImag[ix][iy] = 0;
399 vResultImag[ix][iy] += vLHSImag[ix][iy];
400 if (rRHS.isComplex())
401 vResultImag[ix][iy] -= vRHSImag[ix][iy];
410 ImageFile::addImages (const ImageFile& rRHS, ImageFile& result) const
412 if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) {
413 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]");
417 if (isComplex() || rRHS.isComplex() && ! result.isComplex())
418 result.convertRealToComplex();
420 ImageFileArrayConst vLHS = getArray();
421 ImageFileArrayConst vLHSImag = getImaginaryArray();
422 ImageFileArrayConst vRHS = rRHS.getArray();
423 ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray();
424 ImageFileArray vResult = result.getArray();
425 ImageFileArray vResultImag = result.getImaginaryArray();
427 for (unsigned int ix = 0; ix < m_nx; ix++) {
428 for (unsigned int iy = 0; iy < m_ny; iy++) {
429 vResult[ix][iy] = vLHS[ix][iy] + vRHS[ix][iy];
430 if (result.isComplex()) {
431 vResultImag[ix][iy] = 0;
433 vResultImag[ix][iy] += vLHSImag[ix][iy];
434 if (rRHS.isComplex())
435 vResultImag[ix][iy] += vRHSImag[ix][iy];
444 ImageFile::multiplyImages (const ImageFile& rRHS, ImageFile& result) const
446 if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) {
447 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]");
451 if (isComplex() || rRHS.isComplex() && ! result.isComplex())
452 result.convertRealToComplex();
454 ImageFileArrayConst vLHS = getArray();
455 ImageFileArrayConst vLHSImag = getImaginaryArray();
456 ImageFileArrayConst vRHS = rRHS.getArray();
457 ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray();
458 ImageFileArray vResult = result.getArray();
459 ImageFileArray vResultImag = result.getImaginaryArray();
461 for (unsigned int ix = 0; ix < m_nx; ix++) {
462 for (unsigned int iy = 0; iy < m_ny; iy++) {
463 if (result.isComplex()) {
466 dImag = vLHSImag[ix][iy];
467 std::complex<double> cLHS (vLHS[ix][iy], dImag);
469 if (rRHS.isComplex())
470 dImag = vRHSImag[ix][iy];
471 std::complex<double> cRHS (vRHS[ix][iy], dImag);
472 std::complex<double> cResult = cLHS * cRHS;
473 vResult[ix][iy] = cResult.real();
474 vResultImag[ix][iy] = cResult.imag();
476 vResult[ix][iy] = vLHS[ix][iy] * vRHS[ix][iy];
485 ImageFile::divideImages (const ImageFile& rRHS, ImageFile& result) const
487 if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) {
488 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]");
492 if (isComplex() || rRHS.isComplex() && ! result.isComplex())
493 result.convertRealToComplex();
495 ImageFileArrayConst vLHS = getArray();
496 ImageFileArrayConst vLHSImag = getImaginaryArray();
497 ImageFileArrayConst vRHS = rRHS.getArray();
498 ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray();
499 ImageFileArray vResult = result.getArray();
500 ImageFileArray vResultImag = result.getImaginaryArray();
502 for (unsigned int ix = 0; ix < m_nx; ix++) {
503 for (unsigned int iy = 0; iy < m_ny; iy++) {
504 if (result.isComplex()) {
507 dImag = vLHSImag[ix][iy];
508 std::complex<double> cLHS (vLHS[ix][iy], dImag);
510 if (rRHS.isComplex())
511 dImag = vRHSImag[ix][iy];
512 std::complex<double> cRHS (vRHS[ix][iy], dImag);
513 std::complex<double> cResult = cLHS / cRHS;
514 vResult[ix][iy] = cResult.real();
515 vResultImag[ix][iy] = cResult.imag();
518 vResult[ix][iy] = vLHS[ix][iy] / vRHS[ix][iy];
530 ImageFile::invertPixelValues (ImageFile& result) const
532 if (m_nx != result.nx() || m_ny != result.ny()) {
533 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
537 if (isComplex() && ! result.isComplex())
538 result.convertRealToComplex();
540 ImageFileArrayConst vLHS = getArray();
541 ImageFileArray vResult = result.getArray();
543 for (unsigned int ix = 0; ix < m_nx; ix++) {
544 ImageFileColumnConst in = vLHS[ix];
545 ImageFileColumn out = vResult[ix];
546 for (unsigned int iy = 0; iy < m_ny; iy++)
554 ImageFile::sqrt (ImageFile& result) const
556 if (m_nx != result.nx() || m_ny != result.ny()) {
557 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
561 if (isComplex() && ! result.isComplex())
562 result.convertRealToComplex();
564 bool bComplexOutput = result.isComplex();
565 ImageFileArrayConst vLHS = getArray();
566 if (! bComplexOutput) // check if should convert to complex output
567 for (unsigned int ix = 0; ix < m_nx; ix++)
568 for (unsigned int iy = 0; iy < m_ny; iy++)
569 if (! bComplexOutput && vLHS[ix][iy] < 0) {
570 result.convertRealToComplex();
571 bComplexOutput = true;
575 ImageFileArrayConst vLHSImag = getImaginaryArray();
576 ImageFileArray vResult = result.getArray();
577 ImageFileArray vResultImag = result.getImaginaryArray();
579 for (unsigned int ix = 0; ix < m_nx; ix++) {
580 for (unsigned int iy = 0; iy < m_ny; iy++) {
581 if (result.isComplex()) {
584 dImag = vLHSImag[ix][iy];
585 std::complex<double> cLHS (vLHS[ix][iy], dImag);
586 std::complex<double> cResult = std::sqrt(cLHS);
587 vResult[ix][iy] = cResult.real();
588 vResultImag[ix][iy] = cResult.imag();
590 vResult[ix][iy] = ::sqrt (vLHS[ix][iy]);
599 ImageFile::log (ImageFile& result) const
601 if (m_nx != result.nx() || m_ny != result.ny()) {
602 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
606 if (isComplex() && ! result.isComplex())
607 result.convertRealToComplex();
609 ImageFileArrayConst vLHS = getArray();
610 ImageFileArrayConst vLHSImag = getImaginaryArray();
611 ImageFileArray vResult = result.getArray();
612 ImageFileArray vResultImag = result.getImaginaryArray();
614 for (unsigned int ix = 0; ix < m_nx; ix++) {
615 for (unsigned int iy = 0; iy < m_ny; iy++) {
616 if (result.isComplex()) {
619 dImag = vLHSImag[ix][iy];
620 std::complex<double> cLHS (vLHS[ix][iy], dImag);
621 std::complex<double> cResult = std::log (cLHS);
622 vResult[ix][iy] = cResult.real();
623 vResultImag[ix][iy] = cResult.imag();
625 vResult[ix][iy] = ::log (vLHS[ix][iy]);
634 ImageFile::exp (ImageFile& result) const
636 if (m_nx != result.nx() || m_ny != result.ny()) {
637 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
641 if (isComplex() && ! result.isComplex())
642 result.convertRealToComplex();
644 ImageFileArrayConst vLHS = getArray();
645 ImageFileArrayConst vLHSImag = getImaginaryArray();
646 ImageFileArray vResult = result.getArray();
647 ImageFileArray vResultImag = result.getImaginaryArray();
649 for (unsigned int ix = 0; ix < m_nx; ix++) {
650 for (unsigned int iy = 0; iy < m_ny; iy++) {
651 if (result.isComplex()) {
654 dImag = vLHSImag[ix][iy];
655 std::complex<double> cLHS (vLHS[ix][iy], dImag);
656 std::complex<double> cResult = std::exp (cLHS);
657 vResult[ix][iy] = cResult.real();
658 vResultImag[ix][iy] = cResult.imag();
660 vResult[ix][iy] = ::exp (vLHS[ix][iy]);
669 ImageFile::scaleImage (ImageFile& result) const
671 unsigned int nx = m_nx;
672 unsigned int ny = m_ny;
673 unsigned int newNX = result.nx();
674 unsigned int newNY = result.ny();
676 double dXScale = static_cast<double>(newNX) / static_cast<double>(nx);
677 double dYScale = static_cast<double>(newNY) / static_cast<double>(ny);
679 if (isComplex() && ! result.isComplex())
680 result.convertRealToComplex();
682 ImageFileArrayConst vReal = getArray();
683 ImageFileArrayConst vImag = getImaginaryArray();
684 ImageFileArray vResult = result.getArray();
685 ImageFileArray vResultImag = result.getImaginaryArray();
687 for (unsigned int ix = 0; ix < newNX; ix++) {
688 for (unsigned int iy = 0; iy < newNY; iy++) {
689 double dXPos = ix / dXScale;
690 double dYPos = iy / dYScale;
691 unsigned int scaleNX = static_cast<unsigned int> (dXPos);
692 unsigned int scaleNY = static_cast<unsigned int> (dYPos);
693 double dXFrac = dXPos - scaleNX;
694 double dYFrac = dYPos - scaleNY;
695 if (scaleNX >= nx - 1 || scaleNY >= ny - 1) {
696 vResult[ix][iy] = vReal[scaleNX][scaleNY];
697 if (result.isComplex()) {
699 vResultImag[ix][iy] = vImag[scaleNX][scaleNY];
701 vResultImag[ix][iy] = 0;
704 vResult[ix][iy] = (1 - dXFrac) * (1 - dYFrac) * vReal[scaleNX][scaleNY] +
705 dXFrac * (1 - dYFrac) * vReal[scaleNX+1][scaleNY] +
706 dYFrac * (1 - dXFrac) * vReal[scaleNX][scaleNY+1] +
707 dXFrac * dYFrac * vReal[scaleNX+1][scaleNY+1];
708 if (result.isComplex()) {
710 vResultImag[ix][iy] = (1 - dXFrac) * (1 - dYFrac) * vImag[scaleNX][scaleNY] +
711 dXFrac * (1 - dYFrac) * vImag[scaleNX+1][scaleNY] +
712 dYFrac * (1 - dXFrac) * vImag[scaleNX][scaleNY+1] +
713 dXFrac * dYFrac * vImag[scaleNX+1][scaleNY+1];
715 vResultImag[ix][iy] = 0;
726 ImageFile::fft (ImageFile& result) const
728 if (m_nx != result.nx() || m_ny != result.ny()) {
729 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
733 if (result.dataType() == Array2dFile::DATA_TYPE_REAL) {
734 if (! result.convertRealToComplex ())
738 fftw_complex* in = new fftw_complex [m_nx * m_ny];
740 ImageFileArrayConst vReal = getArray();
741 ImageFileArrayConst vImag = getImaginaryArray();
744 unsigned int iArray = 0;
745 for (ix = 0; ix < m_nx; ix++)
746 for (iy = 0; iy < m_ny; iy++) {
747 in[iArray].re = vReal[ix][iy];
749 in[iArray].im = vImag[ix][iy];
755 fftwnd_plan plan = fftw2d_create_plan (m_nx, m_ny, FFTW_FORWARD, FFTW_IN_PLACE);
757 fftwnd_one (plan, in, NULL);
759 ImageFileArray vRealResult = result.getArray();
760 ImageFileArray vImagResult = result.getImaginaryArray();
762 unsigned int iScale = m_nx * m_ny;
763 for (ix = 0; ix < m_nx; ix++)
764 for (iy = 0; iy < m_ny; iy++) {
765 vRealResult[ix][iy] = in[iArray].re / iScale;
766 vImagResult[ix][iy] = in[iArray].im / iScale;
770 fftwnd_destroy_plan (plan);
774 Fourier::shuffleFourierToNaturalOrder (result);
781 ImageFile::ifft (ImageFile& result) const
783 if (m_nx != result.nx() || m_ny != result.ny()) {
784 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
788 if (result.dataType() == Array2dFile::DATA_TYPE_REAL) {
789 if (! result.convertRealToComplex ())
793 ImageFileArrayConst vReal = getArray();
794 ImageFileArrayConst vImag = getImaginaryArray();
795 ImageFileArray vRealResult = result.getArray();
796 ImageFileArray vImagResult = result.getImaginaryArray();
798 for (ix = 0; ix < m_nx; ix++)
799 for (iy = 0; iy < m_ny; iy++) {
800 vRealResult[ix][iy] = vReal[ix][iy];
802 vImagResult[ix][iy] = vImag[ix][iy];
804 vImagResult[ix][iy] = 0;
807 Fourier::shuffleNaturalToFourierOrder (result);
809 fftw_complex* in = new fftw_complex [m_nx * m_ny];
811 unsigned int iArray = 0;
812 for (ix = 0; ix < m_nx; ix++)
813 for (iy = 0; iy < m_ny; iy++) {
814 in[iArray].re = vRealResult[ix][iy];
815 in[iArray].im = vImagResult[ix][iy];
819 fftwnd_plan plan = fftw2d_create_plan (m_nx, m_ny, FFTW_BACKWARD, FFTW_IN_PLACE);
821 fftwnd_one (plan, in, NULL);
824 for (ix = 0; ix < m_nx; ix++)
825 for (iy = 0; iy < m_ny; iy++) {
826 vRealResult[ix][iy] = in[iArray].re;
827 vImagResult[ix][iy] = in[iArray].im;
831 fftwnd_destroy_plan (plan);
839 ImageFile::fftRows (ImageFile& result) const
841 if (m_nx != result.nx() || m_ny != result.ny()) {
842 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::fftRows]");
846 if (result.dataType() == Array2dFile::DATA_TYPE_REAL) {
847 if (! result.convertRealToComplex ())
851 fftw_complex* in = new fftw_complex [m_nx];
853 ImageFileArrayConst vReal = getArray();
854 ImageFileArrayConst vImag = getImaginaryArray();
856 fftw_plan plan = fftw_create_plan (m_nx, FFTW_FORWARD, FFTW_IN_PLACE);
857 std::complex<double>* pcRow = new std::complex<double> [m_nx];
860 unsigned int iArray = 0;
861 for (iy = 0; iy < m_ny; iy++) {
862 for (ix = 0; ix < m_nx; ix++) {
863 in[ix].re = vReal[ix][iy];
865 in[ix].im = vImag[ix][iy];
870 fftw_one (plan, in, NULL);
872 for (ix = 0; ix < m_nx; ix++)
873 pcRow[ix] = std::complex<double>(in[ix].re, in[ix].im);
875 Fourier::shuffleFourierToNaturalOrder (pcRow, m_nx);
876 for (ix = 0; ix < m_nx; ix++) {
877 vReal[ix][iy] = pcRow[ix].real();
878 vImag[ix][iy] = pcRow[ix].imag();
883 fftw_destroy_plan (plan);
890 ImageFile::ifftRows (ImageFile& result) const
892 if (m_nx != result.nx() || m_ny != result.ny()) {
893 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::fftRows]");
897 if (result.dataType() == Array2dFile::DATA_TYPE_REAL) {
898 if (! result.convertRealToComplex ())
902 fftw_complex* in = new fftw_complex [m_nx];
904 ImageFileArrayConst vReal = getArray();
905 ImageFileArrayConst vImag = getImaginaryArray();
907 fftw_plan plan = fftw_create_plan (m_nx, FFTW_BACKWARD, FFTW_IN_PLACE);
908 std::complex<double>* pcRow = new std::complex<double> [m_nx];
911 unsigned int iArray = 0;
912 for (iy = 0; iy < m_ny; iy++) {
913 for (ix = 0; ix < m_nx; ix++) {
916 dImag = vImag[ix][iy];
917 pcRow[ix] = std::complex<double> (vReal[ix][iy], dImag);
920 Fourier::shuffleNaturalToFourierOrder (pcRow, m_nx);
922 for (ix = 0; ix < m_nx; ix++) {
923 in[ix].re = pcRow[ix].real();
924 in[ix].im = pcRow[ix].imag();
927 fftw_one (plan, in, NULL);
929 for (ix = 0; ix < m_nx; ix++) {
930 vReal[ix][iy] = in[ix].re;
931 vImag[ix][iy] = in[ix].im;
936 fftw_destroy_plan (plan);
943 ImageFile::fftCols (ImageFile& result) const
949 ImageFile::ifftCols (ImageFile& result) const
959 ImageFile::fourier (ImageFile& result) const
961 if (m_nx != result.nx() || m_ny != result.ny()) {
962 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
966 if (! result.isComplex())
967 if (! result.convertRealToComplex ())
970 ImageFileArrayConst vLHS = getArray();
971 ImageFileArrayConst vLHSImag = getImaginaryArray();
972 ImageFileArray vRealResult = result.getArray();
973 ImageFileArray vImagResult = result.getImaginaryArray();
977 // alloc output matrix
978 CTSimComplex** complexOut = new CTSimComplex* [m_nx];
979 for (ix = 0; ix < m_nx; ix++)
980 complexOut[ix] = new CTSimComplex [m_ny];
982 // fourier each x column
983 CTSimComplex* pY = new CTSimComplex [m_ny];
984 for (ix = 0; ix < m_nx; ix++) {
985 for (iy = 0; iy < m_ny; iy++) {
988 dImag = vLHSImag[ix][iy];
989 pY[iy] = std::complex<double>(vLHS[ix][iy], dImag);
991 ProcessSignal::finiteFourierTransform (pY, complexOut[ix], m_ny, ProcessSignal::FORWARD);
995 // fourier each y row
996 CTSimComplex* pX = new CTSimComplex [m_nx];
997 CTSimComplex* complexOutRow = new CTSimComplex [m_nx];
998 for (iy = 0; iy < m_ny; iy++) {
999 for (ix = 0; ix < m_nx; ix++)
1000 pX[ix] = complexOut[ix][iy];
1001 ProcessSignal::finiteFourierTransform (pX, complexOutRow, m_nx, ProcessSignal::FORWARD);
1002 for (ix = 0; ix < m_nx; ix++)
1003 complexOut[ix][iy] = complexOutRow[ix];
1006 delete [] complexOutRow;
1008 for (ix = 0; ix < m_nx; ix++)
1009 for (iy = 0; iy < m_ny; iy++) {
1010 vRealResult[ix][iy] = complexOut[ix][iy].real();
1011 vImagResult[ix][iy] = complexOut[ix][iy].imag();
1014 Fourier::shuffleFourierToNaturalOrder (result);
1016 // delete complexOut matrix
1017 for (ix = 0; ix < m_nx; ix++)
1018 delete [] complexOut[ix];
1019 delete [] complexOut;
1025 ImageFile::inverseFourier (ImageFile& result) const
1027 if (m_nx != result.nx() || m_ny != result.ny()) {
1028 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
1032 if (result.dataType() == Array2dFile::DATA_TYPE_REAL) {
1033 if (! result.convertRealToComplex ())
1037 ImageFileArrayConst vLHSReal = getArray();
1038 ImageFileArrayConst vLHSImag = getImaginaryArray();
1039 ImageFileArray vRealResult = result.getArray();
1040 ImageFileArray vImagResult = result.getImaginaryArray();
1042 unsigned int ix, iy;
1043 // alloc 2d complex output matrix
1044 CTSimComplex** complexOut = new CTSimComplex* [m_nx];
1045 for (ix = 0; ix < m_nx; ix++)
1046 complexOut[ix] = new CTSimComplex [m_ny];
1048 // put input image into result
1049 for (ix = 0; ix < m_nx; ix++)
1050 for (iy = 0; iy < m_ny; iy++) {
1051 vRealResult[ix][iy] = vLHSReal[ix][iy];
1053 vImagResult[ix][iy] = vLHSImag[ix][iy];
1055 vImagResult[ix][iy] = 0;
1058 Fourier::shuffleNaturalToFourierOrder (result);
1060 // ifourier each x column
1061 CTSimComplex* pCol = new CTSimComplex [m_ny];
1062 for (ix = 0; ix < m_nx; ix++) {
1063 for (iy = 0; iy < m_ny; iy++) {
1064 pCol[iy] = std::complex<double> (vRealResult[ix][iy], vImagResult[ix][iy]);
1066 ProcessSignal::finiteFourierTransform (pCol, complexOut[ix], m_ny, ProcessSignal::BACKWARD);
1070 // ifourier each y row
1071 CTSimComplex* complexInRow = new CTSimComplex [m_nx];
1072 CTSimComplex* complexOutRow = new CTSimComplex [m_nx];
1073 for (iy = 0; iy < m_ny; iy++) {
1074 for (ix = 0; ix < m_nx; ix++)
1075 complexInRow[ix] = complexOut[ix][iy];
1076 ProcessSignal::finiteFourierTransform (complexInRow, complexOutRow, m_nx, ProcessSignal::BACKWARD);
1077 for (ix = 0; ix < m_nx; ix++)
1078 complexOut[ix][iy] = complexOutRow[ix];
1080 delete [] complexInRow;
1081 delete [] complexOutRow;
1083 for (ix = 0; ix < m_nx; ix++)
1084 for (iy = 0; iy < m_ny; iy++) {
1085 vRealResult[ix][iy] = complexOut[ix][iy].real();
1086 vImagResult[ix][iy] = complexOut[ix][iy].imag();
1089 // delete complexOut matrix
1090 for (ix = 0; ix < m_nx; ix++)
1091 delete [] complexOut[ix];
1092 delete [] complexOut;
1099 ImageFile::magnitude (ImageFile& result) const
1101 if (m_nx != result.nx() || m_ny != result.ny()) {
1102 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
1106 ImageFileArray vReal = getArray();
1107 ImageFileArray vImag = getImaginaryArray();
1108 ImageFileArray vRealResult = result.getArray();
1110 for (unsigned int ix = 0; ix < m_nx; ix++)
1111 for (unsigned int iy = 0; iy < m_ny; iy++) {
1113 vRealResult[ix][iy] = ::sqrt (vReal[ix][iy] * vReal[ix][iy] + vImag[ix][iy] * vImag[ix][iy]);
1115 vRealResult[ix][iy] = vReal[ix][iy];
1118 if (result.isComplex())
1119 result.convertComplexToReal();
1125 ImageFile::phase (ImageFile& result) const
1127 if (m_nx != result.nx() || m_ny != result.ny()) {
1128 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
1132 ImageFileArray vReal = getArray();
1133 ImageFileArray vImag = getImaginaryArray();
1134 ImageFileArray vRealResult = result.getArray();
1136 for (unsigned int ix = 0; ix < m_nx; ix++)
1137 for (unsigned int iy = 0; iy < m_ny; iy++) {
1139 vRealResult[ix][iy] = ::atan2 (vImag[ix][iy], vReal[ix][iy]);
1141 vRealResult[ix][iy] = 0;
1144 if (result.isComplex())
1145 result.convertComplexToReal();
1151 ImageFile::square (ImageFile& result) const
1153 if (m_nx != result.nx() || m_ny != result.ny()) {
1154 sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]");
1158 if (isComplex() && ! result.isComplex())
1159 result.convertRealToComplex();
1161 ImageFileArrayConst vLHS = getArray();
1162 ImageFileArrayConst vLHSImag = getImaginaryArray();
1163 ImageFileArray vResult = result.getArray();
1164 ImageFileArray vResultImag = result.getImaginaryArray();
1166 for (unsigned int ix = 0; ix < m_nx; ix++) {
1167 for (unsigned int iy = 0; iy < m_ny; iy++) {
1168 if (result.isComplex()) {
1171 dImag = vLHSImag[ix][iy];
1172 std::complex<double> cLHS (vLHS[ix][iy], dImag);
1173 std::complex<double> cResult = cLHS * cLHS;
1174 vResult[ix][iy] = cResult.real();
1175 vResultImag[ix][iy] = cResult.imag();
1177 vResult[ix][iy] = vLHS[ix][iy] * vLHS[ix][iy];
1186 ImageFile::convertFormatNameToID (const char* const formatName)
1188 int formatID = FORMAT_INVALID;
1190 for (int i = 0; i < s_iFormatCount; i++)
1191 if (strcasecmp (formatName, s_aszFormatName[i]) == 0) {
1200 ImageFile::convertFormatIDToName (int formatID)
1202 static const char *formatName = "";
1204 if (formatID >= 0 && formatID < s_iFormatCount)
1205 return (s_aszFormatName[formatID]);
1207 return (formatName);
1211 ImageFile::convertFormatIDToTitle (const int formatID)
1213 static const char *formatTitle = "";
1215 if (formatID >= 0 && formatID < s_iFormatCount)
1216 return (s_aszFormatTitle[formatID]);
1218 return (formatTitle);
1222 ImageFile::exportImage (const char* const pszFormat, const char* const pszFilename, int nxcell, int nycell, double densmin, double densmax)
1224 int iFormatID = convertFormatNameToID (pszFormat);
1225 if (iFormatID == FORMAT_INVALID) {
1226 sys_error (ERR_SEVERE, "Invalid format %s [ImageFile::exportImage]", pszFormat);
1230 if (iFormatID == FORMAT_PGM)
1231 return writeImagePGM (pszFilename, nxcell, nycell, densmin, densmax);
1232 else if (iFormatID == FORMAT_PGMASCII)
1233 return writeImagePGMASCII (pszFilename, nxcell, nycell, densmin, densmax);
1234 else if (iFormatID == FORMAT_PNG)
1235 return writeImagePNG (pszFilename, 8, nxcell, nycell, densmin, densmax);
1236 else if (iFormatID == FORMAT_PNG16)
1237 return writeImagePNG (pszFilename, 16, nxcell, nycell, densmin, densmax);
1239 sys_error (ERR_SEVERE, "Invalid format %s [ImageFile::exportImage]", pszFormat);
1244 ImageFile::writeImagePGM (const char* const outfile, int nxcell, int nycell, double densmin, double densmax)
1249 ImageFileArray v = getArray();
1251 unsigned char* rowp = new unsigned char [nx * nxcell];
1253 if ((fp = fopen (outfile, "wb")) == NULL)
1256 fprintf(fp, "P5\n");
1257 fprintf(fp, "%d %d\n", nx, ny);
1258 fprintf(fp, "255\n");
1260 for (int irow = ny - 1; irow >= 0; irow--) {
1261 for (int icol = 0; icol < nx; icol++) {
1262 int pos = icol * nxcell;
1263 double dens = (v[icol][irow] - densmin) / (densmax - densmin);
1264 dens = clamp (dens, 0., 1.);
1265 for (int p = pos; p < pos + nxcell; p++) {
1266 rowp[p] = static_cast<unsigned int> (dens * 255.);
1269 for (int ir = 0; ir < nycell; ir++) {
1270 for (int ic = 0; ic < nx * nxcell; ic++)
1271 fputc( rowp[ic], fp );
1282 ImageFile::writeImagePGMASCII (const char* const outfile, int nxcell, int nycell, double densmin, double densmax)
1287 ImageFileArray v = getArray();
1289 unsigned char* rowp = new unsigned char [nx * nxcell];
1291 if ((fp = fopen (outfile, "wb")) == NULL)
1294 fprintf(fp, "P2\n");
1295 fprintf(fp, "%d %d\n", nx, ny);
1296 fprintf(fp, "255\n");
1298 for (int irow = ny - 1; irow >= 0; irow--) {
1299 for (int icol = 0; icol < nx; icol++) {
1300 int pos = icol * nxcell;
1301 double dens = (v[icol][irow] - densmin) / (densmax - densmin);
1302 dens = clamp (dens, 0., 1.);
1303 for (int p = pos; p < pos + nxcell; p++) {
1304 rowp[p] = static_cast<unsigned int> (dens * 255.);
1307 for (int ir = 0; ir < nycell; ir++) {
1308 for (int ic = 0; ic < nx * nxcell; ic++)
1309 fprintf(fp, "%d ", rowp[ic]);
1323 ImageFile::writeImagePNG (const char* const outfile, int bitdepth, int nxcell, int nycell, double densmin, double densmax)
1325 double max_out_level = (1 << bitdepth) - 1;
1328 ImageFileArray v = getArray();
1330 unsigned char* rowp = new unsigned char [nx * nxcell * (bitdepth / 8)];
1332 FILE *fp = fopen (outfile, "wb");
1336 png_structp png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL);
1340 png_infop info_ptr = png_create_info_struct (png_ptr);
1342 png_destroy_write_struct (&png_ptr, (png_infopp) NULL);
1347 if (setjmp (png_ptr->jmpbuf)) {
1348 png_destroy_write_struct (&png_ptr, &info_ptr);
1353 png_init_io(png_ptr, fp);
1355 png_set_IHDR (png_ptr, info_ptr, nx * nxcell, ny * nycell, bitdepth, PNG_COLOR_TYPE_GRAY, PNG_INTERLACE_NONE, PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_DEFAULT);
1357 png_write_info(png_ptr, info_ptr);
1358 for (int irow = ny - 1; irow >= 0; irow--) {
1359 png_bytep row_pointer = rowp;
1361 for (int icol = 0; icol < nx; icol++) {
1362 int pos = icol * nxcell;
1363 double dens = (v[icol][irow] - densmin) / (densmax - densmin);
1364 dens = clamp (dens, 0., 1.);
1365 unsigned int outval = static_cast<unsigned int> (dens * max_out_level);
1367 for (int p = pos; p < pos + nxcell; p++) {
1372 rowp[rowpos+1] = (outval >> 8) & 0xFF;
1373 rowp[rowpos] = (outval & 0xFF);
1377 for (int ir = 0; ir < nycell; ir++)
1378 png_write_rows (png_ptr, &row_pointer, 1);
1381 png_write_end (png_ptr, info_ptr);
1382 png_destroy_write_struct (&png_ptr, &info_ptr);
1393 static const int N_GRAYSCALE=256;
1396 ImageFile::writeImageGIF (const char* const outfile, int nxcell, int nycell, double densmin, double densmax)
1398 int gs_indices[N_GRAYSCALE];
1401 ImageFileArray v = getArray();
1403 unsigned char* rowp = new unsigned char [nx * nxcell];
1405 gdImagePtr gif = gdImageCreate(nx * nxcell, ny * nycell);
1406 for (int i = 0; i < N_GRAYSCALE; i++)
1407 gs_indices[i] = gdImageColorAllocate(gif, i, i, i);
1409 int lastrow = ny * nycell - 1;
1410 for (int irow = 0; irow < ny; irow++) {
1411 int rpos = irow * nycell;
1412 for (int ir = rpos; ir < rpos + nycell; ir++) {
1413 for (int icol = 0; icol < nx; icol++) {
1414 int cpos = icol * nxcell;
1415 double dens = (v[icol][irow] - densmin) / (densmax - densmin);
1416 dens = clamp(dens, 0., 1.);
1417 for (int ic = cpos; ic < cpos + nxcell; ic++) {
1418 rowp[ic] = (unsigned int) (dens * (double) (N_GRAYSCALE - 1));
1419 gdImageSetPixel(gif, ic, lastrow - ir, gs_indices[rowp[ic]]);
1426 if ((out = fopen (outfile,"w")) == NULL) {
1427 sys_error(ERR_SEVERE, "Error opening output file %s for writing", outfile);
1430 gdImageGif(gif,out);
1432 gdImageDestroy(gif);