OpenAPS.swift 35 KB

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