Ccorrelation_opticalFlow_fem.h

Go to the documentation of this file.

/**********************************************************************
 * 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 CCORRELATION_OPTICALFLOW_FEM
#define CCORRELATION_OPTICALFLOW_FEM


#include "Ccorrelation_opticalFlow.h"



template<typename T,typename Timg>
class Ccorrelation_opticalFlow_fem: public Ccorrelation_opticalFlow<T,Timg>
{
  
  public:

    Ccorrelation_opticalFlow_fem() : Ccorrelation_opticalFlow<T,Timg>()
    {// constructor
      
      this->class_name = "Correlation : Optical Flow Finite Element Method";

    }
    
    virtual void init(CParameterNetCDF &fp)
    {// init from parameter file
      
      Ccorrelation_opticalFlow<T,Timg>::init(fp);
       
      std::string attribute_name = "median_filter";
      int error = fp.loadAttribute(attribute_name,this->m_medianFilter);
      if (this->m_verbose){printf("\t\t median filter over following domain -> %i\n",this->m_medianFilter);}
      
    }

    virtual void assignL(const Cmesh<T,Timg> &oMesh, CImgList<T> &Ve)
    {//assign approriate size Ve = Grad[dim].N[dim][mode] product at pix

        Ve.assign(oMesh.num_var(), oMesh.num_mod(),1,1,1, 0);
   
    }
   
    virtual void evalGN(const std::vector<T> &grad,const CImgList<T> &N, CImgList<T> &Ve)
    {//eval V[dim][mode] = grad[dim].N[dim][mode] product
    
      cimglist_for(N, dim)
      {
        cimg_forX(N[dim], mode)
        {
          Ve[dim][mode] = grad[dim] * N[dim][mode];
        }
      }
      
    }
   
    virtual void e_ref(const Cimage<Timg> &oImage, const Cmesh<T,Timg> &oMesh, const int &elem, std::vector<T> &ref)
    {// store position of upper left node as element position reference
      
        cimglist_for(oMesh.m_node, dim)
        {
          ref[dim] = oMesh.m_node[dim][oMesh.m_connect[elem].front()]*oImage.m_pix[dim] ;
        }
        
    }
   
    virtual void evalShape(const CImg<Timg> &list, const Cimage<Timg> &oImage, const Cfield<T,Timg> &oField, const std::vector<T> &ref, const std::vector<T> &lCoor, const std::vector<T> &gCoor, CImgList<T> &N)
    { 
      
      std::vector<T> coor;
      coor = lCoor;
      
      (oField.m_pShape)->exec(list, coor, N);
      
    }

    virtual void evalU(const CImgList<T> &solution, const Cmesh<T,Timg> &oMesh, const int &elem, const CImgList<T> &N, std::vector<T> &U)
    {//eval U[dim] at pix
    
      U[2] = 0;//init required if dimension is lower than 3
    
      cimglist_for(N, dim)
      {
        
        U[dim] = 0;
        
        cimg_forX(N[dim], mode)
        {
          U[dim] += solution[dim][oMesh.m_connect[elem][mode]] * N[dim][mode];
        }

      }
      
      
    }
  
};
 
#endif