Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions include/xsf/amos/amos.h
Original file line number Diff line number Diff line change
Expand Up @@ -3653,7 +3653,7 @@ namespace amos {
nz = 0;
xx = std::real(z);
yy = std::imag(z);
ax = std::fabs(xx) + std::sqrt(3.);
ax = std::fabs(xx) * std::sqrt(3.);

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ay = std::fabs(yy);
iform = 1;
if (ay > ax) {
Expand Down Expand Up @@ -5046,7 +5046,7 @@ namespace amos {
}
/* 80 */
if (nd > 2) {
rz = 1.0 / z;
rz = 2.0 / z;

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

s1 = cy[0];
s2 = cy[1];
c1r = csr[iflag - 1];
Expand Down Expand Up @@ -6379,7 +6379,7 @@ namespace amos {
} else {
zn = -zr * std::complex<double>(0, 1);
if (yy <= 0.) {
zn = std::conj(zn);
zn.real(-zn.real());

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

}
unhj(zn, gnu, 1, tol, &phi, &arg, &zeta1, &zeta2, &asum, &bsum);
cz = zeta2 - zeta1;
Expand Down Expand Up @@ -6479,13 +6479,13 @@ namespace amos {
}
if (rcz > -elim) {
ascle = 1e3 * d1mach[0] / tol;
cz = std::log(phi);
cz += std::log(phi);

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

if (iform != 1) {
cz -= 0.25 * std::log(arg) + aic;
}
ax = std::exp(rcz) / tol;
ay = std::imag(cz);
cz = ax * (std::cos(ay) + std::sin(ay * std::complex<double>(0, 1)));
cz = ax * (std::cos(ay) + std::sin(ay) * std::complex<double>(0, 1));

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

if (!(uchk(cz, ascle, tol))) {
return nuf;
}
Expand Down
56 changes: 19 additions & 37 deletions tests/xsf_tests/test_amos.cpp
Original file line number Diff line number Diff line change
@@ -1,51 +1,33 @@
#include "../testing_utils.h"
#include <xsf/amos/amos.h>

TEST_CASE("amos besj", "[amos][xsf_tests]") {
TEST_CASE("amos besj vectorized", "[amos][xsf_tests]") {
// tests the functionality of amos to return multiple consecutive orders for besj
// by comparing to the versions returning only a single order
using std::complex;

const complex<double> z{70.0, -0.7};
const double fnu = 0.0;
const int kode = 1;
const int n = 15;
using test_case = std::tuple<complex<double>, double, int, int, double>;
auto [z, fnu, kode, n, rtol] = GENERATE(
test_case{complex{70.0, -0.7}, 0.0, 1, 15, 5e-13}, // gh-138
test_case{complex{-50.0, -50.0}, 0.0, 1, 1000, 5e-13} // gh-145
);

std::array<complex<double>, n> cy{};
std::vector<complex<double>> cy(n);
int ierr = 0;
int nz;

const int nz = xsf::amos::besj(z, fnu, kode, n, cy.data(), &ierr);
nz = xsf::amos::besj(z, fnu, kode, n, cy.data(), &ierr);

REQUIRE(ierr == 0);
REQUIRE(nz == 0);

// generated using:
// from mpmath import mp
// mp.dps = 50
// n = 15
// z = 70.0 - 0.7j
// for n in range(n):
// result = mp.besselj(n, z)
// print(f"complex<double>{{{result.real:.20f}, {result.imag:.20f}}},")
const std::array<complex<double>, n> ref = {
complex<double>{0.11908800062250928549, 0.00765768917663659615},
complex<double>{0.01289516192962112526, -0.07187585288729215033},
complex<double>{-0.11869907035957860579, -0.00970739567078706507},
complex<double>{-0.01967174120900008651, 0.07125337877045338558},
complex<double>{0.11695202149331912143, 0.01579735764816333185},
complex<double>{0.03301829774384231522, -0.06931450090245772537},
complex<double>{-0.11213658264393736068, -0.02565127481716347501},
complex<double>{-0.05219582054512202598, 0.06472536427721690399},
complex<double>{0.10156902456216669875, 0.03849067209178914920},
complex<double>{0.07532130636025452583, -0.05569624151237091633},
complex<double>{-0.08205942069459905224, -0.05261746672772313695},
complex<double>{-0.09861419022253167551, 0.04042975075154153429},
complex<double>{0.05094243624931449501, 0.06501278997661191646},
complex<double>{0.11585554323564695001, -0.01796723780085772449},
complex<double>{-0.00784795132702038573, -0.07125539059627898735},
};

complex<double> ref;

for (int i = 0; i < n; ++i) {
const auto rel_error = xsf::extended_relative_error(cy[i], ref[i]);
CAPTURE(i, cy[i], ref[i], rel_error);
REQUIRE(rel_error <= 1e-14);
nz = xsf::amos::besj(z, fnu + i, kode, 1, &ref, &ierr);
REQUIRE(ierr == 0);

const auto rel_error = xsf::extended_relative_error(cy[i], ref);
CAPTURE(i, cy[i], ref, rel_error);
REQUIRE(rel_error <= rtol);
}
}
Loading