source: trunk/src/pspio_meshfunc.c @ 70

/*
 Copyright (C) 2011 J. Alberdi, M. Oliveira, Y. Pouillon, and M. Verstraete

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU Lesser General Public License as published by
 the Free Software Foundation; either version 3 of the License, or 
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU Lesser General Public License for more details.

 You should have received a copy of the GNU Lesser General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 $Id$
*/

#include "pspio_meshfunc.h"

#include <stdlib.h>

int pspio_meshfunc_alloc(pspio_meshfunc_t *func, pspio_mesh_t *mesh){
  int i;

  func = (pspio_meshfunc_t *) malloc (sizeof(pspio_meshfunc_t));

  if (func == NULL) {
    return PSPIO_ENOMEM;
  }

  func->f = (double *) malloc (mesh->np * sizeof(double));

  if (func->f == NULL) {
    free (func);
    return PSPIO_ENOMEM;
  }

  for (i = 0; i < mesh->np; i++)
    {
      func->f[i] = 0;
    }
  func->mesh = mesh;
  func->spl = gsl_spline_alloc(gsl_interp_cspline, mesh->np);
  func->acc = gsl_interp_accel_alloc();

  return PSPIO_SUCCESS;
}


int pspio_meshfunc_set(pspio_meshfunc_t *func, double *f){
  int i;
  
  for (i = 0; i < func->mesh->np; i++)
    {
      func->f[i] = f[i];
    }

  gsl_spline_init(func->spl, func->mesh->r, func->f, func->mesh->np);

  return PSPIO_SUCCESS;
}


void pspio_meshfunc_free(pspio_meshfunc_t *func){

  if (func == NULL) return;

  free(func->f);
  gsl_spline_free(func->spl);
  gsl_interp_accel_free(func->acc);
  func->mesh = NULL;
  free(func);

}