TiffWriter.hpp

#ifndef TIFFWRITER_HPP
#define TIFFWRITER_HPP
 
#include <iostream>
#include "tiffio.h"
#include "map_grid.hpp"
#include <string>
 
template<unsigned int dim, typename T>
class TiffWriter
{
    template<typename grid, typename Mem> int write(grid data, std::string file)
    {
        // Grid can have several properties we can save only scalar fields
        // for each properties save one scalar fields
 
        // Tiff files can be saved up to 5D
 
        if (dim > 5)
        {
            std::cerr << "Error Tiff writer support until 5D images" << "\n";
        }
 
        // Open the tiff image
 
        uint32 width;
        uint32 height;
        TIFF *tif = TIFFOpen(file.c_str(),"w");
 
        // if the file is open
 
        if(tif)
        {
            // set width and height for 2D
 
            width = data.getGrid().size(0);
            height = data.getGrid().size(1);
 
            TIFFSetField(tif,TIFFTAG_IMAGEWIDTH, &width);
            TIFFSetField(tif,TIFFTAG_IMAGELENGTH, &height);
 
            // Create the tiff line, in case the grid is CPU, we have only
            // one property and is a scalar, we can directly copy the line
 
            typename boost::fusion::result_of::at<T::type,0>::type first_element_type;
 
            if (typeid(grid).name() == "grid_cpu" && T::num_prop == 1 && boost::is_array<first_element_type>::type::value == true)
            {
                // Get the grid key iterator
 
                grid_key_dx_iterator<dim> key = data.getIterator();
 
                // write all lines
 
                for(int i = 0; i < height ; i++)
                {
                    // select the correct lines
 
                    key.set(1,i);
 
                    // we have only one scalar properties, get the buffer pointer
                    void * buf = &data.template get<0>(key);
 
                    TIFFWriteScanline(tif,buf,i, 0);
                }
            }
            else
            {
                // we have to create the a scan line for each properties and index array
                // each property and index array became a channel
 
                // Get the grid key iterator
 
                grid_key_dx_iterator<dim> key = data.getIterator();
 
                // count how many properties and how many indexes we have
 
                const int n_prp = total_prop<T>;
 
                // write all lines
 
                for(int i = 0; i < height ; i++)
                {
                    // select the correct lines
 
                    key.set(1,i);
 
                    // we have only one scalar properties, get the buffer pointer
                    void * buf = &data.template get<0>(key);
 
                    TIFFWriteScanline(tif,buf,i, 0);
                }
            }
        }
    }
 
};
 
#endif