27 #include <almostEqual.hpp>
30 #ifndef PROSHADE_MATHS
31 #define PROSHADE_MATHS
37 extern void dgesdd_ (
char* jobz,
int* m,
int* n,
double* a,
int* lda,
double* s,
double* u,
int* ldu,
double* vt,
int* ldvt,
double* work,
int* lwork,
double* rwork,
int* iwork,
int* info );
39 extern void zgesdd_ (
char* jobz,
int* m,
int* n, std::complex<double>* a,
int* lda,
double* s, std::complex<double>* u,
int* ldu, std::complex<double>* vt,
int* ldvt, std::complex<double>* work,
int* lwork,
double* rwork,
int* iwork,
int* info );
41 extern void dgeev_ (
char* jobvl,
char* jobvr,
int* n,
double* a,
int* lda,
double* wr,
double* wi,
double* vl,
int* ldvl,
double* vr,
int* ldvr,
double* work,
int* lwork,
int* info );
60 proshade_double radius;
61 proshade_double radiusMax;
62 proshade_double radiusMin;
63 proshade_unsign angularDim;
64 proshade_double representedAngle;
65 proshade_unsign sphereNumber;
67 proshade_double* axesValues;
68 std::vector< std::pair< proshade_unsign,proshade_unsign > > peaks;
70 ProSHADE_rotFun_sphere ( proshade_double rad, proshade_double radRange, proshade_unsign dim, proshade_double repAng, proshade_unsign sphNo );
80 std::vector<std::pair<proshade_unsign,proshade_unsign>>
getPeaks (
void );
88 void findAllPeaks ( proshade_signed noSmNeighbours, std::vector< proshade_double >* allHeights );
112 proshade_double a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33;
113 proshade_double xStartIndex, xRange, yStartIndex, yRange;
123 BicubicInterpolator ( proshade_double** areaToInterpolate, proshade_double xStart, proshade_double yStart )
126 this->xStartIndex = xStart;
127 this->yStartIndex = yStart;
134 this->a00 = areaToInterpolate[1][1];
135 this->a01 = - ( 0.5 * areaToInterpolate[1][0] ) +
136 ( 0.5 * areaToInterpolate[1][2] );
137 this->a02 = areaToInterpolate[1][0] -
138 ( 2.5 * areaToInterpolate[1][1] ) +
139 ( 2.0 * areaToInterpolate[1][2] ) -
140 ( 0.5 * areaToInterpolate[1][3] );
141 this->a03 = - ( 0.5 * areaToInterpolate[1][0] ) +
142 ( 1.5 * areaToInterpolate[1][1] ) -
143 ( 1.5 * areaToInterpolate[1][2] ) +
144 ( 0.5 * areaToInterpolate[1][3] );
145 this->a10 = - ( 0.5 * areaToInterpolate[0][1] ) +
146 ( 0.5 * areaToInterpolate[2][1] );
147 this->a11 = ( 0.25 * areaToInterpolate[0][0] ) -
148 ( 0.25 * areaToInterpolate[0][2] ) -
149 ( 0.25 * areaToInterpolate[2][0] ) +
150 ( 0.25 * areaToInterpolate[2][2] );
151 this->a12 = - ( 0.5 * areaToInterpolate[0][0] ) +
152 ( 1.25 * areaToInterpolate[0][1] ) -
153 areaToInterpolate[0][2] +
154 ( 0.25 * areaToInterpolate[0][3] ) +
155 ( 0.5 * areaToInterpolate[2][0] ) -
156 ( 1.25 * areaToInterpolate[2][1] ) +
157 areaToInterpolate[2][2] -
158 ( 0.25 * areaToInterpolate[2][3] );
159 this->a13 = ( 0.25 * areaToInterpolate[0][0] ) -
160 ( 0.75 * areaToInterpolate[0][1] ) +
161 ( 0.75 * areaToInterpolate[0][2] ) -
162 ( 0.25 * areaToInterpolate[0][3] ) -
163 ( 0.25 * areaToInterpolate[2][0] ) +
164 ( 0.75 * areaToInterpolate[2][1] ) -
165 ( 0.75 * areaToInterpolate[2][2] ) +
166 ( 0.25 * areaToInterpolate[2][3] );
167 this->a20 = areaToInterpolate[0][1] -
168 ( 2.5 * areaToInterpolate[1][1] ) +
169 ( 2.0 * areaToInterpolate[2][1] ) -
170 ( 0.5 * areaToInterpolate[3][1] );
171 this->a21 = - ( 0.5 * areaToInterpolate[0][0] ) +
172 ( 0.5 * areaToInterpolate[0][2] ) +
173 ( 1.25 * areaToInterpolate[1][0] ) -
174 ( 1.25 * areaToInterpolate[1][2] ) -
175 areaToInterpolate[2][0] + areaToInterpolate[2][2] +
176 ( 0.25 * areaToInterpolate[3][0] ) -
177 ( 0.25 * areaToInterpolate[3][2] );
178 this->a22 = areaToInterpolate[0][0] -
179 ( 2.5 * areaToInterpolate[0][1] ) +
180 ( 2.0 * areaToInterpolate[0][2] ) -
181 ( 0.5 * areaToInterpolate[0][3] ) -
182 ( 2.5 * areaToInterpolate[1][0] ) +
183 ( 6.25 * areaToInterpolate[1][1] ) -
184 ( 5.0 * areaToInterpolate[1][2] ) +
185 ( 1.25 * areaToInterpolate[1][3] ) +
186 ( 2.0 * areaToInterpolate[2][0] ) -
187 ( 5.0 * areaToInterpolate[2][1] ) +
188 ( 4.0 * areaToInterpolate[2][2] ) -
189 areaToInterpolate[2][3] -
190 ( 0.5 * areaToInterpolate[3][0] ) +
191 ( 1.25 * areaToInterpolate[3][1] ) -
192 areaToInterpolate[3][2] +
193 ( 0.25 * areaToInterpolate[3][3] );
194 this->a23 = - ( 0.5 * areaToInterpolate[0][0] ) +
195 ( 1.5 * areaToInterpolate[0][1] ) -
196 ( 1.5 * areaToInterpolate[0][2] ) +
197 ( 0.5 * areaToInterpolate[0][3] ) +
198 ( 1.25 * areaToInterpolate[1][0] ) -
199 ( 3.75 * areaToInterpolate[1][1] ) +
200 ( 3.75 * areaToInterpolate[1][2] ) -
201 ( 1.25 * areaToInterpolate[1][3] ) -
202 areaToInterpolate[2][0] +
203 ( 3.0 * areaToInterpolate[2][1] ) -
204 ( 3.0 * areaToInterpolate[2][2] ) +
205 areaToInterpolate[2][3] +
206 ( 0.25 * areaToInterpolate[3][0] ) -
207 ( 0.75 * areaToInterpolate[3][1] ) +
208 ( 0.75 * areaToInterpolate[3][2] ) -
209 ( 0.25 * areaToInterpolate[3][3] );
210 this->a30 = - ( 0.5 * areaToInterpolate[0][1] ) +
211 ( 1.5 * areaToInterpolate[1][1] ) -
212 ( 1.5 * areaToInterpolate[2][1] ) +
213 ( 0.5*areaToInterpolate[3][1] );
214 this->a31 = ( 0.25 * areaToInterpolate[0][0] ) -
215 ( 0.25 * areaToInterpolate[0][2] ) -
216 ( 0.75 * areaToInterpolate[1][0] ) +
217 ( 0.75 * areaToInterpolate[1][2] ) +
218 ( 0.75 * areaToInterpolate[2][0] ) -
219 ( 0.75 * areaToInterpolate[2][2] ) -
220 ( 0.25 * areaToInterpolate[3][0] ) +
221 ( 0.25 * areaToInterpolate[3][2] );
222 this->a32 = - ( 0.5 * areaToInterpolate[0][0] ) +
223 ( 1.25 * areaToInterpolate[0][1] ) -
224 areaToInterpolate[0][2] +
225 ( 0.25 * areaToInterpolate[0][3] ) +
226 ( 1.5 * areaToInterpolate[1][0] ) -
227 ( 3.75 * areaToInterpolate[1][1] ) +
228 ( 3.0 * areaToInterpolate[1][2] ) -
229 ( 0.75 * areaToInterpolate[1][3] ) -
230 ( 1.5 * areaToInterpolate[2][0] ) +
231 ( 3.75 * areaToInterpolate[2][1] ) -
232 ( 3.0 * areaToInterpolate[2][2] ) +
233 ( 0.75 * areaToInterpolate[2][3] ) +
234 ( 0.5 * areaToInterpolate[3][0] ) -
235 ( 1.25 * areaToInterpolate[3][1] ) +
236 areaToInterpolate[3][2] -
237 ( 0.25 * areaToInterpolate[3][3] );
238 this->a33 = ( 0.25 * areaToInterpolate[0][0] ) -
239 ( 0.75 * areaToInterpolate[0][1] ) +
240 ( 0.75 * areaToInterpolate[0][2] ) -
241 ( 0.25 * areaToInterpolate[0][3] ) -
242 ( 0.75 * areaToInterpolate[1][0] ) +
243 ( 2.25 * areaToInterpolate[1][1] ) -
244 ( 2.25 * areaToInterpolate[1][2] ) +
245 ( 0.75 * areaToInterpolate[1][3] ) +
246 ( 0.75 * areaToInterpolate[2][0] ) -
247 ( 2.25 * areaToInterpolate[2][1] ) +
248 ( 2.25 * areaToInterpolate[2][2] ) -
249 ( 0.75 * areaToInterpolate[2][3] ) -
250 ( 0.25 * areaToInterpolate[3][0] ) +
251 ( 0.75 * areaToInterpolate[3][1] ) -
252 ( 0.75 * areaToInterpolate[3][2] ) +
253 ( 0.25 * areaToInterpolate[3][3] );
271 proshade_double
getValue ( proshade_double x, proshade_double y )
274 if ( ( ( x < this->xStartIndex ) || ( x > ( this->xStartIndex + this->xRange ) ) ) ||
275 ( ( y < this->yStartIndex ) || ( y > ( this->yStartIndex + this->yRange ) ) ) )
277 if ( ( x < this->xStartIndex ) || ( x > ( this->xStartIndex + this->xRange ) ) ) { std::cout <<
"PROBLEM WITH LAT" << std::endl; }
278 if ( ( y < this->yStartIndex ) || ( y > ( this->yStartIndex + this->yRange ) ) ) { std::cout <<
"PROBLEM WITH LON" << std::endl; }
280 throw ProSHADE_exception (
"Requested bicubic interpolation outside of pre-computed\n : square.",
"ES00064", __FILE__, __LINE__, __func__,
"The supplied x or y value(s) is outside of the range of\n : the bi-cubic interpolator's pre-computed square. Please\n : make sure the start values were correctly supplied when\n : the constructor was called or create a new interpolator\n : for these values." );
284 proshade_double unitSquareX = ( x - this->xStartIndex ) / this->xRange;
285 proshade_double unitSquareY = ( y - this->yStartIndex ) / this->yRange;
288 proshade_double x2 = std::pow ( unitSquareX, 2.0 );
289 proshade_double x3 = std::pow ( unitSquareX, 3.0 );
290 proshade_double y2 = std::pow ( unitSquareY, 2.0 );
291 proshade_double y3 = std::pow ( unitSquareY, 3.0 );
294 return ( ( this->a00 + this->a01 * unitSquareY + this->a02 * y2 + this->a03 * y3 ) +
295 ( this->a10 + this->a11 * unitSquareY + this->a12 * y2 + this->a13 * y3 ) * unitSquareX +
296 ( this->a20 + this->a21 * unitSquareY + this->a22 * y2 + this->a23 * y3 ) * x2 +
297 ( this->a30 + this->a31 * unitSquareY + this->a32 * y2 + this->a33 * y3 ) * x3 );
301 void complexMultiplication ( proshade_double* r1, proshade_double* i1, proshade_double* r2, proshade_double* i2,
302 proshade_double* retReal, proshade_double* retImag );
304 proshade_double* retReal, proshade_double* retImag );
307 void vectorMeanAndSD ( std::vector<proshade_double>* vec, proshade_double*& ret );
309 void arrayMedianAndIQR ( proshade_double* vec, proshade_unsign vecSize, proshade_double*& ret );
310 proshade_double
pearsonCorrCoeff ( proshade_double* valSet1, proshade_double* valSet2, proshade_unsign length );
312 proshade_unsign taylorSeriesCap );
313 void getGLPolyAtZero ( proshade_unsign order, proshade_double *polyValue, proshade_double *deriValue );
314 void getGLFirstEvenRoot ( proshade_double polyAtZero, proshade_unsign order, proshade_double *abscAtZero,
315 proshade_double *weighAtZero, proshade_unsign taylorSeriesCap );
316 proshade_double
evaluateGLSeries ( proshade_double *series, proshade_double target, proshade_unsign terms );
317 proshade_double
advanceGLPolyValue ( proshade_double from, proshade_double to, proshade_double valAtFrom,
318 proshade_unsign noSteps, proshade_unsign taylorSeriesCap );
320 proshade_unsign taylorSeriesCap );
322 proshade_double* abscissas, proshade_double* weights, proshade_double integralOverRange,
323 proshade_double maxSphereDists );
325 proshade_double* abscissas, proshade_double* weights, proshade_double integralOverRange,
326 proshade_double maxSphereDists, proshade_double* retReal, proshade_double* retImag );
329 void getEulerZXZFromSOFTPosition ( proshade_signed band, proshade_signed x, proshade_signed y, proshade_signed z, proshade_double* eulerAlpha,
330 proshade_double* eulerBeta, proshade_double* eulerGamma );
331 void getSOFTPositionFromEulerZXZ ( proshade_signed band, proshade_double eulerAlpha, proshade_double eulerBeta, proshade_double eulerGamma,
332 proshade_double* x, proshade_double* y, proshade_double* z );
335 void getAxisAngleFromRotationMatrix ( proshade_double* rotMat, proshade_double* x, proshade_double* y, proshade_double* z, proshade_double* ang, proshade_signed verbose = 1 );
336 void getAxisAngleFromRotationMatrix ( std::vector< proshade_double >* rotMat, proshade_double* x, proshade_double* y, proshade_double* z, proshade_double* ang, proshade_signed verbose = 1 );
339 void getEulerZXZFromRotMatrix ( proshade_double* rotMat, proshade_double* eA, proshade_double* eB, proshade_double* eG );
340 void getEulerZXZFromAngleAxis ( proshade_double axX, proshade_double axY, proshade_double axZ, proshade_double axAng, proshade_double* eA,
341 proshade_double* eB, proshade_double* eG );
342 void multiplyTwoSquareMatrices ( proshade_double* A, proshade_double* B, proshade_double* res, proshade_unsign dim = 3 );
344 proshade_double
normalDistributionValue ( proshade_double mean, proshade_double standardDev, proshade_double value );
345 proshade_double
computeDotProduct ( proshade_double* x1, proshade_double* y1, proshade_double* z1, proshade_double* x2, proshade_double* y2,
346 proshade_double* z2 );
347 proshade_double
computeDotProduct ( proshade_double x1, proshade_double y1, proshade_double z1, proshade_double x2, proshade_double y2,
348 proshade_double z2 );
349 proshade_double*
computeCrossProduct ( proshade_double* x1, proshade_double* y1, proshade_double* z1, proshade_double* x2, proshade_double* y2,
350 proshade_double* z2 );
351 proshade_double*
computeCrossProduct ( proshade_double x1, proshade_double y1, proshade_double z1, proshade_double x2, proshade_double y2,
352 proshade_double z2 );
360 proshade_double*
findRotMatMatchingVectors ( proshade_double x1, proshade_double y1, proshade_double z1, proshade_double x2, proshade_double y2,
361 proshade_double z2 );
363 std::vector < proshade_double >
findVectorFromTwoVAndTwoD ( proshade_double x1, proshade_double y1, proshade_double z1, proshade_double x2, proshade_double y2,
364 proshade_double z2, proshade_double dot1, proshade_double dot2 );
365 std::vector < proshade_double >
findVectorFromThreeVAndThreeD ( proshade_double x1, proshade_double y1, proshade_double z1, proshade_double x2, proshade_double y2,
366 proshade_double z2, proshade_double x3, proshade_double y3, proshade_double z3, proshade_double dot1,
367 proshade_double dot2, proshade_double dot3 );
368 std::vector< proshade_double >
multiplyGroupElementMatrices ( std::vector< proshade_double >* el1, std::vector< proshade_double >* el2 );
369 bool rotationMatrixSimilarity ( std::vector< proshade_double >* mat1, std::vector< proshade_double >* mat2, proshade_double tolerance = 0.1 );
370 bool vectorOrientationSimilarity ( proshade_double a1, proshade_double a2, proshade_double a3, proshade_double b1, proshade_double b2,
371 proshade_double b3, proshade_double tolerance = 0.1 );
373 proshade_double b3, proshade_double tolerance = 0.1 );
374 void optimiseAxisBiCubicInterpolation ( proshade_double* bestLattitude, proshade_double* bestLongitude, proshade_double* bestSum, std::vector<proshade_unsign>* sphereList,
375 std::vector<ProSHADE_internal_spheres::ProSHADE_rotFun_sphere*>* sphereMappedRotFun, proshade_double step = 0.05 );
377 std::vector<ProSHADE_internal_maths::BicubicInterpolator*>* interpols, std::vector<ProSHADE_internal_spheres::ProSHADE_rotFun_sphere*>* sphereMappedRotFun );
379 std::vector<ProSHADE_internal_maths::BicubicInterpolator*>* interpols, std::vector<ProSHADE_internal_spheres::ProSHADE_rotFun_sphere*>* sphereMappedRotFun );
381 std::vector<ProSHADE_internal_maths::BicubicInterpolator*>* interpols, std::vector<ProSHADE_internal_spheres::ProSHADE_rotFun_sphere*>* sphereMappedRotFun );
383 std::vector<ProSHADE_internal_maths::BicubicInterpolator*>* interpols, std::vector<ProSHADE_internal_spheres::ProSHADE_rotFun_sphere*>* sphereMappedRotFun );
384 bool isAxisUnique ( std::vector< proshade_double* >* CSymList, proshade_double* axis, proshade_double tolerance = 0.1,
bool improve =
false );
385 bool isAxisUnique ( std::vector< proshade_double* >* CSymList, proshade_double X, proshade_double Y, proshade_double Z, proshade_double fold, proshade_double tolerance );
386 std::vector< proshade_unsign >
findAllPrimes ( proshade_unsign upTo );
387 proshade_double
computeGaussian ( proshade_double val, proshade_double sigma );
388 std::vector < proshade_double >
smoothen1D ( proshade_double step, proshade_signed windowSize, proshade_double sigma, std::vector< proshade_double > data );
389 proshade_single
getResolutionOfReflection ( proshade_single h, proshade_single k, proshade_single l, proshade_single xDim, proshade_single yDim, proshade_single zDim );
390 void binReciprocalSpaceReflections ( proshade_unsign xInds, proshade_unsign yInds, proshade_unsign zInds, proshade_signed* noBin, proshade_signed*& binIndexing );
391 proshade_double
computeFSC ( fftw_complex *fCoeffs1, fftw_complex *fCoeffs2, proshade_unsign xInds, proshade_unsign yInds, proshade_unsign zInds,
392 proshade_signed noBins, proshade_signed* binIndexing, proshade_double**& binData, proshade_signed*& binCounts, proshade_double*& fscByBin );
393 void computeFSCWeightByBin ( proshade_double*& weights1, proshade_double*& weights2, proshade_signed* binIndexing, proshade_double* fscByBin, proshade_signed noBins,
394 proshade_signed xDim, proshade_signed yDim, proshade_signed zDim );
395 proshade_double
computeTheFValue ( proshade_complex* fCoeffs, proshade_double* weights, proshade_signed xDim, proshade_signed yDim, proshade_signed zDim );
396 void computeTrFunDerivatives ( proshade_complex* fCoeffs, proshade_double* weights1, proshade_double* weights2, proshade_signed xDim, proshade_signed yDim, proshade_signed zDim,
397 proshade_double*& firstDers, proshade_double*& secondDers );
398 proshade_double*
computeTrFunStep ( proshade_double* firstDers, proshade_double* secondDers );
399 std::vector< proshade_signed >
findPeaks1D ( std::vector< proshade_double > data );
400 proshade_double
findTopGroupSmooth ( std::vector< proshade_double* >* CSym,
size_t peakPos, proshade_double step, proshade_double sigma, proshade_signed windowSize, proshade_double maxLim = 0.9 );
401 proshade_double
findTopGroupSmooth ( std::vector< std::vector< proshade_double > >* CSym,
size_t peakPos, proshade_double step, proshade_double sigma, proshade_signed windowSize,
402 proshade_double maxLim = 0.9 );
403 void combineFourierForTranslation ( fftw_complex* tmpOut1, fftw_complex* tmpOut2, fftw_complex*& resOut, proshade_unsign xD, proshade_unsign yD, proshade_unsign zD );
404 void findHighestValueInMap ( fftw_complex* resIn, proshade_unsign xD, proshade_unsign yD, proshade_unsign zD, proshade_double* trsX,
405 proshade_double* trsY, proshade_double* trsZ, proshade_double* mapPeak );