/*
 * zernike.h
 *
 * Copyright 2016 Edward V. Emelianov <eddy@sao.ru, edward.emelianoff@gmail.com>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301, USA.
 */
#pragma once
#ifndef __ZERNIKE_H__
#define __ZERNIKE_H__

// default step in coordinate grid
#define DEFAULT_CRD_STEP  (0.05)
// default wavefront unit: lambda
#define DEFAULT_WF_UNIT "meter"
// default wavelength
#define DEFAULT_WAVELENGTH (0.65e-6)
// max power of Zernike polynomial
#define ZERNIKE_MAX_POWER  (100)

typedef struct{
    double r,theta; // polar coordinates
    int idx;        // index of given point in square matrix
} polar;

typedef struct{
    polar *P;       // polar coordinates inside unitary circle
    double **Rpow;  // powers of R
    int N;          // max power of Zernike coeffs
    int Sz;         // size of P
    int WH;         // Width/Height of matrix
} polcrds;

// for `const char * const *units` thanks to http://stackoverflow.com/a/3875555/1965803
typedef struct{
    double wf_coeff;            // multiplier for wavefront units (in .dat files coefficients are in meters)
    const char * const *units;  // symbol units' names
} wf_units;

// additional components to Zernike coefficients
typedef struct{
    int idx;        // Znumber
    double addval;  // additional value
} coeff;

void z_set_scale(double s);
double z_get_scale();

int z_set_step(double step);
double z_get_step();

int z_set_wavelength(double w);
double z_get_wavelength();

int z_set_wfunit(char *u);
char *z_get_wfunit();

double z_get_wfcoeff();
void z_print_wfunits();

void z_set_rotangle(double angle);
double z_get_rotangle();

int z_set_Nzero(int val); // setter when val > 0 and getter elsewhere

void z_set_tozero(int **val);
int z_get_tozero(int **idxs);

void z_set_addcoef(char **list);
int z_get_addcoef(coeff **vals);

void convert_Zidx(int p, int *N, int *M);

polcrds *gen_coords();
void free_coords(polcrds *p);

double **build_rpow(int n, int Sz, polar *P);

double *Zcompose(int Zsz, double *Zidxs, polcrds *P);

int z_save_wavefront(polcrds *P, double *Z, double *std, char *fprefix);
#endif // __ZERNIKE_H__