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 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkDiagonalMatrixSource.h
-------------------------------------------------------------------------
Copyright 2008 Sandia Corporation.
Under the terms of Contract DE-AC04-94AL85000 with Sandia Corporation,
the U.S. Government retains certain rights in this software.
-------------------------------------------------------------------------
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.
=========================================================================*/
/**
* @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 "vtkFiltersSourcesModule.h" // For export macro
#include "vtkArrayDataAlgorithm.h"
class VTKFILTERSSOURCES_EXPORT vtkDiagonalMatrixSource : public vtkArrayDataAlgorithm
{
public:
static vtkDiagonalMatrixSource* New();
vtkTypeMacro(vtkDiagonalMatrixSource, vtkArrayDataAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent) VTK_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() VTK_OVERRIDE;
int RequestData(
vtkInformation*,
vtkInformationVector**,
vtkInformationVector*) VTK_OVERRIDE;
private:
vtkDiagonalMatrixSource(const vtkDiagonalMatrixSource&) VTK_DELETE_FUNCTION;
void operator=(const vtkDiagonalMatrixSource&) VTK_DELETE_FUNCTION;
vtkArray* GenerateDenseArray();
vtkArray* GenerateSparseArray();
int ArrayType;
vtkIdType Extents;
double Diagonal;
double SuperDiagonal;
double SubDiagonal;
char* RowLabel;
char* ColumnLabel;
};
#endif
|