portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-06-12 20:27:06 +01:00
Code Releases Activity

conversion of Grid tensors to nested std::vector in preparation for tensor serialisation

Browse Source
This commit is contained in:
Antonin Portelli
2018-03-06 19:22:03 +00:00
parent c399c2b44d
commit 485c5db0fe
2 changed files with 51 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
lib/serialisation/BaseIO.h
View File

@ -31,6 +31,7 @@ Author: Guido Cossu <guido.cossu@ed.ac.uk>
#define GRID_SERIALISATION_ABSTRACT_READER_H
#include <type_traits>
#include <Grid/tensors/Tensors.h>
namespace Grid {
// Vector IO utilities ///////////////////////////////////////////////////////
@ -69,6 +70,40 @@ namespace Grid {
return os;
}
// convert Grid scalar tensors to std::vectors
template <typename T, typename V>
void tensorToVec(V &v, const T& t)
{
v = t;
}
template <typename T, typename V>
void tensorToVec(V &v, const iScalar<T>& t)
{
tensorToVec(v, t._internal);
}
template <typename T, typename V, int N>
void tensorToVec(std::vector<V> &v, const iVector<T, N>& t)
{
v.resize(N);
for (unsigned int i = 0; i < N; i++)
{
tensorToVec(v[i], t._internal[i]);
}
}
template <typename T, typename V, int N>
void tensorToVec(std::vector<std::vector<V>> &v, const iMatrix<T, N>& t)
{
v.resize(N, std::vector<V>(N));
for (unsigned int i = 0; i < N; i++)
for (unsigned int j = 0; j < N; j++)
{
tensorToVec(v[i][j], t._internal[i][j]);
}
}
// Vector element trait //////////////////////////////////////////////////////
template <typename T>
struct element