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
|
// Author: David Alexander
#pragma once
#include <xmmintrin.h>
#include <boost/numeric/ublas/matrix.hpp>
#include <utility>
#include <vector>
#include <ConsensusCore/Interval.hpp>
#include <ConsensusCore/LFloat.hpp>
#include <ConsensusCore/Matrix/AbstractMatrix.hpp>
#include <ConsensusCore/Types.hpp>
#include <ConsensusCore/Utils.hpp>
namespace ConsensusCore {
#ifndef SWIG
// SWIG gags on this
using boost::numeric::ublas::matrix;
using boost::numeric::ublas::column_major;
#endif // SWIG
typedef matrix<lfloat, column_major> boost_dense_matrix;
class DenseMatrix : public AbstractMatrix, private boost_dense_matrix
{
public: // Constructor, destructor
DenseMatrix(int rows, int cols);
~DenseMatrix();
public: // Nullability
static const DenseMatrix& Null();
bool IsNull() const;
public: // Size information
int Rows() const;
int Columns() const;
public: // Information about entries filled by column
void StartEditingColumn(int j, int hintBegin, int hintEnd);
void FinishEditingColumn(int j, int usedBegin, int usedEnd);
Interval UsedRowRange(int j) const;
bool IsColumnEmpty(int j) const;
int UsedEntries() const;
int AllocatedEntries() const; // an entry may be stored but not filled
public: // Accessors
//
// Note there is no non-const operator()(i, j) provided---for good
// resaon. We need to have set'ing go through a distinct code path
// from get'ing, so that we can have safety checks in the set code
// path. This is not what you get with operator()(i,j) const +
// non-const, since m(i,j) uses the operator according to the
// constness/non-constness of m, not based on whether the operation
// is a set or get. To my knowledge, there is no way to do this in
// C++.
//
const float& operator()(int i, int j) const;
bool IsAllocated(int i, int j) const;
float Get(int i, int j) const;
void Set(int i, int j, float v);
void ClearColumn(int j);
public: // SSE accessors, which access 4 successive entries in a column
__m128 Get4(int i, int j) const;
void Set4(int i, int j, __m128 v);
public:
// Method SWIG clients can use to get a native matrix (e.g. Numpy)
// mat must be filled as a ROW major matrix
void ToHostMatrix(float** mat, int* rows, int* cols) const;
private:
std::vector<Interval> usedRanges_;
int columnBeingEdited_;
void CheckInvariants(int column) const;
};
}
#include <ConsensusCore/Matrix/DenseMatrix-inl.hpp>
|
