tempo2/TKfit.h at master · mattpitkin/tempo2 · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
//  Copyright (C) 2006,2007,2008,2009, George Hobbs, Russell Edwards

#ifndef __TKfit_h
#define __TKfit_h


/*
 *    This file is part of TEMPO2.
 *
 *    TEMPO2 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 3 of the License, or
 *    (at your option) any later version.
 *    TEMPO2 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 TEMPO2.  If not, see <http://www.gnu.org/licenses/>.
 */

/*
 *    If you use TEMPO2 then please acknowledge it by citing
 *    Hobbs, Edwards & Manchester (2006) MNRAS, Vol 369, Issue 2,
 *    pp. 655-672 (bibtex: 2006MNRAS.369..655H)
 *    or Edwards, Hobbs & Manchester (2006) MNRAS, VOl 372, Issue 4,
 *    pp. 1549-1574 (bibtex: 2006MNRAS.372.1549E) when discussing the
 *    timing model.
 */

#include "TKmatrix.h"
#include "TKlongdouble.h"

#ifdef __cplusplus
extern "C" {
#endif
    double TKleastSquares(double* b, double* white_b,
            double** designMatrix, double** white_designMatrix,
            int n,int nf, double tol, char rescale_errors,
            double* outP, double* e, double** CVM);

    double TKrobustLeastSquares(double* b, double* white_b,
            double** designMatrix, double** white_designMatrix,
            int n,int nf, double tol, char rescale_errors,
            double* outP, double* e, double** cvm, char robust);

    double TKconstrainedLeastSquares(double* b, double* white_b,
            double** designMatrix, double** white_designMatrix,
            double** constraintsMatrix,
            int n,int nf, int nconstraints, double tol, char rescale_errors,
            double* outP, double* e, double** cvm);

    double TKrobustConstrainedLeastSquares(double* b, double* white_b,
            double** designMatrix, double** white_designMatrix,
            double** constraintsMatrix,
            int n,int nf, int nconstraints, double tol, char rescale_errors,
            double* outP, double* e, double** cvm, char robust);

    double TKrobustDefConstrainedLeastSquares(double* b, double* white_b,
            double** designMatrix, double** white_designMatrix,
            double** constraintsMatrix,
            int n,int nf, int nconstraints, double tol, char rescale_errors,
            double* outP, double* e, double** cvm, char robust,double* constraint_vals);


    // legacy convinience methods
    void TKleastSquares_svd(double *x,double *y,double *sig,int n,double *p,double *e,int nf,double **cvm, double *chisq, void (*fitFuncs)(double, double [], int),int weight);
    void TKleastSquares_svd_noErr(double *x,double *y,int n,double *p,int nf, void (*fitFuncs)(double, double [], int));


    void TKremovePoly_f(float *px,float *py,int n,int m);
    void TKremovePoly_d(double *px,double *py,int n,int m);
    void TKfindPoly_d(double *px,double *py,int n,int m,double* p);
    void TKfitPoly(double x,double *v,int m);

#ifdef __cplusplus
}
longdouble TKfindMax_Ld(longdouble *x,int n);
#endif

#ifdef __Tempo2_h

void TKleastSquares_single_pulsar(double *x,double *y,int n,double *outP,double *e,int nf,double **cvm, double *chisq, void (*fitFuncs)(double, double [], int,pulsar *,int, int),pulsar *psr,double tol, int *ip,char rescale_errors, double **uinv);

void TKleastSquares_global_pulsar(double **x,double **y,int *n,
        double *outP,double *e,int* nf, int nglobal,double **cvm,
        double *chisq, void (*fitFuncs)(double, double [], int,pulsar *,int,int),
        pulsar *psr,double tol, int **ip,char rescale_errors, double ***uinv, int npsr) ;


void TKfit_getPulsarDesignMatrix(double *x,double *y,int n,int nf,
        void (*fitFuncs)(double, double [], int,pulsar *,int,int),
        pulsar *psr,int *ip, double **uinv, int ipsr,
        double ***OUT_designMatrix,double ***OUT_white_designMatrix,
        double** OUT_b, double** OUT_wb);

// legacy methods
void TKleastSquares_svd_psr_dcm(double *x,double *y,double *sig,int n,double *p,double *e,int nf,double **cvm, double *chisq, void (*fitFuncs)(double, double [], int,pulsar *,int, int),int weight,pulsar *psr,double tol, int *ip,double **uinv);
void TKleastSquares_svd_psr(double *x,double *y,double *sig,int n,double *p,double *e,int nf,double **cvm, double *chisq, void (*fitFuncs)(double, double [], int,pulsar *,int,int),int weight,pulsar *psr,double tol, int *ip);

#endif

#endif