/********************************************************************
**                Image Component Library (ICL)                    **
**                                                                 **
** Copyright (C) 2006-2013 CITEC, University of Bielefeld          **
**                         Neuroinformatics Group                  **
** Website: www.iclcv.org and                                      **
**          http://opensource.cit-ec.de/projects/icl               **
**                                                                 **
** File   : ICLFilter/src/ICLFilter/WeightedSumOp.cpp              **
** Module : ICLFilter                                              **
** Authors: Christof Elbrechter, Robert Haschke, Andre Justus      **
**                                                                 **
**                                                                 **
** GNU LESSER GENERAL PUBLIC LICENSE                               **
** This file may be used under the terms of the GNU Lesser General **
** Public License version 3.0 as published by the                  **
**                                                                 **
** Free Software Foundation and appearing in the file LICENSE.LGPL **
** included in the packaging of this file.  Please review the      **
** following information to ensure the license requirements will   **
** be met: http://www.gnu.org/licenses/lgpl-3.0.txt                **
**                                                                 **
** The development of this software was supported by the           **
** Excellence Cluster EXC 277 Cognitive Interaction Technology.    **
** The Excellence Cluster EXC 277 is a grant of the Deutsche       **
** Forschungsgemeinschaft (DFG) in the context of the German       **
** Excellence Initiative.                                          **
**                                                                 **
********************************************************************/

#include <ICLFilter/WeightedSumOp.h>

using namespace icl::utils;
using namespace icl::core;

namespace icl {
  namespace filter{
    using namespace std;
     // {{{ C++ fallback
    namespace{
      template <class T, class D>
      void apply_ws(const Img<T> *src, Img<D> *dst, const vector<D>& weights) {
       
        const ImgIterator<T> itSrc = src->beginROI(0);
        const ImgIterator<T> itSrcEnd = src->endROI(0);
        ImgIterator<D> itDst = dst->beginROI(0);
        
        D w = weights[0];
        for(;itSrc != itSrcEnd; ++itSrc, ++itDst){
          *itDst = *itSrc * w;
        }
        
        for (int c=src->getChannels()-1; c > 0; c--) {
          w = weights[c];
          const ImgIterator<T> itSrc = src->beginROI(c);
          const ImgIterator<T> itSrcEnd = src->endROI(c);
          ImgIterator<D> itDst = dst->beginROI(0);
          for(;itSrc!=itSrcEnd; ++itSrc, ++itDst){
            *itDst += *itSrc * w;
          }
        }
      }
    }
  
    // }}}
    
    void WeightedSumOp::apply (const ImgBase *poSrc, ImgBase **ppoDst){
      ICLASSERT_RETURN(poSrc);
      ICLASSERT_RETURN(ppoDst);
      ICLASSERT_RETURN( *ppoDst != poSrc );
      ICLASSERT_RETURN( (int)m_vecWeights.size() == poSrc->getChannels() );
      
      
      if(!prepare(ppoDst,
                  poSrc->getDepth() == depth64f ? depth64f : depth32f,
                  getClipToROI() ? poSrc->getROISize() : poSrc->getSize(),
                  formatMatrix,1,getClipToROI() ? Rect(Point::null,poSrc->getROISize()) : poSrc->getROI(),
                  poSrc->getTime()) ) return;
      
      if((*ppoDst)->getDepth() == depth64f){
        switch (poSrc->getDepth()) {
  #define ICL_INSTANTIATE_DEPTH(D) case depth##D: apply_ws(poSrc->asImg<icl##D>(),(*ppoDst)->asImg<icl64f>(),m_vecWeights); break;
          ICL_INSTANTIATE_ALL_DEPTHS;
  #undef ICL_INSTANTIATE_DEPTH
        }
      }else{
        vector<icl32f> v(m_vecWeights.size());
        for(unsigned int i=0;i<m_vecWeights.size();++i){
          v[i] = (float)m_vecWeights[i];
        }
        switch (poSrc->getDepth()) {
  #define ICL_INSTANTIATE_DEPTH(D) case depth##D: apply_ws(poSrc->asImg<icl##D>(),(*ppoDst)->asImg<icl32f>(),v); break;
          ICL_INSTANTIATE_ALL_DEPTHS;
  #undef ICL_INSTANTIATE_DEPTH
        }
      }
    }
  
  } // namespace filter
}