mmio.h 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133
  1. /*
  2. * Matrix Market I/O library for ANSI C
  3. *
  4. * See http://math.nist.gov/MatrixMarket for details.
  5. *
  6. *
  7. */
  8. #ifndef MM_IO_H
  9. #define MM_IO_H
  10. #define MM_MAX_LINE_LENGTH 1025
  11. #define MatrixMarketBanner "%%MatrixMarket"
  12. #define MM_MAX_TOKEN_LENGTH 64
  13. typedef char MM_typecode[4];
  14. char *mm_typecode_to_str(MM_typecode matcode);
  15. int mm_read_banner(FILE *f, MM_typecode *matcode);
  16. int mm_read_mtx_crd_size(FILE *f, int *M, int *N, int *nz);
  17. int mm_read_mtx_array_size(FILE *f, int *M, int *N);
  18. int mm_write_banner(FILE *f, MM_typecode matcode);
  19. int mm_write_mtx_crd_size(FILE *f, int M, int N, int nz);
  20. int mm_write_mtx_array_size(FILE *f, int M, int N);
  21. /********************* MM_typecode query fucntions ***************************/
  22. #define mm_is_matrix(typecode) ((typecode)[0]=='M')
  23. #define mm_is_sparse(typecode) ((typecode)[1]=='C')
  24. #define mm_is_coordinate(typecode)((typecode)[1]=='C')
  25. #define mm_is_dense(typecode) ((typecode)[1]=='A')
  26. #define mm_is_array(typecode) ((typecode)[1]=='A')
  27. #define mm_is_complex(typecode) ((typecode)[2]=='C')
  28. #define mm_is_real(typecode) ((typecode)[2]=='R')
  29. #define mm_is_pattern(typecode) ((typecode)[2]=='P')
  30. #define mm_is_integer(typecode) ((typecode)[2]=='I')
  31. #define mm_is_symmetric(typecode)((typecode)[3]=='S')
  32. #define mm_is_general(typecode) ((typecode)[3]=='G')
  33. #define mm_is_skew(typecode) ((typecode)[3]=='K')
  34. #define mm_is_hermitian(typecode)((typecode)[3]=='H')
  35. int mm_is_valid(MM_typecode matcode); /* too complex for a macro */
  36. /********************* MM_typecode modify fucntions ***************************/
  37. #define mm_set_matrix(typecode) ((*typecode)[0]='M')
  38. #define mm_set_coordinate(typecode) ((*typecode)[1]='C')
  39. #define mm_set_array(typecode) ((*typecode)[1]='A')
  40. #define mm_set_dense(typecode) mm_set_array(typecode)
  41. #define mm_set_sparse(typecode) mm_set_coordinate(typecode)
  42. #define mm_set_complex(typecode)((*typecode)[2]='C')
  43. #define mm_set_real(typecode) ((*typecode)[2]='R')
  44. #define mm_set_pattern(typecode)((*typecode)[2]='P')
  45. #define mm_set_integer(typecode)((*typecode)[2]='I')
  46. #define mm_set_symmetric(typecode)((*typecode)[3]='S')
  47. #define mm_set_general(typecode)((*typecode)[3]='G')
  48. #define mm_set_skew(typecode) ((*typecode)[3]='K')
  49. #define mm_set_hermitian(typecode)((*typecode)[3]='H')
  50. #define mm_clear_typecode(typecode) ((*typecode)[0]=(*typecode)[1]= \
  51. (*typecode)[2]=' ',(*typecode)[3]='G')
  52. #define mm_initialize_typecode(typecode) mm_clear_typecode(typecode)
  53. /********************* Matrix Market error codes ***************************/
  54. #define MM_COULD_NOT_READ_FILE 11
  55. #define MM_PREMATURE_EOF 12
  56. #define MM_NOT_MTX 13
  57. #define MM_NO_HEADER 14
  58. #define MM_UNSUPPORTED_TYPE 15
  59. #define MM_LINE_TOO_LONG 16
  60. #define MM_COULD_NOT_WRITE_FILE 17
  61. /******************** Matrix Market internal definitions ********************
  62. MM_matrix_typecode: 4-character sequence
  63. ojbect sparse/ data storage
  64. dense type scheme
  65. string position: [0] [1] [2] [3]
  66. Matrix typecode: M(atrix) C(oord) R(eal) G(eneral)
  67. A(array) C(omplex) H(ermitian)
  68. P(attern) S(ymmetric)
  69. I(nteger) K(kew)
  70. ***********************************************************************/
  71. #define MM_MTX_STR "matrix"
  72. #define MM_ARRAY_STR "array"
  73. #define MM_DENSE_STR "array"
  74. #define MM_COORDINATE_STR "coordinate"
  75. #define MM_SPARSE_STR "coordinate"
  76. #define MM_COMPLEX_STR "complex"
  77. #define MM_REAL_STR "real"
  78. #define MM_INT_STR "integer"
  79. #define MM_GENERAL_STR "general"
  80. #define MM_SYMM_STR "symmetric"
  81. #define MM_HERM_STR "hermitian"
  82. #define MM_SKEW_STR "skew-symmetric"
  83. #define MM_PATTERN_STR "pattern"
  84. /* high level routines */
  85. int mm_write_mtx_crd(char fname[], int M, int N, int nz, int I[], int J[],
  86. double val[], MM_typecode matcode);
  87. int mm_read_mtx_crd_data(FILE *f, int M, int N, int nz, int I[], int J[],
  88. double val[], MM_typecode matcode);
  89. int mm_read_mtx_crd_entry(FILE *f, int *I, int *J, double *real, double *img,
  90. MM_typecode matcode);
  91. int mm_read_unsymmetric_sparse(const char *fname, int *M_, int *N_, int *nz_,
  92. double **val_, int **I_, int **J_);
  93. #endif