|
1 | 1 | from apparser.cv.models import CvAllData, CvChangeData, CvBox |
2 | | -from apparser.cv.events import Detected, UnDetected, Moved, Resized |
| 2 | +from apparser.cv.events import Detected, Undetected, Moved, Resized |
3 | 3 |
|
4 | 4 |
|
5 | 5 | def _is_moved(box: CvBox, old_box: CvBox) -> bool: |
6 | | - """Check whether a box position changed. |
7 | | -
|
8 | | - :param box: Current box state. |
9 | | - :type box: CvBox |
10 | | - :param old_box: Previous box state. |
11 | | - :type old_box: CvBox |
12 | | - :return: True if the box coordinates changed. |
13 | | - :rtype: bool |
14 | | - """ |
15 | 6 | return abs(box.x - old_box.x) > 0 or abs(box.y - old_box.y) > 0 |
16 | 7 |
|
17 | 8 |
|
18 | 9 | def _is_resized(box: CvBox, old_box: CvBox) -> bool: |
19 | | - """Check whether both box dimensions changed. |
20 | | -
|
21 | | - :param box: Current box state. |
22 | | - :type box: CvBox |
23 | | - :param old_box: Previous box state. |
24 | | - :type old_box: CvBox |
25 | | - :return: True if width and height both changed. |
26 | | - :rtype: bool |
27 | | - """ |
28 | 10 | return abs(box.width - old_box.width) > 0 or abs(box.height - old_box.height) > 0 |
29 | 11 |
|
30 | 12 |
|
@@ -59,7 +41,7 @@ def __get_undetected(self, current_data: CvAllData) -> list[CvChangeData]: |
59 | 41 | :rtype: list[CvChangeData] |
60 | 42 | """ |
61 | 43 | new_ids = [i.track_id for i in current_data.boxes if i.track_id is not None] |
62 | | - return [CvChangeData(UnDetected, i, i) for i in self.__old_data.boxes if |
| 44 | + return [CvChangeData(Undetected, i, i) for i in self.__old_data.boxes if |
63 | 45 | i.track_id not in new_ids and i.track_id is not None] |
64 | 46 |
|
65 | 47 | def check(self, data: CvAllData) -> list[CvChangeData]: |
|
0 commit comments