|
|
import traceback
|
|
|
from SciQLopBindings import PyDataProvider, Product, VectorTimeSerie, ScalarTimeSerie, DataSeriesType
|
|
|
import numpy as np
|
|
|
from speasy.cache import _cache
|
|
|
from speasy.common.datetime_range import DateTimeRange
|
|
|
from datetime import datetime, timedelta, timezone
|
|
|
from speasy.common.variable import SpeasyVariable
|
|
|
from speasy.amda import AMDA
|
|
|
|
|
|
amda = AMDA()
|
|
|
|
|
|
|
|
|
def vp_make_scalar(var=None):
|
|
|
if var is None:
|
|
|
return (((np.array([]), np.array([])), np.array([])), DataSeriesType.SCALAR)
|
|
|
else:
|
|
|
return (((var.time, np.array([])), var.data), DataSeriesType.SCALAR)
|
|
|
|
|
|
class DemoVP(PyDataProvider):
|
|
|
def __init__(self):
|
|
|
super().__init__()
|
|
|
self.register_products([Product("/VP/thb_fgm_gse_mod",[],{"type":"scalar"})])
|
|
|
|
|
|
def get_data(self,metadata,start,stop):
|
|
|
try:
|
|
|
tstart = datetime.fromtimestamp(start, tz=timezone.utc)
|
|
|
tend = datetime.fromtimestamp(stop, tz=timezone.utc)
|
|
|
thb_bs = amda.get_parameter(start_time=tstart, stop_time=tend, parameter_id='thb_bs', method="REST")
|
|
|
thb_bs.data = np.sqrt((thb_bs.data*thb_bs.data).sum(axis=1))
|
|
|
return vp_make_scalar(thb_bs)
|
|
|
except Exception as e:
|
|
|
print(traceback.format_exc())
|
|
|
print(f"Error in {__file__} ",str(e))
|
|
|
return (((np.array([]), np.array([])), np.array([])), ts_type)
|
|
|
|
|
|
|
|
|
t=DemoVP()
|
|
|
|
|
|
|
