SolidHarmonicsStorage.h

/*
 * SolidHarmonicsStorage.h
 *
 *  Created on: Nov 20, 2014
 *      Author: tchipevn, rajats
 */
 
#ifndef SOLIDHARMONICSSTORAGE_H_
#define SOLIDHARMONICSSTORAGE_H_
 
#include "utils/mardyn_assert.h"
 
namespace bhfmm {
class SolidHarmonicsStorage;
 
void swap(SolidHarmonicsStorage & s1, SolidHarmonicsStorage & s2);
 
class SolidHarmonicsStorage {
public:
    SolidHarmonicsStorage(int numRows = 0, bool initializeToZero = true);
 
    SolidHarmonicsStorage(const SolidHarmonicsStorage& s);
 
    ~SolidHarmonicsStorage();
 
    friend void swap(SolidHarmonicsStorage& s1, SolidHarmonicsStorage& s2);
 
    SolidHarmonicsStorage& operator =(SolidHarmonicsStorage s);
 
    SolidHarmonicsStorage& operator+=(const SolidHarmonicsStorage& s);
 
    SolidHarmonicsStorage& operator*=(double scalar);
 
    int getNumRows() const;
 
    int getTotalNumValues() const;
 
    void setToZero();
 
    int index(int l, int m) const {
        mardyn_assert(l >= 0);
        mardyn_assert(m >= 0);
        mardyn_assert(l <= _numRows);
        mardyn_assert(m <= l);
        return l * (l + 1) / 2 + m;
    }
 
    double& getValue(int l, int m) {
        return _values[index(l, m)];
    }
 
    double& getValueSequential(int i) {
        return _values[i];
    }
 
    double getValueConst(int l, int m) const {
        return _values[index(l, m)];
    }
 
    double getValueConstSequential(int i) const {
        return _values[i];
    }
 
private:
    int _numRows;
 
    int _totalNumValues;
 
    double* _values;
};
 
SolidHarmonicsStorage operator+(SolidHarmonicsStorage lhs, const SolidHarmonicsStorage & rhs);
 
SolidHarmonicsStorage operator*(double scalar, SolidHarmonicsStorage s);
 
} /* namespace bhfmm */
 
#endif /* SOLIDHARMONICSSTORAGE_H_ */