FitsWriter.h

Go to the documentation of this file.

/*
 * FitsWriter.h
 *
 *  Created on: May 23, 2017
 *      Author: mschefer
 */
 
#ifndef _SEFRAMEWORK_IMAGE_FITSWRITER_H_
#define _SEFRAMEWORK_IMAGE_FITSWRITER_H_
 
#include "ElementsKernel/Logging.h"
#include "SEFramework/CoordinateSystem/CoordinateSystem.h"
#include "SEFramework/Image/Image.h"
#include "SEFramework/Image/ImageAccessor.h"
#include "SEFramework/Image/WriteableBufferedImage.h"
#include "SEFramework/FITS/FitsImageSource.h"
#include "SEFramework/FITS/TemporaryFitsImageSource.h"
 
namespace SourceXtractor {
 
static Elements::Logging fitsWriterLogger = Elements::Logging::getLogger("FitsWriter");

class FitsWriter {
 
public:

  virtual ~FitsWriter() = default;
 
  template <typename T>
  static void writeFile(const Image<T> &image, const std::string &filename,
                        const std::shared_ptr<CoordinateSystem> coord_system = nullptr, bool append=false) {
    auto target_image = newImage<T>(filename, image.getWidth(), image.getHeight(), coord_system, append);
 
    // FIXME optimize the copy by using tile boundaries, image chunks, etc
    ImageAccessor<T> accessor(image);
    for (int y = 0; y < image.getHeight(); y++) {
      for (int x = 0; x < image.getWidth(); x++) {
        target_image->setValue(x, y, accessor.getValue(x, y));
      }
    }
  }
 
  template <typename T>
  static std::shared_ptr<WriteableImage<T>> newImage(const std::string &filename, int width, int height,
                                                     const std::shared_ptr<CoordinateSystem> coord_system = nullptr, bool append=false) {
    fitsWriterLogger.debug() << "Creating file " << filename;
 
    auto image_source = std::make_shared<FitsImageSource>(filename, width, height, ImageTile::getTypeValue(T()), coord_system, append);
    return WriteableBufferedImage<T>::create(image_source);
  }
 
  template <typename T>
    static std::shared_ptr<WriteableImage<T>> newTemporaryImage(const std::string &pattern, int width, int height) {
    fitsWriterLogger.debug() << "Creating temporary fits file";
 
    auto image_source = std::make_shared<TemporaryFitsImageSource>(pattern, width, height, ImageTile::getTypeValue(T()));
    return WriteableBufferedImage<T>::create(image_source);
  }
 
 
}; /* End of FitsReader class */
 
} /* namespace SourceXtractor */
 
 
 
 
 
#endif /* _SEFRAMEWORK_IMAGE_FITSWRITER_H_ */