CImg_math.h

/**********************************************************************
 * Copyright (C) 2012, The YaDICs Project Developers. 
 * See the COPYRIGHT file at the top-level directory of this distribution ./COPYRIGHT.
 * See ./COPYING file for copying and redistribution conditions.
 * 
 * This file is part of YaDICs.
 * YaDICs is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * YaDICs is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with YaDICs.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * Information about how to use the software are provided at http://yadic.univ-lille1.fr/
 **********************************************************************/


#ifndef CIMG_MATH
#define CIMG_MATH

#include "../CImg/CImg.h"
 using namespace cimg_library;

//---------------------------------------------------------------
template<typename T>
void matrixDisplay(const CImg<T> &a);
template<typename T>
CImg<T> matrixProduct(const CImg<T> &a,const CImg<T> &b);
//---------------------------------------------------------------


template<typename T>
void matrixDisplay(const CImg<T> &a) {
  
  int I(a.height()), J(a.width()), K(a.depth()), L(a.spectrum());
  
  cimg_forXY(a,x,y) {
    if((J-1)==0){
      if(x==0){
        std::cout<<"[";
      }
      std::cout<<a(x,y)<<"]\n";
    }
    else if(x%(J-1)==0&&x!=0){
      std::cout<<a(x,y)<<"]\n";
    }
    else{
      if(x==0){
        std::cout<<"[";
      }
      std::cout<<a(x,y)<<"\t";
    };
  };

}




template<typename T>
CImg<T> matrixProduct(const CImg<T> &a,const CImg<T> &b){
  
  int Ia(a.height()), Ja(a.width()), Ka(a.depth()), La(a.spectrum());
  int Ib(b.height()), Jb(b.width()), Kb(b.depth()), Lb(b.spectrum());
  
  CImg<double> c(Jb, Ia, Ka, La, 0);
  
  if (Ja==Ib){
    cimg_forXY(c,x,y){
        for (int k=0; k<Ja; k++) {
          c(x,y)+=(double)a(k,y)*(double)b(x,k);
        };
      };
      return c;
    }
    else {
      throw CImgInstanceException("Error dimension: enter such a product A(i,j)xB(j,k)\n"); 
    };
}



#endif