SoVRRasterStackReader.h

/*=======================================================================
 *** THE CONTENT OF THIS WORK IS PROPRIETARY TO FEI S.A.S, (FEI S.A.S.),            ***
 ***              AND IS DISTRIBUTED UNDER A LICENSE AGREEMENT.                     ***
 ***                                                                                ***
 ***  REPRODUCTION, DISCLOSURE,  OR USE,  IN WHOLE OR IN PART,  OTHER THAN AS       ***
 ***  SPECIFIED  IN THE LICENSE ARE  NOT TO BE  UNDERTAKEN  EXCEPT WITH PRIOR       ***
 ***  WRITTEN AUTHORIZATION OF FEI S.A.S.                                           ***
 ***                                                                                ***
 ***                        RESTRICTED RIGHTS LEGEND                                ***
 ***  USE, DUPLICATION, OR DISCLOSURE BY THE GOVERNMENT OF THE CONTENT OF THIS      ***
 ***  WORK OR RELATED DOCUMENTATION IS SUBJECT TO RESTRICTIONS AS SET FORTH IN      ***
 ***  SUBPARAGRAPH (C)(1) OF THE COMMERCIAL COMPUTER SOFTWARE RESTRICTED RIGHT      ***
 ***  CLAUSE  AT FAR 52.227-19  OR SUBPARAGRAPH  (C)(1)(II)  OF  THE RIGHTS IN      ***
 ***  TECHNICAL DATA AND COMPUTER SOFTWARE CLAUSE AT DFARS 52.227-7013.             ***
 ***                                                                                ***
 ***                   COPYRIGHT (C) 1996-2023 BY FEI S.A.S,                        ***
 ***                        BORDEAUX, FRANCE                                        ***
 ***                      ALL RIGHTS RESERVED                                       ***
**=======================================================================*/
/*=======================================================================
** Author      : C. Ognier (Jul 2005)
**=======================================================================*/
#ifndef  _SO_VR_RASTER_STACK_READER_
#define  _SO_VR_RASTER_STACK_READER_
 
#ifdef _WIN32
#pragma warning( push )
#pragma warning(disable:4251)
#endif
 
#include <LDM/readers/SoVolumeReader.h>
#include <Inventor/helpers/SbFileHelper.h>
#include <Inventor/image/SoRasterImageFile.h>
#include <Inventor/image/SoRasterReaderSet.h>
#include <Inventor/image/SoRasterImageRW.h>
#include <Inventor/image/SbRasterImage.h>
#include <Inventor/STL/vector>
#include <Inventor/STL/string>
#include <VolumeViz/nodes/SoVolumeData.h>
 
class  SoVRRasterStackReader : public SoVolumeReader  
 {
  SO_FIELDCONTAINER_HEADER(SoVRRasterStackReader);
 
 public:
 
   enum ChannelSelection {
    MAX_CHANNEL = 0,   // Maximum of all channels
    LUMINANCE = 1,     // Luminance
    RED_CHANNEL = 2,   // Red channel
    GREEN_CHANNEL = 3, // Green channel
    BLUE_CHANNEL = 4,  // Blue channel
    ALPHA_CHANNEL = 5, // Alpha channel
    ALL_CHANNELS = 11  // All channels combined
   };
 
  SoVRRasterStackReader();
 
  virtual int setFilename(const SbString& filename);
 
  SbBool setFilenameList(const SbStringList& filenameList);
 
  SbBool setDirectory(const SbString& directory);
 
  virtual ReadError getDataChar(SbBox3f& size, SoDataSet::DataType& type, SbVec3i32& dim) override;
 
  virtual void getSubSlice(const SbBox2i32& subSlice, int sliceNumber, void* data);
 
  void setSize(SbBox3f& size) {m_size = size;};
 
  void setChannelSelection(ChannelSelection cs) {m_channelSelection = cs;};
 
  virtual SoVolumeReader::ReaderType getReaderType ()  {return RASTERSTACK;};
 
  virtual SbBool isRGBA() const;
 
  virtual SbBool isThreadSafe() const;
 
private:
 
  int extractFileList(const SbString& headFile);
  SbBool extractParameters(const char* lineStr);
 
  SoDataSet::DataType findOutDestType(const SbRasterImage::Components nc, const unsigned int numComponent);
 
  void RGBA2HLS(SbVec3f& hls, float* rgba);
 
  float   convertFloat(float   value);
  short   convertShort(short   value);
  int32_t convertInt32(int32_t value);
  unsigned short convertUShort(unsigned short value);
  uint32_t convertUInt32(uint32_t value);
  SbBool needEndianConversion();
 
  int getChannelId(const char* channel);
 
  template <typename T>
  void getSubSliceInternal(const SbBox2i32 &subSlice, int sliceNumber, void* data);
 
  template <typename TSRC, typename T>
  void convertSubSliceInternal(TSRC* srcData, const SbVec2i32 srcSize, const unsigned int srcNumComponent, std::vector<T>& dstVector);
 
  SbBool readImage(int numImage, SbRasterImage& rasterImage, const SbBox2i32& region, SbBool infoOnly);
 
  template <typename T>
  void getSingleChannelData(std::vector<T> srcVector, SbVec2i32& size2s,
    unsigned int xMin, unsigned int xMax,
    unsigned int yMin, unsigned int yMax,
    void* data);
 
  template <typename T>
  T endianConverter(const T value);
 
  SbString  m_listDirectory;
  SbBox3f   m_size;
  SbVec3i32 m_dim;
  SbBool    m_isRawData;
  unsigned int m_nc;
  SoDataSet::DataType m_srcType;
  SoDataSet::DataType m_destType;
  ChannelSelection m_channelSelection;
  std::vector< SbString > m_fileList;
  unsigned int m_rawHeaderSize;
  SbBool m_rawIsBigEndian;
  SbBool m_rawIsBinary;
 
private:
 
  bool m_isInitialized;
 
  /*
   * Internal function to read a region of a raw file into a SbRasterImage
   * @return success
   */
  template <typename T>
  SbBool readRawBuffer(FILE* fp, SbRasterImage& rasterImage, const SbBox2i32& region);
};
 
#ifdef _WIN32
#pragma warning( pop )
#endif
 
#endif /* _SO_VR_RASTER_STACK_READER_ */