mirror of
https://github.com/eddyem/BTA_utils.git
synced 2025-12-06 02:35:13 +03:00
96 lines
2.7 KiB
C
96 lines
2.7 KiB
C
/*
|
|
* 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__
|