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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkBMPImageIO_h
#define itkBMPImageIO_h
#include "ITKIOBMPExport.h"
#include <fstream>
#include "itkImageIOBase.h"
#include <cstdio>
#include "itkRGBPixel.h"
namespace itk
{
/**
* \class BMPImageIO
*
* \brief Read BMPImage file format.
*
* \ingroup IOFilters
*
* \ingroup ITKIOBMP
*/
class ITKIOBMP_EXPORT BMPImageIO : public ImageIOBase
{
public:
ITK_DISALLOW_COPY_AND_MOVE(BMPImageIO);
/** Standard class type aliases. */
using Self = BMPImageIO;
using Superclass = ImageIOBase;
using Pointer = SmartPointer<Self>;
using RGBPixelType = RGBPixel<unsigned char>; // Palette is only unsigned char in BMP files
using PaletteType = std::vector<RGBPixelType>;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(BMPImageIO);
/** Getter for the FileLowerLeft attribute. */
itkGetConstMacro(FileLowerLeft, bool);
/** Getter for the BMPCompression attribute. */
itkGetConstMacro(BMPCompression, long);
/** Getter for the ColorPalette attribute. */
itkGetConstReferenceMacro(ColorPalette, PaletteType);
/*-------- This part of the interfaces deals with reading data. ----- */
/** Determine the file type. Returns true if this ImageIO can read the
* file specified. */
bool
CanReadFile(const char *) override;
/** Set the spacing and dimension information for the set filename. */
void
ReadImageInformation() override;
/** Reads the data from disk into the memory buffer provided. */
void
Read(void * buffer) override;
/*-------- This part of the interfaces deals with writing data. ----- */
/** Determine the file type. Returns true if this ImageIO can write the
* file specified. */
bool
CanWriteFile(const char *) override;
/** Set the spacing and dimension information for the set filename. */
void
WriteImageInformation() override;
/** Writes the data to disk from the memory buffer provided. Make sure
* that the IORegions has been set properly. */
void
Write(const void * buffer) override;
BMPImageIO();
~BMPImageIO() override;
void
PrintSelf(std::ostream & os, Indent indent) const override;
private:
void
SwapBytesIfNecessary(void * buffer, SizeValueType numberOfPixels);
/** This methods ensures that the endianness is respected */
void
Write32BitsInteger(unsigned int value);
void
Write16BitsInteger(unsigned short value);
RGBPixelType
GetColorPaletteEntry(const unsigned char entry) const;
std::ifstream m_Ifstream{};
std::ofstream m_Ofstream{};
long m_BitMapOffset{ 0 };
bool m_FileLowerLeft{ false };
short m_Depth{ 8 };
unsigned short m_NumberOfColors{ 0 };
unsigned int m_ColorPaletteSize{ 0 };
long m_BMPCompression{ 0 };
unsigned long m_BMPDataSize{ 0 };
PaletteType m_ColorPalette{};
};
} // end namespace itk
#endif // itkBMPImageIO_h
|
