// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
/**
 * @class   vtkImageImport
 * @brief   Import data from a C array.
 *
 * vtkImageImport provides methods needed to import image data from a source
 * independent of VTK, such as a simple C array or a third-party pipeline.
 * Note that the VTK convention is for the image voxel index (0,0,0) to be
 * the lower-left corner of the image, while most 2D image formats use
 * the upper-left corner.  You can use vtkImageFlip to correct the
 * orientation after the image has been loaded into VTK.
 * Note that is also possible to import the raw data from a Python string
 * instead of from a C array. The array applies on scalar point data only, not
 * on cell data.
 * @sa
 * vtkImageExport
 */

#ifndef vtkImageImport_h
#define vtkImageImport_h

#include "vtkIOImageModule.h" // For export macro
#include "vtkImageAlgorithm.h"

VTK_ABI_NAMESPACE_BEGIN
class VTKIOIMAGE_EXPORT vtkImageImport : public vtkImageAlgorithm
{
public:
  static vtkImageImport* New();
  vtkTypeMacro(vtkImageImport, vtkImageAlgorithm);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  /**
   * Import data and make an internal copy of it.  If you do not want
   * VTK to copy the data, then use SetImportVoidPointer instead (do
   * not use both).  Give the size of the data array in bytes.
   */
  void CopyImportVoidPointer(void* ptr, vtkIdType size);

  ///@{
  /**
   * Set the pointer from which the image data is imported.  VTK will
   * not make its own copy of the data, it will access the data directly
   * from the supplied array.  VTK will not attempt to delete the data
   * nor modify the data.
   */
  void SetImportVoidPointer(void* ptr);
  void* GetImportVoidPointer() { return this->ImportVoidPointer; }
  ///@}

  /**
   * Set the pointer from which the image data is imported.  Set save to 1
   * (the default) unless you want VTK to delete the array via C++ delete
   * when the vtkImageImport object is deallocated.  VTK will not make its
   * own copy of the data, it will access the data directly from the
   * supplied array.
   */
  void SetImportVoidPointer(void* ptr, int save);

  ///@{
  /**
   * Set/Get the data type of pixels in the imported data.  This is used
   * as the scalar type of the Output.  Default: Short.
   */
  vtkSetMacro(DataScalarType, int);
  void SetDataScalarTypeToDouble() { this->SetDataScalarType(VTK_DOUBLE); }
  void SetDataScalarTypeToFloat() { this->SetDataScalarType(VTK_FLOAT); }
  void SetDataScalarTypeToInt() { this->SetDataScalarType(VTK_INT); }
  void SetDataScalarTypeToShort() { this->SetDataScalarType(VTK_SHORT); }
  void SetDataScalarTypeToUnsignedShort() { this->SetDataScalarType(VTK_UNSIGNED_SHORT); }
  void SetDataScalarTypeToUnsignedChar() { this->SetDataScalarType(VTK_UNSIGNED_CHAR); }
  vtkGetMacro(DataScalarType, int);
  const char* GetDataScalarTypeAsString()
  {
    return vtkImageScalarTypeNameMacro(this->DataScalarType);
  }
  ///@}

  ///@{
  /**
   * Set/Get the number of scalar components, for RGB images this must be 3.
   * Default: 1.
   */
  vtkSetMacro(NumberOfScalarComponents, int);
  vtkGetMacro(NumberOfScalarComponents, int);
  ///@}

  ///@{
  /**
   * Get/Set the extent of the data buffer.  The dimensions of your data
   * must be equal to (extent[1]-extent[0]+1) * (extent[3]-extent[2]+1) *
   * (extent[5]-DataExtent[4]+1).  For example, for a 2D image use
   * (0,width-1, 0,height-1, 0,0).
   */
  vtkSetVector6Macro(DataExtent, int);
  vtkGetVector6Macro(DataExtent, int);
  void SetDataExtentToWholeExtent() { this->SetDataExtent(this->GetWholeExtent()); }
  ///@}

  ///@{
  /**
   * Set/Get the spacing (typically in mm) between image voxels.
   * Default: (1.0, 1.0, 1.0).
   */
  vtkSetVector3Macro(DataSpacing, double);
  vtkGetVector3Macro(DataSpacing, double);
  ///@}

  ///@{
  /**
   * Set/Get the origin of the data, i.e. the coordinates (usually in mm)
   * of voxel (0,0,0).  Default: (0.0, 0.0, 0.0).
   */
  vtkSetVector3Macro(DataOrigin, double);
  vtkGetVector3Macro(DataOrigin, double);
  ///@}

  ///@{
  /**
   * Set/Get the direction of the data, i.e. the 3x3 matrix to rotate
   * the coordinates from index space (ijk) to physical space (xyz).
   * Default: Identity Matrix (1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0)
   */
  vtkSetVectorMacro(DataDirection, double, 9);
  vtkGetVectorMacro(DataDirection, double, 9);
  ///@}

  ///@{
  /**
   * Get/Set the whole extent of the image.  This is the largest possible
   * extent.  Set the DataExtent to the extent of the image in the buffer
   * pointed to by the ImportVoidPointer.
   */
  vtkSetVector6Macro(WholeExtent, int);
  vtkGetVector6Macro(WholeExtent, int);
  ///@}

  /**
   * Propagates the update extent through the callback if it is set.
   */
  int RequestUpdateExtent(vtkInformation* request, vtkInformationVector** inputVector,
    vtkInformationVector* outputVector) override;
  /**
   * Override vtkAlgorithm
   */
  int ComputePipelineMTime(vtkInformation* request, vtkInformationVector** inInfoVec,
    vtkInformationVector* outInfoVec, int requestFromOutputPort, vtkMTimeType* mtime) override;

  ///@{
  /**
   * Set/get the scalar array name for this data set. Initial value is
   * "scalars".
   */
  vtkSetStringMacro(ScalarArrayName);
  vtkGetStringMacro(ScalarArrayName);
  ///@}

  ///@{
  /**
   * These are function pointer types for the pipeline connection
   * callbacks.  See further documentation on each individual callback.
   */
  typedef void (*UpdateInformationCallbackType)(void*);
  typedef int (*PipelineModifiedCallbackType)(void*);
  typedef int* (*WholeExtentCallbackType)(void*);
  typedef double* (*SpacingCallbackType)(void*);
  typedef double* (*OriginCallbackType)(void*);
  typedef double* (*DirectionCallbackType)(void*);
  typedef const char* (*ScalarTypeCallbackType)(void*);
  typedef int (*NumberOfComponentsCallbackType)(void*);
  typedef void (*PropagateUpdateExtentCallbackType)(void*, int*);
  typedef void (*UpdateDataCallbackType)(void*);
  typedef int* (*DataExtentCallbackType)(void*);
  typedef void* (*BufferPointerCallbackType)(void*);
  ///@}

  ///@{
  /**
   * Set/Get the callback for propagating UpdateInformation calls to a
   * third-party pipeline.  The callback should make sure that the
   * third-party pipeline information is up to date.
   */
  vtkSetMacro(UpdateInformationCallback, UpdateInformationCallbackType);
  vtkGetMacro(UpdateInformationCallback, UpdateInformationCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for checking whether the third-party
   * pipeline has been modified since the last invocation of the
   * callback.  The callback should return 1 for modified, and 0 for
   * not modified.  The first call should always return modified.
   */
  vtkSetMacro(PipelineModifiedCallback, PipelineModifiedCallbackType);
  vtkGetMacro(PipelineModifiedCallback, PipelineModifiedCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the whole extent of the input
   * image from a third-party pipeline.  The callback should return a
   * vector of six integers describing the extent of the whole image
   * (x1 x2 y1 y2 z1 z2).
   */
  vtkSetMacro(WholeExtentCallback, WholeExtentCallbackType);
  vtkGetMacro(WholeExtentCallback, WholeExtentCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the spacing of the input image
   * from a third-party pipeline.  The callback should return a vector
   * of three double values describing the spacing (dx dy dz).
   */
  vtkSetMacro(SpacingCallback, SpacingCallbackType);
  vtkGetMacro(SpacingCallback, SpacingCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the origin of the input image
   * from a third-party pipeline.  The callback should return a vector
   * of three double values describing the origin (x0 y0 z0).
   */
  vtkSetMacro(OriginCallback, OriginCallbackType);
  vtkGetMacro(OriginCallback, OriginCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the direction of the input image
   * from a third-party pipeline.  The callback should return a vector
   * of nine double values describing the rotation from ijk to xyz.
   */
  vtkSetMacro(DirectionCallback, DirectionCallbackType);
  vtkGetMacro(DirectionCallback, DirectionCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the scalar value type of the
   * input image from a third-party pipeline.  The callback should
   * return a string with the name of the type.
   */
  vtkSetMacro(ScalarTypeCallback, ScalarTypeCallbackType);
  vtkGetMacro(ScalarTypeCallback, ScalarTypeCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the number of components of the
   * input image from a third-party pipeline.  The callback should
   * return an integer with the number of components.
   */
  vtkSetMacro(NumberOfComponentsCallback, NumberOfComponentsCallbackType);
  vtkGetMacro(NumberOfComponentsCallback, NumberOfComponentsCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for propagating the pipeline update extent
   * to a third-party pipeline.  The callback should take a vector of
   * six integers describing the extent.  This should cause the
   * third-party pipeline to provide data which contains at least this
   * extent after the next UpdateData callback.
   */
  vtkSetMacro(PropagateUpdateExtentCallback, PropagateUpdateExtentCallbackType);
  vtkGetMacro(PropagateUpdateExtentCallback, PropagateUpdateExtentCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for propagating UpdateData calls to a
   * third-party pipeline.  The callback should make sure the
   * third-party pipeline is up to date.
   */
  vtkSetMacro(UpdateDataCallback, UpdateDataCallbackType);
  vtkGetMacro(UpdateDataCallback, UpdateDataCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting the data extent of the input
   * image from a third-party pipeline.  The callback should return a
   * vector of six integers describing the extent of the buffered
   * portion of the image (x1 x2 y1 y2 z1 z2).  The buffer location
   * should be set with the BufferPointerCallback.
   */
  vtkSetMacro(DataExtentCallback, DataExtentCallbackType);
  vtkGetMacro(DataExtentCallback, DataExtentCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the callback for getting a pointer to the data buffer of
   * an image from a third-party pipeline.  The callback should return
   * a pointer to the beginning of the buffer.  The extent of the
   * buffer should be set with the DataExtentCallback.
   */
  vtkSetMacro(BufferPointerCallback, BufferPointerCallbackType);
  vtkGetMacro(BufferPointerCallback, BufferPointerCallbackType);
  ///@}

  ///@{
  /**
   * Set/Get the user data which will be passed as the first argument
   * to all of the third-party pipeline callbacks.
   */
  vtkSetMacro(CallbackUserData, void*);
  vtkGetMacro(CallbackUserData, void*);
  ///@}

  ///@{
  /**
   * Invoke the appropriate callbacks
   */
  int InvokePipelineModifiedCallbacks();
  void InvokeUpdateInformationCallbacks();
  void InvokeExecuteInformationCallbacks();
  void InvokeExecuteDataCallbacks();
  void LegacyCheckWholeExtent();
  ///@}

protected:
  vtkImageImport();
  ~vtkImageImport() override;

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

  void* ImportVoidPointer;
  int SaveUserArray;

  int NumberOfScalarComponents;
  int DataScalarType;

  int WholeExtent[6];
  int DataExtent[6];
  double DataSpacing[3];
  double DataOrigin[3];
  double DataDirection[9];

  char* ScalarArrayName;
  void* CallbackUserData;

  UpdateInformationCallbackType UpdateInformationCallback;
  PipelineModifiedCallbackType PipelineModifiedCallback;
  WholeExtentCallbackType WholeExtentCallback;
  SpacingCallbackType SpacingCallback;
  OriginCallbackType OriginCallback;
  DirectionCallbackType DirectionCallback;
  ScalarTypeCallbackType ScalarTypeCallback;
  NumberOfComponentsCallbackType NumberOfComponentsCallback;
  PropagateUpdateExtentCallbackType PropagateUpdateExtentCallback;
  UpdateDataCallbackType UpdateDataCallback;
  DataExtentCallbackType DataExtentCallback;
  BufferPointerCallbackType BufferPointerCallback;

  void ExecuteDataWithInformation(vtkDataObject* d, vtkInformation* outInfo) override;

private:
  vtkImageImport(const vtkImageImport&) = delete;
  void operator=(const vtkImageImport&) = delete;
};

VTK_ABI_NAMESPACE_END
#endif