vtkSTLReader.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#ifndef vtkSTLReader_h
#define vtkSTLReader_h

#include "vtkAbstractPolyDataReader.h"
#include "vtkIOGeometryModule.h" // For export macro

VTK_ABI_NAMESPACE_BEGIN
class vtkCellArray;
class vtkFloatArray;
class vtkIncrementalPointLocator;
class vtkPoints;
class vtkResourceParser;
class vtkResourceStream;

class VTKIOGEOMETRY_EXPORT vtkSTLReader : public vtkAbstractPolyDataReader
{
public:
  vtkTypeMacro(vtkSTLReader, vtkAbstractPolyDataReader);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  static vtkSTLReader* New();


  static bool CanReadFile(VTK_FILEPATH const char* filename);
  static bool CanReadFile(vtkResourceStream* stream);

  vtkMTimeType GetMTime() override;


  vtkSetMacro(RelaxedConformance, bool);
  vtkGetMacro(RelaxedConformance, bool);
  vtkBooleanMacro(RelaxedConformance, bool);


  vtkSetMacro(Merging, vtkTypeBool);
  vtkGetMacro(Merging, vtkTypeBool);
  vtkBooleanMacro(Merging, vtkTypeBool);


  vtkSetMacro(ScalarTags, vtkTypeBool);
  vtkGetMacro(ScalarTags, vtkTypeBool);
  vtkBooleanMacro(ScalarTags, vtkTypeBool);


  void SetLocator(vtkIncrementalPointLocator* locator);
  vtkGetObjectMacro(Locator, vtkIncrementalPointLocator);

  vtkGetStringMacro(Header);

  vtkGetObjectMacro(BinaryHeader, vtkUnsignedCharArray);

protected:
  vtkSTLReader();
  ~vtkSTLReader() override;

  vtkIncrementalPointLocator* NewDefaultLocator();

  vtkSetStringMacro(Header);
  virtual void SetBinaryHeader(vtkUnsignedCharArray* binaryHeader);

  vtkTypeBool Merging = true;
  vtkTypeBool ScalarTags = false;
  vtkIncrementalPointLocator* Locator = nullptr;
  char* Header = nullptr;
  vtkUnsignedCharArray* BinaryHeader = nullptr;

  int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;

private:
  bool RelaxedConformance = true;

  vtkSTLReader(const vtkSTLReader&) = delete;
  void operator=(const vtkSTLReader&) = delete;

  bool ReadBinarySTL(vtkResourceStream* stream, vtkPoints*, vtkCellArray*);
  bool ReadASCIISTL(
    vtkResourceParser* parser, vtkPoints*, vtkCellArray*, vtkFloatArray* scalars = nullptr);

  static bool ReadBinaryHeader(vtkResourceStream* stream, vtkUnsignedCharArray* header);
  static bool ReadBinaryTrisField(vtkResourceStream* stream, uint32_t& numTrisField);
  static bool ReadBinaryTrisFile(vtkResourceStream* stream, vtkTypeInt64& numTrisFile);
};

VTK_ABI_NAMESPACE_END
#endif