FitsImageSource.h

Go to the documentation of this file.

 
/*
 * FitsImageSource.h
 *
 *  Created on: Feb 21, 2018
 *      Author: mschefer
 */
 
#ifndef _SEFRAMEWORK_IMAGE_FITSIMAGESOURCE_H_
#define _SEFRAMEWORK_IMAGE_FITSIMAGESOURCE_H_
 
#include <memory>
#include <vector>
#include <map>
 
#include <boost/lexical_cast.hpp>
 
#include "FilePool/FileManager.h"
#include "SEFramework/CoordinateSystem/CoordinateSystem.h"
#include "SEFramework/Image/ImageSourceWithMetadata.h"
#include "SEFramework/FITS/FitsFile.h"
#include "SEUtils/VariantCast.h"
 
 
namespace SourceXtractor {
 
using Euclid::FilePool::FileManager;
using Euclid::FilePool::FileHandler;
 
class FitsImageSource : public ImageSource, public std::enable_shared_from_this<ImageSource> {
public:
 
 
  explicit FitsImageSource(const std::string& filename, int hdu_number = 0,
                  ImageTile::ImageType image_type = ImageTile::AutoType,
                  std::shared_ptr<FileManager> manager = FileManager::getDefault());
 
  FitsImageSource(const std::string& filename, int width, int height,
                  ImageTile::ImageType image_type,
                  const std::shared_ptr<CoordinateSystem> coord_system = nullptr,
                  bool append = false,
                  bool empty_primary = false,
                  std::shared_ptr<FileManager> manager = FileManager::getDefault());
 
  virtual ~FitsImageSource() = default;
 
  std::string getRepr() const override {
    return m_filename;
  }
 
  int getWidth() const override {
    return m_width;
  }
 
  int getHeight() const override {
    return m_height;
  }
 
  int getDepth() const {
    return m_depth;
  }
 
  void setLayer(int layer);
 
  std::shared_ptr<ImageTile> getImageTile(int x, int y, int width, int height) const override;
 
  void saveTile(ImageTile& tile) override;
 
  template<typename TT>
  bool readFitsKeyword(const std::string& header_keyword, TT& out_value) const {
    auto& headers = getMetadata();
    auto i = headers.find(header_keyword);
    if (i != headers.end()) {
      out_value = VariantCast<TT>(i->second.m_value);
      return true;
    }
    return false;
  }
 
  int getHDU() const {
    return m_hdu_number;
  }
 
  ImageTile::ImageType getType() const override {
    return m_image_type;
  }
 
  std::unique_ptr<std::vector<char>> getFitsHeaders(int& number_of_records) const;
 
  const std::map<std::string, MetadataEntry>& getMetadata() const override;
 
  void setMetadata(const std::string& key, const MetadataEntry& value) override;
 
private:
  void switchHdu(fitsfile *fptr, int hdu_number) const;
 
  int getDataType() const;
 
  int getImageType() const;
 
  std::string m_filename;
  std::shared_ptr<FileManager> m_file_manager;
  std::shared_ptr<FileHandler> m_handler;
 
  int m_hdu_number;
 
  int m_width;
  int m_height;
  int m_depth;
  ImageTile::ImageType m_image_type;
 
  int m_current_layer;
};
 
}
 
 
#endif /* _SEFRAMEWORK_IMAGE_FITSIMAGESOURCE_H_ */