vav.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465
  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 sp.realtime_temperature > max(
  150. 27.0, sp.temperature_target
  151. ) or sp.realtime_temperature < min(21.0, sp.temperature_target):
  152. if sp.temperature_target > 0.0:
  153. bad_spaces.append(sp)
  154. if bad_spaces:
  155. virtual_diff = (
  156. self.equipment.virtual_target_temperature
  157. - self.equipment.virtual_realtime_temperature
  158. )
  159. if self.season == Season.cooling:
  160. bad_spaces = sorted(bad_spaces, key=attrgetter("diff"))
  161. worst = bad_spaces[0]
  162. if worst.diff <= 0:
  163. if worst.diff < virtual_diff:
  164. self.equipment.virtual_target_temperature = (
  165. worst.temperature_target
  166. )
  167. self.equipment.virtual_realtime_temperature = (
  168. worst.realtime_temperature
  169. )
  170. else:
  171. self.equipment.virtual_target_temperature = (
  172. min(21.0, worst.temperature_target) + 0.5
  173. )
  174. self.equipment.virtual_realtime_temperature = (
  175. worst.realtime_temperature
  176. )
  177. elif self.season == Season.heating:
  178. bad_spaces = sorted(bad_spaces, key=attrgetter("diff"), reverse=True)
  179. worst = bad_spaces[0]
  180. if worst.diff >= 0:
  181. if worst.diff > virtual_diff:
  182. self.equipment.virtual_target_temperature = (
  183. worst.temperature_target
  184. )
  185. self.equipment.virtual_realtime_temperature = (
  186. worst.realtime_temperature
  187. )
  188. else:
  189. self.equipment.virtual_target_temperature = (
  190. max(27.0, worst.temperature_target) - 0.5
  191. )
  192. self.equipment.virtual_realtime_temperature = (
  193. worst.realtime_temperature
  194. )
  195. return (
  196. self.equipment.virtual_target_temperature,
  197. self.equipment.virtual_realtime_temperature,
  198. )
  199. async def run(self) -> None:
  200. await self.build_virtual_temperature()
  201. temperature_set, temperature_realtime = await self.rectify()
  202. await self.get_supply_air_flow_set(temperature_set, temperature_realtime)
  203. self.equipment.running_status = True
  204. class VAVControllerV3(VAVControllerV2):
  205. def __init__(self, vav_params: VAVBox, season: Season):
  206. super(VAVControllerV3, self).__init__(vav_params)
  207. self.season = season
  208. def get_valid_spaces(self) -> List:
  209. valid_spaces = list()
  210. for sp in self.equipment.spaces:
  211. if sp.realtime_temperature > 0.0 and sp.temperature_target > 22.0:
  212. valid_spaces.append(sp)
  213. return valid_spaces
  214. async def build_virtual_temperature(self) -> None:
  215. valid_spaces = self.get_valid_spaces()
  216. if not valid_spaces:
  217. virtual_realtime, virtual_target = np.NAN, np.NAN
  218. else:
  219. sorted_spaces = sorted(
  220. valid_spaces, key=lambda x: x.vav_temporary_update_time
  221. )
  222. if sorted_spaces[-1].vav_temporary_update_time > get_time_str(
  223. 60 * 60 * 2, flag="ago"
  224. ):
  225. virtual_realtime = sorted_spaces[-1].realtime_temperature
  226. virtual_target = sorted_spaces[-1].temperature_target
  227. else:
  228. virtual_realtime, virtual_target = 0.0, 0.0
  229. total_weight = 0.0
  230. for sp in valid_spaces:
  231. temp_weight = sp.vav_default_weight
  232. virtual_realtime += sp.realtime_temperature * temp_weight
  233. virtual_target += sp.temperature_target * temp_weight
  234. total_weight += temp_weight
  235. if total_weight == 0:
  236. for sp in valid_spaces:
  237. virtual_realtime += sp.realtime_temperature
  238. virtual_target += sp.temperature_target
  239. virtual_realtime /= len(valid_spaces)
  240. virtual_target /= len(valid_spaces)
  241. else:
  242. virtual_realtime /= total_weight
  243. virtual_target /= total_weight
  244. self.equipment.virtual_realtime_temperature = virtual_realtime
  245. self.equipment.virtual_target_temperature = virtual_target
  246. class VAVControllerV4(VAVControllerV3):
  247. def __init__(self, vav_params: VAVBox, season: Season, return_air_temp: float):
  248. super().__init__(vav_params, season)
  249. self.return_air_temp = return_air_temp
  250. def get_next_temp_set(
  251. self, virtual_realtime_temp: float, virtual_target_temp: float
  252. ) -> float:
  253. if np.isnan(virtual_realtime_temp) or np.isnan(virtual_target_temp):
  254. next_temp_set = np.NAN
  255. else:
  256. next_temp_set = (
  257. virtual_target_temp + self.return_air_temp - virtual_realtime_temp
  258. )
  259. self.equipment.setting_temperature = next_temp_set
  260. return next_temp_set
  261. async def run(self) -> None:
  262. await self.build_virtual_temperature()
  263. temperature_set, temperature_realtime = await self.rectify()
  264. self.get_next_temp_set(temperature_set, temperature_realtime)
  265. self.equipment.running_status = True
  266. async def fetch_vav_control_params(project_id: str, equipment_id: str) -> Dict:
  267. async with AsyncClient() as client:
  268. duo_duo = Duoduo(client, project_id)
  269. platform = DataPlatformService(client, project_id)
  270. season = await duo_duo.get_season()
  271. served_spaces = await duo_duo.get_space_by_equipment(equipment_id)
  272. space_objects = []
  273. realtime_supply_air_temperature_list = []
  274. for sp in served_spaces:
  275. sp_id = sp.get("id")
  276. transfer = SpaceInfoService(client, project_id, sp_id)
  277. current_target = await transfer.get_current_temperature_target()
  278. realtime_temperature = await platform.get_realtime_temperature(sp_id)
  279. related_equipment = await transfer.get_equipment()
  280. equipment_objects = []
  281. for eq in related_equipment:
  282. if eq.get("category") == "ACATFC":
  283. speed = await platform.get_fan_speed(eq.get("id"))
  284. temp_fcu_params = {"id": eq.get("id"), "air_valve_speed": speed}
  285. fcu = FCU(**temp_fcu_params)
  286. equipment_objects.append(fcu)
  287. elif eq.get("category") == "ACATAH":
  288. realtime_supply_air_temperature_list.append(
  289. await platform.get_realtime_supply_air_temperature(eq.get("id"))
  290. )
  291. temp_space_params = {
  292. "id": sp_id,
  293. "realtime_temperature": realtime_temperature,
  294. "equipment": equipment_objects,
  295. "temperature_target": current_target,
  296. "diff": current_target - realtime_temperature,
  297. }
  298. space = Space(**temp_space_params)
  299. space_objects.append(space)
  300. realtime_supply_air_temperature = np.array(
  301. realtime_supply_air_temperature_list
  302. ).mean()
  303. realtime_supply_air_flow = await platform.get_realtime_supply_air_flow(
  304. equipment_id
  305. )
  306. lower_limit, upper_limit = await platform.get_air_flow_limit(equipment_id)
  307. vav_params = {
  308. "id": equipment_id,
  309. "spaces": space_objects,
  310. "supply_air_temperature": realtime_supply_air_temperature,
  311. "supply_air_flow": realtime_supply_air_flow,
  312. "supply_air_flow_lower_limit": lower_limit,
  313. "supply_air_flow_upper_limit": upper_limit,
  314. "season": season,
  315. }
  316. return vav_params
  317. @logger.catch()
  318. async def get_vav_control_v1(project: str, equipment_id: str) -> VAVBox:
  319. vav_params = await fetch_vav_control_params(project, equipment_id)
  320. vav = VAVBox(**vav_params)
  321. vav_controller = VAVController(vav)
  322. await vav_controller.run()
  323. regulated_vav = vav_controller.get_results()
  324. return regulated_vav
  325. @logger.catch()
  326. async def get_vav_control_v2(db: Session, project_id: str, equipment_id: str) -> VAVBox:
  327. vav_params = await fetch_vav_control_params(project_id, equipment_id)
  328. vav = VAVBox(**vav_params)
  329. weights = get_weights_by_vav(db, equipment_id)
  330. vav_controller = VAVControllerV2(
  331. vav, [SpaceWeight.from_orm(weight) for weight in weights], vav_params["season"]
  332. )
  333. await vav_controller.run()
  334. regulated_vav = vav_controller.get_results()
  335. return regulated_vav
  336. @logger.catch()
  337. async def build_acatva_instructions(
  338. params: ACATVAInstructionsRequest,
  339. ) -> ACATVAInstructions:
  340. space_params = []
  341. for sp in params.spaces:
  342. temp_sp = Space(**sp.dict())
  343. temp_sp.diff = temp_sp.temperature_target - temp_sp.realtime_temperature
  344. space_params.append(temp_sp)
  345. if params.supply_air_temperature == -1:
  346. if params.acatah_supply_air_temperature == -1:
  347. supply_air_temperature = np.NAN
  348. else:
  349. supply_air_temperature = params.acatah_supply_air_temperature
  350. else:
  351. supply_air_temperature = params.supply_air_temperature
  352. vav_params = VAVBox(
  353. spaces=space_params,
  354. supply_air_temperature=supply_air_temperature,
  355. supply_air_flow=params.supply_air_flow,
  356. supply_air_flow_lower_limit=params.supply_air_flow_lower_limit,
  357. supply_air_flow_upper_limit=params.supply_air_flow_upper_limit,
  358. )
  359. controller = VAVControllerV3(vav_params=vav_params, season=Season(params.season))
  360. await controller.run()
  361. regulated_vav = controller.get_results()
  362. instructions = ACATVAInstructions(
  363. supply_air_flow_set=regulated_vav.supply_air_flow_set,
  364. virtual_realtime_temperature=regulated_vav.virtual_realtime_temperature,
  365. virtual_temperature_target_set=regulated_vav.virtual_target_temperature,
  366. )
  367. return instructions
  368. @logger.catch()
  369. async def build_acatva_instructions_for_JM(
  370. params: ACATVAInstructionsRequestV2,
  371. ) -> float:
  372. # Control logic for Jiaming.
  373. space_params = []
  374. for sp in params.spaces:
  375. temp_sp = Space(**sp.dict())
  376. if temp_sp.temperature_target:
  377. temp_sp.diff = temp_sp.temperature_target - temp_sp.realtime_temperature
  378. else:
  379. temp_sp.diff = 0.0
  380. space_params.append(temp_sp)
  381. vav_params = VAVBox(spaces=space_params)
  382. return_air_temp = (
  383. np.NAN if params.return_air_temp == -1.0 else params.return_air_temp
  384. )
  385. controller = VAVControllerV4(
  386. vav_params=vav_params,
  387. season=Season(params.season),
  388. return_air_temp=return_air_temp,
  389. )
  390. await controller.run()
  391. regulated_vav = controller.get_results()
  392. next_temp_set = regulated_vav.setting_temperature
  393. if next_temp_set:
  394. if np.isnan(next_temp_set):
  395. next_temp_set = -1.0
  396. else:
  397. next_temp_set = -1.0
  398. return next_temp_set