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
|
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright 2008 Sandia Corporation
// SPDX-License-Identifier: LicenseRef-BSD-3-Clause-Sandia-USGov
/**
* @class vtkDiagonalMatrixSource
* @brief generates a sparse or dense square matrix
* with user-specified values for the diagonal, superdiagonal, and subdiagonal.
*
* @par Thanks:
* Developed by Timothy M. Shead (tshead@sandia.gov) at Sandia National Laboratories.
*/
#ifndef vtkDiagonalMatrixSource_h
#define vtkDiagonalMatrixSource_h
#include "vtkArrayDataAlgorithm.h"
#include "vtkFiltersSourcesModule.h" // For export macro
VTK_ABI_NAMESPACE_BEGIN
class vtkArray;
class VTKFILTERSSOURCES_EXPORT vtkDiagonalMatrixSource : public vtkArrayDataAlgorithm
{
public:
static vtkDiagonalMatrixSource* New();
vtkTypeMacro(vtkDiagonalMatrixSource, vtkArrayDataAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) override;
// Determines whether the output matrix will be dense or sparse
enum StorageType
{
DENSE,
SPARSE
};
vtkGetMacro(ArrayType, int);
vtkSetMacro(ArrayType, int);
///@{
/**
* Stores the extents of the output matrix (which is square)
*/
vtkGetMacro(Extents, vtkIdType);
vtkSetMacro(Extents, vtkIdType);
///@}
///@{
/**
* Stores the value that will be assigned to diagonal elements (default: 1)
*/
vtkGetMacro(Diagonal, double);
vtkSetMacro(Diagonal, double);
///@}
///@{
/**
* Stores the value that will be assigned to superdiagonal elements (default: 0)
*/
vtkGetMacro(SuperDiagonal, double);
vtkSetMacro(SuperDiagonal, double);
///@}
///@{
/**
* Stores the value that will be assigned to subdiagonal elements (default: 0)
*/
vtkGetMacro(SubDiagonal, double);
vtkSetMacro(SubDiagonal, double);
///@}
///@{
/**
* Controls the output matrix row dimension label.
* Default: "rows"
*/
vtkGetStringMacro(RowLabel);
vtkSetStringMacro(RowLabel);
///@}
///@{
/**
* Controls the output matrix column dimension label.
* Default: "columns"
*/
vtkGetStringMacro(ColumnLabel);
vtkSetStringMacro(ColumnLabel);
///@}
protected:
vtkDiagonalMatrixSource();
~vtkDiagonalMatrixSource() override;
int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector*) override;
private:
vtkDiagonalMatrixSource(const vtkDiagonalMatrixSource&) = delete;
void operator=(const vtkDiagonalMatrixSource&) = delete;
vtkArray* GenerateDenseArray();
vtkArray* GenerateSparseArray();
int ArrayType;
vtkIdType Extents;
double Diagonal;
double SuperDiagonal;
double SubDiagonal;
char* RowLabel;
char* ColumnLabel;
};
VTK_ABI_NAMESPACE_END
#endif
|