{-# LANGUAGE PostfixOperators #-}
{-# LANGUAGE ScopedTypeVariables #-}
module Math.NumberTheory.Zeta.Dirichlet
( betas
, betasEven
, betasOdd
) where
import Data.ExactPi
import Data.List.Infinite (Infinite(..), (....))
import Data.List.NonEmpty (NonEmpty(..))
import qualified Data.List.Infinite as Inf
import Data.Ratio ((%))
import Math.NumberTheory.Recurrences (euler, factorial)
import Math.NumberTheory.Zeta.Hurwitz (zetaHurwitz)
import Math.NumberTheory.Zeta.Utils (skipOdds)
betasOdd :: Infinite ExactPi
betasOdd :: Infinite ExactPi
betasOdd = forall a b c.
(a -> b -> c) -> Infinite a -> Infinite b -> Infinite c
Inf.zipWith Integer -> Rational -> ExactPi
Exact ((Integer
1, Integer
3)....) forall a b. (a -> b) -> a -> b
$ forall a b c d e.
(a -> b -> c -> d -> e)
-> Infinite a
-> Infinite b
-> Infinite c
-> Infinite d
-> Infinite e
Inf.zipWith4
(\Rational
sgn Integer
denom Integer
eul Integer
twos -> Rational
sgn forall a. Num a => a -> a -> a
* (Integer
eul forall a. Integral a => a -> a -> Ratio a
% (Integer
twos forall a. Num a => a -> a -> a
* Integer
denom)))
(forall a. NonEmpty a -> Infinite a
Inf.cycle (Rational
1 forall a. a -> [a] -> NonEmpty a
:| [-Rational
1]))
(forall a. Infinite a -> Infinite a
skipOdds forall a. (Num a, Enum a) => Infinite a
factorial)
(forall a. Infinite a -> Infinite a
skipOdds forall a. Integral a => Infinite a
euler)
(forall a. (a -> a) -> a -> Infinite a
Inf.iterate (Integer
4 *) Integer
4)
betasEven :: forall a. (Floating a, Ord a) => a -> Infinite a
betasEven :: forall a. (Floating a, Ord a) => a -> Infinite a
betasEven a
eps = (a
1 forall a. Fractional a => a -> a -> a
/ a
2) forall a. a -> Infinite a -> Infinite a
:< Infinite a
hurwitz
where
hurwitz :: Infinite a
hurwitz :: Infinite a
hurwitz =
forall a b c d.
(a -> b -> c -> d)
-> Infinite a -> Infinite b -> Infinite c -> Infinite d
Inf.zipWith3 (\a
quarter a
threeQuarters a
four ->
(a
quarter forall a. Num a => a -> a -> a
- a
threeQuarters) forall a. Fractional a => a -> a -> a
/ a
four)
(forall a. Infinite a -> Infinite a
Inf.tail forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Infinite a -> Infinite a
skipOdds forall a b. (a -> b) -> a -> b
$ forall a. (Floating a, Ord a) => a -> a -> Infinite a
zetaHurwitz a
eps a
0.25)
(forall a. Infinite a -> Infinite a
Inf.tail forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Infinite a -> Infinite a
skipOdds forall a b. (a -> b) -> a -> b
$ forall a. (Floating a, Ord a) => a -> a -> Infinite a
zetaHurwitz a
eps a
0.75)
(forall a. (a -> a) -> a -> Infinite a
Inf.iterate (a
16 *) a
16)
betas :: (Floating a, Ord a) => a -> Infinite a
betas :: forall a. (Floating a, Ord a) => a -> Infinite a
betas a
eps = a
e forall a. a -> Infinite a -> Infinite a
:< a
o forall a. a -> Infinite a -> Infinite a
:< forall a. (a -> a -> a) -> Infinite a -> Infinite a
Inf.scanl1 forall {a}. (Ord a, Fractional a) => a -> a -> a
f (forall a. Infinite a -> Infinite a -> Infinite a
Inf.interleave Infinite a
es Infinite a
os)
where
a
e :< Infinite a
es = forall a. (Floating a, Ord a) => a -> Infinite a
betasEven a
eps
a
o :< Infinite a
os = forall a b. (a -> b) -> Infinite a -> Infinite b
Inf.map (forall a. Fractional a => (a -> a -> Bool) -> [Rational] -> a
getRationalLimit (\a
a a
b -> forall a. Num a => a -> a
abs (a
a forall a. Num a => a -> a -> a
- a
b) forall a. Ord a => a -> a -> Bool
< a
eps) forall b c a. (b -> c) -> (a -> b) -> a -> c
. ExactPi -> [Rational]
rationalApproximations) Infinite ExactPi
betasOdd
f :: a -> a -> a
f a
x a
y = a
1 forall a. Ord a => a -> a -> a
`min` (a
y forall a. Ord a => a -> a -> a
`max` (a
1 forall a. Num a => a -> a -> a
+ (a
x forall a. Num a => a -> a -> a
- a
1) forall a. Fractional a => a -> a -> a
/ a
2))