GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Cleaned up the implementation of digamma

Browse Source
This commit is contained in:
Benoit Steiner 2016-04-13 14:24:06 -07:00
parent d5de1a8220
commit 2986253259
1 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
Eigen/src/Core/SpecialFunctions.h
View File

@ -281,20 +281,18 @@ struct digamma_impl {
*/ */
Scalar p, q, nz, s, w, y; Scalar p, q, nz, s, w, y;
bool negative; bool negative = false;
const Scalar maxnum = NumTraits<Scalar>::infinity(); const Scalar maxnum = NumTraits<Scalar>::infinity();
const Scalar m_pi = EIGEN_PI; const Scalar m_pi(EIGEN_PI);
negative = 0; const Scalar zero = Scalar(0);
nz = 0.0; const Scalar one = Scalar(1);
const Scalar half = Scalar(0.5);
const Scalar zero = 0.0; nz = zero;
const Scalar one = 1.0;
const Scalar half = 0.5;
if (x <= zero) { if (x <= zero) {
negative = one; negative = true;
q = x; q = x;
p = numext::floor(q); p = numext::floor(q);
if (p == q) { if (p == q) {
@ -463,7 +461,7 @@ template <typename Scalar>
struct igammac_impl { struct igammac_impl {
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
static Scalar run(Scalar a, Scalar x) { static Scalar run(Scalar a, Scalar x) {
/* igamc() /* igamc()
* *
* Incomplete gamma integral (modified for Eigen) * Incomplete gamma integral (modified for Eigen)
* *
@ -618,7 +616,7 @@ template <typename Scalar>
struct igamma_impl { struct igamma_impl {
EIGEN_DEVICE_FUNC EIGEN_DEVICE_FUNC
static Scalar run(Scalar a, Scalar x) { static Scalar run(Scalar a, Scalar x) {
/* igam() /* igam()
* Incomplete gamma integral * Incomplete gamma integral
* *
* *