test_multiple.c

Go to the documentation of this file.

/*====================================================================*/
/*====================================================================*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <strings.h>
#include <limits.h>
#include <math.h>
#include "../gsl_fit.h"

/*====================================================================*/
#include "test_functions.h"
/*====================================================================*/
double d[3][2];
/*====================================================================*/
void load(gsl_fit *gft, char *filename);

/*====================================================================*/
/*====================================================================*/
/*====================================================================*/
/*====================================================================*/
int main(int argc, char *argv[])
{
  double        param[4], error[4];
  unsigned int  fixed[4];
  gsl_fit *my_gft=NULL;

  /*========================================================
  **  Prototype for how this might work: 
  */
  
  /*========================================================
  **  1. Allocate and pick method from the several types available 
  */
  //my_gft = gsl_fit_alloc(LEAST_SQUARES, LMSDR, 100, MULTIPLE);
  my_gft = gsl_fit_alloc(MULTIDIM_MINIMIZAION, CONJUGATE_FR, 500, MULTIPLE);
  //my_gft = gsl_fit_alloc(SIMULATED_ANNEAL, 0, 0, MULTIPLE);

  /*========================================================
  **  2. Load the data.
  **  In this example the relationship bewteen the data sets are
  **  known beforehand. It can be summarized as:
  **     for data set i, multiply the height of peak j by d[i][j]
  **
  ** Other examples might be:
  **     for data set i, add d[i][j] to the position of peak j 
  **     for data set i, the width of peak j is divided by d[i][j]
  */
  load(my_gft, "test_data/data0.txt");
  load(my_gft, "test_data/data1.txt");
  load(my_gft, "test_data/data2.txt");
  
  d[0][0] =  1.0; d[0][1] = -1.0;
  d[1][0] =  1.0; d[1][1] =  1.0;
  d[2][0] = -1.0; d[2][1] =  1.0;
   
  /*========================================================
  **  4. Make an inital guess.
  */
  /* The first function added is function 0 */
  param[0] = 10.472;   fixed[0] =    0;
  param[1] = 18.627;   fixed[1] =    0;
  param[2] =  8.150;   fixed[2] =    0;
  param[3] =  1.150;   fixed[3] =    0;
  gsl_fit_add_model_multiple(my_gft, mpearson_area, param, fixed, 4); 

  /* The second function added is function 1 */
  param[0] =  9.323;   fixed[0] =    0;
  param[1] = 70.033;   fixed[1] =    0;
  param[2] =  3.704;   fixed[2] =    0;
  param[3] = 15.150;   fixed[3] =    0;
  gsl_fit_add_model_multiple(my_gft, mpearson_area, param, fixed, 4);

  /* The first function added is function 2 */
  param[0] =  0.0012;   fixed[0] =    0;
  param[1] =  1.1000;   fixed[1] =    0;
  gsl_fit_add_model_multiple(my_gft, mline, param, fixed, 2);
  
  /*========================================================*/
  gsl_fit_print(my_gft, "test_data/start.txt");

  /*========================================================
  ** 5. Do the fitting
  */
  gsl_fit_do(my_gft);

  /*========================================================*/
  gsl_fit_print(my_gft, "test_data/end.txt");

  /*========================================================
  **  6. Get the results
  */
  gsl_fit_get_result(my_gft, 0, param, error, 4);
  gsl_fit_get_result(my_gft, 1, param, error, 4);
  gsl_fit_get_result(my_gft, 2, param, error, 2);

  /*========================================================
  **  7. Free the container
  */
  gsl_fit_free(my_gft);
  
  return EXIT_SUCCESS;
}
/*====================================================================*/
/*====================================================================*/
/*====================================================================*/
void load(gsl_fit *gft, char *filename)
{
  int                i, j;
  double             d[3];
  double            *x, *y, *s;
  char               c[120];
  FILE              *fp;

  /*======================================  
  ** open the file for reading 
  */
  if (!(fp = fopen(filename, "r")))
  {
    fprintf(stderr,"Can not open file.\n");
    exit(EXIT_FAILURE);
  }

  /*======================================  
  ** Count the number of records by using
  ** the sucessful conversion of three column data
  */
  bzero(c, 120); j = 0;
  while(fgets(c, 119, fp) != NULL)
  {
    i = sscanf(c, "%lf %lf %lf", &d[0], &d[1], &d[2]);
    if (i == 3) j++;
    bzero(c,120);
  }
  
  /*======================================  
  ** rewind the file pointer to the start of the file 
  */
  rewind(fp);
  
  /*======================================  
  ** allocate the memory to hold the data  
  */
  x = (double *)malloc(j*sizeof(double));
  y = (double *)malloc(j*sizeof(double));
  s = (double *)malloc(j*sizeof(double));
  
  /*======================================  
  ** read the data into memory  
  */
  bzero(c, 120); j = 0;
  while(fgets(c, 119, fp) != NULL)
  {
    i = sscanf(c, "%lf %lf %lf", &x[j], &y[j], &s[j]);
    if (i == 3) j++;
    bzero(c,120);
  }
  
  /*======================================  
  ** done with the file pointer 
  */
  fclose(fp); 
  
  /*====================================== 
  ** Add the data set to the gsl container.
  */
  gsl_fit_add_dataset(gft, x, y, s, j);
  
  /*====================================== 
  ** Free the used memory
  */
  free(x);
  free(y);
  free(s);
  
  /*======================================*/
  return;
}
/*====================================================================*/
/*====================================================================*/
/*====================================================================*/
#include "test_functions.c"

---