ECCE @ EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ptwXY_misc.cc
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file ptwXY_misc.cc
1 /*
2 # <<BEGIN-copyright>>
3 # <<END-copyright>>
4 */
5 
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <cmath>
9 #include <float.h>
10 
11 #include "ptwXY.h"
12 
13 #if defined __cplusplus
14 #include <cmath>
15 #include "G4Log.hh"
16 namespace GIDI {
17 using namespace GIDI;
18 #endif
19 
21  void *argList, int level, int checkForRoots, double eps );
22 static nfu_status ptwXY_createFromFunctionZeroCrossing( ptwXYPoints *ptwXY, double x1, double y1, double x2, double y2,
23  ptwXY_createFromFunction_callback func, void *argList, double eps );
25  void *argList, int level, int checkForRoots );
26 static nfu_status ptwXY_applyFunctionZeroCrossing( ptwXYPoints *ptwXY1, double y1, double y2, ptwXYPoint *p1, ptwXYPoint *p2,
27  ptwXY_applyFunction_callback func, void *argList );
28 /*
29 ************************************************************
30 */
31 void ptwXY_update_biSectionMax( ptwXYPoints *ptwXY1, double oldLength ) {
32 
33  ptwXY1->biSectionMax = ptwXY1->biSectionMax - 1.442695 * G4Log( ptwXY1->length / oldLength ); /* 1.442695 = 1 / std::log( 2. ) */
34  if( ptwXY1->biSectionMax < 0 ) ptwXY1->biSectionMax = 0;
36 }
37 /*
38 ************************************************************
39 */
40 ptwXYPoints *ptwXY_createFromFunction( int n, double *xs, ptwXY_createFromFunction_callback func, void *argList, double accuracy, int checkForRoots,
41  int biSectionMax, nfu_status *status ) {
42 
43  int64_t i;
44  double x1, y1, x2, y2, eps = ClosestAllowXFactor * DBL_EPSILON;
45  ptwXYPoints *ptwXY;
46  ptwXYPoint *p1, *p2;
47 
48  *status = nfu_Okay;
49  if( n < 2 ) { *status = nfu_tooFewPoints; return( NULL ); }
50  for( i = 1; i < n; i++ ) {
51  if( xs[i-1] >= xs[i] ) *status = nfu_XNotAscending;
52  }
53  if( *status == nfu_XNotAscending ) return( NULL );
54 
55  x1 = xs[0];
56  if( ( *status = func( x1, &y1, argList ) ) != nfu_Okay ) return( NULL );
57  if( ( ptwXY = ptwXY_new( ptwXY_interpolationLinLin, NULL, biSectionMax, accuracy, 500, 50, status, 0 ) ) == NULL ) return( NULL );
58  for( i = 1; i < n; i++ ) {
59  if( ( *status = ptwXY_setValueAtX_overrideIfClose( ptwXY, x1, y1, eps, 0 ) ) != nfu_Okay ) goto err;
60  x2 = xs[i];
61  if( ( *status = func( x2, &y2, argList ) ) != nfu_Okay ) goto err;
62  if( ( *status = ptwXY_createFromFunctionBisect( ptwXY, x1, y1, x2, y2, func, argList, 0, checkForRoots, eps ) ) != nfu_Okay ) goto err;
63  x1 = x2;
64  y1 = y2;
65  }
66  if( ( *status = ptwXY_setValueAtX_overrideIfClose( ptwXY, x2, y2, eps, 1 ) ) != nfu_Okay ) goto err;
67 
68  if( checkForRoots ) {
69  if( ( *status = ptwXY_simpleCoalescePoints( ptwXY ) ) != nfu_Okay ) goto err;
70  for( i = ptwXY->length - 1, p2 = NULL; i >= 0; i--, p2 = p1 ) { /* Work backward so lower points are still valid if a new point is added. */
71  p1 = &(ptwXY->points[i]);
72  if( p2 != NULL ) {
73  if( ( p1->y * p2->y ) < 0. ) {
74  if( ( *status = ptwXY_createFromFunctionZeroCrossing( ptwXY, p1->x, p1->y, p2->x, p2->y, func, argList, eps ) ) != nfu_Okay ) goto err;
75  }
76  }
77  }
78  }
79 
80  return( ptwXY );
81 
82 err:
83  ptwXY_free( ptwXY );
84  return( NULL );
85 }
86 /*
87 ************************************************************
88 */
89 ptwXYPoints *ptwXY_createFromFunction2( ptwXPoints *xs, ptwXY_createFromFunction_callback func, void *argList, double accuracy, int checkForRoots,
90  int biSectionMax, nfu_status *status ) {
91 
92  return( ptwXY_createFromFunction( (int) xs->length, xs->points, func, argList, accuracy, checkForRoots, biSectionMax, status ) );
93 }
94 /*
95 ************************************************************
96 */
98  void *argList, int level, int checkForRoots, double eps ) {
99 
100  nfu_status status;
101  double x, y, f;
102 
103  if( ( x2 - x1 ) < ClosestAllowXFactor * DBL_EPSILON * ( std::fabs( x1 ) + std::fabs( x2 ) ) ) return( nfu_Okay );
104  if( level >= ptwXY->biSectionMax ) return( nfu_Okay );
105  x = 0.5 * ( x1 + x2 );
106  if( ( status = ptwXY_interpolatePoint( ptwXY->interpolation, x, &y, x1, y1, x2, y2 ) ) != nfu_Okay ) return( status );
107  if( ( status = func( x, &f, argList ) ) != nfu_Okay ) return( status );
108  if( std::fabs( f - y ) <= 0.8 * std::fabs( f * ptwXY->accuracy ) ) return( nfu_Okay );
109  if( ( status = ptwXY_createFromFunctionBisect( ptwXY, x1, y1, x, f, func, argList, level + 1, checkForRoots, eps ) ) ) return( status );
110  if( ( status = ptwXY_setValueAtX_overrideIfClose( ptwXY, x, f, eps, 0 ) ) != nfu_Okay ) return( status );
111  return( ptwXY_createFromFunctionBisect( ptwXY, x, f, x2, y2, func, argList, level + 1, checkForRoots, eps ) );
112 }
113 /*
114 ************************************************************
115 */
116 static nfu_status ptwXY_createFromFunctionZeroCrossing( ptwXYPoints *ptwXY, double x1, double y1, double x2, double y2,
117  ptwXY_createFromFunction_callback func, void *argList, double eps ) {
118 
119  //For coverity #63077
120  if ( y2 == y1 ) return ( nfu_badInput );
121 
122  int i;
123  double x, y;
124  nfu_status status;
125 
126  for( i = 0; i < 16; i++ ) {
127  if( y2 == y1 ) break;
128  x = ( y2 * x1 - y1 * x2 ) / ( y2 - y1 );
129  if( x <= x1 ) x = x1 + 0.1 * ( x2 - x1 );
130  if( x >= x2 ) x = x2 - 0.1 * ( x2 - x1 );
131  if( ( status = func( x, &y, argList ) ) != nfu_Okay ) return( status );
132  if( y == 0 ) break;
133  if( y1 * y < 0 ) {
134  x2 = x;
135  y2 = y; }
136  else {
137  x1 = x;
138  y1 = y;
139  }
140  }
141  return( ptwXY_setValueAtX_overrideIfClose( ptwXY, x, 0., eps, 1 ) );
142 }
143 /*
144 ************************************************************
145 */
146 nfu_status ptwXY_applyFunction( ptwXYPoints *ptwXY1, ptwXY_applyFunction_callback func, void *argList, int checkForRoots ) {
147 
148  int64_t i, originalLength = ptwXY1->length, notFirstPass = 0;
149  double y1, y2 = 0;
150  nfu_status status;
151  ptwXYPoint p1, p2;
152 
153  checkForRoots = checkForRoots && ptwXY1->biSectionMax;
154  if( ptwXY1->status != nfu_Okay ) return( ptwXY1->status );
157  if( ( status = ptwXY_simpleCoalescePoints( ptwXY1 ) ) != nfu_Okay ) return( status );
158  for( i = originalLength - 1; i >= 0; i-- ) {
159  y1 = ptwXY1->points[i].y;
160  if( ( status = func( &(ptwXY1->points[i]), argList ) ) != nfu_Okay ) return( status );
161  p1 = ptwXY1->points[i];
162  if( notFirstPass ) {
163  if( ( status = ptwXY_applyFunction2( ptwXY1, y1, y2, &p1, &p2, func, argList, 0, checkForRoots ) ) != nfu_Okay ) return( status );
164  }
165  notFirstPass = 1;
166  p2 = p1;
167  y2 = y1;
168  }
169  ptwXY_update_biSectionMax( ptwXY1, (double) originalLength );
170  return( status );
171 }
172 /*
173 ************************************************************
174 */
176  void *argList, int level, int checkForRoots ) {
177 
178  double y;
179  ptwXYPoint p;
180  nfu_status status;
181 
182  if( ( p2->x - p1->x ) < ClosestAllowXFactor * DBL_EPSILON * ( std::fabs( p1->x ) + std::fabs( p2->x ) ) ) return( nfu_Okay );
183  if( level >= ptwXY1->biSectionMax ) goto checkForZeroCrossing;
184  p.x = 0.5 * ( p1->x + p2->x );
185  if( ( status = ptwXY_interpolatePoint( ptwXY1->interpolation, p.x, &y, p1->x, y1, p2->x, y2 ) ) != nfu_Okay ) return( status );
186  p.y = y;
187  if( ( status = func( &p, argList ) ) != nfu_Okay ) return( status );
188  if( std::fabs( ( p.x - p1->x ) * ( p2->y - p1->y ) + ( p2->x - p1->x ) * ( p1->y - p.y ) ) <= 0.8 * std::fabs( ( p2->x - p1->x ) * p.y * ptwXY1->accuracy ) )
189  goto checkForZeroCrossing;
190  if( ( status = ptwXY_setValueAtX( ptwXY1, p.x, p.y ) ) != nfu_Okay ) return( status );
191  if( ( status = ptwXY_applyFunction2( ptwXY1, y1, y, p1, &p, func, argList, level + 1, checkForRoots ) ) ) return( status );
192  return( ptwXY_applyFunction2( ptwXY1, y, y2, &p, p2, func, argList, level + 1, checkForRoots ) );
193 
194 checkForZeroCrossing:
195  if( checkForRoots && ( ( p1->y * p2->y ) < 0. ) ) return( ptwXY_applyFunctionZeroCrossing( ptwXY1, y1, y2, p1, p2, func, argList ) );
196  return( nfu_Okay );
197 }
198 /*
199 ************************************************************
200 */
202  ptwXY_applyFunction_callback func, void *argList ) {
203 
204  int i;
205  double y, x1 = p1->x, x2 = p2->x, nY1 = p1->y, nY2 = p2->y, refY = 0.5 * ( std::fabs( p1->y ) + std::fabs( p2->y ) );
206  ptwXYPoint p;
207  nfu_status status;
208 
209  //For coverity #63074
210  if ( nY2 == nY1 ) return ( nfu_badInput );
211 
212  for( i = 0; i < 6; i++ ) {
213  if( nY2 == nY1 ) break;
214  p.x = ( nY2 * x1 - nY1 * x2 ) / ( nY2 - nY1 );
215  if( p.x <= x1 ) p.x = 0.5 * ( x1 + x2 );
216  if( p.x >= x2 ) p.x = 0.5 * ( x1 + x2 );
217  if( ( status = ptwXY_interpolatePoint( ptwXY1->interpolation, p.x, &y, p1->x, y1, p2->x, y2 ) ) != nfu_Okay ) return( status );
218  p.y = y;
219  if( ( status = func( &p, argList ) ) != nfu_Okay ) return( status );
220  if( p.y == 0 ) break;
221  if( 0.5 * refY < std::fabs( p.y ) ) break;
222  refY = std::fabs( p.y );
223  if( p1->y * p.y < 0 ) {
224  x2 = p.x;
225  nY2 = p.y; }
226  else {
227  x1 = p.x;
228  nY1 = p.y;
229  }
230  }
231  return( ptwXY_setValueAtX( ptwXY1, p.x, 0. ) );
232 }
233 /*
234 ************************************************************
235 */
236 ptwXYPoints *ptwXY_fromString( char const *str, ptwXY_interpolation interpolation, ptwXY_interpolationOtherInfo const *interpolationOtherInfo,
237  double biSectionMax, double accuracy, char **endCharacter, nfu_status *status ) {
238 
239  int64_t numberConverted;
240  double *doublePtr;
241  ptwXYPoints *ptwXY = NULL;
242 
243  if( ( *status = nfu_stringToListOfDoubles( str, &numberConverted, &doublePtr, endCharacter ) ) != nfu_Okay ) return( NULL );
244  *status = nfu_oddNumberOfValues;
245  if( ( numberConverted % 2 ) == 0 )
246  ptwXY = ptwXY_create( interpolation, interpolationOtherInfo, biSectionMax, accuracy, numberConverted, 10, numberConverted / 2, doublePtr, status, 0 );
247  nfu_free( doublePtr );
248  return( ptwXY );
249 }
250 /*
251 ************************************************************
252 */
253 void ptwXY_showInteralStructure( ptwXYPoints *ptwXY, FILE *f, int printPointersAsNull ) {
254 
255  int64_t i, n = ptwXY_getNonOverflowLength( ptwXY );
256  ptwXYPoint *point = ptwXY->points;
257  ptwXYOverflowPoint *overflowPoint;
258 
259  fprintf( f, "status = %d interpolation = %d length = %d allocatedSize = %d\n",
260  (int) ptwXY->status, (int) ptwXY->interpolation, (int) ptwXY->length, (int) ptwXY->allocatedSize );
261  fprintf( f, "userFlag = %d biSectionMax = %.8e accuracy = %.2e minFractional_dx = %.6e\n",
262  ptwXY->userFlag, ptwXY->biSectionMax, ptwXY->accuracy, ptwXY->minFractional_dx );
263  fprintf( f, "interpolationString = %s\n", ptwXY->interpolationOtherInfo.interpolationString );
264  fprintf( f, "getValueFunc is NULL = %d. argList is NULL = %d.\n",
265  ptwXY->interpolationOtherInfo.getValueFunc == NULL, ptwXY->interpolationOtherInfo.argList == NULL );
266  fprintf( f, " overflowLength = %d overflowAllocatedSize = %d mallocFailedSize = %d\n",
267  (int) ptwXY->overflowLength, (int) ptwXY->overflowAllocatedSize, (int) ptwXY->mallocFailedSize );
268  fprintf( f, " Points data, points = %20p\n", ( printPointersAsNull ? NULL : (void*)ptwXY->points ) );
269  for( i = 0; i < n; i++, point++ ) fprintf( f, " %14.7e %14.7e\n", point->x, point->y );
270  fprintf( f, " Overflow points data; %20p\n", ( printPointersAsNull ? NULL : (void*)&(ptwXY->overflowHeader) ) );
271  for( overflowPoint = ptwXY->overflowHeader.next; overflowPoint != &(ptwXY->overflowHeader); overflowPoint = overflowPoint->next ) {
272  fprintf( f, " %14.7e %14.7e %8d %20p %20p %20p\n", overflowPoint->point.x, overflowPoint->point.y, (int) overflowPoint->index,
273  (void*) ( printPointersAsNull ? NULL : overflowPoint ), (void*) ( printPointersAsNull ? NULL : overflowPoint->prior ),
274  (void*) ( printPointersAsNull ? NULL : overflowPoint->next ) );
275  }
276  fprintf( f, " Points in order\n" );
277  for( i = 0; i < ptwXY->length; i++ ) {
278  point = ptwXY_getPointAtIndex( ptwXY, i );
279  fprintf( f, " %14.7e %14.7e\n", point->x, point->y );
280  }
281 }
282 /*
283 ************************************************************
284 */
285 void ptwXY_simpleWrite( ptwXYPoints *ptwXY, FILE *f, char *format ) {
286 
287  int64_t i;
288  ptwXYPoint *point;
289 
290  for( i = 0; i < ptwXY->length; i++ ) {
291  point = ptwXY_getPointAtIndex( ptwXY, i );
292  fprintf( f, format, point->x, point->y );
293  }
294 }
295 /*
296 ************************************************************
297 */
298 void ptwXY_simplePrint( ptwXYPoints *ptwXY, char *format ) {
299 
300  ptwXY_simpleWrite( ptwXY, stdout, format );
301 }
302 
303 #if defined __cplusplus
304 }
305 #endif