From c854bf8109ddce134fc0e6fc5c27e7e648bc7734 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Sat, 19 May 2018 17:16:50 +0000 Subject: [PATCH] Merge branch 'stretching-devel' of ssh://git.penic.eu:29418/trisurf-ng into stretching-devel --- src/shcomplex.c | 48 +++++++++++++++++++++++++++++++++++++++++++++++- 1 files changed, 47 insertions(+), 1 deletions(-) diff --git a/src/shcomplex.c b/src/shcomplex.c index d9488ca..bd3a733 100644 --- a/src/shcomplex.c +++ b/src/shcomplex.c @@ -1,8 +1,13 @@ +/* vim: set ts=4 sts=4 sw=4 noet : */ #include<math.h> #include<stdlib.h> #include<gsl/gsl_complex.h> #include<gsl/gsl_complex_math.h> #include<gsl/gsl_sf_legendre.h> + +#include<gsl/gsl_matrix.h> +#include<gsl/gsl_vector.h> +#include<gsl/gsl_linalg.h> #include "general.h" #include "sh.h" #include "shcomplex.h" @@ -64,7 +69,7 @@ if(sph->co != NULL) free(sph->co); if(sph->ulm !=NULL) free(sph->ulm); if(sph->ulmComplex !=NULL) free(sph->ulmComplex); - + if(sph->sumUlm2 !=NULL) free(sph->sumUlm2); if(sph->Ylmi!=NULL) { for(i=0;i<sph->l;i++){ if(sph->Ylmi[i]!=NULL){ @@ -125,3 +130,44 @@ return TS_SUCCESS; } + +ts_double calculateKc(ts_vesicle *vesicle, ts_int lmin, ts_int lmax){ + ts_int min=lmin; + ts_int max=lmax; //vesicle->sphHarmonics->l-3; + ts_long i,j; + ts_double retval, bval; + gsl_matrix *A=gsl_matrix_alloc(max-min,2); + gsl_vector *tau=gsl_vector_alloc(2); + gsl_vector *b=gsl_vector_alloc(max-min); + gsl_vector *x=gsl_vector_alloc(2); + gsl_vector *res=gsl_vector_alloc(max-min); + + //solving (A^T*A)*x=A^T*b + //fill the data for matrix A and vector b + for(i=min;i<max;i++){ + gsl_matrix_set(A, i-min,0,(ts_double)((i-1)*(i+2))); + gsl_matrix_set(A, i-min,1,(ts_double)((i-1)*(i+2)*(i+1)*i)); +// fprintf(stderr,"%e %e\n", gsl_matrix_get(A,i-min,0), gsl_matrix_get(A,i-min,1)); + bval=0.0; + //average for m from 0..l (only positive m's) + for(j=0;j<=i;j++){ + bval+=vesicle->sphHarmonics->sumUlm2[i][(j+i)]; + } + bval=bval/(ts_double)vesicle->sphHarmonics->N/(ts_double)(i+1); + + gsl_vector_set(b,i-min,1.0/bval); +// fprintf(stderr,"%e\n", 1.0/gsl_vector_get(b,i-min)); + } +// fprintf(stderr,"b[2]=%e\n",gsl_vector_get(b,1)); + gsl_linalg_QR_decomp(A,tau); + gsl_linalg_QR_lssolve(A,tau,b,x,res); +// fprintf(stderr,"kc=%e\n",gsl_vector_get(x,1)); + retval=gsl_vector_get(x,1); + gsl_matrix_free(A); + gsl_vector_free(tau); + gsl_vector_free(b); + gsl_vector_free(x); + gsl_vector_free(res); + + return retval; +} -- Gitblit v1.9.3