Cfield_opticalFlow_integrated_block.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 CFIELD_OPTICALFLOW_INTEGRATED_BLOCK
#define CFIELD_OPTICALFLOW_INTEGRATED_BLOCK


//-----------------------------CIMG_NETCDF-------------------------------
#define cimg_plugin  "plugins/add_fileformat.h"
#define cimg_use_netcdf

#ifdef cimg_use_netcdf
 #include "../NetCDF.Tool/struct_parameter_NetCDF.h"
// #define     cimg_plugin1 "plugins/netcdf_file_format4CImg1.h"
// #define cimglist_plugin1 "plugins/netcdf_file_format4CImgList1.h"
 #define     cimg_plugin2 "plugins/netcdf_file_format4CImg2.h"
 #define cimglist_plugin2 "plugins/netcdf_file_format4CImgList2.h"
 #include "../CImg.Tool/CImg_NetCDF.h"
#endif
//-----------------------------CIMG_NETCDF-------------------------------

#include "Cfield_opticalFlow_integrated.h"

template<typename T, typename Timg>
class Cfield_opticalFlow_integrated_block: public Cfield_opticalFlow_integrated<T,Timg>
{
  
  public:
    
    Cfield_opticalFlow_integrated_block() : Cfield_opticalFlow_integrated<T,Timg>()
    {
      this->class_name = "Field : Optical Flow Integrated per Block";
      
    }
    
    virtual void init(CParameterNetCDF &fp)
    {
      Cfield_opticalFlow_integrated<T,Timg>::init(fp);

    }  

    virtual void exec(const Cimage<Timg> &oImage, const Cmesh<T,Timg> &oMesh, CImgList<T> &modes_prev, CshapeFunction<T,Timg> &oShape_prev, const std::vector<T> &ref_prev, const CImgList<T> &nodes_prev)
    {
      this->m_mode.assign(this->m_pShape->m_dof, oMesh.grid_dims(0),oMesh.grid_dims(1),oMesh.grid_dims(2),1, 0);

      Cfield<T,Timg>::exec(oImage, oMesh, modes_prev, oShape_prev, ref_prev, nodes_prev); 
      
    }
    
    virtual void project(const Cmesh<T,Timg> &oMesh, const Cimage<Timg> &oImage, CImgList<T> &modes_prev, CshapeFunction<T,Timg> &oShape_prev, const std::vector<T> &ref_prev, const CImgList<T> &nodes_prev, CImgList<T> &modes_cur)
    {//project modes on new base
    
      //Check if rigid body translations are required within the current shape function. If yes, [U,V,W] at nodes are transfered to [Tx,Ty,Tz] at element centroid.
      std::vector<int> md, pos;
      md.clear();pos.clear();
      for (int i = 0; i<this->m_pShape->m_mods.size(); i++)
      {
        if (this->m_pShape->m_mods[i].compare(0,2,"Tx") == 0)
        {pos.push_back(i);md.push_back(0);}
        if (this->m_pShape->m_mods[i].compare(0,2,"Ty") == 0)
        {pos.push_back(i);md.push_back(1);}
        if (this->m_pShape->m_mods[i].compare(0,2,"Tz") == 0)
        {pos.push_back(i);md.push_back(2);}
      }

      if ((oShape_prev.m_name).compare(0,5,"OFFEM") == 0)
      {//previous step was a FEM type
        
        for (int i=0; i<pos.size(); i++)
        {//linear interpolation of required rigid body motions
         
          modes_prev[md[i]].resize(this->grid_dims(0),this->grid_dims(1),this->grid_dims(2),this->grid_dims(3),3);
          modes_cur[pos[i]] = modes_prev[md[i]];
          
        }
        
      }
      else if((oShape_prev.m_name).compare(0,2,"IC") == 0)
      {
        
        //build U map on previous mesh
        CImgList<T> U(oMesh.num_var(), modes_prev[0].width(), modes_prev[0].height(), modes_prev[0].depth(), 1, 0);
        cimglist_for(U, mode)
        {U[mode] = modes_prev[mode+2];}
        
        for (int i=0; i<pos.size(); i++)
        {//linear interpolation of required rigid body motions
         
          U[md[i]].resize(this->grid_dims(0),this->grid_dims(1),this->grid_dims(2),this->grid_dims(3),3);
          modes_cur[pos[i]] = U[md[i]];
          
        }
        
      }
      else if((oShape_prev.m_name).compare(0,4,"OFIB") == 0)
      {
        
        //One assumes that modes doesn't change from one OFIB to another
        if (modes_cur.size() == modes_prev.size())
        {this->Field2Field(modes_prev, modes_cur);}
        else
        {printf("warning : modes have been changes for multi-scale OFIB strategy!!!");}
        
      }
      else if((oShape_prev.m_name).compare(0,3,"OFI") == 0)
      {
      
        //build U map on previous mesh
        CImgList<T> U(oMesh.num_var(), modes_cur[0].width(), modes_cur[0].height(), modes_cur[0].depth(), 1, 0);
        this->OFIU(oMesh, oImage, oShape_prev, ref_prev, nodes_prev, modes_prev, U);
        
        for (int i=0; i<pos.size(); i++)
        {//linear interpolation of required rigid body motions
          modes_cur[pos[i]] = U[md[i]];
        }
        
      }
      
    }


};

#endif