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
|
// -*- mode: C++; c-indent-level: 4; c-basic-offset: 4; indent-tabs-mode: nil; -*-
//
// upper_tri.h: Rcpp R/C++ interface class library -- upper.tri
//
// Copyright (C) 2010 - 2017 Dirk Eddelbuettel and Romain Francois
// Copyright (C) 2017 Dirk Eddelbuettel, Romain Francois, and Nathan Russell
//
// This file is part of Rcpp.
//
// Rcpp is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 2 of the License, or
// (at your option) any later version.
//
// Rcpp is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Rcpp. If not, see <http://www.gnu.org/licenses/>.
#ifndef Rcpp__sugar__upper_tri_h
#define Rcpp__sugar__upper_tri_h
namespace Rcpp {
namespace sugar {
template <int RTYPE, bool NA, typename T>
class UpperTri : public MatrixBase<LGLSXP, false, UpperTri<RTYPE, NA, T> > {
public:
typedef Rcpp::MatrixBase<RTYPE, NA, T> MatBase;
UpperTri(const T& lhs, bool diag)
: nr(lhs.nrow()),
nc(lhs.ncol()),
getter(diag ? (&UpperTri::get_diag_true) : (&UpperTri::get_diag_false))
{}
inline int operator()(int i, int j) const { return get(i, j); }
inline R_xlen_t size() const { return static_cast<R_xlen_t>(nr) * nc; }
inline int nrow() const { return nr; }
inline int ncol() const { return nc; }
private:
typedef bool (UpperTri::*Method)(int, int) const;
int nr, nc;
Method getter;
inline bool get_diag_true(int i, int j) const { return i <= j; }
inline bool get_diag_false(int i, int j) const { return i < j; }
inline bool get(int i, int j) const { return (this->*getter)(i, j); }
};
} // sugar
template <int RTYPE, bool NA, typename T>
inline sugar::UpperTri<RTYPE, NA, T>
upper_tri(const Rcpp::MatrixBase<RTYPE, NA, T>& lhs, bool diag = false) {
return sugar::UpperTri<RTYPE, NA, T>(lhs, diag);
}
} // Rcpp
#endif // Rcpp__sugar__upper_tri_h
|
