Squashed 'third_party/cimg/' content from commit 4b66369
Change-Id: I7454d9107a08dba899fd4659731733049165ae0a
git-subtree-dir: third_party/cimg
git-subtree-split: 4b66369ab4e34a46119d6c43e9adce061bb40f4b
diff --git a/examples/wavelet_atrous.cpp b/examples/wavelet_atrous.cpp
new file mode 100644
index 0000000..5f5628d
--- /dev/null
+++ b/examples/wavelet_atrous.cpp
@@ -0,0 +1,191 @@
+/*
+ #
+ # File : wavelet_atrous.cpp
+ # ( C++ source file )
+ #
+ # Description : Performs a 2D or 3D 'a trous' wavelet transform
+ # (using a cubic spline) on an image or a video sequence.
+ # This file is a part of the CImg Library project.
+ # ( http://cimg.eu )
+ #
+ # Author : Renaud Peteri
+ # ( Renaud.Peteri(at)mines-paris.org )
+ # Andrea Onofri
+ # ( Andrea.Onofri(at)teletu.it )
+ #
+ # Institution : CWI, Amsterdam
+ #
+ # Date : February 2005
+ #
+ # References : Starck, J.-L., Murtagh, F. and Bijaoui, A.,
+ # Image Processing and Data Analysis: The Multiscale Approach,
+ # Cambridge University Press, 1998.
+ # (Hardback and softback, ISBN 0-521-59084-1 and 0-521-59914-8.)
+ #
+ # License : CeCILL v2.0
+ # ( http://www.cecill.info/licences/Licence_CeCILL_V2-en.html )
+ #
+ # This software is governed by the CeCILL license under French law and
+ # abiding by the rules of distribution of free software. You can use,
+ # modify and/ or redistribute the software under the terms of the CeCILL
+ # license as circulated by CEA, CNRS and INRIA at the following URL
+ # "http://www.cecill.info".
+ #
+ # As a counterpart to the access to the source code and rights to copy,
+ # modify and redistribute granted by the license, users are provided only
+ # with a limited warranty and the software's author, the holder of the
+ # economic rights, and the successive licensors have only limited
+ # liability.
+ #
+ # In this respect, the user's attention is drawn to the risks associated
+ # with loading, using, modifying and/or developing or reproducing the
+ # software by the user in light of its specific status of free software,
+ # that may mean that it is complicated to manipulate, and that also
+ # therefore means that it is reserved for developers and experienced
+ # professionals having in-depth computer knowledge. Users are therefore
+ # encouraged to load and test the software's suitability as regards their
+ # requirements in conditions enabling the security of their systems and/or
+ # data to be ensured and, more generally, to use and operate it in the
+ # same conditions as regards security.
+ #
+ # The fact that you are presently reading this means that you have had
+ # knowledge of the CeCILL license and that you accept its terms.
+ #
+*/
+
+#include "CImg.h"
+using namespace cimg_library;
+#ifndef cimg_imagepath
+#define cimg_imagepath "img/"
+#endif
+
+// Define convolution mask.
+CImg<float> mask(const unsigned char dirIdx, const unsigned char scale) {
+ const int
+ d1 = 1 << (scale-1),
+ d2 = 1 << scale,
+ c = d2,
+ vecLen = (1 << (scale + 1)) + 1;
+
+ const float
+ valC = 0.375f, // 6/16
+ valD1 = 0.25f, // 4/16
+ valD2 = 0.0625f; // 1/16
+
+ switch(dirIdx) {
+ case 0 : { // x
+ CImg<float> m(vecLen,1,1,1,0);
+ m(c) = valC;
+ m(c - d1) = m(c + d1) = valD1;
+ m(c - d2) = m(c + d2) = valD2;
+ return m;
+ }
+ case 1: { // y
+ CImg<float> m(1,vecLen,1,1,0);
+ m(0,c) = valC;
+ m(0,c - d1) = m(0,c + d1) = valD1;
+ m(0,c - d2) = m(0,c + d2) = valD2;
+ return m;
+ }
+ case 2: { // t
+ CImg<float> m(1,1,vecLen,1,0);
+ m(0,0,c) = valC;
+ m(0,0,c - d1) = m(0,0,c + d1) = valD1;
+ m(0,0,c - d2) = m(0,0,c + d2) = valD2;
+ return m;
+ }
+ default: throw CImgException("Error, unknow decompostion axe, dirIdx = '%c'.",dirIdx);
+ }
+}
+
+/*------------------
+ Main procedure
+ ----------------*/
+int main(int argc,char **argv) {
+ cimg_usage("Perform an 'a trous' wavelet transform (using a cubic spline) on an image or on a video sequence.\n"
+ "This wavelet transform is undecimated and produces 2 images/videos at each scale. For an example of\n"
+ "decomposition on a video, try -i img/trees.inr (sequence from the MIT).\n"
+ "\t(Type -h for help)");
+
+ // Read command line parameters
+ const char
+ *name_i = cimg_option("-i",cimg_imagepath "parrot.ppm","Input image or video"),
+ *name_o = cimg_option("-o","","Name of the multiscale analysis output"),
+ *axe_dec = cimg_option("-axe",(char*)0,
+ "Perform the multiscale decomposition in just one direction ('x', 'y' or 't')");
+ const unsigned int
+ s = cimg_option("-s",3,"Scale of decomposition");
+
+ const bool help = cimg_option("-h",false,"Display Help");
+ if (help) std::exit(0);
+
+ // Initialize Image Data
+ std::fprintf(stderr," - Load image sequence '%s'...\n",cimg::basename(name_i));
+ const CImg<float> texture_in(name_i);
+ CImg<float> mask_conv;
+ CImgList<float> res(s,texture_in.width(),texture_in.height(),texture_in.depth());
+ CImgList<float> wav(s,texture_in.width(),texture_in.height(),texture_in.depth());
+ cimglist_for(res,l) { res(l).fill(0.0); wav(l).fill(0.0); }
+ unsigned int i;
+
+ int firstDirIdx = 0,lastDirIdx = 2;
+ if (axe_dec) { // The multiscale decomposition will be performed in just one direction
+ char c = cimg::lowercase(axe_dec[0]);
+ switch(c) {
+ case 'x': firstDirIdx = 0; break;
+ case 'y': firstDirIdx = 1; break;
+ case 't': firstDirIdx = 2; break;
+ default: throw CImgException("Error, unknow decompostion axe '%c', try 'x', 'y' or 't'",c);
+ }
+ lastDirIdx = firstDirIdx; // Only one direction
+ }
+
+ for (i = 0; i<s; i++) {
+ std::fprintf(stderr," - Performing scale %u ...\n",i + 1);
+ if (i==0) { res(i) = texture_in;} else { res(i) = res(i - 1); }
+ for (int di = firstDirIdx; di<=lastDirIdx; di++) {
+ mask_conv = mask((unsigned char)di,(unsigned char)(i + 1));
+ res(i) = res(i).get_convolve(mask_conv);
+ }
+ if (i==0) { wav(i) = texture_in - res(i); } // res(0) and wav(0) are the 1st scale of decompostion
+ else { wav(i) = res(i - 1) - res(i); }
+ }
+
+ if (*name_o) {
+ // Save the Multi-Scale Analysis.
+ std::fprintf(stderr," - Saving of all output sequences : %s in the msa/ directory... \n",cimg::basename(name_o));
+ int count = 1; // res0 = original image
+ char filename[256] = "", filename_wav[256] = "";
+ char STmp[16] = "";
+ const int err = std::system("mkdir msa");
+ if (!err) for (i = 0; i<s; i++) {
+ std::strcpy( filename, "msa/res" );
+ std::strcpy( filename_wav, "msa/wav" );
+ if (count<10) { std::strcat( filename, "0" ); std::strcat( filename_wav, "0" ); }
+ std::sprintf(STmp,"%d_",count);
+ std::strcat(filename,STmp); std::strcat(filename_wav,STmp);
+ std::strcat(filename,name_o); std::strcat(filename_wav,name_o);
+ res(i).save(filename);
+ wav(i).save(filename_wav);
+ count++;
+ }
+ }
+
+ // Result visualization.
+ const float col[] = { 255, 255, 255 };
+ for (i = 0; i<s; i++) {
+ res[i].normalize(0,255).draw_text(2,2,"Scale %d",col,0,1,13,i);
+ wav[i].normalize(0,255).draw_text(2,2,"Scale %d",col,0,1,13,i);
+ }
+
+ CImgDisplay disp(res,"Approximations levels by increasing scale",0);
+ CImgDisplay disp2(wav,"Wavelet coefficients by increasing scale",0);
+ while (!disp.is_closed() && !disp.is_keyQ() && !disp.is_keyESC() &&
+ !disp2.is_closed() && !disp2.is_keyQ() && !disp2.is_keyESC()) {
+ if (disp.is_resized()) disp.resize().display(res);
+ if (disp2.is_resized()) disp2.resize().display(wav);
+ CImgDisplay::wait(disp,disp2);
+ }
+
+ return 0;
+}