| 1 | /* | 
|---|
| 2 | * MatrixContainer.hpp | 
|---|
| 3 | * | 
|---|
| 4 | *  Created on: Sep 15, 2011 | 
|---|
| 5 | *      Author: heber | 
|---|
| 6 | */ | 
|---|
| 7 |  | 
|---|
| 8 | #ifndef MATRIXCONTAINER_HPP_ | 
|---|
| 9 | #define MATRIXCONTAINER_HPP_ | 
|---|
| 10 |  | 
|---|
| 11 | // include config.h | 
|---|
| 12 | #ifdef HAVE_CONFIG_H | 
|---|
| 13 | #include <config.h> | 
|---|
| 14 | #endif | 
|---|
| 15 |  | 
|---|
| 16 | #include <vector> | 
|---|
| 17 |  | 
|---|
| 18 | #include "boost/serialization/vector.hpp" | 
|---|
| 19 |  | 
|---|
| 20 | /** Contains a sequence of matrices parsed from file. | 
|---|
| 21 | * | 
|---|
| 22 | */ | 
|---|
| 23 | class MatrixContainer { | 
|---|
| 24 | public: | 
|---|
| 25 | typedef std::vector<std::vector<double> > MatrixArray; | 
|---|
| 26 | typedef std::vector<MatrixArray> ArrayOfMatrices; | 
|---|
| 27 | typedef std::vector<std::string> StringVector; | 
|---|
| 28 | typedef std::vector<int> IntVector; | 
|---|
| 29 | typedef std::vector<IntVector> VectorOfIntVectors; | 
|---|
| 30 |  | 
|---|
| 31 | ArrayOfMatrices Matrix; | 
|---|
| 32 | VectorOfIntVectors Indices; | 
|---|
| 33 | StringVector Header; | 
|---|
| 34 | int MatrixCounter; | 
|---|
| 35 | IntVector RowCounter; | 
|---|
| 36 | IntVector ColumnCounter; | 
|---|
| 37 |  | 
|---|
| 38 | MatrixContainer(); | 
|---|
| 39 | virtual ~MatrixContainer(); | 
|---|
| 40 |  | 
|---|
| 41 | bool InitialiseIndices(class MatrixContainer *Matrix = NULL); | 
|---|
| 42 | bool ParseMatrix(std::istream &input, int skiplines, int skipcolumns, size_t MatrixNr); | 
|---|
| 43 | bool AddMatrix(const std::string &header, const MatrixArray &matrix, size_t MatrixNr); | 
|---|
| 44 | virtual bool ParseFragmentMatrix(const std::string name, const std::string prefix, std::string suffix, int skiplines, int skipcolumns); | 
|---|
| 45 | bool AllocateMatrix(StringVector GivenHeader, int MCounter, IntVector RCounter, IntVector CCounter); | 
|---|
| 46 | bool ResetMatrix(); | 
|---|
| 47 | double FindMinValue(); | 
|---|
| 48 | double FindMaxValue(); | 
|---|
| 49 | bool SetLastMatrix(double value, int skipcolumns); | 
|---|
| 50 | bool SetLastMatrix(const MatrixArray &values, int skipcolumns); | 
|---|
| 51 | //bool ParseIndices(); | 
|---|
| 52 | //bool SumSubValues(); | 
|---|
| 53 | bool SumSubManyBodyTerms(class MatrixContainer &Matrix, class KeySetsContainer &KeySets, int Order); | 
|---|
| 54 | bool WriteTotalFragments(const std::string name, const std::string prefix); | 
|---|
| 55 | bool WriteLastMatrix(const std::string name, const std::string prefix, const std::string suffix); | 
|---|
| 56 |  | 
|---|
| 57 | bool operator==(const MatrixContainer &other) const; | 
|---|
| 58 |  | 
|---|
| 59 | bool operator!=(const MatrixContainer &other) const { | 
|---|
| 60 | return !(*this == other); | 
|---|
| 61 | } | 
|---|
| 62 |  | 
|---|
| 63 | private: | 
|---|
| 64 | friend class boost::serialization::access; | 
|---|
| 65 | // serialization | 
|---|
| 66 | template<class Archive> | 
|---|
| 67 | void serialize(Archive & ar, const unsigned int version) | 
|---|
| 68 | { | 
|---|
| 69 | ar & Matrix; | 
|---|
| 70 | ar & Indices; | 
|---|
| 71 | ar & Header; | 
|---|
| 72 | ar & MatrixCounter; | 
|---|
| 73 | ar & RowCounter; | 
|---|
| 74 | ar & ColumnCounter; | 
|---|
| 75 | } | 
|---|
| 76 | }; | 
|---|
| 77 |  | 
|---|
| 78 | std::ostream & operator << (std::ostream &ost, const MatrixContainer &m); | 
|---|
| 79 |  | 
|---|
| 80 | #endif /* MATRIXCONTAINER_HPP_ */ | 
|---|