Homec4science
Diffusion lammps (efficient_neuronet) Browse

lammps/lib/poems/colmatmap.cpp
7d77aea42d30efficient_neuronet

Last Change
Raw File

colmatmap.cpp
View Options

/*
*_________________________________________________________________________*
* POEMS: PARALLELIZABLE OPEN SOURCE EFFICIENT MULTIBODY SOFTWARE *
* DESCRIPTION: SEE READ-ME *
* FILE NAME: colmatmap.cpp *
* AUTHORS: See Author List *
* GRANTS: See Grants List *
* COPYRIGHT: (C) 2005 by Authors as listed in Author's List *
* LICENSE: Please see License Agreement *
* DOWNLOAD: Free at www.rpi.edu/~anderk5 *
* ADMINISTRATOR: Prof. Kurt Anderson *
* Computational Dynamics Lab *
* Rensselaer Polytechnic Institute *
* 110 8th St. Troy NY 12180 *
* CONTACT: anderk5@rpi.edu *
*_________________________________________________________________________*/
#include "colmatmap.h"
#include "colmatrix.h"
#include <iostream>
#include <cstdlib>
using namespace std;
ColMatMap::ColMatMap(){
numrows = 0;
elements = 0;
}
ColMatMap::~ColMatMap(){
delete [] elements;
}
ColMatMap::ColMatMap(const ColMatMap& A){ // copy constructor
numrows = 0;
elements = 0;
Dim(A.numrows);
for(int i=0;i<numrows;i++)
elements[i] = A.elements[i];
}
ColMatMap::ColMatMap(ColMatrix& A){ // copy constructor
numrows = 0;
elements = 0;
Dim(A.GetNumRows());
for(int i=0;i<numrows;i++)
elements[i] = A.GetElementPointer(i);
}
/*
ColMatrix::ColMatrix(const VirtualMatrix& A){ // copy constructor
if( A.GetNumCols() != 1 ){
cerr << "error trying to write a 2D matrix to a collumn" << endl;
exit(1);
}
numrows = 0;
elements = 0;
Dim(A.GetNumRows());
for(int i=0;i<numrows;i++)
elements[i] = A.BasicGet(i,0);
}
*/
ColMatMap::ColMatMap(int m){ // size constructor
numrows = 0;
elements = 0;
Dim(m);
}
void ColMatMap::Dim(int m){
delete [] elements;
numrows = m;
elements = new double* [m];
}
void ColMatMap::Const(double value){
for(int i=0;i<numrows;i++)
*(elements[i]) = value;
}
MatrixType ColMatMap::GetType() const{
return COLMATMAP;
}
ostream& ColMatMap::WriteData(ostream& c) const{ //output
c << numrows << ' ';
for(int i=0;i<numrows;i++)
c << *(elements[i]) << ' ';
return c;
}
double& ColMatMap::operator_1int (int i){ // array access
if((i>numrows) || (i<1)){
cerr << "matrix index invalid in operator ()" << endl;
exit(1);
}
return *(elements[i-1]);
}
double ColMatMap::Get_1int(int i) const{
if((i>numrows) || (i<1)){
cerr << "matrix index exceeded in Get" << endl;
exit(1);
}
return *(elements[i-1]);
}
void ColMatMap::Set_1int(int i, double value){
if((i>numrows) || (i<1)){
cerr << "matrix index exceeded in Set" << endl;
exit(1);
}
*(elements[i-1]) = value;
}
double ColMatMap::BasicGet_1int(int i) const{
return *(elements[i]);
}
void ColMatMap::SetElementPointer(int i, double* p){
elements[i] = p;
}
double* ColMatMap::GetElementPointer(int i){
return elements[i];
}
void ColMatMap::BasicSet_1int(int i, double value){
*(elements[i]) = value;
}
void ColMatMap::BasicIncrement_1int(int i, double value){
*(elements[i]) += value;
}
void ColMatMap::AssignVM(const VirtualMatrix& A){
if(A.GetNumRows() != numrows){
cerr << "dimension mismatch in ColMatMap assignment" << endl;
exit(0);
}
if( A.GetNumCols() != 1 ){
cerr << "error trying to write a 2D matrix to a collumn" << endl;
exit(1);
}
for(int i=0;i<numrows;i++)
*(elements[i]) = A.BasicGet(i,0);
}
ColMatMap& ColMatMap::operator=(const ColMatMap& A){ // assignment operator
if(A.numrows != numrows){
cerr << "dimension mismatch in ColMatMap assignment" << endl;
exit(0);
}
for(int i=0;i<numrows;i++)
*(elements[i]) = *(A.elements[i]);
return *this;
}
ColMatMap& ColMatMap::operator=(const ColMatrix& A){ // assignment operator
if(A.GetNumRows() != numrows){
cerr << "dimension mismatch in ColMatMap assignment" << endl;
exit(0);
}
for(int i=0;i<numrows;i++)
*(elements[i]) = A.BasicGet(i);
return *this;
}
ColMatMap& ColMatMap::operator=(const VirtualColMatrix& A){ // overloaded =
if(A.GetNumRows() != numrows){
cerr << "dimension mismatch in ColMatMap assignment" << endl;
exit(0);
}
for(int i=0;i<numrows;i++)
*(elements[i]) = A.BasicGet(i);
return *this;
}
ColMatMap& ColMatMap::operator=(const VirtualMatrix& A){ // overloaded =
if(A.GetNumRows() != numrows){
cerr << "dimension mismatch in ColMatMap assignment" << endl;
exit(0);
}
if( A.GetNumCols() != 1 ){
cerr << "error trying to write a 2D matrix to a collumn" << endl;
exit(1);
}
for(int i=0;i<numrows;i++)
*(elements[i]) = A.BasicGet(i,0);
return *this;
}
ColMatMap& ColMatMap::operator*=(double b){
for(int i=0;i<numrows;i++)
*(elements[i]) *= b;
return *this;
}
c4science · Help