/
SaveFITS.h
68 lines (50 loc) · 2.2 KB
/
SaveFITS.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2016 ISIS Rutherford Appleton Laboratory UKRI,
// NScD Oak Ridge National Laboratory, European Spallation Source,
// Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#pragma once
#include "MantidAPI/Algorithm.h"
#include "MantidAPI/MatrixWorkspace_fwd.h"
#include "MantidDataHandling/DllConfig.h"
namespace Mantid {
namespace DataHandling {
/**
SaveFITS : Save images in FITS formats.
*/
class MANTID_DATAHANDLING_DLL SaveFITS final : public API::Algorithm {
public:
const std::string name() const override final;
int version() const override final;
const std::vector<std::string> seeAlso() const override { return {"LoadFITS", "SaveNXTomo"}; }
const std::string category() const override final;
const std::string summary() const override final;
private:
void init() override final;
void exec() override final;
std::map<std::string, std::string> validateInputs() override;
void saveFITSImage(const API::MatrixWorkspace_sptr &img, const std::string &filename);
void writeFITSHeaderBlock(const API::MatrixWorkspace_sptr &img, std::ofstream &file);
void writeFITSImageMatrix(const API::MatrixWorkspace_sptr &img, std::ofstream &file);
void writeFITSHeaderEntry(const std::string &hdr, std::ofstream &file);
std::string makeBitDepthHeader(size_t depth) const;
void writeFITSHeaderAxesSizes(const API::MatrixWorkspace_sptr &img, std::ofstream &file);
void writePaddingFITSHeaders(size_t count, std::ofstream &file);
static const size_t g_maxBitDepth;
static const std::array<int, 3> g_bitDepths;
static const size_t g_maxBytesPP;
// size of header entries in bytes
static const size_t g_maxLenHdr;
// pre-defined header contents
static const std::string g_FITSHdrEnd;
static const std::string g_FITSHdrFirst;
static const std::string g_bitDepthPre;
static const std::string g_bitDepthPost;
static const std::string g_FITSHdrAxes;
static const std::string g_FITSHdrExtensions;
static const std::string g_FITSHdrRefComment1;
static const std::string g_FITSHdrRefComment2;
};
} // namespace DataHandling
} // namespace Mantid