PyFitsFile.cpp

Go to the documentation of this file.

#include "SEUtils/VariantCast.h"
#include <SEImplementation/PythonConfig/PyFitsFile.h>
#include <boost/algorithm/string.hpp>

namespace SourceXtractor {

PyFitsFile::PyFitsFile(const std::string& filename) : m_filename(filename) {
  m_file = std::make_shared<FitsFile>(filename, false);
}

std::vector<int> PyFitsFile::getImageHdus() const {
  std::vector<int> v;

  for(auto num : m_file->getImageHdus()) {
    v.push_back(num-1);
  }

  return v;
}

std::vector<int> PyFitsFile::getDimensions(int hdu) const {
  return m_file->getDimensions(hdu+1);
}


std::map<std::string, std::string> PyFitsFile::getHeaders(int hdu) const {
  const auto& headers = m_file->getHDUHeaders(hdu+1);
  std::map<std::string, std::string> headers_str;

  for (auto& i : headers) {
    auto key = i.first;
    auto value = VariantCast<std::string>(i.second.m_value);
    boost::trim(key);
    boost::trim(value);
    headers_str[key] = value;
  }

  return headers_str;
}

}