File: bnmatrxw.h

package info (click to toggle)
arpack%2B%2B 2.3-10
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye, forky, sid, trixie
  • size: 4,556 kB
  • sloc: cpp: 16,612; sh: 8,819; ansic: 2,312; makefile: 258
file content (68 lines) | stat: -rw-r--r-- 1,151 bytes parent folder | download | duplicates (8)
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
/*
   ARPACK++ v1.2 2/20/2000
   c++ interface to ARPACK code.

   MODULE BNMatrxW.h
   Class template for the n by n matrix A with entries

          A(i,j) = k*(s(i))*(t(j)-1), if j-nl<= i <= j;
                   k*(t(j))*(s(i)-1), if j < i <= j+nu,

   where s(i) = i/(n+1), t(j) = j/(n+1) and k = 1/(n+1).       

   ARPACK Authors
      Richard Lehoucq
      Danny Sorensen
      Chao Yang
      Dept. of Computational & Applied Mathematics
      Rice University
      Houston, Texas
*/

#ifndef BNMATRXW_H
#define BNMATRXW_H


template<class ARFLOAT>
void MatrixW(int n, int nl, int nu, ARFLOAT* &A)
{

  int     i, j, p;
  ARFLOAT h;

  // Defining h.

  h = 1.0/ARFLOAT(n+1);

  // Creating output vector A.

  A = new ARFLOAT[n*(nl+nu+1)];

  p = 0;

  for (j=1; j<=n; j++) {

    for (i=j-nu; i<=j; i++) {
      if (i<1) {
        A[p++] = 0.0;
      }
      else {
        A[p++] = h*(ARFLOAT(j)*h-1.0)*(ARFLOAT(i)*h);
      }
    }

    for(i=j+1; i<=(j+nl); i++) {
      if (i>n) {
        A[p++] = 0.0;
      }
      else {
        A[p++]  = h*(ARFLOAT(i)*h-1.0)*(ARFLOAT(j)*h);
      }
    }

  }

} // MatrixW.

#endif // BNMATRXW_H