Commit 9c637adc authored by Alvarez, Gonzalo's avatar Alvarez, Gonzalo
Browse files

Matrix::rotate() added

parent 6cc1abc9

src/CrsMatrix.h

+2 −0
Original line number Diff line number Diff line
@@ -854,6 +854,8 @@ externalProduct(CrsMatrix<T>& B, 
	// -------------------------------------------------

	std::vector<int> nnz_B_row( nrow_B );



	assert(nrow_A*nrow_eye <= permutationFull.size());



	for( SizeType ia=0; ia < nrow_A; ia++) {



		SizeType nnz_row = A.getRowPtr(ia+1) - A.getRowPtr(ia);

src/Matrix.h

+41 −0
Original line number Diff line number Diff line
@@ -30,6 +30,7 @@ Please see full open source license included in file LICENSE. 
#include "Mpi.h"

#include "Io/IoSerializerStub.h"

#include <fstream>

#include "BLAS.h"



namespace PsimagLite {
@@ -794,6 +795,46 @@ bool isZero(const Matrix<T>& m) 
	return true;

}



template<typename T>

void rotate(Matrix<T>& m, const Matrix<T>& transform)

{

	Matrix<T> C(transform.cols(), m.cols());



	// C = transform^\dagger * m

	psimag::BLAS::GEMM('C',

	                   'N',

	                   transform.cols(),

	                   m.cols(),

	                   m.rows(),

	                   1.0,

	                   &(transform(0, 0)),

	                   transform.rows(),

	                   &(m(0, 0)),

	                   m.rows(),

	                   0.0,

	                   &(C(0, 0)),

	                   C.rows());





	// m = C * transform

	m.clear();

	m.resize(C.rows(), transform.cols());

	psimag::BLAS::GEMM('N',

	                   'N',

	                   C.rows(),

	                   transform.cols(),

	                   transform.rows(),

	                   1.0,

	                   &(C(0, 0)),

	                   C.rows(),

	                   &(transform(0, 0)),

	                   transform.rows(),

	                   0.0,

	                   &(m(0, 0)),

	                   m.rows());



}



// closures start



template<typename T1,typename T2>