toeplitz_rshp.c

Go to the documentation of this file.

#include "toeplitz.h"

//r1.1 - Frederic Dauvergne (APC)
//This is the reshaping routines to build the optimal data structure when needed. 
//The index functions find the right index number of the data location for a choosen
//transformation.


int fctid_mat2vect(int i, int id0, int n, int lambda)
{
  int I,J, i_out;
  int distcorrmin= lambda-1;
  int rfirst=id0%n;

  if (i==-1) 
    return (-1);
  

  I = (i+rfirst)%(n+distcorrmin);
  J = (i+rfirst)/(n+distcorrmin);

  if (I<n)
    i_out = I-rfirst+J*n;  
  else
    i_out = -1; //not defined. value is zero.


  return i_out;
}


int fctid_mat2vect_inv(int i, int id0, int n, int lambda)
{
  int I,J, i_out;
  int distcorrmin= lambda-1;
  int rfirst=id0%n;

  if (i==-1)
    i_out = -1; //not defined. value is zero.

  I = (i+rfirst)%(n);
  J = (i+rfirst)/(n);

  i_out = I-rfirst+J*(n+distcorrmin);

  return i_out;
}


int fctid_concatcol(int i, int id0, int n, int m, int l, int lconc, int lambda, int *nocol, int nbcol)
{
  int I,J, i_out;
  int distcorrmin= lambda-1;
  int rfirst=id0%n;

  if (i==-1)
    return (-1);

  if (i>=lconc)
    return (-2);//this indice not define. It shouldn't be used

  I = (i+rfirst)%(n);
  J = (i+rfirst)/(n);

    i_out = I-rfirst+nocol[J]*(n);


  return i_out;
}


int fctid_concatcol_inv(int i, int id0, int n, int m, int l, int lconc, int lambda, int *nocol_inv, int nbcol)
{
  int I,J, i_out;
  int distcorrmin= lambda-1;
  int rfirst=id0%n;

  if (i==-1)
    return (-1);

  if (i>=l)
    return (-2);//this indice not define. It shouldn't be used

  I = (i+rfirst)%(n);
  J = (i+rfirst)/(n);

  if (nocol_inv[J]==(-1))
    i_out = -1;
  else
    i_out = I-rfirst+nocol_inv[J]*(n);


  return i_out;
}



int fctid_vect2nfftblock(int i, int v1_size, int fft_size, int nfft, int lambda)
{

  int I,J, i_out;
  int distcorrmin= lambda-1;

  if (i==-1)
    return (-1);

  I = (i)%(fft_size);
  J = (i)/(fft_size);

  i_out = (I-distcorrmin) + J*(fft_size-2*distcorrmin) ;

  if (i_out<0 || i_out>=v1_size)
    i_out = -1;


  return i_out;
}


int is_needconcat(int *nocol, int nbcol)
{
  int i;
  int ip=nocol[0];
  for(i=1;i<nbcol;i++) {
    if (nocol[i]!=(ip+i)) 
      return 1;
  }


  return 0;
}


int fctid_vect2nfftblock_inv(int i, int v1_size, int fft_size, int nfft, int lambda)
{

  int I,J, i_out;
  int distcorrmin= lambda-1;

  if (i<0 || i>=v1_size)
    return (-2);

  I = (i)%(fft_size-2*distcorrmin);
  J = (i)/(fft_size-2*distcorrmin);

  i_out = (I+distcorrmin) + J*(fft_size) ;

  return i_out;
}


int define_rshp_size(int flag_format_rshp, int fft_size, int nfft, int v1_size, int vedge_size, int *nrshp, int *mrshp, int *lrshp)
{

  if (flag_format_rshp==2) {
    *nrshp=fft_size;
    *mrshp=nfft;
    *lrshp=(*nrshp)*(*mrshp);
  }
  else if (flag_format_rshp==1) {
    *nrshp=v1_size;
    *mrshp=1;
    *lrshp=(*nrshp)*(*mrshp);
  }
  else if (flag_format_rshp==0) { //this case appear only if flag_shortcut_nbcol_eq_1==0
    *nrshp=vedge_size;
    *mrshp=1;
    *lrshp=vedge_size;
  }
  else {//error not a good flag_format_rshp
  }

  return 0;
}


int build_nocol_inv(int *nocol, int nbcol, int m)  //ncol_inv to define as parameters
{
 int i;
 int *nocol_inv;
  nocol_inv = (int *) calloc(m, sizeof(double));

  for(i=0;i<m;i++)
    nocol_inv[i]=-1;
  for(i=0;i<nbcol;i++)
    nocol_inv[nocol[i]]=i;


  return 0;
}


int build_reshape(double *Vin, int *nocol, int nbcol, int lconc, int n, int m, int id0, int l, int lambda, int nfft, double *Vrshp, int nrshp, int mrshp, int lrshp, int flag_format_rshp)
{

  int i;
  int rfirst=id0%n;
  int i_out1, i_out2, i_out3;
  int distcorrmin=lambda-1;

  int v1_size;
  int fft_size;

  int idf = id0+l-1;
  int lconc0;

  FILE *file;
  file = stdout;

  v1_size=lconc+(distcorrmin)*(nbcol-1);
  fft_size = ceil(1.0*v1_size/nfft)+2*distcorrmin;

 //used transformation
if (VERBOSE) {
  fprintf(file, "fctid_concatcol: \t %d\n", (is_needconcat(nocol, nbcol)==1));
  fprintf(file, "fctid_mat2vect: \t %d\n", (nbcol>1)); 
  fprintf(file, "fctid_vect2nfftblock \t %d\n", (nfft>1));
}


  for(i=0;i<lrshp;i++) {

  if (nfft>1)
    i_out1 = fctid_vect2nfftblock( i, v1_size, fft_size, nfft, lambda);
  else
    i_out1 = i;

  if (nbcol>1)
    i_out2 = fctid_mat2vect(i_out1 , rfirst, n, lambda);
  else
    i_out2 = i_out1;

  if (is_needconcat(nocol, nbcol)==1)
    i_out3 = fctid_concatcol(i_out2, id0, n, m, l, lconc, lambda, nocol, nbcol);
  else
    i_out3 = i_out2;


  if (i_out3==-1)
    Vrshp[i]=0;
  else
    Vrshp[i]=Vin[i_out3];

  }//end for


  return 0;
}


int extract_result(double *Vout, int *nocol, int nbcol, int lconc, int n, int m, int id0, int l, int lambda, int nfft, double *Vrshp, int nrshp, int mrshp, int lrshp, int flag_format_rshp)
{  

  int i;
  int rfirst=id0%n;
  int i_out1, i_out2, i_out3;
  int i_in1;
  int distcorrmin=lambda-1;

  int v1_size;
  int fft_size;

  FILE *file;
  file = stdout;

  v1_size=lconc+(distcorrmin)*(nbcol-1);
  fft_size = ceil(1.0*v1_size/nfft)+2*distcorrmin;

 //used transformation
if (VERBOSE) {
  fprintf(file, "fctid_concatcol: \t %d\n", (is_needconcat(nocol, nbcol)==1));
  fprintf(file, "fctid_mat2vect: \t %d\n", (nbcol>1));
  fprintf(file, "fctid_vect2nfftblock \t %d\n", (nfft>1));
}

  int lcol;
  int j,k;

  for(i=0;i<lconc;i++) { 

  if (is_needconcat(nocol, nbcol)==1)
    i_in1=fctid_concatcol(i, id0, n, m, l, lconc, lambda, nocol, nbcol);
  else
    i_in1 = i;

  if (nbcol>1)
    i_out2 = fctid_mat2vect_inv(i , rfirst, n, lambda);
  else
    i_out2 = i_out1;

  if (nfft>1)
    i_out3 = fctid_vect2nfftblock_inv(i_out2, v1_size, fft_size, nfft, lambda);
  else
    i_out3 = i_out2;

  if (i_out3==-1)
    Vout[i]=-1;
  else if (i_out3==-2)
    Vout[i]=-2;
  else
      Vout[i_in1] = Vrshp[i_out3];
  }


  return 0;
}