File: demo.cc

package info (click to toggle)
suitesparse 1%3A7.10.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 254,920 kB
  • sloc: ansic: 1,134,743; cpp: 46,133; makefile: 4,875; fortran: 2,087; java: 1,826; sh: 996; ruby: 725; python: 495; asm: 371; sed: 166; awk: 44
file content (82 lines) | stat: -rw-r--r-- 1,758 bytes parent folder | download | duplicates (2)
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
//------------------------------------------------------------------------------
// SuiteSparse/TestConfig/SPQR/demo.cc
//------------------------------------------------------------------------------

// Copyright (c) 2024, Timothy A. Davis, All Rights Reserved.
// SPDX-License-Identifier: BSD-3-clause

//------------------------------------------------------------------------------

// TestConfig program

#include <iostream>

#include "SuiteSparseQR_C.h"

int main (void)
{
    #define N 2
    #define NNZ 4
    int64_t n = N;
    int64_t nzmax = NNZ;
    int64_t Ap[N+1];
    int64_t Ai[NNZ];
    double Ax[NNZ];
    Ap [0] = 0;
    Ap [1] = 2;
    Ap [2] = NNZ;
    Ai [0] = 0;
    Ai [1] = 1;
    Ai [2] = 0;
    Ai [3] = 1;
    Ax [0] = 11.0;
    Ax [1] = 21.0;
    Ax [2] = 12.0;
    Ax [3] = 22.0;

    double b [N] = {8., 45.};

    cholmod_sparse *A, A_struct;
    cholmod_dense *B, B_struct;
    cholmod_dense *X;

    // make a shallow CHOLMOD copy of A
    A = &A_struct;
    A->nrow = n;
    A->ncol = n;
    A->p = Ap;
    A->i = Ai;
    A->x = Ax;
    A->z = nullptr;
    A->nzmax = NNZ;
    A->packed = true;
    A->sorted = true;
    A->nz = nullptr;
    A->itype = CHOLMOD_LONG;
    A->dtype = CHOLMOD_DOUBLE;
    A->xtype = CHOLMOD_REAL;
    A->stype = 0;

    // make a shallow CHOLMOD copy of b
    B = &B_struct;
    B->nrow = n;
    B->ncol = 1;
    B->x = b;
    B->z = nullptr;
    B->d = n;
    B->nzmax = n;
    B->dtype = CHOLMOD_DOUBLE;
    B->xtype = CHOLMOD_REAL;

    cholmod_common cc;
    cholmod_l_start (&cc);

    X = SuiteSparseQR_C_backslash_default (A, B, &cc);
    cc.print = 5 ;
    cholmod_l_print_dense (X, "X from QR", &cc);
    cholmod_l_free_dense (&X, &cc) ;

    cholmod_l_finish (&cc);

    return 0;
}