matz_svd_routcos

Fonction de calcul
fr

##
matz_svd

### Contenu du fichier

    # include "scicos_block4.h"
    # include "../machine.h"
    #include <stdio.h>
    extern int C2F(zgesvd)();
    extern int C2F(dlaset)();

    #if WIN32
    #define NULL    0
    #endif

    #ifndef min
    #define min(a,b) ((a) <= (b) ? (a) : (b))
    #endif

    #ifndef max
    #define max(a,b) ((a) >= (b) ? (a) : (b))
    #endif

    typedef struct
    {         double *l0;
    	  double *LA;
    	  double *LU;
              double *LSV;
              double *LVT;
              double *dwork;
              double *rwork;
    } mat_sdv_struct ;
    void matz_svd(scicos_block *block,int flag)
    {
     double *ur,*ui;
     double *y1r,*y2r,*y3r;
     double *y1i,*y3i;
     int nu,mu;
     int info;
     int i,j,ij,ji,ii,lwork,rw;
     mat_sdv_struct *ptr;
     mu=GetInPortRows(block,1);
     nu=GetInPortCols(block,1);
     ur=GetRealInPortPtrs(block,1);
     ui=GetImagInPortPtrs(block,1);
     y1r=GetRealOutPortPtrs(block,1);
     y2r=GetRealOutPortPtrs(block,2);
     y3r=GetRealOutPortPtrs(block,3);
     y1i=GetImagOutPortPtrs(block,1);
     //y2i=GetImagOutPortPtrs(block,2);
     y3i=GetImagOutPortPtrs(block,3);
     lwork=max(3*min(mu,nu)+max(mu,nu),5*min(mu,nu)-4);
     rw=5*min(mu,nu);
                 /*init : initialization*/
    if (flag==4)
       {if((*(block->work)=(mat_sdv_struct*) scicos_malloc(sizeof(mat_sdv_struct)))==NULL)
    	{set_block_error(-16);
    	 return;}
        ptr=*(block->work);
        if((ptr->l0=(double*) scicos_malloc(sizeof(double)))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->LA=(double*) scicos_malloc(sizeof(double)*(2*mu*nu)))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->LU=(double*) scicos_malloc(sizeof(double)*(2*mu*mu)))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->LA);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->LSV=(double*) scicos_malloc(sizeof(double)*(min(mu,nu))))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->LU);
    	 scicos_free(ptr->LA);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->LVT=(double*) scicos_malloc(sizeof(double)*(2*nu*nu)))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->LSV);
    	 scicos_free(ptr->LU);
    	 scicos_free(ptr->LA);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->dwork=(double*) scicos_malloc(sizeof(double)*2*lwork))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->LVT);
    	 scicos_free(ptr->LSV);
    	 scicos_free(ptr->LU);
    	 scicos_free(ptr->LA);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
        if((ptr->rwork=(double*) scicos_malloc(sizeof(double)*2*rw))==NULL)
    	{set_block_error(-16);
    	 scicos_free(ptr->dwork);
    	 scicos_free(ptr->LVT);
    	 scicos_free(ptr->LSV);
    	 scicos_free(ptr->LU);
    	 scicos_free(ptr->LA);
    	 scicos_free(ptr->l0);
    	 scicos_free(ptr);
    	 return;}
       }

           /* Terminaison */
    else if (flag==5)
       {ptr=*(block->work);
        if((ptr->rwork)!=NULL){
        	scicos_free(ptr->l0);
        	scicos_free(ptr->LA);
        	scicos_free(ptr->LU);
        	scicos_free(ptr->LSV);
        	scicos_free(ptr->LVT);
        	scicos_free(ptr->dwork);
        	scicos_free(ptr);
        	return;}
       }

    else
       {
        ptr=*(block->work);
        for (i=0;i<(mu*nu);i++)
    	{ptr->LA[2*i]=ur[i];
    	 ptr->LA[2*i+1]=ui[i];}
        C2F(zgesvd)("A","A",&mu,&nu,ptr->LA,&mu,ptr->LSV,ptr->LU,&mu,ptr->LVT,&nu,ptr->dwork,&lwork,ptr->rwork,&info);
         if (info !=0)
           {if (flag!=6)
       	{set_block_error(-7);
            return;}}

        *(ptr->l0)=0;
         C2F(dlaset)("F",&mu,&nu,ptr->l0,ptr->l0,y2r,&mu);
        for (i=0;i<min(mu,nu);i++)
    	{ii=i+i*mu;
    	 *(y2r+ii)=*(ptr->LSV+i);}
        for (j=0;j<nu;j++)
    	{for (i=j;i<nu;i++)
    		{ij=i+j*nu;
    	 	 ji=j+i*nu;
    		 *(y3r+ij)=*(ptr->LVT+2*ji);
    		 *(y3r+ji)=*(ptr->LVT+2*ij);
    		 *(y3i+ij)=-(*(ptr->LVT+2*ji+1));
    		 *(y3i+ji)=-(*(ptr->LVT+2*ij+1));}
    	}
        for (i=0;i<mu*mu;i++)
    	{*(y1r+i)=*(ptr->LU+2*i);
    	 *(y1i+i)=*(ptr->LU+2*i+1);}
       }
    }