FitsOutput.h

Go to the documentation of this file.

/*
 * FitsOutput.h
 *
 *  Created on: Feb 8, 2022
 *      Author: mschefer
 */
 
#ifndef _SEIMPLEMENTATION_OUTPUT_FITSOUTPUT_H_
#define _SEIMPLEMENTATION_OUTPUT_FITSOUTPUT_H_
 
#include "Table/FitsWriter.h"
 
#include "SEImplementation/Output/FlushableOutput.h"
 
namespace SourceXtractor {
 
class FitsOutput : public FlushableOutput {
 
public:
  FitsOutput (const std::string& filename, SourceToRowConverter source_to_row, size_t flush_size)
    : FlushableOutput(source_to_row, flush_size), m_filename(filename), m_part_nb(0) {
    m_fits_writer = std::make_shared<Euclid::Table::FitsWriter>(m_filename, true);
    m_fits_writer->setHduName("CATALOG");
  }
 
  void nextPart() override {
    m_part_nb++;
    std::stringstream hdu_name;
    hdu_name << "CATALOG_" << m_part_nb;
 
    m_fits_writer = std::make_shared<Euclid::Table::FitsWriter>(m_filename);
    m_fits_writer->setHduName(hdu_name.str());
  }
 
protected:
  void writeRows(const std::vector<Euclid::Table::Row>& rows) override {
    Euclid::Table::Table table {rows};
    m_fits_writer->addData(table);
  }
 
private:
  std::string m_filename;
  int m_part_nb;
 
  std::shared_ptr<Euclid::Table::FitsWriter> m_fits_writer;
};
 
}
 
#endif /* _SEIMPLEMENTATION_OUTPUT_FITSOUTPUT_H_ */