Actual source code: ex131.c
2: static char help[] = "Tests MatMult() on MatLoad() matrix \n\n";
4: #include <petscmat.h>
6: int main(int argc, char **args)
7: {
8: Mat A;
9: Vec x, b;
10: PetscViewer fd; /* viewer */
11: char file[PETSC_MAX_PATH_LEN]; /* input file name */
12: PetscBool flg;
14: PetscFunctionBeginUser;
15: PetscCall(PetscInitialize(&argc, &args, (char *)0, help));
16: /* Determine file from which we read the matrix A */
17: PetscCall(PetscOptionsGetString(NULL, NULL, "-f", file, sizeof(file), &flg));
18: PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must indicate binary file with the -f option");
20: /* Load matrix A */
21: PetscCall(PetscViewerBinaryOpen(PETSC_COMM_WORLD, file, FILE_MODE_READ, &fd));
22: PetscCall(MatCreate(PETSC_COMM_WORLD, &A));
23: PetscCall(MatLoad(A, fd));
24: flg = PETSC_FALSE;
25: PetscCall(VecCreate(PETSC_COMM_WORLD, &x));
26: PetscCall(PetscOptionsGetString(NULL, NULL, "-vec", file, sizeof(file), &flg));
27: if (flg) {
28: if (file[0] == '0') {
29: PetscInt m;
30: PetscScalar one = 1.0;
31: PetscCall(PetscInfo(0, "Using vector of ones for RHS\n"));
32: PetscCall(MatGetLocalSize(A, &m, NULL));
33: PetscCall(VecSetSizes(x, m, PETSC_DECIDE));
34: PetscCall(VecSetFromOptions(x));
35: PetscCall(VecSet(x, one));
36: }
37: } else {
38: PetscCall(VecLoad(x, fd));
39: PetscCall(PetscViewerDestroy(&fd));
40: }
41: PetscCall(VecDuplicate(x, &b));
42: PetscCall(MatMult(A, x, b));
44: /* Print (for testing only) */
45: PetscCall(MatView(A, 0));
46: PetscCall(VecView(b, 0));
47: /* Free data structures */
48: PetscCall(MatDestroy(&A));
49: PetscCall(VecDestroy(&x));
50: PetscCall(VecDestroy(&b));
51: PetscCall(PetscFinalize());
52: return 0;
53: }