vtkHDRReader.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   Visualization Toolkit
  Module:    vtkHDRReader.h
 
  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
 
     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.
 
=========================================================================*/
#ifndef vtkHDRReader_h
#define vtkHDRReader_h
 
#include "vtkIOImageModule.h" // For export macro
#include "vtkImageReader.h"
#include <string>
#include <vector>
 
class VTKIOIMAGE_EXPORT vtkHDRReader : public vtkImageReader
{
public:
  static vtkHDRReader* New();
  vtkTypeMacro(vtkHDRReader, vtkImageReader);
 
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  enum FormatType
  {
    FORMAT_32BIT_RLE_RGBE = 0,
    FORMAT_32BIT_RLE_XYZE
  };
 
 
  vtkGetMacro(Format, int);
 
 
  vtkGetMacro(Gamma, double);
 
 
  vtkGetMacro(Exposure, double);
 
 
  vtkGetMacro(PixelAspect, double);
 
  int CanReadFile(const char* fname) override;
 
  const char* GetFileExtensions() override { return ".hdr .pic"; }
 
  const char* GetDescriptiveName() override { return "Radiance HDR"; }
 
protected:
  vtkHDRReader();
  ~vtkHDRReader() override;
 
  std::string ProgramType;
  FormatType Format;
  double Gamma;
  double Exposure;
  double PixelAspect;
 
  bool FlippedX = false;
 
  bool SwappedAxis = false;
 
  void ExecuteInformation() override;
  void ExecuteDataWithInformation(vtkDataObject* out, vtkInformation* outInfo) override;
  bool HDRReaderUpdateSlice(float* outPtr, int* outExt);
  void HDRReaderUpdate(vtkImageData* data, float* outPtr);
 
  bool HasError(istream* is);
 
  int GetWidth() const;
  int GetHeight() const;
 
  bool ReadHeaderData();
 
  void ConvertAllDataFromRGBToXYZ(float* outPtr, int size);
 
  void FillOutPtrRLE(int* outExt, float*& outPtr, std::vector<unsigned char>& lineBuffer);
  void FillOutPtrNoRLE(int* outExt, float*& outPtr, std::vector<unsigned char>& lineBuffer);
 
  bool ReadAllFileNoRLE(istream* is, float* outPtr, int decrPtr, int* outExt);
 
  bool ReadLineRLE(istream* is, unsigned char* lineBufferPtr);
 
  void RGBE2Float(unsigned char rgbe[4], float& r, float& g, float& b);
 
  static void XYZ2RGB(const float convertMatrix[3][3], float& r, float& g, float& b);
 
private:
  vtkHDRReader(const vtkHDRReader&) = delete;
  void operator=(const vtkHDRReader&) = delete;
};
#endif