Rethink damage data type, work on damage sensors

src/Dodge/Placement/Instance/Sensor.hs

@@ -1,5 +1,6 @@
 module Dodge.Placement.Instance.Sensor
     ( lightSensor
+    , damageSensor
     ) where
 import Color
 import Dodge.LightSource
@@ -14,41 +15,42 @@ import Control.Lens
 import Data.Either
 
 damageSensor 
-    :: (DamageType -> Either Int Int) -- Left gets sensed, Right does damage
+    :: DamageType -- Left gets sensed, Right does damage
     -> Float
     -> (Machine -> World -> World)
     -> PlacementSpot -> Placement
-damageSensor damF wdth upf ps = pContID ps ( PutLS theLS)
+damageSensor damF wdth upf ps = pContID ps (PutLS theLS)
     $ \lsid -> Just $ spNoID ps $ PutMachine yellow (reverse $ square wdth) defaultMachine
         { _mcDraw = sensorSPic wdth
-        , _mcUpdate = \mc w -> upf mc $ sensorUpdate damF mc w
+        , _mcUpdate = \mc -> upf mc . sensorUpdate damF mc
         , _mcLSs = [lsid]
         }
   where
     theLS = lsPosCol (V3 0 0 30) 0.1
 
 lightSensor :: Float -> (Machine -> World -> World) -> PlacementSpot -> Placement
-lightSensor = damageSensor senseLasering
+lightSensor = damageSensor Lasering
 
-senseLasering :: DamageType -> Either Int Int
-senseLasering Lasering {_dmAmount = x} = Left x
-senseLasering _ = Right 0
-
-sensorUpdate :: (DamageType -> Either Int Int) -> Machine -> World -> World
+sensorUpdate :: DamageType -> Machine -> World -> World
 sensorUpdate damF mc w = w & machines . ix mcid %~ upmc
     & lightSources . ix lsid %~ upls
   where
-    upmc = ( mcSensor %~ \x' -> min 1000 (max 0 (x' - 5 + newSense)) )
+    upmc = ( mcSensorAmount %~ \x' -> min 1000 (max 0 (x' - 5 + newSense)) )
         . ( mcHP -~ sum dam )
         . (mcDamage .~ [])
-    x = _mcSensor mc
+    x = _mcSensorAmount mc
     mcid = _mcID mc
     lsid = head (_mcLSs mc)
-    (senseData,dam) = partitionEithers $ map damF $ _mcDamage mc
+    (senseData,dam) = partitionEithers $ map (damageUsing damF) $ _mcDamage mc
     newSense = sum senseData
     ni = fromIntegral x / 1000
     upls = lsParam . lsCol .~ V3 ni ni ni
 
+damageUsing :: DamageType -> Damage -> Either Int Int
+damageUsing dt dm
+    | _dmType dm == dt = Left $ _dmAmount dm
+    | otherwise = Right 0
+
 sensorSPic :: Float -> Machine -> SPic
 sensorSPic wdth _ = ( colorSH yellow $ upperPrismPoly 25 (square wdth)
     , mempty )