OpenAPS.swift 36 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880
  1. import Combine
  2. import CoreData
  3. import Foundation
  4. import JavaScriptCore
  5. final class OpenAPS {
  6. private let jsWorker = JavaScriptWorker()
  7. private let processQueue = DispatchQueue(label: "OpenAPS.processQueue", qos: .utility)
  8. private let storage: FileStorage
  9. let context = CoreDataStack.shared.newTaskContext()
  10. let jsonConverter = JSONConverter()
  11. init(storage: FileStorage) {
  12. self.storage = storage
  13. }
  14. // Helper function to convert a Decimal? to NSDecimalNumber?
  15. func decimalToNSDecimalNumber(_ value: Decimal?) -> NSDecimalNumber? {
  16. guard let value = value else { return nil }
  17. return NSDecimalNumber(decimal: value)
  18. }
  19. // Use the helper function for cleaner code
  20. func processDetermination(_ determination: Determination) {
  21. context.perform {
  22. let newOrefDetermination = OrefDetermination(context: self.context)
  23. newOrefDetermination.id = UUID()
  24. newOrefDetermination.totalDailyDose = self.decimalToNSDecimalNumber(determination.tdd)
  25. newOrefDetermination.insulinSensitivity = self.decimalToNSDecimalNumber(determination.isf)
  26. newOrefDetermination.currentTarget = self.decimalToNSDecimalNumber(determination.current_target)
  27. newOrefDetermination.eventualBG = determination.eventualBG.map(NSDecimalNumber.init)
  28. newOrefDetermination.deliverAt = determination.deliverAt
  29. newOrefDetermination.insulinForManualBolus = self.decimalToNSDecimalNumber(determination.insulinForManualBolus)
  30. newOrefDetermination.carbRatio = self.decimalToNSDecimalNumber(determination.carbRatio)
  31. newOrefDetermination.glucose = self.decimalToNSDecimalNumber(determination.bg)
  32. newOrefDetermination.reservoir = self.decimalToNSDecimalNumber(determination.reservoir)
  33. newOrefDetermination.insulinReq = self.decimalToNSDecimalNumber(determination.insulinReq)
  34. newOrefDetermination.temp = determination.temp?.rawValue ?? "absolute"
  35. newOrefDetermination.rate = self.decimalToNSDecimalNumber(determination.rate)
  36. newOrefDetermination.reason = determination.reason
  37. newOrefDetermination.duration = Int16(determination.duration ?? 0)
  38. newOrefDetermination.iob = self.decimalToNSDecimalNumber(determination.iob)
  39. newOrefDetermination.threshold = self.decimalToNSDecimalNumber(determination.threshold)
  40. newOrefDetermination.minDelta = self.decimalToNSDecimalNumber(determination.minDelta)
  41. newOrefDetermination.sensitivityRatio = self.decimalToNSDecimalNumber(determination.sensitivityRatio)
  42. newOrefDetermination.expectedDelta = self.decimalToNSDecimalNumber(determination.expectedDelta)
  43. newOrefDetermination.cob = Int16(Int(determination.cob ?? 0))
  44. newOrefDetermination.manualBolusErrorString = self.decimalToNSDecimalNumber(determination.manualBolusErrorString)
  45. newOrefDetermination.tempBasal = determination.insulin?.temp_basal.map { NSDecimalNumber(decimal: $0) }
  46. newOrefDetermination.scheduledBasal = determination.insulin?.scheduled_basal.map { NSDecimalNumber(decimal: $0) }
  47. newOrefDetermination.bolus = determination.insulin?.bolus.map { NSDecimalNumber(decimal: $0) }
  48. newOrefDetermination.smbToDeliver = determination.units.map { NSDecimalNumber(decimal: $0) }
  49. newOrefDetermination.carbsRequired = Int16(Int(determination.carbsReq ?? 0))
  50. if let predictions = determination.predictions {
  51. ["iob": predictions.iob, "zt": predictions.zt, "cob": predictions.cob, "uam": predictions.uam]
  52. .forEach { type, values in
  53. if let values = values {
  54. let forecast = Forecast(context: self.context)
  55. forecast.id = UUID()
  56. forecast.type = type
  57. forecast.date = Date()
  58. forecast.orefDetermination = newOrefDetermination
  59. for (index, value) in values.enumerated() {
  60. let forecastValue = ForecastValue(context: self.context)
  61. forecastValue.index = Int32(index)
  62. forecastValue.value = Int32(value)
  63. forecast.addToForecastValues(forecastValue)
  64. }
  65. newOrefDetermination.addToForecasts(forecast)
  66. }
  67. }
  68. }
  69. self.attemptToSaveContext()
  70. }
  71. }
  72. func attemptToSaveContext() {
  73. do {
  74. guard context.hasChanges else { return }
  75. try context.save()
  76. } catch {
  77. print(error.localizedDescription)
  78. }
  79. }
  80. // fetch glucose to pass it to the meal function and to determine basal
  81. private func fetchAndProcessGlucose() -> String {
  82. var glucoseAsJSON: String?
  83. context.performAndWait {
  84. let results = CoreDataStack.shared.fetchEntities(
  85. ofType: GlucoseStored.self,
  86. onContext: context,
  87. predicate: NSPredicate.predicateForSixHoursAgo,
  88. key: "date",
  89. ascending: false,
  90. fetchLimit: 72,
  91. batchSize: 24
  92. )
  93. // convert to json
  94. glucoseAsJSON = self.jsonConverter.convertToJSON(results)
  95. }
  96. return glucoseAsJSON ?? "{}"
  97. }
  98. private func fetchAndProcessCarbs() -> String {
  99. // perform fetch AND conversion on the same thread
  100. // if we do it like this we do not change the thread and do not have to pass the objectIDs
  101. var carbsAsJSON: String?
  102. context.performAndWait {
  103. let results = CoreDataStack.shared.fetchEntities(
  104. ofType: CarbEntryStored.self,
  105. onContext: context,
  106. predicate: NSPredicate.predicateForOneDayAgo,
  107. key: "date",
  108. ascending: false
  109. )
  110. // convert to json
  111. carbsAsJSON = self.jsonConverter.convertToJSON(results)
  112. }
  113. return carbsAsJSON ?? "{}"
  114. }
  115. private func fetchPumpHistoryObjectIDs() -> [NSManagedObjectID]? {
  116. context.performAndWait {
  117. let results = CoreDataStack.shared.fetchEntities(
  118. ofType: PumpEventStored.self,
  119. onContext: context,
  120. predicate: NSPredicate.pumpHistoryLast24h,
  121. key: "timestamp",
  122. ascending: false,
  123. batchSize: 50
  124. )
  125. return results.map(\.objectID)
  126. }
  127. }
  128. private func parsePumpHistory(_ pumpHistoryObjectIDs: [NSManagedObjectID]) -> String {
  129. // Return an empty JSON object if the list of object IDs is empty
  130. guard !pumpHistoryObjectIDs.isEmpty else { return "{}" }
  131. // Execute all operations on the background context
  132. let jsonResult = context.performAndWait {
  133. // Load the pump events from the object IDs
  134. let pumpHistory: [PumpEventStored] = pumpHistoryObjectIDs
  135. .compactMap { context.object(with: $0) as? PumpEventStored }
  136. // Create the DTOs
  137. let dtos: [PumpEventDTO] = pumpHistory.flatMap { event -> [PumpEventDTO] in
  138. var eventDTOs: [PumpEventDTO] = []
  139. if let bolusDTO = event.toBolusDTOEnum() {
  140. eventDTOs.append(bolusDTO)
  141. }
  142. if let tempBasalDTO = event.toTempBasalDTOEnum() {
  143. eventDTOs.append(tempBasalDTO)
  144. }
  145. if let tempBasalDurationDTO = event.toTempBasalDurationDTOEnum() {
  146. eventDTOs.append(tempBasalDurationDTO)
  147. }
  148. return eventDTOs
  149. }
  150. // Convert the DTOs to JSON
  151. return jsonConverter.convertToJSON(dtos)
  152. }
  153. // Return the JSON result
  154. return jsonResult
  155. }
  156. func determineBasal(currentTemp: TempBasal, clock: Date = Date()) -> Future<Determination?, Never> {
  157. Future { promise in
  158. self.processQueue.async {
  159. debug(.openAPS, "Start determineBasal")
  160. // clock
  161. self.storage.save(clock, as: Monitor.clock)
  162. let pass = self.loadFileFromStorage(name: Monitor.clock)
  163. // temp_basal
  164. let tempBasal = currentTemp.rawJSON
  165. self.storage.save(tempBasal, as: Monitor.tempBasal)
  166. let pumpHistoryObjectIDs = self.fetchPumpHistoryObjectIDs() ?? []
  167. let pumpHistoryJSON = self.parsePumpHistory(pumpHistoryObjectIDs)
  168. // carbs
  169. let carbsAsJSON = self.fetchAndProcessCarbs()
  170. /// glucose
  171. let glucoseAsJSON = self.fetchAndProcessGlucose()
  172. /// profile
  173. let profile = self.loadFileFromStorage(name: Settings.profile)
  174. let basalProfile = self.loadFileFromStorage(name: Settings.basalProfile)
  175. /// meal
  176. let meal = self.meal(
  177. pumphistory: pumpHistoryJSON,
  178. profile: profile,
  179. basalProfile: basalProfile,
  180. clock: pass,
  181. carbs: carbsAsJSON,
  182. glucose: glucoseAsJSON
  183. )
  184. self.storage.save(meal, as: Monitor.meal)
  185. // iob
  186. let autosens = self.loadFileFromStorage(name: Settings.autosense)
  187. let iob = self.iob(
  188. pumphistory: pumpHistoryJSON,
  189. profile: profile,
  190. clock: pass,
  191. autosens: autosens.isEmpty ? .null : autosens
  192. )
  193. self.storage.save(iob, as: Monitor.iob)
  194. // determine-basal
  195. let reservoir = self.loadFileFromStorage(name: Monitor.reservoir)
  196. let preferences = self.loadFileFromStorage(name: Settings.preferences)
  197. // oref2
  198. let oref2_variables = self.oref2()
  199. let orefDetermination = self.determineBasal(
  200. glucose: glucoseAsJSON,
  201. currentTemp: tempBasal,
  202. iob: iob,
  203. profile: profile,
  204. autosens: autosens.isEmpty ? .null : autosens,
  205. meal: meal,
  206. microBolusAllowed: true,
  207. reservoir: reservoir,
  208. pumpHistory: pumpHistoryJSON,
  209. preferences: preferences,
  210. basalProfile: basalProfile,
  211. oref2_variables: oref2_variables
  212. )
  213. debug(.openAPS, "Determinated: \(orefDetermination)")
  214. if var determination = Determination(from: orefDetermination) {
  215. determination.timestamp = determination.deliverAt ?? clock
  216. self.storage.save(determination, as: Enact.suggested)
  217. // save to core data asynchronously
  218. self.processDetermination(determination)
  219. if determination.tdd ?? 0 > 0 {
  220. self.context.perform {
  221. let saveToTDD = TDD(context: self.context)
  222. saveToTDD.timestamp = determination.timestamp ?? Date()
  223. saveToTDD.tdd = (determination.tdd ?? 0) as NSDecimalNumber?
  224. do {
  225. guard self.context.hasChanges else { return }
  226. try self.context.save()
  227. } catch {
  228. print(error.localizedDescription)
  229. }
  230. let saveTarget = Target(context: self.context)
  231. saveTarget.current = (determination.current_target ?? 100) as NSDecimalNumber?
  232. do {
  233. guard self.context.hasChanges else { return }
  234. try self.context.save()
  235. } catch {
  236. print(error.localizedDescription)
  237. }
  238. }
  239. }
  240. promise(.success(determination))
  241. } else {
  242. promise(.success(nil))
  243. }
  244. }
  245. }
  246. }
  247. func oref2() -> RawJSON {
  248. context.performAndWait {
  249. let preferences = storage.retrieve(OpenAPS.Settings.preferences, as: Preferences.self)
  250. var hbt_ = preferences?.halfBasalExerciseTarget ?? 160
  251. let wp = preferences?.weightPercentage ?? 1
  252. let smbMinutes = (preferences?.maxSMBBasalMinutes ?? 30) as NSDecimalNumber
  253. let uamMinutes = (preferences?.maxUAMSMBBasalMinutes ?? 30) as NSDecimalNumber
  254. let tenDaysAgo = Date().addingTimeInterval(-10.days.timeInterval)
  255. let twoHoursAgo = Date().addingTimeInterval(-2.hours.timeInterval)
  256. var uniqueEvents = [TDD]()
  257. let requestTDD = TDD.fetchRequest() as NSFetchRequest<TDD>
  258. requestTDD.predicate = NSPredicate(format: "timestamp > %@ AND tdd > 0", tenDaysAgo as NSDate)
  259. let sortTDD = NSSortDescriptor(key: "timestamp", ascending: true)
  260. requestTDD.sortDescriptors = [sortTDD]
  261. try? uniqueEvents = context.fetch(requestTDD)
  262. var sliderArray = [TempTargetsSlider]()
  263. let requestIsEnbled = TempTargetsSlider.fetchRequest() as NSFetchRequest<TempTargetsSlider>
  264. let sortIsEnabled = NSSortDescriptor(key: "date", ascending: false)
  265. requestIsEnbled.sortDescriptors = [sortIsEnabled]
  266. // requestIsEnbled.fetchLimit = 1
  267. try? sliderArray = context.fetch(requestIsEnbled)
  268. var overrideArray = [Override]()
  269. let requestOverrides = Override.fetchRequest() as NSFetchRequest<Override>
  270. let sortOverride = NSSortDescriptor(key: "date", ascending: false)
  271. requestOverrides.sortDescriptors = [sortOverride]
  272. // requestOverrides.fetchLimit = 1
  273. try? overrideArray = context.fetch(requestOverrides)
  274. var tempTargetsArray = [TempTargets]()
  275. let requestTempTargets = TempTargets.fetchRequest() as NSFetchRequest<TempTargets>
  276. let sortTT = NSSortDescriptor(key: "date", ascending: false)
  277. requestTempTargets.sortDescriptors = [sortTT]
  278. requestTempTargets.fetchLimit = 1
  279. try? tempTargetsArray = context.fetch(requestTempTargets)
  280. let total = uniqueEvents.compactMap({ each in each.tdd as? Decimal ?? 0 }).reduce(0, +)
  281. var indeces = uniqueEvents.count
  282. // Only fetch once. Use same (previous) fetch
  283. let twoHoursArray = uniqueEvents.filter({ ($0.timestamp ?? Date()) >= twoHoursAgo })
  284. var nrOfIndeces = twoHoursArray.count
  285. let totalAmount = twoHoursArray.compactMap({ each in each.tdd as? Decimal ?? 0 }).reduce(0, +)
  286. var temptargetActive = tempTargetsArray.first?.active ?? false
  287. let isPercentageEnabled = sliderArray.first?.enabled ?? false
  288. var useOverride = overrideArray.first?.enabled ?? false
  289. var overridePercentage = Decimal(overrideArray.first?.percentage ?? 100)
  290. var unlimited = overrideArray.first?.indefinite ?? true
  291. var disableSMBs = overrideArray.first?.smbIsOff ?? false
  292. let currentTDD = (uniqueEvents.last?.tdd ?? 0) as Decimal
  293. if indeces == 0 {
  294. indeces = 1
  295. }
  296. if nrOfIndeces == 0 {
  297. nrOfIndeces = 1
  298. }
  299. let average2hours = totalAmount / Decimal(nrOfIndeces)
  300. let average14 = total / Decimal(indeces)
  301. let weight = wp
  302. let weighted_average = weight * average2hours + (1 - weight) * average14
  303. var duration: Decimal = 0
  304. var newDuration: Decimal = 0
  305. var overrideTarget: Decimal = 0
  306. if useOverride {
  307. duration = (overrideArray.first?.duration ?? 0) as Decimal
  308. overrideTarget = (overrideArray.first?.target ?? 0) as Decimal
  309. let advancedSettings = overrideArray.first?.advancedSettings ?? false
  310. let addedMinutes = Int(duration)
  311. let date = overrideArray.first?.date ?? Date()
  312. if date.addingTimeInterval(addedMinutes.minutes.timeInterval) < Date(),
  313. !unlimited
  314. {
  315. useOverride = false
  316. let saveToCoreData = Override(context: self.context)
  317. saveToCoreData.enabled = false
  318. saveToCoreData.date = Date()
  319. saveToCoreData.duration = 0
  320. saveToCoreData.indefinite = false
  321. saveToCoreData.percentage = 100
  322. do {
  323. guard self.context.hasChanges else { return "{}" }
  324. try self.context.save()
  325. } catch {
  326. print(error.localizedDescription)
  327. }
  328. }
  329. }
  330. if !useOverride {
  331. unlimited = true
  332. overridePercentage = 100
  333. duration = 0
  334. overrideTarget = 0
  335. disableSMBs = false
  336. }
  337. if temptargetActive {
  338. var duration_ = 0
  339. var hbt = Double(hbt_)
  340. var dd = 0.0
  341. if temptargetActive {
  342. duration_ = Int(truncating: tempTargetsArray.first?.duration ?? 0)
  343. hbt = tempTargetsArray.first?.hbt ?? Double(hbt_)
  344. let startDate = tempTargetsArray.first?.startDate ?? Date()
  345. let durationPlusStart = startDate.addingTimeInterval(duration_.minutes.timeInterval)
  346. dd = durationPlusStart.timeIntervalSinceNow.minutes
  347. if dd > 0.1 {
  348. hbt_ = Decimal(hbt)
  349. temptargetActive = true
  350. } else {
  351. temptargetActive = false
  352. }
  353. }
  354. }
  355. if currentTDD > 0 {
  356. let averages = Oref2_variables(
  357. average_total_data: average14,
  358. weightedAverage: weighted_average,
  359. past2hoursAverage: average2hours,
  360. date: Date(),
  361. isEnabled: temptargetActive,
  362. presetActive: isPercentageEnabled,
  363. overridePercentage: overridePercentage,
  364. useOverride: useOverride,
  365. duration: duration,
  366. unlimited: unlimited,
  367. hbt: hbt_,
  368. overrideTarget: overrideTarget,
  369. smbIsOff: disableSMBs,
  370. advancedSettings: overrideArray.first?.advancedSettings ?? false,
  371. isfAndCr: overrideArray.first?.isfAndCr ?? false,
  372. isf: overrideArray.first?.isf ?? false,
  373. cr: overrideArray.first?.cr ?? false,
  374. smbIsAlwaysOff: overrideArray.first?.smbIsAlwaysOff ?? false,
  375. start: (overrideArray.first?.start ?? 0) as Decimal,
  376. end: (overrideArray.first?.end ?? 0) as Decimal,
  377. smbMinutes: (overrideArray.first?.smbMinutes ?? smbMinutes) as Decimal,
  378. uamMinutes: (overrideArray.first?.uamMinutes ?? uamMinutes) as Decimal
  379. )
  380. storage.save(averages, as: OpenAPS.Monitor.oref2_variables)
  381. return self.loadFileFromStorage(name: Monitor.oref2_variables)
  382. } else {
  383. let averages = Oref2_variables(
  384. average_total_data: 0,
  385. weightedAverage: 1,
  386. past2hoursAverage: 0,
  387. date: Date(),
  388. isEnabled: temptargetActive,
  389. presetActive: isPercentageEnabled,
  390. overridePercentage: overridePercentage,
  391. useOverride: useOverride,
  392. duration: duration,
  393. unlimited: unlimited,
  394. hbt: hbt_,
  395. overrideTarget: overrideTarget,
  396. smbIsOff: disableSMBs,
  397. advancedSettings: overrideArray.first?.advancedSettings ?? false,
  398. isfAndCr: overrideArray.first?.isfAndCr ?? false,
  399. isf: overrideArray.first?.isf ?? false,
  400. cr: overrideArray.first?.cr ?? false,
  401. smbIsAlwaysOff: overrideArray.first?.smbIsAlwaysOff ?? false,
  402. start: (overrideArray.first?.start ?? 0) as Decimal,
  403. end: (overrideArray.first?.end ?? 0) as Decimal,
  404. smbMinutes: (overrideArray.first?.smbMinutes ?? smbMinutes) as Decimal,
  405. uamMinutes: (overrideArray.first?.uamMinutes ?? uamMinutes) as Decimal
  406. )
  407. storage.save(averages, as: OpenAPS.Monitor.oref2_variables)
  408. return self.loadFileFromStorage(name: Monitor.oref2_variables)
  409. }
  410. }
  411. }
  412. func autosense() -> Future<Autosens?, Never> {
  413. Future { promise in
  414. self.processQueue.async {
  415. debug(.openAPS, "Start autosens")
  416. // pump history
  417. let pumpHistoryObjectIDs = self.fetchPumpHistoryObjectIDs() ?? []
  418. let pumpHistoryJSON = self.parsePumpHistory(pumpHistoryObjectIDs)
  419. // carbs
  420. let carbsAsJSON = self.fetchAndProcessCarbs()
  421. /// glucose
  422. let glucoseAsJSON = self.fetchAndProcessGlucose()
  423. let profile = self.loadFileFromStorage(name: Settings.profile)
  424. let basalProfile = self.loadFileFromStorage(name: Settings.basalProfile)
  425. let tempTargets = self.loadFileFromStorage(name: Settings.tempTargets)
  426. let autosensResult = self.autosense(
  427. glucose: glucoseAsJSON,
  428. pumpHistory: pumpHistoryJSON,
  429. basalprofile: basalProfile,
  430. profile: profile,
  431. carbs: carbsAsJSON,
  432. temptargets: tempTargets
  433. )
  434. debug(.openAPS, "AUTOSENS: \(autosensResult)")
  435. if var autosens = Autosens(from: autosensResult) {
  436. autosens.timestamp = Date()
  437. self.storage.save(autosens, as: Settings.autosense)
  438. promise(.success(autosens))
  439. } else {
  440. promise(.success(nil))
  441. }
  442. }
  443. }
  444. }
  445. func autotune(categorizeUamAsBasal: Bool = false, tuneInsulinCurve: Bool = false) -> Future<Autotune?, Never> {
  446. Future { promise in
  447. self.processQueue.async {
  448. debug(.openAPS, "Start autotune")
  449. // pump history
  450. let pumpHistoryObjectIDs = self.fetchPumpHistoryObjectIDs() ?? []
  451. let pumpHistoryJSON = self.parsePumpHistory(pumpHistoryObjectIDs)
  452. /// glucose
  453. let glucoseAsJSON = self.fetchAndProcessGlucose()
  454. let profile = self.loadFileFromStorage(name: Settings.profile)
  455. let pumpProfile = self.loadFileFromStorage(name: Settings.pumpProfile)
  456. // carbs
  457. let carbsAsJSON = self.fetchAndProcessCarbs()
  458. let autotunePreppedGlucose = self.autotunePrepare(
  459. pumphistory: pumpHistoryJSON,
  460. profile: profile,
  461. glucose: glucoseAsJSON,
  462. pumpprofile: pumpProfile,
  463. carbs: carbsAsJSON,
  464. categorizeUamAsBasal: categorizeUamAsBasal,
  465. tuneInsulinCurve: tuneInsulinCurve
  466. )
  467. debug(.openAPS, "AUTOTUNE PREP: \(autotunePreppedGlucose)")
  468. let previousAutotune = self.storage.retrieve(Settings.autotune, as: RawJSON.self)
  469. let autotuneResult = self.autotuneRun(
  470. autotunePreparedData: autotunePreppedGlucose,
  471. previousAutotuneResult: previousAutotune ?? profile,
  472. pumpProfile: pumpProfile
  473. )
  474. debug(.openAPS, "AUTOTUNE RESULT: \(autotuneResult)")
  475. if let autotune = Autotune(from: autotuneResult) {
  476. self.storage.save(autotuneResult, as: Settings.autotune)
  477. promise(.success(autotune))
  478. } else {
  479. promise(.success(nil))
  480. }
  481. }
  482. }
  483. }
  484. func makeProfiles(useAutotune: Bool) -> Future<Autotune?, Never> {
  485. Future { promise in
  486. debug(.openAPS, "Start makeProfiles")
  487. self.processQueue.async {
  488. var preferences = self.loadFileFromStorage(name: Settings.preferences)
  489. if preferences.isEmpty {
  490. preferences = Preferences().rawJSON
  491. }
  492. let pumpSettings = self.loadFileFromStorage(name: Settings.settings)
  493. let bgTargets = self.loadFileFromStorage(name: Settings.bgTargets)
  494. let basalProfile = self.loadFileFromStorage(name: Settings.basalProfile)
  495. let isf = self.loadFileFromStorage(name: Settings.insulinSensitivities)
  496. let cr = self.loadFileFromStorage(name: Settings.carbRatios)
  497. let tempTargets = self.loadFileFromStorage(name: Settings.tempTargets)
  498. let model = self.loadFileFromStorage(name: Settings.model)
  499. let autotune = useAutotune ? self.loadFileFromStorage(name: Settings.autotune) : .empty
  500. let freeaps = self.loadFileFromStorage(name: FreeAPS.settings)
  501. let pumpProfile = self.makeProfile(
  502. preferences: preferences,
  503. pumpSettings: pumpSettings,
  504. bgTargets: bgTargets,
  505. basalProfile: basalProfile,
  506. isf: isf,
  507. carbRatio: cr,
  508. tempTargets: tempTargets,
  509. model: model,
  510. autotune: RawJSON.null,
  511. freeaps: freeaps
  512. )
  513. let profile = self.makeProfile(
  514. preferences: preferences,
  515. pumpSettings: pumpSettings,
  516. bgTargets: bgTargets,
  517. basalProfile: basalProfile,
  518. isf: isf,
  519. carbRatio: cr,
  520. tempTargets: tempTargets,
  521. model: model,
  522. autotune: autotune.isEmpty ? .null : autotune,
  523. freeaps: freeaps
  524. )
  525. self.storage.save(pumpProfile, as: Settings.pumpProfile)
  526. self.storage.save(profile, as: Settings.profile)
  527. if let tunedProfile = Autotune(from: profile) {
  528. promise(.success(tunedProfile))
  529. return
  530. }
  531. promise(.success(nil))
  532. }
  533. }
  534. }
  535. // MARK: - Private
  536. private func iob(pumphistory: JSON, profile: JSON, clock: JSON, autosens: JSON) -> RawJSON {
  537. dispatchPrecondition(condition: .onQueue(processQueue))
  538. return jsWorker.inCommonContext { worker in
  539. worker.evaluate(script: Script(name: Prepare.log))
  540. worker.evaluate(script: Script(name: Bundle.iob))
  541. worker.evaluate(script: Script(name: Prepare.iob))
  542. return worker.call(function: Function.generate, with: [
  543. pumphistory,
  544. profile,
  545. clock,
  546. autosens
  547. ])
  548. }
  549. }
  550. private func meal(pumphistory: JSON, profile: JSON, basalProfile: JSON, clock: JSON, carbs: JSON, glucose: JSON) -> RawJSON {
  551. dispatchPrecondition(condition: .onQueue(processQueue))
  552. return jsWorker.inCommonContext { worker in
  553. worker.evaluate(script: Script(name: Prepare.log))
  554. worker.evaluate(script: Script(name: Bundle.meal))
  555. worker.evaluate(script: Script(name: Prepare.meal))
  556. return worker.call(function: Function.generate, with: [
  557. pumphistory,
  558. profile,
  559. clock,
  560. glucose,
  561. basalProfile,
  562. carbs
  563. ])
  564. }
  565. }
  566. private func autotunePrepare(
  567. pumphistory: JSON,
  568. profile: JSON,
  569. glucose: JSON,
  570. pumpprofile: JSON,
  571. carbs: JSON,
  572. categorizeUamAsBasal: Bool,
  573. tuneInsulinCurve: Bool
  574. ) -> RawJSON {
  575. dispatchPrecondition(condition: .onQueue(processQueue))
  576. return jsWorker.inCommonContext { worker in
  577. worker.evaluate(script: Script(name: Prepare.log))
  578. worker.evaluate(script: Script(name: Bundle.autotunePrep))
  579. worker.evaluate(script: Script(name: Prepare.autotunePrep))
  580. return worker.call(function: Function.generate, with: [
  581. pumphistory,
  582. profile,
  583. glucose,
  584. pumpprofile,
  585. carbs,
  586. categorizeUamAsBasal,
  587. tuneInsulinCurve
  588. ])
  589. }
  590. }
  591. private func autotuneRun(
  592. autotunePreparedData: JSON,
  593. previousAutotuneResult: JSON,
  594. pumpProfile: JSON
  595. ) -> RawJSON {
  596. dispatchPrecondition(condition: .onQueue(processQueue))
  597. return jsWorker.inCommonContext { worker in
  598. worker.evaluate(script: Script(name: Prepare.log))
  599. worker.evaluate(script: Script(name: Bundle.autotuneCore))
  600. worker.evaluate(script: Script(name: Prepare.autotuneCore))
  601. return worker.call(function: Function.generate, with: [
  602. autotunePreparedData,
  603. previousAutotuneResult,
  604. pumpProfile
  605. ])
  606. }
  607. }
  608. private func determineBasal(
  609. glucose: JSON,
  610. currentTemp: JSON,
  611. iob: JSON,
  612. profile: JSON,
  613. autosens: JSON,
  614. meal: JSON,
  615. microBolusAllowed: Bool,
  616. reservoir: JSON,
  617. pumpHistory: JSON,
  618. preferences: JSON,
  619. basalProfile: JSON,
  620. oref2_variables: JSON
  621. ) -> RawJSON {
  622. dispatchPrecondition(condition: .onQueue(processQueue))
  623. return jsWorker.inCommonContext { worker in
  624. worker.evaluate(script: Script(name: Prepare.log))
  625. worker.evaluate(script: Script(name: Prepare.determineBasal))
  626. worker.evaluate(script: Script(name: Bundle.basalSetTemp))
  627. worker.evaluate(script: Script(name: Bundle.getLastGlucose))
  628. worker.evaluate(script: Script(name: Bundle.determineBasal))
  629. if let middleware = self.middlewareScript(name: OpenAPS.Middleware.determineBasal) {
  630. worker.evaluate(script: middleware)
  631. }
  632. return worker.call(
  633. function: Function.generate,
  634. with: [
  635. iob,
  636. currentTemp,
  637. glucose,
  638. profile,
  639. autosens,
  640. meal,
  641. microBolusAllowed,
  642. reservoir,
  643. Date(),
  644. pumpHistory,
  645. preferences,
  646. basalProfile,
  647. oref2_variables
  648. ]
  649. )
  650. }
  651. }
  652. private func autosense(
  653. glucose: JSON,
  654. pumpHistory: JSON,
  655. basalprofile: JSON,
  656. profile: JSON,
  657. carbs: JSON,
  658. temptargets: JSON
  659. ) -> RawJSON {
  660. dispatchPrecondition(condition: .onQueue(processQueue))
  661. return jsWorker.inCommonContext { worker in
  662. worker.evaluate(script: Script(name: Prepare.log))
  663. worker.evaluate(script: Script(name: Bundle.autosens))
  664. worker.evaluate(script: Script(name: Prepare.autosens))
  665. return worker.call(
  666. function: Function.generate,
  667. with: [
  668. glucose,
  669. pumpHistory,
  670. basalprofile,
  671. profile,
  672. carbs,
  673. temptargets
  674. ]
  675. )
  676. }
  677. }
  678. private func exportDefaultPreferences() -> RawJSON {
  679. dispatchPrecondition(condition: .onQueue(processQueue))
  680. return jsWorker.inCommonContext { worker in
  681. worker.evaluate(script: Script(name: Prepare.log))
  682. worker.evaluate(script: Script(name: Bundle.profile))
  683. worker.evaluate(script: Script(name: Prepare.profile))
  684. return worker.call(function: Function.exportDefaults, with: [])
  685. }
  686. }
  687. private func makeProfile(
  688. preferences: JSON,
  689. pumpSettings: JSON,
  690. bgTargets: JSON,
  691. basalProfile: JSON,
  692. isf: JSON,
  693. carbRatio: JSON,
  694. tempTargets: JSON,
  695. model: JSON,
  696. autotune: JSON,
  697. freeaps: JSON
  698. ) -> RawJSON {
  699. dispatchPrecondition(condition: .onQueue(processQueue))
  700. return jsWorker.inCommonContext { worker in
  701. worker.evaluate(script: Script(name: Prepare.log))
  702. worker.evaluate(script: Script(name: Bundle.profile))
  703. worker.evaluate(script: Script(name: Prepare.profile))
  704. return worker.call(
  705. function: Function.generate,
  706. with: [
  707. pumpSettings,
  708. bgTargets,
  709. isf,
  710. basalProfile,
  711. preferences,
  712. carbRatio,
  713. tempTargets,
  714. model,
  715. autotune,
  716. freeaps
  717. ]
  718. )
  719. }
  720. }
  721. private func loadJSON(name: String) -> String {
  722. try! String(contentsOf: Foundation.Bundle.main.url(forResource: "json/\(name)", withExtension: "json")!)
  723. }
  724. private func loadFileFromStorage(name: String) -> RawJSON {
  725. storage.retrieveRaw(name) ?? OpenAPS.defaults(for: name)
  726. }
  727. private func middlewareScript(name: String) -> Script? {
  728. if let body = storage.retrieveRaw(name) {
  729. return Script(name: "Middleware", body: body)
  730. }
  731. if let url = Foundation.Bundle.main.url(forResource: "javascript/\(name)", withExtension: "") {
  732. return Script(name: "Middleware", body: try! String(contentsOf: url))
  733. }
  734. return nil
  735. }
  736. static func defaults(for file: String) -> RawJSON {
  737. let prefix = file.hasSuffix(".json") ? "json/defaults" : "javascript"
  738. guard let url = Foundation.Bundle.main.url(forResource: "\(prefix)/\(file)", withExtension: "") else {
  739. return ""
  740. }
  741. return (try? String(contentsOf: url)) ?? ""
  742. }
  743. func processAndSave(forecastData: [String: [Int]]) {
  744. let currentDate = Date()
  745. context.perform {
  746. for (type, values) in forecastData {
  747. self.createForecast(type: type, values: values, date: currentDate, context: self.context)
  748. }
  749. do {
  750. guard self.context.hasChanges else { return }
  751. try self.context.save()
  752. } catch {
  753. print(error.localizedDescription)
  754. }
  755. }
  756. }
  757. func createForecast(type: String, values: [Int], date: Date, context: NSManagedObjectContext) {
  758. let forecast = Forecast(context: context)
  759. forecast.id = UUID()
  760. forecast.date = date
  761. forecast.type = type
  762. for (index, value) in values.enumerated() {
  763. let forecastValue = ForecastValue(context: context)
  764. forecastValue.value = Int32(value)
  765. forecastValue.index = Int32(index)
  766. forecastValue.forecast = forecast
  767. }
  768. }
  769. }