From ad94741c5b96f1e82836aec00606b1894ae5ce64 Mon Sep 17 00:00:00 2001
From: Dima Gerasimov <karlicoss@gmail.com>
Date: Tue, 14 Nov 2017 08:24:11 +0000
Subject: [PATCH] draw timestamps on axis

---
 TODO.org   |  3 +++
 analyse.py | 79 +++++++++++++++++++++++++++++++++++++++++++++---------
 2 files changed, 69 insertions(+), 13 deletions(-)

diff --git a/TODO.org b/TODO.org
index 238e5bd..62f6537 100644
--- a/TODO.org
+++ b/TODO.org
@@ -13,3 +13,6 @@ https://github.com/GlenCrawford/ruby_jawbone
 ** time_created: apparently starto of sleep (leftmost point of plot)
 ** time_updated: rightmost point of plot
 ** details -> asleep_time, awake_time
+
+
+* TODO align multiple on the same plot/picture to see patterns
diff --git a/analyse.py b/analyse.py
index 1a09b88..22eabf6 100755
--- a/analyse.py
+++ b/analyse.py
@@ -6,6 +6,8 @@ from backup_config import SLEEPS_FILE, GRAPHS_DIR
 from datetime import datetime
 from datetime import date
 
+fromtimestamp = datetime.fromtimestamp
+
 XID = str # TODO how to shared with backup thing?
 
 class SleepEntry:
@@ -28,10 +30,13 @@ class SleepEntry:
 
     # TODO take timezones into account?
     def created(self) -> datetime:
-        return datetime.fromtimestamp(self.js['time_created'])
+        return fromtimestamp(self.js['time_created'])
 
     def completed(self) -> datetime:
-        return datetime.fromtimestamp(self.js['time_completed'])
+        return fromtimestamp(self.js['time_completed'])
+
+    def asleep(self) -> datetime:
+        return fromtimestamp(self._details()['asleep_time'])
 
     def graph(self) -> str:
         return os.path.join(GRAPHS_DIR, self.xid() + ".png")
@@ -59,22 +64,70 @@ import numpy as np # type: ignore
 import matplotlib.pyplot as plt # type: ignore
 # pip install imageio
 from imageio import imread # type: ignore
+from scipy.misc import imresize # type: ignore
 
 img = imread(sleep.graph())
 # all of them are 300x300 images apparently
-(a, b, _) = img.shape
+size = img.shape
+(height, width, depth) = size
+size = (height, width * 5, depth)
+(height, width, depth) = size
+
 # print(img.shape)
 # print(img.size)
-print(a)
-print(b)
 
-np.random.seed(0)
-x = np.random.uniform(0.0, a, 100)
-y = np.random.uniform(0.0, b, 100)
-plt.scatter(x,y,zorder=1)
-plt.imshow(img,zorder=0)
-plt.title(str(sleep))
-plt.text(0, 0, str(sleep.created().time()))
-plt.text(300, 0, str(sleep.completed().time()))
+img = imresize(img, size)
+
+def hhmm(time: datetime):
+    return time.strftime("%H:%M")
+
+
+def xpos(time: datetime) -> float:
+    tick = span / width
+    fromstart = time - sleep.created()
+    return fromstart / tick
+
+print(xpos(sleep.asleep()))
+
+import matplotlib.dates as mdates # type: ignore
+from matplotlib.axes import Axes # type: ignore
+from matplotlib.ticker import MultipleLocator, FixedLocator # type: ignore
+
+fig = plt.figure(figsize=(10, 5))
+axes: Axes = fig.add_subplot(1,1,1)
+
+xlims = [sleep.created(), sleep.completed()]
+xlims = [mdates.date2num(i) for i in xlims]
+# axes.figure(figsize=(10, 5))
+axes.set_xlim(xlims)
+
+ylims = [0, 100]
+axes.set_ylim(ylims)
+# axes.set_xlim((sleep.created(), sleep.completed()))
+
+hhmm_fmt = mdates.DateFormatter('%H:%M')
+axes.xaxis.set_major_formatter(hhmm_fmt)
+ticks = [
+    sleep.created(),
+    sleep.asleep(),
+    sleep.completed(),
+]
+axes.xaxis.set_ticks(ticks)
+# axes.set_
+# plt.gca().xaxis.set_major_formatter(myFmt)
+# plt.gca().xaxis_date()
+
+# axes.imshow(img, zorder=0)
+# plt.figure(figsize=(10, 5))
+plt.imshow(img, zorder=0, extent=[
+    xlims[0], xlims[1],
+    ylims[0], ylims[1],
+]
+, aspect='auto'
+)
+# plt.title(str(sleep))
+
+# plt.text(sleep.asleep(), 0, hhmm(sleep.asleep()))
+
 plt.show()