sum_code.h

#include "sum.h"
#include "data.h"
 
 
template<unsigned int Nunmod, class Unmod, unsigned int Nsum, class Sum, bool Magn>
bool RecoSum<Nunmod,Unmod,Nsum,Sum,Magn>::process(RecoData& rd, RecoController& controller) {
  Log<Reco> odinlog(c_label(),"process");
 
  // create new view of input data
  ComplexData<Nsum+Nunmod>& indata=rd.data(Rank<Nsum+Nunmod>());
 
  TinyVector<int,Nsum+Nunmod> inshape(indata.shape());
 
  TinyVector<int,Nsum> sumshape;
  for(int i=0; i<int(Nsum); i++) sumshape(i)=inshape(i);
 
  TinyVector<int,Nunmod> outshape;
  for(int i=0; i<int(Nunmod); i++) outshape(i)=inshape(Nsum+i);
 
  ODINLOG(odinlog,normalDebug) << "indata.shape()/outshape/sumshape=" << indata.shape() << "/" << outshape << "/" << sumshape << STD_endl;
 
  // summed output data with same (padded) shape as input data
  TinyVector<int,Nsum+Nunmod> outshape_padded=1;
  for(unsigned int i=0; i<Nunmod; i++) outshape_padded(Nsum+i)=outshape(i);
 
 
  // create domains for retrieving input data from the correct position
  // initialize unmodified part with outshape
  TinyVector<int,Nsum+Nunmod> lowin=0;
  TinyVector<int,Nsum+Nunmod> uppin=outshape_padded-1;
 
  TinyVector<int,Nsum+Nunmod> lowout=0;
  TinyVector<int,Nsum+Nunmod> uppout=outshape_padded-1;
 
   // output domain, the same for each dataset
  RectDomain<Nsum+Nunmod> outdomain(lowout,uppout);
 
  ComplexData<Nsum+Nunmod> outdata_padded(outshape_padded); outdata_padded=STD_complex(0.0);
 
  for(int isum=0; isum<product(sumshape); isum++) {
 
    // create Nsum-dim index from linear index
    TinyVector<int,Nsum> sumindex=index2extent(sumshape, isum);
 
    // set input inidices
    for(unsigned int i=0; i<Nsum; i++) lowin(i)=uppin(i)=sumindex(i);
 
    // sum data
    RectDomain<Nsum+Nunmod> indomain(lowin,uppin);
 
    if(Magn) outdata_padded(outdomain)+=float2real(cabs(indata(indomain)));
    else     outdata_padded(outdomain)+=indata(indomain);
  }
 
  ComplexData<Nunmod> outdata(outshape);
  outdata_padded.convert_to(outdata); // bring to output shape
  rd.data(Rank<Nunmod>()).reference(outdata);
 
  ODINLOG(odinlog,normalDebug) << " outshape=" << outshape << STD_endl;
 
  modify_coord(rd.coord());
 
  if(Magn) rd.mode=RecoData::real_data;
 
  if(!execute_next_step(rd,controller)) return false;
 
  return true;
}