Cformat_float_float.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 CFORMAT_FLOAT_FLOAT
#define CFORMAT_FLOAT_FLOAT

#include "Cformat.h"

class Cformat_float_float: public Cformat
{

  public:
 
  Cformat_float_float() : Cformat()
  {// constructor
    
    this->class_name = "Format : float float";

  }    

  virtual void exec(int argc, char *argv[])
  {

    //output name
    std::string ofile = cimg_option("-o","./","ofile");
    
    //create image container object
    Cimage_factory<float> image_factory("image_list");
    Cimage<float>* pImage = image_factory.create(argc,argv);
  
    //create sequence correlation object
    Csequence<float,float>* pSequence = new Csequence<float,float>;
    pSequence->init(argc, argv);

    for (int s=0; s<pImage->num_samples(); s++)
    {

      frame_number = s;
    
      pImage->exec(argc, argv, s);//load images
    
      //Test if the image sequence is (A-B, B-C...) or (A-B, C-D...) as in PIV. In the last case, pSequence->m_dataField must be clear. Otherwise, the second correlation sequence will be treated with the previous field as init.
      if (pImage->m_mode != DEPENDENT){pSequence->m_modes.clear();}
      pSequence->exec(argc, argv, *pImage);//perform correlation sequences
    
      this->save(*pSequence, *pImage, ofile, frame_number);
      
    }//end image list
  
    CImg<T> res(pSequence->m_meanRes.size(),1,1,1,0);
    cimg_forX(res,step){res[step] = pSequence->m_meanRes[step];}
    pImage->save_netcdf(res, ofile + "residue_mean", 0);
  
    std::cout<<"calcul done\n";

    delete pImage;
    delete pSequence;

  }
 
  void save(Csequence<float,float> &oSequence, Cimage<float> &oImage, const std::string &ofile, int &s)
  {
   
    std::cout<<"-------------------------------\n";
    std::cout<<"\n\nData saving\n\n";
    std::cout<<"-------------------------------\n";
    
        
    std::string sample_string;
    std::ostringstream sample_index;  
    
    
    oImage.save_netcdf(oSequence.m_modes, ofile + "modes", s);

    if (!oSequence.m_correlograms.is_empty())
    {
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(oSequence.m_correlograms, ofile + "correlograms", s);
      }
      else
      {
        sample_index.width(5);
        sample_index.fill('0');
        sample_index << s;

        sample_string = ofile + "correlograms_"+sample_index.str()+".tiff";
        oSequence.m_correlograms.save_tiff(sample_string.c_str());
        
        sample_index.str("");
        
      }
      
    }

    if (!oSequence.m_residues.is_empty())
    {
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(oSequence.m_residues, ofile + "residue", s);
      }
      else
      {
        
        sample_index.width(5);
        sample_index.fill('0');
        sample_index << s;
        
        sample_string = ofile + "residue_"+sample_index.str()+".tiff";
        
        oSequence.m_residues.save_tiff(sample_string.c_str());
        
        sample_index.str("");
          
        }
    }
    
    if (!oSequence.m_deformed.is_empty())
    {
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(oSequence.m_deformed, ofile + "deform", s);
      }
      else
      {
        sample_index.width(5);
        sample_index.fill('0');
        sample_index << s;
        
        sample_string = ofile + "deform_"+sample_index.str()+".tiff";
        
        oSequence.m_deformed.save_tiff(sample_string.c_str());
        
        sample_index.str("");
      }
    }

    if (!oSequence.m_fields.is_empty())
    {
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(oSequence.m_fields, ofile + "fields", s);
      }
      else
      {
        cimglist_for(oSequence.m_fields, l)
        {
          
          sample_index.width(5);
          sample_index.fill('0');
          sample_index << s;
          
          if(l == 0){sample_string = ofile + "U_"+sample_index.str()+".tiff";}
          if(l == 1){sample_string = ofile + "V_"+sample_index.str()+".tiff";}
          if(l == 2){sample_string = ofile + "W_"+sample_index.str()+".tiff";}
          
          oSequence.m_fields[l].save_tiff(sample_string.c_str());
          
          sample_index.str("");
          
        }
        
      }
      
    }
    
    if (!oSequence.m_node.is_empty() and s == 0)
    {
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(oSequence.m_node, ofile + "nodes", s);
      }
      else
      {
        cimglist_for(oSequence.m_node, l)
        {
          
          if(l == 0){sample_string = ofile + "X.tiff";}
          if(l == 1){sample_string = ofile + "Y.tiff";}
          if(l == 2){sample_string = ofile + "Z.tiff";}
          
          oSequence.m_node[l].save_tiff(sample_string.c_str());
          
        }
        
      }
    }
      
    if (!oImage.m_mask.is_empty() and s == 0)
    {
      
      CImg<int> msk(oImage.m_mask.get_resize(oSequence.m_node[0],1));
     
      if (oImage.m_oFormat!="tiff")
      {
        oImage.save_netcdf(msk, ofile + "mask_modes", s);
      }
      else
      {
        sample_string = ofile + "mask_modes_"+".tiff";
        msk.save_tiff(sample_string.c_str());
      }
      
    }

    
  }

 
};

#endif