You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.2 KiB
48 lines
1.2 KiB
5 years ago
|
from analytic_types import ModelCache, TimeSeries
|
||
5 years ago
|
from analytic_types.segment import Segment
|
||
|
|
||
4 years ago
|
from enum import Enum
|
||
5 years ago
|
from typing import List, Optional, Tuple
|
||
5 years ago
|
|
||
5 years ago
|
import utils.meta
|
||
5 years ago
|
|
||
4 years ago
|
class Bound(Enum):
|
||
|
ALL = 'ALL'
|
||
|
UPPER = 'UPPER'
|
||
|
LOWER = 'LOWER'
|
||
|
|
||
5 years ago
|
class DetectionResult:
|
||
|
|
||
|
def __init__(
|
||
|
self,
|
||
5 years ago
|
cache: Optional[ModelCache] = None,
|
||
|
segments: Optional[List[Segment]] = None,
|
||
5 years ago
|
last_detection_time: int = None
|
||
5 years ago
|
):
|
||
5 years ago
|
if cache is None:
|
||
|
cache = {}
|
||
|
if segments is None:
|
||
|
segments = []
|
||
5 years ago
|
self.cache = cache
|
||
|
self.segments = segments
|
||
|
self.last_detection_time = last_detection_time
|
||
5 years ago
|
|
||
|
# TODO: use @utils.meta.JSONClass (now it can't serialize list of objects)
|
||
|
def to_json(self):
|
||
|
return {
|
||
|
'cache': self.cache,
|
||
|
'segments': list(map(lambda segment: segment.to_json(), self.segments)),
|
||
5 years ago
|
'lastDetectionTime': self.last_detection_time
|
||
5 years ago
|
}
|
||
5 years ago
|
|
||
|
@utils.meta.JSONClass
|
||
|
class ProcessingResult():
|
||
|
|
||
5 years ago
|
def __init__(
|
||
|
self,
|
||
|
lower_bound: Optional[TimeSeries] = None,
|
||
|
upper_bound: Optional[TimeSeries] = None,
|
||
|
):
|
||
|
self.lower_bound = lower_bound
|
||
|
self.upper_bound = upper_bound
|