X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=libctsim%2Fimagefile.cpp;h=0cbd67826ae194750b64cb8df9a70f3dbbdaeeac;hp=7f1c6a03085a2f9e51a50f32872becbff5296f2b;hb=9b2bb510160bdb56f04847f5b55ab61dd8a47976;hpb=08f34bf3ba14d4f436f4d2ef0ee5af1d6eb266ac diff --git a/libctsim/imagefile.cpp b/libctsim/imagefile.cpp index 7f1c6a0..0cbd678 100644 --- a/libctsim/imagefile.cpp +++ b/libctsim/imagefile.cpp @@ -9,7 +9,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: imagefile.cpp,v 1.4 2000/06/25 17:32:24 kevin Exp $ +** $Id: imagefile.cpp,v 1.29 2001/01/02 05:34:57 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 @@ -27,132 +27,149 @@ #include "ct.h" +const int ImageFile::FORMAT_INVALID = -1; +const int ImageFile::FORMAT_PGM = 0; +const int ImageFile::FORMAT_PGMASCII = 1; +#ifdef HAVE_PNG +const int ImageFile::FORMAT_PNG = 2; +const int ImageFile::FORMAT_PNG16 = 3; +#endif + +const char* ImageFile::s_aszFormatName[] = +{ + {"pgm"}, + {"pgmascii"}, +#ifdef HAVE_PNG + {"png"}, + {"png16"}, +#endif +}; + +const char* ImageFile::s_aszFormatTitle[] = +{ + {"PGM"}, + {"PGM ASCII"}, + {"PNG"}, + {"PNG 16-bit"}, +}; + +const int ImageFile::s_iFormatCount = sizeof(s_aszFormatName) / sizeof(const char*); + + -/////////////////////////////////////////////////////////////////////////// -// CLASS IMPLEMENTATION -// -// Name: Array2dFileLabel -// Purpose: Labels for Array2dFiles -/////////////////////////////////////////////////////////////////////////// - -void -Array2dFileLabel::init (void) -{ - m_calcTime = 0; - m_labelType = L_EMPTY; - TIMEDATE td; - td_get_tmdt (&td); - m_year = td.d.year; - m_month = td.d.month; - m_day = td.d.date; - m_hour = td.t.hour; - m_minute = td.t.minute; - m_second = td.t.second; -} - -Array2dFileLabel::Array2dFileLabel() -{ - init(); -} - -Array2dFileLabel::Array2dFileLabel(const char* const str, double ctime = 0.) - : m_strLabel (str) -{ - init(); - - m_labelType = L_USER; - m_calcTime = ctime; -} - -Array2dFileLabel::Array2dFileLabel(const int type, const char* const str, double ctime = 0.) - : m_strLabel (str) -{ - init(); - - m_labelType = type; - m_calcTime = ctime; -} - -Array2dFileLabel::~Array2dFileLabel() -{ -} - -const string& -Array2dFileLabel::getDateString (void) const -{ - ostringstream oss; - oss << m_month <<"/"<< m_day <<"/"<< m_year << " " << m_hour <<":"<< m_minute <<":"<< m_second; - m_strDate = oss.str(); - return m_strDate; -} - - -/* FILE - * image.c Routines for managing images - */ - +F32Image::F32Image (int nx, int ny, int dataType) +: Array2dFile (nx, ny, sizeof(kfloat32), Array2dFile::PIXEL_FLOAT32, dataType) +{ +} + +F32Image::F32Image (void) +: Array2dFile() +{ + setPixelFormat (Array2dFile::PIXEL_FLOAT32); + setPixelSize (sizeof(kfloat32)); + setDataType (Array2dFile::DATA_TYPE_REAL); +} + +F64Image::F64Image (int nx, int ny, int dataType) +: Array2dFile (nx, ny, sizeof(kfloat64), Array2dFile::PIXEL_FLOAT64, dataType) +{ +} + +F64Image::F64Image (void) +: Array2dFile () +{ + setPixelFormat (PIXEL_FLOAT64); + setPixelSize (sizeof(kfloat64)); + setDataType (Array2dFile::DATA_TYPE_REAL); +} void -ImageFile::filterResponse (const char* const domainName, double bw, const char* const filterName, double filt_param) -{ - int hx = (mNX - 1) / 2; - int hy = (mNY - 1) / 2; - ImageFileArray v = getArray(); - SignalFilter filter (filterName, domainName, bw, filt_param); - +ImageFile::filterResponse (const char* const domainName, double bw, const char* const filterName, double filt_param, double dInputScale, double dOutputScale) +{ + ImageFileArray v = getArray(); + SignalFilter filter (filterName, domainName, bw, filt_param); + +#if 1 + int iXCenter, iYCenter; + if (isEven (m_nx)) + iXCenter = m_nx / 2; + else + iXCenter = (m_nx - 1) / 2; + if (isEven (m_ny)) + iYCenter = m_ny / 2; + else + iYCenter = (m_ny - 1) / 2; + + for (int ix = 0; ix < m_nx; ix++) + for (int iy = 0; iy < m_ny; iy++) { + long lD2 = ((ix - iXCenter) * (ix - iXCenter)) + ((iy - iYCenter) * (iy - iYCenter)); + double r = ::sqrt (static_cast(lD2)) * dInputScale; + v[ix][iy] = filter.response (r) * dOutputScale; + } +#else + int hx = (m_nx - 1) / 2; + int hy = (m_ny - 1) / 2; + for (int i = -hx; i <= hx; i++) { for (int j = -hy; j <= hy; j++) { - double r = sqrt (i * i + j * j); + double r = ::sqrt (i * i + j * j); v[i+hx][j+hy] = filter.response (r); } - } + } +#endif } int -ImageFile::display (void) +ImageFile::display (void) const { - ImageFileValue pmin, pmax; - - getPixelValueRange (pmin, pmax); - - return (displayScaling (1, pmin, pmax)); + double pmin, pmax; + + getMinMax (pmin, pmax); + + return (displayScaling (1, pmin, pmax)); } int -ImageFile::displayScaling (const int scale, const ImageFileValue pmin, const ImageFileValue pmax) +ImageFile::displayScaling (const int scale, const ImageFileValue pmin, const ImageFileValue pmax) const { - int grayscale[256]; - int nx = mNX; - int ny = mNY; - ImageFileArray v = getArray(); - + int nx = m_nx; + int ny = m_ny; + ImageFileArrayConst v = getArray(); + if (v == NULL || nx == 0 || ny == 0) + return 0; + #if HAVE_G2_H - int pens [nx * ny * scale * scale ]; - - double view_scale = 255 / (pmax - pmin); - int id_X11 = g2_open_X11 (nx * scale, ny * scale); - - for (int i = 0; i < 256; i++) { - double cval = i / 255.; - grayscale[i] = g2_ink (id_X11, cval, cval, cval); - } - - for (int i= 0, iy = ny - 1; iy >= 0; iy--) { - for (int ix = 0; ix < nx; ix++) { - int cval = static_cast((v[ix][iy] - pmin) * view_scale); - if (cval < 0) - cval = 0; - else if (cval > 255) - cval = 255; - pens[i++] = grayscale[cval]; - } + int* pPens = new int [nx * ny * scale * scale ]; + + double view_scale = 255 / (pmax - pmin); + int id_X11 = g2_open_X11 (nx * scale, ny * scale); + int grayscale[256]; + for (int i = 0; i < 256; i++) { + double cval = i / 255.; + grayscale[i] = g2_ink (id_X11, cval, cval, cval); + } + + for (int iy = ny - 1; iy >= 0; iy--) { + int iRowPos = ((ny - 1 - iy) * scale) * (nx * scale); + for (int ix = 0; ix < nx; ix++) { + int cval = static_cast((v[ix][iy] - pmin) * view_scale); + if (cval < 0) + cval = 0; + else if (cval > 255) + cval = 255; + for (int sy = 0; sy < scale; sy++) + for (int sx = 0; sx < scale; sx++) + pPens[iRowPos+(sy * nx * scale)+(sx + (ix * scale))] = grayscale[cval]; } - - g2_image (id_X11, 0., 0., nx, ny, pens); - - return (id_X11); - + } + + g2_image (id_X11, 0., 0., nx * scale, ny * scale, pPens); + + delete pPens; + return (id_X11); +#else + return 0; #endif } @@ -171,135 +188,1125 @@ ImageFile::displayScaling (const int scale, const ImageFileValue pmin, const Ima bool ImageFile::comparativeStatistics (const ImageFile& imComp, double& d, double& r, double& e) const { - if (imComp.nx() != mNX && imComp.ny() != mNY) { - sys_error (ERR_WARNING, "Image sizes differ [ImageFile::comparativeStatistics]"); - return false; - } - ImageFileArray v = getArray(); - ImageFileArray vComp = imComp.getArray(); - - double myMean = 0.; - for (int ix = 0; ix < mNX; ix++) { - for (int iy = 0; iy < mNY; iy++) { - myMean += v[ix][iy]; - } + if (imComp.nx() != m_nx && imComp.ny() != m_ny) { + sys_error (ERR_WARNING, "Image sizes differ [ImageFile::comparativeStatistics]"); + return false; + } + ImageFileArrayConst v = getArray(); + if (v == NULL || m_nx == 0 || m_ny == 0) + return false; + + ImageFileArrayConst vComp = imComp.getArray(); + + double myMean = 0.; + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + myMean += v[ix][iy]; } - myMean /= (mNX * mNY); - - double sqErrorSum = 0.; - double absErrorSum = 0.; - double sqDiffFromMeanSum = 0.; - double absValueSum = 0.; - for (int ix = 0; ix < mNX; ix++) { - for (int iy = 0; iy < mNY; iy++) { - double diff = v[ix][iy] - vComp[ix][iy]; - sqErrorSum += diff * diff; - absErrorSum += fabs(diff); - double diffFromMean = v[ix][iy] - myMean; - sqDiffFromMeanSum += diffFromMean * diffFromMean; - absValueSum += fabs(v[ix][iy]); - } + } + myMean /= (m_nx * m_ny); + + double sqErrorSum = 0.; + double absErrorSum = 0.; + double sqDiffFromMeanSum = 0.; + double absValueSum = 0.; + for (unsigned int ix2 = 0; ix2 < m_nx; ix2++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + double diff = v[ix2][iy] - vComp[ix2][iy]; + sqErrorSum += diff * diff; + absErrorSum += fabs(diff); + double diffFromMean = v[ix2][iy] - myMean; + sqDiffFromMeanSum += diffFromMean * diffFromMean; + absValueSum += fabs(v[ix2][iy]); } - - d = sqrt (sqErrorSum / sqDiffFromMeanSum); - r = absErrorSum / absValueSum; - - int hx = mNX / 2; - int hy = mNY / 2; - double eMax = -1; - for (int ix = 0; ix < hx; ix++) { - for (int iy = 0; iy < hy; iy++) { - double avgPixel = 0.25 * (v[2*ix][2*iy] + v[2*ix+1][2*iy] + v[2*ix][2*iy+1] + v[2*ix+1][2*iy+1]); - double avgPixelComp = 0.25 * (vComp[2*ix][2*iy] + vComp[2*ix+1][2*iy] + vComp[2*ix][2*iy+1] + vComp[2*ix+1][2*iy+1]); - double error = fabs (avgPixel - avgPixelComp); - if (error > eMax) - eMax = error; - } + } + + d = ::sqrt (sqErrorSum / sqDiffFromMeanSum); + r = absErrorSum / absValueSum; + + int hx = m_nx / 2; + int hy = m_ny / 2; + double eMax = -1; + for (int ix3 = 0; ix3 < hx; ix3++) { + for (int iy = 0; iy < hy; iy++) { + 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]); + 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]); + double error = fabs (avgPixel - avgPixelComp); + if (error > eMax) + eMax = error; } - - e = eMax; - - return true; + } + + e = eMax; + + return true; } bool -ImageFile::printComparativeStatistics (const ImageFile& imComp, ostream& os) const +ImageFile::printComparativeStatistics (const ImageFile& imComp, std::ostream& os) const { double d, r, e; - + if (comparativeStatistics (imComp, d, r, e)) { - os << " Normalized root mean squared distance (d): " << d << endl; - os << " Normalized mean absolute distance (r): " << r << endl; - os << "Worst case distance (2x2 pixel average) (e): " << e << endl; + os << " Normalized root mean squared distance (d): " << d << std::endl; + os << " Normalized mean absolute distance (r): " << r << std::endl; + os << "Worst case distance (2x2 pixel average) (e): " << e << std::endl; + return true; } + return false; } void -ImageFile::printStatistics (ostream& os) const +ImageFile::printStatistics (std::ostream& os) const { - double min, max, mean, mode, median, stddev; + double min, max, mean, mode, median, stddev; + + statistics (min, max, mean, mode, median, stddev); + if (isComplex()) + os << "Real Component Statistics" << std::endl; + + os << " min: " << min << std::endl; + os << " max: " << max << std::endl; + os << " mean: " << mean << std::endl; + os << " mode: " << mode << std::endl; + os << "median: " << median << std::endl; + os << "stddev: " << stddev << std::endl; + + if (isComplex()) { + statistics (getImaginaryArray(), min, max, mean, mode, median, stddev); + os << std::endl << "Imaginary Component Statistics" << std::endl; + os << " min: " << min << std::endl; + os << " max: " << max << std::endl; + os << " mean: " << mean << std::endl; + os << " mode: " << mode << std::endl; + os << "median: " << median << std::endl; + os << "stddev: " << stddev << std::endl; + } +} - statistics (min, max, mean, mode, median, stddev); - os << " min: " << min << endl; - os << " max: " << max << endl; - os << " mean: " << mean << endl; - os << " mode: " << mode << endl; - os << "median: " << median << endl; - os << "stddev: " << stddev << endl; +void +ImageFile::statistics (double& min, double& max, double& mean, double& mode, double& median, double& stddev) const +{ + ImageFileArrayConst v = getArray(); + statistics (v, min, max, mean, mode, median, stddev); } +void +ImageFile::statistics (ImageFileArrayConst v, double& min, double& max, double& mean, double& mode, double& median, double& stddev) const +{ + int nx = m_nx; + int ny = m_ny; + + if (v == NULL || nx == 0 || ny == 0) + return; + + std::vector vecImage; + int iVec = 0; + vecImage.resize (nx * ny); + for (int ix = 0; ix < nx; ix++) { + for (int iy = 0; iy < ny; iy++) + vecImage[iVec++] = v[ix][iy]; + } + + vectorNumericStatistics (vecImage, nx * ny, min, max, mean, mode, median, stddev); +} + void -ImageFile::statistics (double& min, double& max, double& mean, double& mode, double& median, double& stddev) const +ImageFile::getMinMax (double& min, double& max) const { - int nx = mNX; - int ny = mNY; - ImageFileArray v = getArray(); - - mean = 0; - min = v[0][0]; - max = v[0][0]; - for (int ix = 0; ix < nx; ix++) { - for (int iy = 0; iy < ny; iy++) { - if (v[ix][iy] > max) - max = v[ix][iy]; - if (v[ix][iy] < min) - min = v[ix][iy]; - mean += v[ix][iy]; - } + int nx = m_nx; + int ny = m_ny; + ImageFileArrayConst v = getArray(); + + if (v == NULL || nx == 0 || ny == 0) + return; + + min = v[0][0]; + max = v[0][0]; + for (int ix = 0; ix < nx; ix++) { + for (int iy = 0; iy < ny; iy++) { + if (v[ix][iy] > max) + max = v[ix][iy]; + if (v[ix][iy] < min) + min = v[ix][iy]; } - mean /= (nx * ny); + } +} + +bool +ImageFile::convertRealToComplex () +{ + if (dataType() != Array2dFile::DATA_TYPE_REAL) + return false; + + if (! reallocRealToComplex()) + return false; + + ImageFileArray vImag = getImaginaryArray(); + for (unsigned int ix = 0; ix < m_nx; ix++) { + ImageFileColumn vCol = vImag[ix]; + for (unsigned int iy = 0; iy < m_ny; iy++) + *vCol++ = 0; + } + + return true; +} + +bool +ImageFile::convertComplexToReal () +{ + if (dataType() != Array2dFile::DATA_TYPE_COMPLEX) + return false; + + ImageFileArray vReal = getArray(); + ImageFileArray vImag = getImaginaryArray(); + for (unsigned int ix = 0; ix < m_nx; ix++) { + ImageFileColumn vRealCol = vReal[ix]; + ImageFileColumn vImagCol = vImag[ix]; + for (unsigned int iy = 0; iy < m_ny; iy++) { + CTSimComplex c (*vRealCol, *vImagCol); + *vRealCol++ = std::abs (c); + vImagCol++; + } + } + + return reallocComplexToReal(); +} + +bool +ImageFile::subtractImages (const ImageFile& rRHS, ImageFile& result) const +{ + if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]"); + return false; + } + + if (isComplex() || rRHS.isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArrayConst vRHS = rRHS.getArray(); + ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + vResult[ix][iy] = vLHS[ix][iy] - vRHS[ix][iy]; + if (result.isComplex()) { + vResultImag[ix][iy] = 0; + if (isComplex()) + vResultImag[ix][iy] += vLHSImag[ix][iy]; + if (rRHS.isComplex()) + vResultImag[ix][iy] -= vRHSImag[ix][iy]; + } + } + } + + return true; +} - static const int nbin = 256; - int hist[ nbin ] = {0}; - double spread = max - min; - mode = 0; - stddev = 0; - for (int ix = 0; ix < nx; ix++) { - for (int iy = 0; iy < ny; iy++) { - int b = static_cast((((v[ix][iy] - min) / spread) * (nbin - 1)) + 0.5); - hist[b]++; - double diff = (v[ix][iy] - mean); - stddev += diff * diff; - } +bool +ImageFile::addImages (const ImageFile& rRHS, ImageFile& result) const +{ + if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]"); + return false; + } + + if (isComplex() || rRHS.isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArrayConst vRHS = rRHS.getArray(); + ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + vResult[ix][iy] = vLHS[ix][iy] + vRHS[ix][iy]; + if (result.isComplex()) { + vResultImag[ix][iy] = 0; + if (isComplex()) + vResultImag[ix][iy] += vLHSImag[ix][iy]; + if (rRHS.isComplex()) + vResultImag[ix][iy] += vRHSImag[ix][iy]; + } + } + } + + return true; +} + +bool +ImageFile::multiplyImages (const ImageFile& rRHS, ImageFile& result) const +{ + if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]"); + return false; + } + + if (isComplex() || rRHS.isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArrayConst vRHS = rRHS.getArray(); + ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cRHS (vRHS[ix][iy], 0); + if (rRHS.isComplex()) + cRHS.imag (vRHSImag[ix][iy]); + std::complex cResult = cLHS * cRHS; + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else + vResult[ix][iy] = vLHS[ix][iy] * vRHS[ix][iy]; + } + } + + + return true; +} + +bool +ImageFile::divideImages (const ImageFile& rRHS, ImageFile& result) const +{ + if (m_nx != rRHS.nx() || m_ny != rRHS.ny() || m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::subtractImage]"); + return false; + } + + if (isComplex() || rRHS.isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArrayConst vRHS = rRHS.getArray(); + ImageFileArrayConst vRHSImag = rRHS.getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cRHS (vRHS[ix][iy], 0); + if (rRHS.isComplex()) + cRHS.imag (vRHSImag[ix][iy]); + std::complex cResult = cLHS / cRHS; + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else { + if (vRHS != 0) + vResult[ix][iy] = vLHS[ix][iy] / vRHS[ix][iy]; + else + vResult[ix][iy] = 0; + } + } + } + + + return true; +} + + +bool +ImageFile::invertPixelValues (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArray vResult = result.getArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + ImageFileColumnConst in = vLHS[ix]; + ImageFileColumn out = vResult[ix]; + for (unsigned int iy = 0; iy < m_ny; iy++) + *out++ = - *in++; + } + + return true; +} + +bool +ImageFile::sqrt (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + bool bComplexOutput = result.isComplex(); + ImageFileArrayConst vLHS = getArray(); + if (! bComplexOutput) // check if should convert to complex output + for (unsigned int ix = 0; ix < m_nx; ix++) + for (unsigned int iy = 0; iy < m_ny; iy++) + if (! bComplexOutput && vLHS[ix][iy] < 0) { + result.convertRealToComplex(); + bComplexOutput = true; + break; + } + + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cResult = std::sqrt(cLHS); + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else + vResult[ix][iy] = ::sqrt (vLHS[ix][iy]); + } + } + + + return true; +} + +bool +ImageFile::log (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cResult = std::log (cLHS); + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else + vResult[ix][iy] = ::log (vLHS[ix][iy]); + } + } + + + return true; +} + +bool +ImageFile::exp (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cResult = std::exp (cLHS); + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else + vResult[ix][iy] = ::exp (vLHS[ix][iy]); + } + } + + + return true; +} + +bool +ImageFile::scaleImage (ImageFile& result) const +{ + unsigned int nx = m_nx; + unsigned int ny = m_ny; + unsigned int newNX = result.nx(); + unsigned int newNY = result.ny(); + + double dXScale = static_cast(newNX) / static_cast(nx); + double dYScale = static_cast(newNY) / static_cast(ny); + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vReal = getArray(); + ImageFileArrayConst vImag = getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < newNX; ix++) { + for (unsigned int iy = 0; iy < newNY; iy++) { + unsigned int scaleNX = static_cast (ix / dXScale); + unsigned int scaleNY = static_cast (iy / dYScale); + vResult[ix][iy] = vReal[scaleNX][scaleNY]; + if (result.isComplex()) { + if (isComplex()) + vResultImag[ix][iy] = vImag[scaleNX][scaleNY]; + else + vResultImag[ix][iy] = 0; + } + } + } + + return true; +} + +#ifdef HAVE_FFTW +bool +ImageFile::fft (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (result.dataType() == Array2dFile::DATA_TYPE_REAL) { + if (! result.convertRealToComplex ()) + return false; + } + + fftw_complex* in = new fftw_complex [m_nx * m_ny]; + + ImageFileArrayConst vReal = getArray(); + ImageFileArrayConst vImag = getImaginaryArray(); + + unsigned int ix, iy; + unsigned int iArray = 0; + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + in[iArray].re = vReal[ix][iy]; + if (isComplex()) + in[iArray].im = vImag[ix][iy]; + else + in[iArray].im = 0; + iArray++; + } + + fftwnd_plan plan = fftw2d_create_plan (m_nx, m_ny, FFTW_FORWARD, FFTW_IN_PLACE); + + fftwnd_one (plan, in, NULL); + + ImageFileArray vRealResult = result.getArray(); + ImageFileArray vImagResult = result.getImaginaryArray(); + iArray = 0; + unsigned int iScale = m_nx * m_ny; + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = in[iArray].re / iScale; + vImagResult[ix][iy] = in[iArray].im / iScale; + iArray++; + } + + fftwnd_destroy_plan (plan); + delete in; + + + Fourier::shuffleFourierToNaturalOrder (result); + + return true; +} + + +bool +ImageFile::ifft (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (result.dataType() == Array2dFile::DATA_TYPE_REAL) { + if (! result.convertRealToComplex ()) + return false; + } + + ImageFileArrayConst vReal = getArray(); + ImageFileArrayConst vImag = getImaginaryArray(); + ImageFileArray vRealResult = result.getArray(); + ImageFileArray vImagResult = result.getImaginaryArray(); + unsigned int ix, iy; + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = vReal[ix][iy]; + if (isComplex()) + vImagResult[ix][iy] = vImag[ix][iy]; + else + vImagResult[ix][iy] = 0; + } + + Fourier::shuffleNaturalToFourierOrder (result); + + fftw_complex* in = new fftw_complex [m_nx * m_ny]; + + unsigned int iArray = 0; + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + in[iArray].re = vRealResult[ix][iy]; + in[iArray].im = vImagResult[ix][iy]; + iArray++; + } + + fftwnd_plan plan = fftw2d_create_plan (m_nx, m_ny, FFTW_BACKWARD, FFTW_IN_PLACE); + + fftwnd_one (plan, in, NULL); + + iArray = 0; + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = in[iArray].re; + vImagResult[ix][iy] = in[iArray].im; + iArray++; + } + + fftwnd_destroy_plan (plan); + + delete in; + + return true; +} +#endif // HAVE_FFTW + + + +bool +ImageFile::fourier (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (! result.isComplex()) + if (! result.convertRealToComplex ()) + return false; + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vRealResult = result.getArray(); + ImageFileArray vImagResult = result.getImaginaryArray(); + + unsigned int ix, iy; + + // alloc output matrix + CTSimComplex** complexOut = new CTSimComplex* [m_nx]; + for (ix = 0; ix < m_nx; ix++) + complexOut[ix] = new CTSimComplex [m_ny]; + + // fourier each x column +#if 1 + CTSimComplex* pY = new CTSimComplex [m_ny]; + for (ix = 0; ix < m_nx; ix++) { + for (iy = 0; iy < m_ny; iy++) { + pY[iy].real (vLHS[ix][iy]); + if (isComplex()) + pY[iy].imag (vLHSImag[ix][iy]); + else + pY[iy].imag (0); + } + ProcessSignal::finiteFourierTransform (pY, complexOut[ix], m_ny, ProcessSignal::FORWARD); + } +#else + double* pY = new double [m_ny]; + for (ix = 0; ix < m_nx; ix++) { + for (iy = 0; iy < m_ny; iy++) { + pY[iy] = vLHS[ix][iy]; + } + ProcessSignal::finiteFourierTransform (pY, complexOut[ix], m_ny, ProcessSignal::FORWARD); + } +#endif + delete [] pY; + + // fourier each y row + CTSimComplex* pX = new CTSimComplex [m_nx]; + CTSimComplex* complexOutRow = new CTSimComplex [m_nx]; + for (iy = 0; iy < m_ny; iy++) { + for (ix = 0; ix < m_nx; ix++) + pX[ix] = complexOut[ix][iy]; + ProcessSignal::finiteFourierTransform (pX, complexOutRow, m_nx, ProcessSignal::FORWARD); + for (ix = 0; ix < m_nx; ix++) + complexOut[ix][iy] = complexOutRow[ix]; + } + delete [] pX; + delete [] complexOutRow; + + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = complexOut[ix][iy].real(); + vImagResult[ix][iy] = complexOut[ix][iy].imag(); + } + + Fourier::shuffleFourierToNaturalOrder (result); + + // delete complexOut matrix + for (ix = 0; ix < m_nx; ix++) + delete [] complexOut[ix]; + delete [] complexOut; + + return true; +} + +bool +ImageFile::inverseFourier (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (result.dataType() == Array2dFile::DATA_TYPE_REAL) { + if (! result.convertRealToComplex ()) + return false; + } + + ImageFileArrayConst vLHSReal = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vRealResult = result.getArray(); + ImageFileArray vImagResult = result.getImaginaryArray(); + + unsigned int ix, iy; + // alloc 2d complex output matrix + CTSimComplex** complexOut = new CTSimComplex* [m_nx]; + for (ix = 0; ix < m_nx; ix++) + complexOut[ix] = new CTSimComplex [m_ny]; + + // put input image into result + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = vLHSReal[ix][iy]; + if (isComplex()) + vImagResult[ix][iy] = vLHSImag[ix][iy]; + else + vImagResult[ix][iy] = 0; + } + + Fourier::shuffleNaturalToFourierOrder (result); + + // ifourier each x column + CTSimComplex* pCol = new CTSimComplex [m_ny]; + for (ix = 0; ix < m_nx; ix++) { + for (iy = 0; iy < m_ny; iy++) { + pCol[iy].real (vRealResult[ix][iy]); + pCol[iy].imag (vImagResult[ix][iy]); + } + ProcessSignal::finiteFourierTransform (pCol, complexOut[ix], m_ny, ProcessSignal::BACKWARD); + } + delete [] pCol; + + // ifourier each y row + CTSimComplex* complexInRow = new CTSimComplex [m_nx]; + CTSimComplex* complexOutRow = new CTSimComplex [m_nx]; + for (iy = 0; iy < m_ny; iy++) { + for (ix = 0; ix < m_nx; ix++) + complexInRow[ix] = complexOut[ix][iy]; + ProcessSignal::finiteFourierTransform (complexInRow, complexOutRow, m_nx, ProcessSignal::BACKWARD); + for (ix = 0; ix < m_nx; ix++) + complexOut[ix][iy] = complexOutRow[ix]; + } + delete [] complexInRow; + delete [] complexOutRow; + + for (ix = 0; ix < m_nx; ix++) + for (iy = 0; iy < m_ny; iy++) { + vRealResult[ix][iy] = complexOut[ix][iy].real(); + vImagResult[ix][iy] = complexOut[ix][iy].imag(); + } + + // delete complexOut matrix + for (ix = 0; ix < m_nx; ix++) + delete [] complexOut[ix]; + delete [] complexOut; + + return true; +} + + +bool +ImageFile::magnitude (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + ImageFileArray vReal = getArray(); + ImageFileArray vImag = getImaginaryArray(); + ImageFileArray vRealResult = result.getArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (isComplex()) + vRealResult[ix][iy] = ::sqrt (vReal[ix][iy] * vReal[ix][iy] + vImag[ix][iy] * vImag[ix][iy]); + else + vRealResult[ix][iy] = vReal[ix][iy]; + } + + if (result.isComplex()) + result.convertComplexToReal(); + + return true; +} + +bool +ImageFile::phase (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + ImageFileArray vReal = getArray(); + ImageFileArray vImag = getImaginaryArray(); + ImageFileArray vRealResult = result.getArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (isComplex()) + vRealResult[ix][iy] = ::atan2 (vImag[ix][iy], vReal[ix][iy]); + else + vRealResult[ix][iy] = 0; + } + + if (result.isComplex()) + result.convertComplexToReal(); + + return true; +} + +bool +ImageFile::square (ImageFile& result) const +{ + if (m_nx != result.nx() || m_ny != result.ny()) { + sys_error (ERR_WARNING, "Difference sizes of images [ImageFile::invertPixelValues]"); + return false; + } + + if (isComplex() && ! result.isComplex()) + result.convertRealToComplex(); + + ImageFileArrayConst vLHS = getArray(); + ImageFileArrayConst vLHSImag = getImaginaryArray(); + ImageFileArray vResult = result.getArray(); + ImageFileArray vResultImag = result.getImaginaryArray(); + + for (unsigned int ix = 0; ix < m_nx; ix++) { + for (unsigned int iy = 0; iy < m_ny; iy++) { + if (result.isComplex()) { + std::complex cLHS (vLHS[ix][iy], 0); + if (isComplex()) + cLHS.imag (vLHSImag[ix][iy]); + std::complex cResult = cLHS * cLHS; + vResult[ix][iy] = cResult.real(); + vResultImag[ix][iy] = cResult.imag(); + } else + vResult[ix][iy] = vLHS[ix][iy] * vLHS[ix][iy]; + } + } + + + return true; +} + +int +ImageFile::convertFormatNameToID (const char* const formatName) +{ + int formatID = FORMAT_INVALID; + + for (int i = 0; i < s_iFormatCount; i++) + if (strcasecmp (formatName, s_aszFormatName[i]) == 0) { + formatID = i; + break; + } + + return (formatID); +} + +const char* +ImageFile::convertFormatIDToName (int formatID) +{ + static const char *formatName = ""; + + if (formatID >= 0 && formatID < s_iFormatCount) + return (s_aszFormatName[formatID]); + + return (formatName); +} + +const char* +ImageFile::convertFormatIDToTitle (const int formatID) +{ + static const char *formatTitle = ""; + + if (formatID >= 0 && formatID < s_iFormatCount) + return (s_aszFormatTitle[formatID]); + + return (formatTitle); +} + +bool +ImageFile::exportImage (const char* const pszFormat, const char* const pszFilename, int nxcell, int nycell, double densmin, double densmax) +{ + int iFormatID = convertFormatNameToID (pszFormat); + if (iFormatID == FORMAT_INVALID) { + sys_error (ERR_SEVERE, "Invalid format %s [ImageFile::exportImage]", pszFormat); + return false; + } + + if (iFormatID == FORMAT_PGM) + return writeImagePGM (pszFilename, nxcell, nycell, densmin, densmax); + else if (iFormatID == FORMAT_PGMASCII) + return writeImagePGMASCII (pszFilename, nxcell, nycell, densmin, densmax); + else if (iFormatID == FORMAT_PNG) + return writeImagePNG (pszFilename, 8, nxcell, nycell, densmin, densmax); + else if (iFormatID == FORMAT_PNG16) + return writeImagePNG (pszFilename, 16, nxcell, nycell, densmin, densmax); + + sys_error (ERR_SEVERE, "Invalid format %s [ImageFile::exportImage]", pszFormat); + return false; +} + +bool +ImageFile::writeImagePGM (const char* const outfile, int nxcell, int nycell, double densmin, double densmax) +{ + FILE *fp; + int nx = m_nx; + int ny = m_ny; + ImageFileArray v = getArray(); + + unsigned char* rowp = new unsigned char [nx * nxcell]; + + if ((fp = fopen (outfile, "wb")) == NULL) + return false; + + fprintf(fp, "P5\n"); + fprintf(fp, "%d %d\n", nx, ny); + fprintf(fp, "255\n"); + + for (int irow = ny - 1; irow >= 0; irow--) { + for (int icol = 0; icol < nx; icol++) { + int pos = icol * nxcell; + double dens = (v[icol][irow] - densmin) / (densmax - densmin); + dens = clamp (dens, 0., 1.); + for (int p = pos; p < pos + nxcell; p++) { + rowp[p] = static_cast (dens * 255.); + } + } + for (int ir = 0; ir < nycell; ir++) { + for (int ic = 0; ic < nx * nxcell; ic++) + fputc( rowp[ic], fp ); } - stddev = sqrt(stddev / (nx * ny)); + } + + delete rowp; + fclose(fp); + + return true; +} - int max_binindex = 0; - int max_bin = -1; - for (int ibin = 0; ibin < nbin; ibin++) { - if (hist[ibin] > max_bin) { - max_bin = hist[ibin]; - max_binindex = ibin; +bool +ImageFile::writeImagePGMASCII (const char* const outfile, int nxcell, int nycell, double densmin, double densmax) +{ + FILE *fp; + int nx = m_nx; + int ny = m_ny; + ImageFileArray v = getArray(); + + unsigned char* rowp = new unsigned char [nx * nxcell]; + + if ((fp = fopen (outfile, "wb")) == NULL) + return false; + + fprintf(fp, "P2\n"); + fprintf(fp, "%d %d\n", nx, ny); + fprintf(fp, "255\n"); + + for (int irow = ny - 1; irow >= 0; irow--) { + for (int icol = 0; icol < nx; icol++) { + int pos = icol * nxcell; + double dens = (v[icol][irow] - densmin) / (densmax - densmin); + dens = clamp (dens, 0., 1.); + for (int p = pos; p < pos + nxcell; p++) { + rowp[p] = static_cast (dens * 255.); } } + for (int ir = 0; ir < nycell; ir++) { + for (int ic = 0; ic < nx * nxcell; ic++) + fprintf(fp, "%d ", rowp[ic]); + fprintf(fp, "\n"); + } + } + + delete rowp; + fclose(fp); + + return true; +} - mode = (max_binindex * spread / (nbin - 1)) + min; + +#ifdef HAVE_PNG +bool +ImageFile::writeImagePNG (const char* const outfile, int bitdepth, int nxcell, int nycell, double densmin, double densmax) +{ + double max_out_level = (1 << bitdepth) - 1; + int nx = m_nx; + int ny = m_ny; + ImageFileArray v = getArray(); + + unsigned char* rowp = new unsigned char [nx * nxcell * (bitdepth / 8)]; + + FILE *fp = fopen (outfile, "wb"); + if (! fp) + return false; + + png_structp png_ptr = png_create_write_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL); + if (! png_ptr) + return false; + + png_infop info_ptr = png_create_info_struct (png_ptr); + if (! info_ptr) { + png_destroy_write_struct (&png_ptr, (png_infopp) NULL); + fclose (fp); + return false; + } + + if (setjmp (png_ptr->jmpbuf)) { + png_destroy_write_struct (&png_ptr, &info_ptr); + fclose (fp); + return false; + } + + png_init_io(png_ptr, fp); + + 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); + + png_write_info(png_ptr, info_ptr); + for (int irow = ny - 1; irow >= 0; irow--) { + png_bytep row_pointer = rowp; + + for (int icol = 0; icol < nx; icol++) { + int pos = icol * nxcell; + double dens = (v[icol][irow] - densmin) / (densmax - densmin); + dens = clamp (dens, 0., 1.); + unsigned int outval = static_cast (dens * max_out_level); - median = 0.; + for (int p = pos; p < pos + nxcell; p++) { + if (bitdepth == 8) + rowp[p] = outval; + else { + int rowpos = p * 2; + rowp[rowpos] = (outval >> 8) & 0xFF; + rowp[rowpos+1] = (outval & 0xFF); + } + } + } + for (int ir = 0; ir < nycell; ir++) + png_write_rows (png_ptr, &row_pointer, 1); + } + + png_write_end (png_ptr, info_ptr); + png_destroy_write_struct (&png_ptr, &info_ptr); + delete rowp; + + fclose(fp); + + return true; } +#endif + +#ifdef HAVE_GD +#include "gd.h" +static const int N_GRAYSCALE=256; +bool +ImageFile::writeImageGIF (const char* const outfile, int nxcell, int nycell, double densmin, double densmax) +{ + int gs_indices[N_GRAYSCALE]; + int nx = m_nx; + int ny = m_ny; + ImageFileArray v = getArray(); + + unsigned char* rowp = new unsigned char [nx * nxcell]; + + gdImagePtr gif = gdImageCreate(nx * nxcell, ny * nycell); + for (int i = 0; i < N_GRAYSCALE; i++) + gs_indices[i] = gdImageColorAllocate(gif, i, i, i); + + int lastrow = ny * nycell - 1; + for (int irow = 0; irow < ny; irow++) { + int rpos = irow * nycell; + for (int ir = rpos; ir < rpos + nycell; ir++) { + for (int icol = 0; icol < nx; icol++) { + int cpos = icol * nxcell; + double dens = (v[icol][irow] - densmin) / (densmax - densmin); + dens = clamp(dens, 0., 1.); + for (int ic = cpos; ic < cpos + nxcell; ic++) { + rowp[ic] = (unsigned int) (dens * (double) (N_GRAYSCALE - 1)); + gdImageSetPixel(gif, ic, lastrow - ir, gs_indices[rowp[ic]]); + } + } + } + } + + FILE *out; + if ((out = fopen (outfile,"w")) == NULL) { + sys_error(ERR_FATAL, "Error opening output file %s for writing", outfile); + return false; + } + gdImageGif(gif,out); + fclose(out); + gdImageDestroy(gif); + + delete rowp; + + return true; +} +#endif