00001
00013 #ifndef CHMAT_H
00014 #define CHMAT_H
00015
00016
00017 #include "libDC.h"
00018
00024 class chmat : public sqmat {
00025 protected:
00027 mat Ch;
00028 public:
00029
00030 void opupdt ( const vec &v, double w );
00031 mat to_mat() const;
00032 void mult_sym ( const mat &C );
00033 void mult_sym ( const mat &C , chmat &U ) const;
00034 void mult_sym_t ( const mat &C );
00035 void mult_sym_t ( const mat &C, chmat &U ) const;
00036 double logdet() const;
00037 vec sqrt_mult ( const vec &v ) const;
00038 double qform ( const vec &v ) const;
00039 double invqform ( const vec &v ) const;
00040 void clear();
00042 void add ( const chmat &A2, double w=1.0 ) {};
00044 void inv ( chmat &Inv ) const { ( Inv.Ch ) = itpp::inv ( Ch ).T();};
00045 ;
00046
00047
00049 virtual ~chmat() {};
00051 chmat ( const int dim0 ) : sqmat ( dim0 ),Ch ( dim0,dim0 ) {};
00053 chmat ( const vec &v) : sqmat ( v.length() ),Ch ( diag(sqrt(v)) ) {};
00055 chmat ( const chmat &Ch0) : sqmat ( Ch0.dim),Ch ( Ch0.dim,Ch0.dim ) {Ch=Ch0.Ch;};
00057 chmat ( const mat &M ) : sqmat ( M.rows() ),Ch ( M.rows(),M.cols() ) {
00058 mat Q;
00059 it_assert_debug ( M.rows() ==M.cols(),"chmat:: input matrix must be square!" );
00060 Ch=chol ( M );
00061 };
00063 chmat ( const chmat &M, const ivec &perm ):sqmat(M.rows()){it_error("not implemneted");};
00065 mat & _Ch() {return Ch;}
00067 void setD ( const vec &nD ) {Ch=diag ( sqrt(nD) );}
00069 void setD ( const vec &nD, int i ) {for ( int j=i;j<nD.length();j++ ) {Ch( j,j ) =sqrt(nD ( j-i ));}}
00070
00072 chmat& operator += ( const chmat &A2 );
00073 chmat& operator -= ( const chmat &A2 ) ;
00074 chmat& operator * ( const chmat &A2 );
00075 chmat& operator * ( const double &d ){Ch*d; return *this;};
00076 chmat& operator = ( const chmat &A2 ){Ch=A2.Ch;dim=A2.dim;return *this;}
00077 };
00078
00079
00082 inline chmat& chmat::operator += ( const chmat &A2 ) {this->add ( A2 );return *this;}
00084 inline chmat& chmat::operator -= ( const chmat &A2 ) {this->add ( A2,-1.0 );return *this;}
00085
00086 #endif // CHMAT_H