Devices/mTaskDevice.icl

   1 implementation module Devices.mTaskDevice
   2
   3 from StdFunc import flip
   4 import Generics.gCons
   5 import mTaskInterpret
   6 import iTasks
   7 import iTasksTTY
   8 import TTY
   9 import qualified Data.Map as DM
  10 import Utils.SDS
  11 import Utils.Devices
  12
  13 import GenBimap
  14 import Devices.mTaskSerial
  15 import Devices.mTaskTCP
  16 import iTasks._Framework.Store
  17 import iTasks.UI.Definition, iTasks.UI.Editor, iTasks.UI.Editor.Builtin, iTasks.UI.Editor.Common, iTasks.UI.Layout.Default, iTasks.UI.Layout.Common
  18
  19 from Data.Func import $
  20
  21 derive class iTask MTaskDevice, MTaskResource, MTaskDeviceSpec, MTaskMSGRecv, MTaskMSGSend, BCShare
  22 derive conses MTaskResource, TTYSettings, BaudRate, Parity, ByteSize, TCPSettings
  23 derive consName MTaskResource, TTYSettings, BaudRate, Parity, ByteSize, TCPSettings
  24
  25 instance == MTaskDevice where
  26         (==) a b = a.deviceChannels == b.deviceChannels
  27
  28 startupDevices :: Task [MTaskDevice]
  29 startupDevices = upd (map reset) deviceStore
  30         where reset d = {d & deviceTask=Nothing, deviceTasks=[], deviceError=Nothing}
  31
  32 makeDevice :: String MTaskResource -> Task MTaskDevice
  33 makeDevice name res = get randomInt @ \rand->{MTaskDevice
  34                 |deviceChannels=name +++ toString rand
  35                 ,deviceName=name
  36                 ,deviceTasks=[]
  37                 ,deviceTask=Nothing
  38                 ,deviceError=Nothing
  39                 ,deviceData=res
  40                 ,deviceSpec=Nothing}
  41
  42 getSynFun :: MTaskResource -> ((Shared Channels) -> Task ())
  43 getSynFun (TCPDevice t) = synFun t
  44 getSynFun (SerialDevice t) = synFun t
  45
  46 addDevice :: (Shared [MTaskDevice]) (MTaskDevice (Shared Channels) -> Task ()) -> Task String
  47 addDevice devices processFun
  48         =   enterChoice "Device type" [] (map consName{|*|} deviceTypes)
  49         >&^ \sh->whileUnchanged sh $ \mty->case mty of
  50                 Nothing = viewInformation "No type selected yet" [] ""
  51                 Just ty = enterInformation "Name" [] -&&- deviceSettings ty
  52                         >>= \(name, settings)->makeDevice name settings
  53                         >>= \dev->upd (\l->[dev:l]) devices
  54                         >>| connectDevice processFun dev
  55                         @! ""
  56         where
  57                 deviceSettings "SerialDevice" = getmTaskSerialDevice
  58                 deviceSettings "TCPDevice" = getmTaskTCPDevice
  59
  60                 deviceTypes :: [MTaskResource]
  61                 deviceTypes = conses{|*|}
  62
  63 connectDevice :: (MTaskDevice (Shared Channels) -> Task ()) MTaskDevice -> Task ()
  64 connectDevice pf d = let ch = channels d in appendTopLevelTask 'DM'.newMap True
  65         (pf d ch -||- catchAll (getSynFun d.deviceData ch) errorHandle)
  66         >>= \tid->withDevices d (\d->{d&deviceTask=Just tid,deviceError=Nothing}) @! ()
  67         where
  68                 errorHandle e = withDevices d (\d->{d&deviceTask=Nothing,deviceError=Just e})
  69
  70 manageDevices :: (MTaskDevice (Shared Channels) -> Task ()) [MTaskDevice] -> Task ()
  71 manageDevices processFun ds = anyTask [
  72                 addDevice deviceStore processFun <<@ Title "Add new device" @! ():
  73                         [viewDevice processFun d
  74                                 <<@ Title d.deviceName\\d<-ds]]
  75         <<@ ArrangeWithTabs @! ()
  76
  77 viewDevice :: (MTaskDevice (Shared Channels) -> Task ()) MTaskDevice -> Task ()
  78 viewDevice pf d = forever $ traceValue "viewDevice" >>| anyTask
  79                 [viewInformation "Device settings" [] d @! ()
  80                 ,viewSharedInformation "Channels" [ViewAs dropEmpty] (channels d) @! ()
  81                 ,forever $
  82                         enterChoice "Delete task on device" [ChooseFromGrid id] d.deviceTasks
  83                         >>* [OnAction (Action "Delete") $ ifValue (\t->t.ident <> -1) (deviceTaskDelete d)]
  84                         @! ()
  85                 ] <<@ ArrangeHorizontal
  86                 >>* [OnAction (Action "Delete Device") (always $ deleteDevice d):
  87                                 if (isJust d.deviceTask) []
  88                                 [OnAction (Action "Connect") (always $ connectDevice pf d)]]
  89         where
  90                 dropEmpty (r,s,ss) = (filter ((=!=)MTEmpty) r,s,ss)
  91
  92 deleteDevice :: MTaskDevice -> Task ()
  93 deleteDevice d = upd (\(r,s,ss)->(r,s,True)) (channels d)
  94         >>| maybe (treturn ()) (flip removeTask topLevelTasks) d.deviceTask
  95         >>| upd (filter ((==)d)) deviceStore
  96         @! ()
  97
  98 sendToDevice :: String (Main (ByteCode () Stmt)) (MTaskDevice, MTaskInterval) -> Task ()
  99 sendToDevice wta mTask (device, timeout) =
 100                 traceValue "starting to send"
 101         >>| get bcStateStore @ toMessages timeout o toRealByteCode (unMain mTask)
 102         >>= \(msgs, st1)->traceValue "messages generated"
 103         >>| set st1 bcStateStore
 104         >>| traceValue "bcstate store updated"
 105         >>| toSDSRecords st1
 106         >>= \sdss->traceValue "Shares created"
 107         >>| set sdss sdsStore//MTaskShareaddToSDSShare
 108         >>| traceValue "Shares store updated"
 109         >>| sendMessages msgs device
 110         >>| traceValue "Messages sent"
 111         >>| makeTask wta -1
 112         >>= \t->traceValue "Task made"
 113         >>| withDevices device (addTask t)
 114         >>| traceValue "Tasks share updated"
 115         @! ()
 116         where
 117                 sharename i = device.deviceChannels +++ "-" +++ toString i
 118                 toSDSRecords st = sequence "" [makeShare wta sdsi sdsval\\{sdsi,sdspub,sdsval}<-st.sdss]// | sdspub]
 119
 120                 addTask :: MTaskTask MTaskDevice -> MTaskDevice
 121                 addTask task device = {device & deviceTasks=[task:device.deviceTasks]}
 122
 123 sendMessages :: [MTaskMSGSend] -> (MTaskDevice -> Task Channels)
 124 sendMessages msgs = upd (\(r,s,ss)->(r,msgs++s,ss)) o channels
 125
 126 withDevices :: MTaskDevice (MTaskDevice -> MTaskDevice) -> Task ()
 127 withDevices a trans = upd (map \b->if (b == a) (trans b) b) deviceStore @! ()
 128
 129 deviceTaskAcked :: MTaskDevice Int -> Task ()
 130 deviceTaskAcked dev i
 131         = withDevices dev (\d->{d&deviceTasks=ackFirst d.deviceTasks})
 132         where
 133                 ackFirst :: [MTaskTask] -> [MTaskTask]
 134                 ackFirst [] = []
 135                 ackFirst [t:ts] = if (t.ident == -1)
 136                         [{t & ident=i}:ts] [t:ackFirst ts]
 137
 138 deviceTaskDelete :: MTaskDevice MTaskTask -> Task ()
 139 deviceTaskDelete dev task = sendMessages [MTTaskDel task.ident] dev
 140         >>| upd (\ds->filter ((<>) dev) ds) deviceStore @! ()
 141
 142 deviceTaskDeleteAcked :: MTaskDevice Int -> Task ()
 143 deviceTaskDeleteAcked d i = withDevices d $ deleteTask
 144         where deleteTask d = {d & deviceTasks=[s\\s<-d.deviceTasks | i <> s.ident]}
 145
 146 deviceAddSpec :: MTaskDevice MTaskDeviceSpec -> Task ()
 147 deviceAddSpec d s = withDevices d $ \r->{r&deviceSpec=Just s}