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