module Petzval.Examples where

import Control.Lens
import Data.Either
import Petzval.Optics
import Petzval.Optics.RTM
import Petzval.System
import Petzval.Trace
import Petzval.Calculations
import Linear
import Numeric.AD.Mode (Scalar, Mode)

import qualified Debug.Trace 

bk7 = SellemeierMat [ (1.03961212,  6.00069867e-3 )
                    , (0.231792344, 2.00179144e-2 )
                    , (1.01046945,  103.560653 )]
n_sk16 = SellemeierMat [ (1.343177740, 0.007046873)
                       , (0.241144399, 0.0229005000)
                       , (0.994317969, 92.75085260)]

n_ssk8 = SellemeierMat [ (1.44857867, 1.17965926e-01)
                       , (1.06937528, 8.69310149E-03)
                       , (4.21566593E-02, 1.11300666E+02) ]

system1 :: Num a => [Element SellemeierMat a]
system1 =
  [ Stop{_thickness = 0, _outsideRadius=5}
  , Surface{_material = bk7, _thickness = 10, _roc = 100, _outsideRadius=10}
  , Surface{_material = air, _thickness = 95, _roc = -100, _outsideRadius=10}
  ]

system2 :: Fractional a => [Element ConstMat a]
system2 =
  [
    Surface{_material = s_sk16, _outsideRadius=11.5, _roc=42.98790, _thickness = 4}
  , Surface{_material = air, _outsideRadius = 11.5, _roc = -248.07740, _thickness = 10.51018}
  , Surface{_material = s_f4, _outsideRadius = 9.852, _roc = -38.21035, _thickness = 2.5}
  , Surface{_material = air, _outsideRadius = 8.885, _roc = 43.95894, _thickness = 0}
  , Stop{_outsideRadius = 8.6762522794, _thickness = 9.86946}
  , Surface{_material=s_sk16, _outsideRadius = 11, _roc=656.66349, _thickness = 4.5}
  , Surface{_material = air, _outsideRadius = 11, _roc = -33.50754, _thickness = 86.48643}
  ]
  where s_sk16 = ConstMat 1.620408
        s_f4 = ConstMat 1.616589

trace1 :: Show a => String -> a -> a
trace1 msg = (msg++) . show >>= Debug.Trace.trace
trace1 _ = id
merit :: (Material mat, RealFloat a, Scalar a ~ Double, Epsilon a, Mode a, Show a) => [Element mat a] -> a
-- merit :: (Material mat) => [Element mat Double] -> Double
merit system = result 
  where baked = bake 0.587618 system
        ep = trace1 "EP: " $ entrancePupil baked
        result = sum
          . map (^2)
          . map (\ angle -> rmsSize
                  . toListOf (each._pos._xy)
                  -- . (\x -> x :: [Ray a])
                  . rights
                  . map fst
                  . map (trace1 "Trace: " . raytrace baked . trace1 "Ray: " . createRay Nothing ep angle . trace1 "Pupil: ")
                  $ spiralPattern 10)
          $ [0, 7.07, 15]

cast1 system = result
  where baked = bake 0.587618 system
        ep = trace1 "EP: " $ entrancePupil baked
        result = raytrace baked . trace1 "Ray: " . createRay Nothing ep 15 . trace1 "Pupil: "