xref: /petsc/src/snes/interface/noise/snesdnest.c (revision ca8e9878d215fbd0d6b1ef802734bc6b1f5fa99c)
163dd3a1aSKris Buschelman 
254f251a9SBarry Smith /* fnoise/snesdnest.F -- translated by f2c (version 20020314).
354f251a9SBarry Smith */
4c6db04a5SJed Brown #include <petscsys.h>
554f251a9SBarry Smith #define FALSE_ 0
654f251a9SBarry Smith #define TRUE_ 1
754f251a9SBarry Smith 
854f251a9SBarry Smith /*  Noise estimation routine, written by Jorge More'.  Details are below. */
954f251a9SBarry Smith 
10*ca8e9878SJed Brown /* Subroutine */ PetscErrorCode SNESNoise_dnest_(PetscInt *nf, double *fval,double *h__,double *fnoise, double *fder2, double *hopt, PetscInt *info, double *eps)
1154f251a9SBarry Smith {
1254f251a9SBarry Smith     /* Initialized data */
1354f251a9SBarry Smith 
1454f251a9SBarry Smith     static double const__[15] = { .71,.41,.23,.12,.063,.033,.018,.0089,
1554f251a9SBarry Smith 	    .0046,.0024,.0012,6.1e-4,3.1e-4,1.6e-4,8e-5 };
1654f251a9SBarry Smith 
1754f251a9SBarry Smith     /* System generated locals */
18a7cc72afSBarry Smith     PetscInt i__1;
1954f251a9SBarry Smith     double d__1, d__2, d__3, d__4;
2054f251a9SBarry Smith 
2154f251a9SBarry Smith 
2254f251a9SBarry Smith     /* Local variables */
2354f251a9SBarry Smith     static double emin, emax;
24a7cc72afSBarry Smith     static PetscInt dsgn[6];
258e653690SSatish Balay     static double f_max, f_min, stdv;
26a7cc72afSBarry Smith     static PetscInt i__, j;
2754f251a9SBarry Smith     static double scale;
28a7cc72afSBarry Smith     static PetscInt mh;
29a7cc72afSBarry Smith     static PetscInt cancel[6], dnoise;
3054f251a9SBarry Smith     static double err2, est1, est2, est3, est4;
3154f251a9SBarry Smith 
3254f251a9SBarry Smith /*     ********** */
3354f251a9SBarry Smith 
3454f251a9SBarry Smith /*     Subroutine dnest */
3554f251a9SBarry Smith 
3654f251a9SBarry Smith /*     This subroutine estimates the noise in a function */
3754f251a9SBarry Smith /*     and provides estimates of the optimal difference parameter */
3854f251a9SBarry Smith /*     for a forward-difference approximation. */
3954f251a9SBarry Smith 
4054f251a9SBarry Smith /*     The user must provide a difference parameter h, and the */
4154f251a9SBarry Smith /*     function value at nf points centered around the current point. */
4254f251a9SBarry Smith /*     For example, if nf = 7, the user must provide */
4354f251a9SBarry Smith 
4454f251a9SBarry Smith /*        f(x-2*h), f(x-h), f(x), f(x+h),  f(x+2*h), */
4554f251a9SBarry Smith 
4654f251a9SBarry Smith /*     in the array fval. The use of nf = 7 function evaluations is */
4754f251a9SBarry Smith /*     recommended. */
4854f251a9SBarry Smith 
4954f251a9SBarry Smith /*     The noise in the function is roughly defined as the variance in */
5054f251a9SBarry Smith /*     the computed value of the function. The noise in the function */
5154f251a9SBarry Smith /*     provides valuable information. For example, function values */
5254f251a9SBarry Smith /*     smaller than the noise should be considered to be zero. */
5354f251a9SBarry Smith 
5454f251a9SBarry Smith /*     This subroutine requires an initial estimate for h. Under estimates */
5554f251a9SBarry Smith /*     are usually preferred. If noise is not detected, the user should */
5654f251a9SBarry Smith /*     increase or decrease h according to the ouput value of info. */
5754f251a9SBarry Smith /*     In most cases, the subroutine detects noise with the initial */
5854f251a9SBarry Smith /*     value of h. */
5954f251a9SBarry Smith 
6054f251a9SBarry Smith /*     The subroutine statement is */
6154f251a9SBarry Smith 
6254f251a9SBarry Smith /*       subroutine dnest(nf,fval,h,hopt,fnoise,info,eps) */
6354f251a9SBarry Smith 
6454f251a9SBarry Smith /*     where */
6554f251a9SBarry Smith 
6654f251a9SBarry Smith /*       nf is an int variable. */
6754f251a9SBarry Smith /*         On entry nf is the number of function values. */
6854f251a9SBarry Smith /*         On exit nf is unchanged. */
6954f251a9SBarry Smith 
7054f251a9SBarry Smith /*       f is a double precision array of dimension nf. */
7154f251a9SBarry Smith /*         On entry f contains the function values. */
7254f251a9SBarry Smith /*         On exit f is overwritten. */
7354f251a9SBarry Smith 
7454f251a9SBarry Smith /*       h is a double precision variable. */
7554f251a9SBarry Smith /*         On entry h is an estimate of the optimal difference parameter. */
7654f251a9SBarry Smith /*         On exit h is unchanged. */
7754f251a9SBarry Smith 
7854f251a9SBarry Smith /*       fnoise is a double precision variable. */
7954f251a9SBarry Smith /*         On entry fnoise need not be specified. */
8054f251a9SBarry Smith /*         On exit fnoise is set to an estimate of the function noise */
8154f251a9SBarry Smith /*            if noise is detected; otherwise fnoise is set to zero. */
8254f251a9SBarry Smith 
8354f251a9SBarry Smith /*       hopt is a double precision variable. */
8454f251a9SBarry Smith /*         On entry hopt need not be specified. */
8554f251a9SBarry Smith /*         On exit hopt is set to an estimate of the optimal difference */
8654f251a9SBarry Smith /*            parameter if noise is detected; otherwise hopt is set to zero. */
8754f251a9SBarry Smith 
8854f251a9SBarry Smith /*       info is an int variable. */
8954f251a9SBarry Smith /*         On entry info need not be specified. */
9054f251a9SBarry Smith /*         On exit info is set as follows: */
9154f251a9SBarry Smith 
9254f251a9SBarry Smith /*            info = 1  Noise has been detected. */
9354f251a9SBarry Smith 
9454f251a9SBarry Smith /*            info = 2  Noise has not been detected; h is too small. */
9554f251a9SBarry Smith /*                      Try 100*h for the next value of h. */
9654f251a9SBarry Smith 
9754f251a9SBarry Smith /*            info = 3  Noise has not been detected; h is too large. */
9854f251a9SBarry Smith /*                      Try h/100 for the next value of h. */
9954f251a9SBarry Smith 
10054f251a9SBarry Smith /*            info = 4  Noise has been detected but the estimate of hopt */
10154f251a9SBarry Smith /*                      is not reliable; h is too small. */
10254f251a9SBarry Smith 
10354f251a9SBarry Smith /*       eps is a double precision work array of dimension nf. */
10454f251a9SBarry Smith 
10554f251a9SBarry Smith /*     MINPACK-2 Project. April 1997. */
10654f251a9SBarry Smith /*     Argonne National Laboratory. */
10754f251a9SBarry Smith /*     Jorge J. More'. */
10854f251a9SBarry Smith 
10954f251a9SBarry Smith /*     ********** */
11054f251a9SBarry Smith     /* Parameter adjustments */
11154f251a9SBarry Smith     --eps;
11254f251a9SBarry Smith     --fval;
11354f251a9SBarry Smith 
11454f251a9SBarry Smith     /* Function Body */
11554f251a9SBarry Smith     *fnoise = 0.;
11654f251a9SBarry Smith     *fder2 = 0.;
11754f251a9SBarry Smith     *hopt = 0.;
11854f251a9SBarry Smith /*     Compute an estimate of the second derivative and */
11954f251a9SBarry Smith /*     determine a bound on the error. */
12054f251a9SBarry Smith     mh = (*nf + 1) / 2;
12154f251a9SBarry Smith     est1 = (fval[mh + 1] - fval[mh] * 2 + fval[mh - 1]) / *h__ / *h__;
12254f251a9SBarry Smith     est2 = (fval[mh + 2] - fval[mh] * 2 + fval[mh - 2]) / (*h__ * 2) / (*h__ *
12354f251a9SBarry Smith 	     2);
12454f251a9SBarry Smith     est3 = (fval[mh + 3] - fval[mh] * 2 + fval[mh - 3]) / (*h__ * 3) / (*h__ *
12554f251a9SBarry Smith 	     3);
12654f251a9SBarry Smith     est4 = (est1 + est2 + est3) / 3;
12754f251a9SBarry Smith /* Computing MAX */
12854f251a9SBarry Smith /* Computing PETSCMAX */
12954f251a9SBarry Smith     d__3 = PetscMax(est1,est2);
13054f251a9SBarry Smith /* Computing MIN */
13154f251a9SBarry Smith     d__4 = PetscMin(est1,est2);
13254f251a9SBarry Smith     d__1 = PetscMax(d__3,est3) - est4, d__2 = est4 - PetscMin(d__4,est3);
13354f251a9SBarry Smith     err2 = PetscMax(d__1,d__2);
13454f251a9SBarry Smith /*      write (2,123) est1, est2, est3 */
13554f251a9SBarry Smith /* 123  format ('Second derivative estimates', 3d12.2) */
13654f251a9SBarry Smith     if (err2 <= PetscAbsScalar(est4) * .1) {
13754f251a9SBarry Smith 	*fder2 = est4;
13854f251a9SBarry Smith     } else if (err2 < PetscAbsScalar(est4)) {
13954f251a9SBarry Smith 	*fder2 = est3;
14054f251a9SBarry Smith     } else {
14154f251a9SBarry Smith 	*fder2 = 0.;
14254f251a9SBarry Smith     }
14354f251a9SBarry Smith /*     Compute the range of function values. */
1448e653690SSatish Balay     f_min = fval[1];
1458e653690SSatish Balay     f_max = fval[1];
14654f251a9SBarry Smith     i__1 = *nf;
14754f251a9SBarry Smith     for (i__ = 2; i__ <= i__1; ++i__) {
14854f251a9SBarry Smith /* Computing MIN */
1498e653690SSatish Balay 	d__1 = f_min, d__2 = fval[i__];
1508e653690SSatish Balay 	f_min = PetscMin(d__1,d__2);
15154f251a9SBarry Smith /* Computing MAX */
1528e653690SSatish Balay 	d__1 = f_max, d__2 = fval[i__];
1538e653690SSatish Balay 	f_max = PetscMax(d__1,d__2);
15454f251a9SBarry Smith     }
15554f251a9SBarry Smith /*     Construct the difference table. */
15654f251a9SBarry Smith     dnoise = FALSE_;
15754f251a9SBarry Smith     for (j = 1; j <= 6; ++j) {
15854f251a9SBarry Smith 	dsgn[j - 1] = FALSE_;
15954f251a9SBarry Smith 	cancel[j - 1] = FALSE_;
16054f251a9SBarry Smith 	scale = 0.;
16154f251a9SBarry Smith 	i__1 = *nf - j;
16254f251a9SBarry Smith 	for (i__ = 1; i__ <= i__1; ++i__) {
16354f251a9SBarry Smith 	    fval[i__] = fval[i__ + 1] - fval[i__];
16454f251a9SBarry Smith 	    if (fval[i__] == 0.) {
16554f251a9SBarry Smith 		cancel[j - 1] = TRUE_;
16654f251a9SBarry Smith 	    }
16754f251a9SBarry Smith /* Computing MAX */
16854f251a9SBarry Smith 	    d__2 = scale, d__3 = (d__1 = fval[i__], PetscAbsScalar(d__1));
16954f251a9SBarry Smith 	    scale = PetscMax(d__2,d__3);
17054f251a9SBarry Smith 	}
17154f251a9SBarry Smith /*        Compute the estimates for the noise level. */
17254f251a9SBarry Smith 	if (scale == 0.) {
17354f251a9SBarry Smith 	    stdv = 0.;
17454f251a9SBarry Smith 	} else {
17554f251a9SBarry Smith 	    stdv = 0.;
17654f251a9SBarry Smith 	    i__1 = *nf - j;
17754f251a9SBarry Smith 	    for (i__ = 1; i__ <= i__1; ++i__) {
17854f251a9SBarry Smith /* Computing 2nd power */
17954f251a9SBarry Smith 		d__1 = fval[i__] / scale;
18054f251a9SBarry Smith 		stdv += d__1 * d__1;
18154f251a9SBarry Smith 	    }
18254f251a9SBarry Smith 	    stdv = scale * PetscSqrtScalar(stdv / (*nf - j));
18354f251a9SBarry Smith 	}
18454f251a9SBarry Smith 	eps[j] = const__[j - 1] * stdv;
18554f251a9SBarry Smith /*        Determine differences in sign. */
18654f251a9SBarry Smith 	i__1 = *nf - j - 1;
18754f251a9SBarry Smith 	for (i__ = 1; i__ <= i__1; ++i__) {
18854f251a9SBarry Smith /* Computing MIN */
18954f251a9SBarry Smith 	    d__1 = fval[i__], d__2 = fval[i__ + 1];
19054f251a9SBarry Smith /* Computing MAX */
19154f251a9SBarry Smith 	    d__3 = fval[i__], d__4 = fval[i__ + 1];
19254f251a9SBarry Smith 	    if (PetscMin(d__1,d__2) < 0. && PetscMax(d__3,d__4) > 0.) {
19354f251a9SBarry Smith 		dsgn[j - 1] = TRUE_;
19454f251a9SBarry Smith 	    }
19554f251a9SBarry Smith 	}
19654f251a9SBarry Smith     }
19754f251a9SBarry Smith /*     First requirement for detection of noise. */
19854f251a9SBarry Smith     dnoise = dsgn[3];
19954f251a9SBarry Smith /*     Check for h too small or too large. */
20054f251a9SBarry Smith     *info = 0;
2018e653690SSatish Balay     if (f_max == f_min) {
20254f251a9SBarry Smith 	*info = 2;
20354f251a9SBarry Smith     } else /* if(complicated condition) */ {
20454f251a9SBarry Smith /* Computing MIN */
2058e653690SSatish Balay 	d__1 = PetscAbsScalar(f_max), d__2 = PetscAbsScalar(f_min);
2068e653690SSatish Balay 	if (f_max - f_min > PetscMin(d__1,d__2) * .1) {
20754f251a9SBarry Smith 	    *info = 3;
20854f251a9SBarry Smith 	}
20954f251a9SBarry Smith     }
21054f251a9SBarry Smith     if (*info != 0) {
211a7cc72afSBarry Smith 	PetscFunctionReturn(0);
21254f251a9SBarry Smith     }
21354f251a9SBarry Smith /*     Determine the noise level. */
21454f251a9SBarry Smith /* Computing MIN */
21554f251a9SBarry Smith     d__1 = PetscMin(eps[4],eps[5]);
21654f251a9SBarry Smith     emin = PetscMin(d__1,eps[6]);
21754f251a9SBarry Smith /* Computing MAX */
21854f251a9SBarry Smith     d__1 = PetscMax(eps[4],eps[5]);
21954f251a9SBarry Smith     emax = PetscMax(d__1,eps[6]);
22054f251a9SBarry Smith     if (emax <= emin * 4 && dnoise) {
22154f251a9SBarry Smith 	*fnoise = (eps[4] + eps[5] + eps[6]) / 3;
22254f251a9SBarry Smith 	if (*fder2 != 0.) {
22354f251a9SBarry Smith 	    *info = 1;
22454f251a9SBarry Smith 	    *hopt = PetscSqrtScalar(*fnoise / PetscAbsScalar(*fder2)) * 1.68;
22554f251a9SBarry Smith 	} else {
22654f251a9SBarry Smith 	    *info = 4;
22754f251a9SBarry Smith 	    *hopt = *h__ * 10;
22854f251a9SBarry Smith 	}
229a7cc72afSBarry Smith 	PetscFunctionReturn(0);
23054f251a9SBarry Smith     }
23154f251a9SBarry Smith /* Computing MIN */
23254f251a9SBarry Smith     d__1 = PetscMin(eps[3],eps[4]);
23354f251a9SBarry Smith     emin = PetscMin(d__1,eps[5]);
23454f251a9SBarry Smith /* Computing MAX */
23554f251a9SBarry Smith     d__1 = PetscMax(eps[3],eps[4]);
23654f251a9SBarry Smith     emax = PetscMax(d__1,eps[5]);
23754f251a9SBarry Smith     if (emax <= emin * 4 && dnoise) {
23854f251a9SBarry Smith 	*fnoise = (eps[3] + eps[4] + eps[5]) / 3;
23954f251a9SBarry Smith 	if (*fder2 != 0.) {
24054f251a9SBarry Smith 	    *info = 1;
24154f251a9SBarry Smith 	    *hopt = PetscSqrtScalar(*fnoise / PetscAbsScalar(*fder2)) * 1.68;
24254f251a9SBarry Smith 	} else {
24354f251a9SBarry Smith 	    *info = 4;
24454f251a9SBarry Smith 	    *hopt = *h__ * 10;
24554f251a9SBarry Smith 	}
246a7cc72afSBarry Smith 	PetscFunctionReturn(0);
24754f251a9SBarry Smith     }
24854f251a9SBarry Smith /*     Noise not detected; decide if h is too small or too large. */
24954f251a9SBarry Smith     if (! cancel[3]) {
25054f251a9SBarry Smith 	if (dsgn[3]) {
25154f251a9SBarry Smith 	    *info = 2;
25254f251a9SBarry Smith 	} else {
25354f251a9SBarry Smith 	    *info = 3;
25454f251a9SBarry Smith 	}
255a7cc72afSBarry Smith 	PetscFunctionReturn(0);
25654f251a9SBarry Smith     }
25754f251a9SBarry Smith     if (! cancel[2]) {
25854f251a9SBarry Smith 	if (dsgn[2]) {
25954f251a9SBarry Smith 	    *info = 2;
26054f251a9SBarry Smith 	} else {
26154f251a9SBarry Smith 	    *info = 3;
26254f251a9SBarry Smith 	}
263a7cc72afSBarry Smith 	PetscFunctionReturn(0);
26454f251a9SBarry Smith     }
26554f251a9SBarry Smith /*     If there is cancelllation on the third and fourth column */
26654f251a9SBarry Smith /*     then h is too small */
26754f251a9SBarry Smith     *info = 2;
268a7cc72afSBarry Smith     PetscFunctionReturn(0);
26954f251a9SBarry Smith /*      if (cancel .or. dsgn(3)) then */
27054f251a9SBarry Smith /*         info = 2 */
27154f251a9SBarry Smith /*      else */
27254f251a9SBarry Smith /*         info = 3 */
27354f251a9SBarry Smith /*      end if */
27454f251a9SBarry Smith } /* dnest_ */
27554f251a9SBarry Smith 
276