draw predict

analysis_db.py

@@ -5,14 +5,32 @@ from util.debug import debug_print
 from draw_db import draw_lists, min_max_list, up_down_list
 from util.date_util import format_date
 
+MIN_FROM = 0
+MIN_TO = 25
+LIGHT_FROM = 0
+LIGHT_TO = 5
+ACC_FROM = 0
+ACC_TO = 10
+
 
 def get_filename(file_obj):
     return file_obj.name
 
 
-def read_db(db_file):
+def read_db(db_file, min_from, min_to, light_from, light_to, acc_from, acc_to):
     db_name = get_filename(db_file)
-    debug_print('read_db, db_name = ', db_name)
+    global MIN_FROM
+    global MIN_TO
+    global LIGHT_FROM
+    global LIGHT_TO
+    global ACC_FROM
+    global ACC_TO
+    MIN_FROM = min_from
+    MIN_TO = min_to
+    LIGHT_FROM = light_from
+    LIGHT_TO = light_to
+    ACC_FROM = acc_from
+    ACC_TO = acc_to
     return read_section(db_name, 'tb_section')
 
 
@@ -62,6 +80,20 @@ def clamp_list(_list, _min, _max):
     return [clamp(x, _min, _max) for x in _list]
 
 
+def predict_sleeping(min_list, light_list, acc_list):
+    res = []
+
+    cnt = min(len(min_list), len(light_list), len(acc_list))
+    for idx in range(cnt):
+        min_from_to = MIN_FROM <= min_list[idx] <= MIN_TO
+        light_from_to = LIGHT_FROM <= light_list[idx] <= LIGHT_TO
+        acc_from_to = ACC_FROM <= acc_list[idx] <= ACC_TO
+        sleeping = 1 if min_from_to and light_from_to and acc_from_to else 0
+        res.append(sleeping)
+
+    return res
+
+
 def read_sample(conn, section_id, tb_name):
     sql_str = ' SELECT section_id, _sample_id, _sample_end_id, section_date, section_end_date FROM ' + str(
         tb_name) + ' WHERE section_id == ' + str(section_id)
@@ -81,6 +113,7 @@ def read_sample(conn, section_id, tb_name):
         acc_list = []
         light_list = []
         screen_list = []
+        predict_list = []
         sample_sql = "SELECT _id, date, max, min, acc_max_dx, acc_max_dy, acc_max_dz, avg, max_snoring FROM sample_table WHERE _id >= " + str(
             _sample_id) + ' AND ' + ' _id <= ' + str(_sample_end_id)
         sample_cursor = conn.execute(sample_sql)
@@ -109,7 +142,11 @@ def read_sample(conn, section_id, tb_name):
         if len(screen_list) > 0:
             screen_list = min_max_list(screen_list, 1, 11)
 
-        return draw_lists(title_str, date_list, max=max_list, min=min_list, acc=acc_list, screen=screen_list, light=light_list)
+        if len(min_list) > 0 and len(light_list) > 0 and len(acc_list) > 0:
+            predict_list = predict_sleeping(min_list, light_list, acc_list)
+
+        return draw_lists(title_str, date_list, max=max_list, min=min_list, acc=acc_list, screen=screen_list,
+                          light=light_list, predict=predict_list)
 
 
 if __name__ == '__main__':

app.py

@@ -10,12 +10,19 @@ def plot_pens(alpha):
     plt.scatter(x=df_pens['bill_length_mm'], y=df_pens['bill_depth_mm'])
     return fig
 
+
+min_from = gr.Slider(label="min_from", value=0, minimum=0, maximum=120, step=1)
+min_to = gr.Slider(label="min_to", value=25, minimum=0, maximum=120, step=1)
+light_from = gr.Slider(label="light_from", value=0, minimum=0, maximum=120, step=1)
+light_to = gr.Slider(label="light_to", value=5, minimum=0, maximum=120, step=1)
+acc_from = gr.Slider(label="acc_from", value=0, minimum=0, maximum=120, step=1)
+acc_to = gr.Slider(label="acc_to", value=10, minimum=0, maximum=120, step=1)
 iface = gr.Interface(
     fn=read_db,
-    inputs=['file'],
+    inputs=['file', min_from, min_to, light_from, light_to, acc_from, acc_to],
     outputs=['plot', 'plot', 'plot', 'plot', 'plot', 'plot', 'plot', 'plot', 'plot', 'plot'],
     title="Sleep Data test",
     description="Let's talk pens.",
     article="Talk more about Penguins here, shall we?",
     theme='peach'
-).launch()
+).launch()

draw_db.py

@@ -31,12 +31,16 @@ def draw_lists(title='', date_list=None, **kv_list):
 
     idx_list = 0
     idx_axs = 0
-    rows = math.ceil(len(kv_list.items()) / 3)
+    rows = math.ceil(len(kv_list.items()) / 3) + 1
     fig, axs = plt.subplots(rows, 1, sharex=True)
-    for key, value in kv_list.items():
-        if type(key) is str and type(value) is list:
-            axs[idx_axs].plot(date_list, value, label=key)
-            axs[idx_axs].legend()
+    for key, values in kv_list.items():
+        if type(key) is str and type(values) is list:
+            if key == 'predict':
+                axs[rows - 1].plot(date_list, values, label=key)
+                axs[rows - 1].legend()
+            else:
+                axs[idx_axs].plot(date_list, values, label=key)
+                axs[idx_axs].legend()
         idx_list = idx_list + 1
         idx_axs = int(idx_list / 3)