working on tensor class

This commit is contained in:
Jiri Pittner 2024-04-02 17:55:07 +02:00
parent 2194a785da
commit 2f48effddc
2 changed files with 47 additions and 24 deletions

View File

@ -16,25 +16,27 @@
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#include <iostream>
#include "tensor.h"
#include "laerror.h"
#include <iostream>
#include "qsort.h"
#include "miscfunc.h"
namespace LA {
template<typename T>
LA_largeindex Tensor<T>::index(const SUPERINDEX &I)
LA_largeindex Tensor<T>::index(int *sign, const SUPERINDEX &I)
{
//check index structure and ranges
#ifndef DEBUG
#ifdef DEBUG
if(I.size()!=shape.size()) laerror("mismatch in the number of tensor index groups");
for(int i=0; i<I.size; ++i)
for(int i=0; i<I.size(); ++i)
{
if(shape[i].number!=I[i].size()) {std::cerr<<"error in index group no. "<<i<<std::endl; laerror("mismatch in the size of tensor index group");}
for(int j=0; j<shape[i].number; ++j)
{
if(I[i][j] <shape[i].offset || I[i][j] >= shape[i].offset+shape[i].size())
if(I[i][j] <shape[i].offset || I[i][j] >= shape[i].offset+shape[i].range)
{
std::cerr<<"error in index group no. "<<i<<" index no. "<<j<<std::endl;
laerror("tensor index out of range");

View File

@ -35,6 +35,18 @@
namespace LA {
template<typename T>
class Signedpointer
{
T *ptr;
int sgn;
public:
Signedpointer(const T *p, int s) : ptr(p),sgn(s) {};
//@@@@@@operations on singedpointer as LHS of the non-const tensor.operator() expressions
};
typedef int LA_index;
typedef int LA_largeindex;
@ -42,7 +54,7 @@ typedef class indexgroup {
int number; //number of indices
int symmetry; //-1 0 or 1
LA_index offset; //indices start at
LA_index size; //indices span this range
LA_index range; //indices span this range
} INDEXGROUP;
typedef NRVec<LA_index> FLATINDEX; //all indices but in a single vector
@ -51,26 +63,31 @@ typedef NRVec<NRVec<LA_index> > SUPERINDEX; //all indices in the INDEXGROUP stru
template<typename T>
class Tensor {
//essential data
NRVec<indexgroup> shape;
NRVec<T> data;
//redundant data to facilitate efficient indexing
NRVec<LA_largeindex> cumsizes; //cumulative sizes of symmetry index groups
private:
LA_largeindex index(const SUPERINDEX &I); //map the tensor indices to the position in data
LA_largeindex index(const FLATINDEX &I); //map the tensor indices to the position in data
LA_largeindex vindex(int i1,va_list args); //map list of indices to the position in data @@@must call va_end
LA_largeindex index(int *sign, const SUPERINDEX &I); //map the tensor indices to the position in data
LA_largeindex index(int *sign, const FLATINDEX &I); //map the tensor indices to the position in data
LA_largeindex vindex(int *sign, int i1, va_list args); //map list of indices to the position in data @@@must call va_end
public:
Tensor() {};
Tensor(const NRVec<indexgroup> &s) : shape(s), data((int)size()) {data.clear();};
int rank() const; //is computed from shape
LA_largeindex size() const; //expensive, is computed from shape
Tensor(const NRVec<indexgroup> &s) : shape(s), data((int)getsize()) {data.clear();};
int getrank() const; //is computed from shape
LA_largeindex getsize(); //set redundant data and return total size
void copyonwrite() {shape.copyonwrite(); data.copyonwrite();};
inline T& operator()(const SUPERINDEX &I) {return data[index(I)];};
inline const T& operator()(const SUPERINDEX &I) const {return data[index(I)];};
inline T& operator()(const FLATINDEX &I) {return data[index(I)];};
inline const T& operator()(const FLATINDEX &I) const {return data[index(I)];};
inline T& operator()(int i1...) {va_list args; va_start(args,i1); return data[vindex(i1,args)];};
inline const T& operator()(int i1...) const {va_list args; va_start(args,i1); return data[vindex(i1,args)];};
inline Signedpointer<T> operator()(const SUPERINDEX &I) {int sign; LA_largeindex i=index(&sign,I); return Signedpointer<T>(&data[i],sign);};
inline const T& operator()(const SUPERINDEX &I) const {int sign; LA_largeindex i=index(&sign,I); if(sign==0) return 0; return sign>0 ?data[i] : -data[i];};
inline Signedpointer<T> operator()(const FLATINDEX &I) {int sign; LA_largeindex i=index(&sign,I); return Signedpointer<T>(&data[i],sign);};
inline const T& operator()(const FLATINDEX &I) const {int sign; LA_largeindex i=index(&sign,I); if(sign==0) return 0; return sign>0 ?data[i] : -data[i];};
inline Signedpointer<T> operator()(int i1...) {va_list args; int sign; LA_largeindex i; va_start(args,i1); i= vindex(&sign, i1,args); return Signedpointer<T>(&data[i],sign); };
inline const T& operator()(int i1...) const {va_list args; ; int sign; LA_largeindex i; va_start(args,i1); i= vindex(&sign, i1,args); if(sign==0) return 0; return sign>0 ?data[i] : -data[i];};
//NOTE: for sign==0 data[i] can be undefined pointer, avoid dereferencing it
//@@@TODO - unwinding to full size in a specified index
//@@@TODO - contractions - basic and efficient
@ -81,7 +98,7 @@ public:
template<typename T>
int Tensor<T>:: rank() const
int Tensor<T>:: getrank() const
{
int r=0;
for(int i=0; i<shape.size(); ++i)
@ -92,24 +109,28 @@ for(int i=0; i<shape.size(); ++i)
return r;
}
template<typename T>
LA_largeindex Tensor<T>::size() const
LA_largeindex Tensor<T>::getsize()
{
cumsizes.resize(shape.size());
LA_largeindex s=1;
for(int i=0; i<shape.size(); ++i)
{
if(shape[i].number==0) laerror("empty index group");
if(shape[i].size==0) return 0;
if(shape[i].range==0) return 0;
cumsizes[i]=s;
switch(shape[i].symmetry)
{
case 0:
s *= longpow(shape[i].size,shape[i].number);
s *= longpow(shape[i].range,shape[i].number);
break;
case 1:
s *= binom(shape[i].size+shape[i].number-1,shape[i].number);
s *= simplicial(shape[i].number,shape[i].range);
break;
case -1:
s *= binom(shape[i].size,shape[i].number);
s *= simplicial(shape[i].number,shape[i].range-shape[i].number+1);
break;
default:
laerror("illegal index group symmetry");