vav.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472
  1. # -*- coding: utf-8 -*-
  2. from operator import attrgetter
  3. from typing import Dict, List, Optional, Tuple
  4. import numpy as np
  5. from fastapi import HTTPException
  6. from httpx import AsyncClient
  7. from loguru import logger
  8. from sqlalchemy.orm import Session
  9. from app.controllers.equipment.controller import EquipmentController
  10. from app.crud.space.weight import get_weights_by_vav
  11. from app.models.domain.devices import (
  12. ACATVAInstructionsRequest,
  13. ACATVAInstructionsRequestV2,
  14. )
  15. from app.schemas.equipment import VAVBox, FCU
  16. from app.schemas.instructions import ACATVAInstructions
  17. from app.schemas.sapce_weight import SpaceWeight
  18. from app.schemas.space import Space
  19. from app.services.platform import DataPlatformService
  20. from app.services.transfer import Duoduo, SpaceInfoService, Season
  21. from app.utils.date import get_time_str
  22. class VAVController(EquipmentController):
  23. def __init__(self, equipment: VAVBox):
  24. super(VAVController, self).__init__()
  25. self.equipment = equipment
  26. async def get_strategy(self):
  27. strategy = "Plan A"
  28. for space in self.equipment.spaces:
  29. for eq in space.equipment:
  30. if isinstance(eq, FCU):
  31. strategy = "Plan B"
  32. break
  33. return strategy
  34. async def build_virtual_temperature(self) -> Tuple[float, float]:
  35. target_list, realtime_list = [], []
  36. buffer_list = []
  37. strategy = await self.get_strategy()
  38. for space in self.equipment.spaces:
  39. if not np.isnan(space.temperature_target):
  40. target_list.append(space.temperature_target)
  41. realtime_list.append(space.realtime_temperature)
  42. if strategy == "Plan B":
  43. for eq in space.equipment:
  44. if isinstance(eq, FCU):
  45. buffer = (4 - eq.air_valve_speed) / 4
  46. buffer_list.append(buffer)
  47. break
  48. total_target = buffer_list + target_list
  49. total_realtime = buffer_list + realtime_list
  50. if total_target and total_realtime:
  51. target_result = np.array(total_target).sum() / len(target_list)
  52. realtime_result = np.array(total_realtime).sum() / len(realtime_list)
  53. self.equipment.setting_temperature = target_result
  54. else:
  55. target_result, realtime_result = np.NAN, np.NAN
  56. return target_result, realtime_result
  57. async def get_supply_air_flow_set(
  58. self, temperature_set: float, temperature_realtime: float
  59. ) -> float:
  60. if np.isnan(temperature_set) or np.isnan(temperature_realtime):
  61. supply_air_flow_set = 0.0
  62. else:
  63. temperature_supply = self.equipment.supply_air_temperature
  64. if np.isnan(temperature_supply):
  65. temperature_supply = 19.0
  66. try:
  67. ratio = abs(
  68. 1
  69. + (temperature_realtime - temperature_set)
  70. / (temperature_set - temperature_supply)
  71. )
  72. except ZeroDivisionError:
  73. ratio = 1
  74. supply_air_flow_set = self.equipment.supply_air_flow * ratio
  75. supply_air_flow_set = max(
  76. self.equipment.supply_air_flow_lower_limit, supply_air_flow_set
  77. )
  78. supply_air_flow_set = min(
  79. self.equipment.supply_air_flow_upper_limit, supply_air_flow_set
  80. )
  81. self.equipment.supply_air_flow_set = supply_air_flow_set
  82. self.equipment.virtual_target_temperature = temperature_set
  83. self.equipment.virtual_realtime_temperature = temperature_realtime
  84. return supply_air_flow_set
  85. async def run(self):
  86. temperature_set, temperature_realtime = await self.build_virtual_temperature()
  87. await self.get_supply_air_flow_set(temperature_set, temperature_realtime)
  88. self.equipment.running_status = True
  89. def get_results(self):
  90. return self.equipment
  91. class VAVControllerV2(VAVController):
  92. def __init__(
  93. self,
  94. equipment: VAVBox,
  95. weights: Optional[List[SpaceWeight]] = None,
  96. season: Optional[Season] = None,
  97. ):
  98. super(VAVControllerV2, self).__init__(equipment)
  99. self.weights = weights
  100. self.season = season
  101. async def build_virtual_temperature(self) -> None:
  102. valid_spaces = []
  103. weights = []
  104. for sp in self.equipment.spaces:
  105. if sp.realtime_temperature > 0.0 and sp.temperature_target > 0.0:
  106. valid_spaces.append(sp)
  107. for weight in self.weights:
  108. if weight.space_id == sp.id:
  109. weights.append(weight)
  110. if valid_spaces:
  111. weights = sorted(weights, key=lambda x: x.temporary_weight_update_time)
  112. if weights[-1].temporary_weight_update_time > get_time_str(
  113. 60 * 60 * 2, flag="ago"
  114. ):
  115. # If someone has submitted a feedback in past two hours, meet the need.
  116. weight_dic = {weight.space_id: 0.0 for weight in weights}
  117. weight_dic.update({weights[-1].space_id: weights[-1].temporary_weight})
  118. else:
  119. weight_dic = {
  120. weight.space_id: weight.default_weight for weight in weights
  121. }
  122. total_weight_value = 0.0
  123. for v in weight_dic.values():
  124. total_weight_value += v
  125. if total_weight_value > 0:
  126. weight_dic = {
  127. k: v / total_weight_value for k, v in weight_dic.items()
  128. }
  129. else:
  130. weight_dic.update({list(weight_dic.keys())[0]: 1.0})
  131. try:
  132. virtual_target, virtual_realtime = 0.0, 0.0
  133. for sp in valid_spaces:
  134. virtual_target += sp.temperature_target * weight_dic.get(sp.id)
  135. virtual_realtime += sp.realtime_temperature * weight_dic.get(sp.id)
  136. except KeyError:
  137. logger.error(f"{self.equipment.id} has wrong vav-space relation")
  138. raise HTTPException(
  139. status_code=404, detail="This VAV box has wrong eq-sp relation"
  140. )
  141. self.equipment.virtual_target_temperature = virtual_target
  142. self.equipment.virtual_realtime_temperature = virtual_realtime
  143. else:
  144. self.equipment.virtual_target_temperature = np.NAN
  145. self.equipment.virtual_realtime_temperature = np.NAN
  146. async def rectify(self) -> Tuple[float, float]:
  147. bad_spaces = list()
  148. for sp in self.equipment.spaces:
  149. if self.season == Season.heating:
  150. if sp.realtime_temperature > max(
  151. 26.0, sp.temperature_target
  152. ) or sp.realtime_temperature < min(20.0, sp.temperature_target):
  153. if sp.temperature_target > 0.0:
  154. bad_spaces.append(sp)
  155. elif self.season == Season.cooling:
  156. if sp.realtime_temperature > max(
  157. 27.0, sp.temperature_target
  158. ) or sp.realtime_temperature < min(22.0, sp.temperature_target):
  159. if sp.temperature_target > 0.0:
  160. bad_spaces.append(sp)
  161. if bad_spaces:
  162. virtual_diff = (
  163. self.equipment.virtual_target_temperature
  164. - self.equipment.virtual_realtime_temperature
  165. )
  166. if self.season == Season.cooling:
  167. bad_spaces = sorted(bad_spaces, key=attrgetter("diff"))
  168. worst = bad_spaces[0]
  169. if worst.diff <= 0:
  170. if worst.diff < virtual_diff:
  171. self.equipment.virtual_target_temperature = (
  172. worst.temperature_target
  173. )
  174. self.equipment.virtual_realtime_temperature = (
  175. worst.realtime_temperature
  176. )
  177. else:
  178. self.equipment.virtual_target_temperature = (
  179. min(22.0, worst.temperature_target) + 0.5
  180. )
  181. self.equipment.virtual_realtime_temperature = (
  182. worst.realtime_temperature
  183. )
  184. elif self.season == Season.heating:
  185. bad_spaces = sorted(bad_spaces, key=attrgetter("diff"), reverse=True)
  186. worst = bad_spaces[0]
  187. if worst.diff >= 0:
  188. if worst.diff > virtual_diff:
  189. self.equipment.virtual_target_temperature = (
  190. worst.temperature_target
  191. )
  192. self.equipment.virtual_realtime_temperature = (
  193. worst.realtime_temperature
  194. )
  195. else:
  196. self.equipment.virtual_target_temperature = (
  197. max(26.0, worst.temperature_target) - 0.5
  198. )
  199. self.equipment.virtual_realtime_temperature = (
  200. worst.realtime_temperature
  201. )
  202. return (
  203. self.equipment.virtual_target_temperature,
  204. self.equipment.virtual_realtime_temperature,
  205. )
  206. async def run(self) -> None:
  207. await self.build_virtual_temperature()
  208. temperature_set, temperature_realtime = await self.rectify()
  209. await self.get_supply_air_flow_set(temperature_set, temperature_realtime)
  210. self.equipment.running_status = True
  211. class VAVControllerV3(VAVControllerV2):
  212. def __init__(self, vav_params: VAVBox, season: Season):
  213. super(VAVControllerV3, self).__init__(vav_params)
  214. self.season = season
  215. def get_valid_spaces(self) -> List:
  216. valid_spaces = list()
  217. for sp in self.equipment.spaces:
  218. if sp.realtime_temperature > 0.0 and sp.temperature_target > 22.0:
  219. valid_spaces.append(sp)
  220. return valid_spaces
  221. async def build_virtual_temperature(self) -> None:
  222. valid_spaces = self.get_valid_spaces()
  223. if not valid_spaces:
  224. virtual_realtime, virtual_target = np.NAN, np.NAN
  225. else:
  226. sorted_spaces = sorted(
  227. valid_spaces, key=lambda x: x.vav_temporary_update_time
  228. )
  229. if sorted_spaces[-1].vav_temporary_update_time > get_time_str(
  230. 60 * 60 * 2, flag="ago"
  231. ):
  232. virtual_realtime = sorted_spaces[-1].realtime_temperature
  233. virtual_target = sorted_spaces[-1].temperature_target
  234. else:
  235. virtual_realtime, virtual_target = 0.0, 0.0
  236. total_weight = 0.0
  237. for sp in valid_spaces:
  238. temp_weight = sp.vav_default_weight
  239. virtual_realtime += sp.realtime_temperature * temp_weight
  240. virtual_target += sp.temperature_target * temp_weight
  241. total_weight += temp_weight
  242. if total_weight == 0:
  243. for sp in valid_spaces:
  244. virtual_realtime += sp.realtime_temperature
  245. virtual_target += sp.temperature_target
  246. virtual_realtime /= len(valid_spaces)
  247. virtual_target /= len(valid_spaces)
  248. else:
  249. virtual_realtime /= total_weight
  250. virtual_target /= total_weight
  251. self.equipment.virtual_realtime_temperature = virtual_realtime
  252. self.equipment.virtual_target_temperature = virtual_target
  253. class VAVControllerV4(VAVControllerV3):
  254. def __init__(self, vav_params: VAVBox, season: Season, return_air_temp: float):
  255. super().__init__(vav_params, season)
  256. self.return_air_temp = return_air_temp
  257. def get_next_temp_set(
  258. self, virtual_realtime_temp: float, virtual_target_temp: float
  259. ) -> float:
  260. if np.isnan(virtual_realtime_temp) or np.isnan(virtual_target_temp):
  261. next_temp_set = np.NAN
  262. else:
  263. next_temp_set = (
  264. virtual_target_temp + self.return_air_temp - virtual_realtime_temp
  265. )
  266. self.equipment.setting_temperature = next_temp_set
  267. return next_temp_set
  268. async def run(self) -> None:
  269. await self.build_virtual_temperature()
  270. temperature_set, temperature_realtime = await self.rectify()
  271. self.get_next_temp_set(temperature_set, temperature_realtime)
  272. self.equipment.running_status = True
  273. async def fetch_vav_control_params(project_id: str, equipment_id: str) -> Dict:
  274. async with AsyncClient() as client:
  275. duo_duo = Duoduo(client, project_id)
  276. platform = DataPlatformService(client, project_id)
  277. season = await duo_duo.get_season()
  278. served_spaces = await duo_duo.get_space_by_equipment(equipment_id)
  279. space_objects = []
  280. realtime_supply_air_temperature_list = []
  281. for sp in served_spaces:
  282. sp_id = sp.get("id")
  283. transfer = SpaceInfoService(client, project_id, sp_id)
  284. current_target = await transfer.get_current_temperature_target()
  285. realtime_temperature = await platform.get_realtime_temperature(sp_id)
  286. related_equipment = await transfer.get_equipment()
  287. equipment_objects = []
  288. for eq in related_equipment:
  289. if eq.get("category") == "ACATFC":
  290. speed = await platform.get_fan_speed(eq.get("id"))
  291. temp_fcu_params = {"id": eq.get("id"), "air_valve_speed": speed}
  292. fcu = FCU(**temp_fcu_params)
  293. equipment_objects.append(fcu)
  294. elif eq.get("category") == "ACATAH":
  295. realtime_supply_air_temperature_list.append(
  296. await platform.get_realtime_supply_air_temperature(eq.get("id"))
  297. )
  298. temp_space_params = {
  299. "id": sp_id,
  300. "realtime_temperature": realtime_temperature,
  301. "equipment": equipment_objects,
  302. "temperature_target": current_target,
  303. "diff": current_target - realtime_temperature,
  304. }
  305. space = Space(**temp_space_params)
  306. space_objects.append(space)
  307. realtime_supply_air_temperature = np.array(
  308. realtime_supply_air_temperature_list
  309. ).mean()
  310. realtime_supply_air_flow = await platform.get_realtime_supply_air_flow(
  311. equipment_id
  312. )
  313. lower_limit, upper_limit = await platform.get_air_flow_limit(equipment_id)
  314. vav_params = {
  315. "id": equipment_id,
  316. "spaces": space_objects,
  317. "supply_air_temperature": realtime_supply_air_temperature,
  318. "supply_air_flow": realtime_supply_air_flow,
  319. "supply_air_flow_lower_limit": lower_limit,
  320. "supply_air_flow_upper_limit": upper_limit,
  321. "season": season,
  322. }
  323. return vav_params
  324. @logger.catch()
  325. async def get_vav_control_v1(project: str, equipment_id: str) -> VAVBox:
  326. vav_params = await fetch_vav_control_params(project, equipment_id)
  327. vav = VAVBox(**vav_params)
  328. vav_controller = VAVController(vav)
  329. await vav_controller.run()
  330. regulated_vav = vav_controller.get_results()
  331. return regulated_vav
  332. @logger.catch()
  333. async def get_vav_control_v2(db: Session, project_id: str, equipment_id: str) -> VAVBox:
  334. vav_params = await fetch_vav_control_params(project_id, equipment_id)
  335. vav = VAVBox(**vav_params)
  336. weights = get_weights_by_vav(db, equipment_id)
  337. vav_controller = VAVControllerV2(
  338. vav, [SpaceWeight.from_orm(weight) for weight in weights], vav_params["season"]
  339. )
  340. await vav_controller.run()
  341. regulated_vav = vav_controller.get_results()
  342. return regulated_vav
  343. @logger.catch()
  344. async def build_acatva_instructions(
  345. params: ACATVAInstructionsRequest,
  346. ) -> ACATVAInstructions:
  347. space_params = []
  348. for sp in params.spaces:
  349. temp_sp = Space(**sp.dict())
  350. temp_sp.diff = temp_sp.temperature_target - temp_sp.realtime_temperature
  351. space_params.append(temp_sp)
  352. if params.supply_air_temperature == -1:
  353. if params.acatah_supply_air_temperature == -1:
  354. supply_air_temperature = np.NAN
  355. else:
  356. supply_air_temperature = params.acatah_supply_air_temperature
  357. else:
  358. supply_air_temperature = params.supply_air_temperature
  359. vav_params = VAVBox(
  360. spaces=space_params,
  361. supply_air_temperature=supply_air_temperature,
  362. supply_air_flow=params.supply_air_flow,
  363. supply_air_flow_lower_limit=params.supply_air_flow_lower_limit,
  364. supply_air_flow_upper_limit=params.supply_air_flow_upper_limit,
  365. )
  366. controller = VAVControllerV3(vav_params=vav_params, season=Season(params.season))
  367. await controller.run()
  368. regulated_vav = controller.get_results()
  369. instructions = ACATVAInstructions(
  370. supply_air_flow_set=regulated_vav.supply_air_flow_set,
  371. virtual_realtime_temperature=regulated_vav.virtual_realtime_temperature,
  372. virtual_temperature_target_set=regulated_vav.virtual_target_temperature,
  373. )
  374. return instructions
  375. @logger.catch()
  376. async def build_acatva_instructions_for_JM(
  377. params: ACATVAInstructionsRequestV2,
  378. ) -> float:
  379. # Control logic for Jiaming.
  380. space_params = []
  381. for sp in params.spaces:
  382. temp_sp = Space(**sp.dict())
  383. if temp_sp.temperature_target:
  384. temp_sp.diff = temp_sp.temperature_target - temp_sp.realtime_temperature
  385. else:
  386. temp_sp.diff = 0.0
  387. space_params.append(temp_sp)
  388. vav_params = VAVBox(spaces=space_params)
  389. return_air_temp = (
  390. np.NAN if params.return_air_temp == -1.0 else params.return_air_temp
  391. )
  392. controller = VAVControllerV4(
  393. vav_params=vav_params,
  394. season=Season(params.season),
  395. return_air_temp=return_air_temp,
  396. )
  397. await controller.run()
  398. regulated_vav = controller.get_results()
  399. next_temp_set = regulated_vav.setting_temperature
  400. if next_temp_set:
  401. if np.isnan(next_temp_set):
  402. next_temp_set = -1.0
  403. else:
  404. next_temp_set = -1.0
  405. return next_temp_set