From 1f18e89c8d59d5e76b9ced11877583e1f9bdd5f6 Mon Sep 17 00:00:00 2001
From: Michael Fogleman <fogleman@gmail.com>
Date: Mon, 22 Jan 2018 16:23:10 -0500
Subject: [PATCH] tweaks

---
 axi/device.py   |  4 ++--
 examples/nes.py | 21 ++++++++++++---------
 2 files changed, 14 insertions(+), 11 deletions(-)

diff --git a/axi/device.py b/axi/device.py
index 62bc781..ba3a32d 100644
--- a/axi/device.py
+++ b/axi/device.py
@@ -25,9 +25,9 @@ PEN_DOWN_POSITION = 45
 PEN_DOWN_SPEED = 150
 PEN_DOWN_DELAY = 0
 
-ACCELERATION = 6
+ACCELERATION = 4
 MAX_VELOCITY = 6
-CORNER_FACTOR = 0.005 * 2
+CORNER_FACTOR = 0.005 * 3
 
 VID_PID = '04D8:FD92'
 
diff --git a/examples/nes.py b/examples/nes.py
index dfd239b..b16f9fd 100644
--- a/examples/nes.py
+++ b/examples/nes.py
@@ -5,12 +5,14 @@ import numpy as np
 import os
 import sys
 
-NUMBER = 1
-TITLE = 'Five Seconds of Donkey Kong'
-LABEL = '#%d' % NUMBER
+NUMBER = 4
+TITLE = 'Five Seconds of Balloon Fight'
+LABEL = '#%s' % NUMBER
 
-COLUMNS = 6
+COLUMNS = 8
 SECONDS = 5
+FRAME_OFFSET = 120
+MIN_CHANGES = 1
 
 def stack_drawings(ds, spacing=0):
     result = axi.Drawing()
@@ -55,13 +57,14 @@ def main():
 
     # trim to SECONDS worth of data
     n = len(data[0])
-    m = SECONDS * 60 / 2
-    a = max(0, int(n // 2 - m))
-    b = min(n, int(n // 2 + m))
+    m = int(SECONDS * 60 / 2)
+    mid = int(n // 2) + FRAME_OFFSET
+    a = max(0, mid - m)
+    b = min(n, mid + m)
     data = [x[a:b] for x in data]
 
     # remove addresses with too few values
-    data = [x for x in data if len(set(x)) > 1]
+    data = [x for x in data if len(set(x)) > MIN_CHANGES]
 
     print '%d series that changed' % len(data)
 
@@ -101,7 +104,7 @@ def main():
     print d.bounds
 
     # save outputs
-    dirname = 'nes/%d' % NUMBER
+    dirname = 'nes/%s' % NUMBER
     try:
         os.makedirs(dirname)
     except Exception: