1 files changed, 68 insertions, 8 deletions

diff --git a/public/bundle.js b/public/bundle.js
index b45f099..09955f1 100644
--- a/public/bundle.js
+++ b/public/bundle.js
@@ -266,11 +266,21 @@ var Select = function (_Component) {
     key: 'render',
     value: function render() {
       var value = this.props.opt[this.props.name];
-      var options = this.props.options.map(function (key, i) {
+      var options = (this.props.options || []).map(function (key, i) {
+        var name = void 0,
+            value = void 0;
+        if (typeof key == 'string') {
+          name = key.length < 4 ? key.toUpperCase() : key;
+          value = key;
+        } else {
+          var frames = Math.round(key.count / 30) + ' s.';
+          name = key.name.replace(/_/g, ' ') + ' (' + frames + ')';
+          value = key.name;
+        }
         return (0, _preact.h)(
           'option',
-          { value: key, key: i },
-          capitalize(key)
+          { value: value, key: i },
+          name
         );
       });
       return (0, _preact.h)(
@@ -555,7 +565,7 @@ var app = (0, _preact.h)(
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.list_sequences = exports.list_checkpoints = exports.set_param = exports.get_params = undefined;
+exports.list_sequences = exports.list_epochs = exports.list_checkpoints = exports.set_param = exports.get_params = undefined;
 
 var _socket = __webpack_require__(/*! ../socket */ "./app/client/socket.js");
 
@@ -579,6 +589,11 @@ var list_checkpoints = exports.list_checkpoints = function list_checkpoints() {
   return { type: 'LOADING_CHECKPOINTS' };
 };
 
+var list_epochs = exports.list_epochs = function list_epochs(path) {
+  socket.list_epochs(path);
+  return { type: 'LOADING_EPOCHS' };
+};
+
 var list_sequences = exports.list_sequences = function list_sequences() {
   socket.list_sequences();
   return { type: 'LOADING_SEQUENCES' };
@@ -691,9 +706,16 @@ var App = function (_Component) {
   }
 
   _createClass(App, [{
+    key: 'componentWillUpdate',
+    value: function componentWillUpdate(nextProps) {
+      console.log('willupdate', nextProps.opt);
+      if (nextProps.opt.checkpoint_name !== this.props.opt.checkpoint_name) {
+        this.props.actions.list_epochs(nextProps.opt.checkpoint_name);
+      }
+    }
+  }, {
     key: 'render',
     value: function render() {
-      var props = this.props;
       return (0, _preact.h)(
         'div',
         { className: 'app' },
@@ -714,6 +736,21 @@ var App = function (_Component) {
                 name: 'send_image',
                 title: 'view mode',
                 options: ['a', 'b', 'sequence', 'recursive']
+              }),
+              (0, _preact.h)(_select2.default, {
+                name: 'checkpoint_name',
+                title: 'checkpoint',
+                options: this.props.checkpoints
+              }),
+              (0, _preact.h)(_select2.default, {
+                name: 'epoch',
+                title: 'epoch',
+                options: this.props.epochs
+              }),
+              (0, _preact.h)(_select2.default, {
+                name: 'sequence_name',
+                title: 'sequence',
+                options: this.props.sequences
               })
             )
           ),
@@ -843,7 +880,12 @@ var App = function (_Component) {
 }(_preact.Component);
 
 var mapStateToProps = function mapStateToProps(state) {
-  return {};
+  return {
+    opt: state.live.opt,
+    checkpoints: state.live.checkpoints,
+    epochs: state.live.epochs,
+    sequences: state.live.sequences
+  };
 };
 
 var mapDispatchToProps = function mapDispatchToProps(dispatch, ownProps) {
@@ -881,6 +923,7 @@ var liveInitialState = {
   error: null,
   opt: {},
   checkpoints: [],
+  checkpoint_dir: ['latest'],
   sequences: []
 };
 
@@ -911,6 +954,11 @@ var liveReducer = function liveReducer() {
         checkpoints: action.checkpoints
       });
 
+    case 'LIST_CHECKPOINT_DIR':
+      return _extends({}, state, {
+        checkpoint_dir: action.checkpoint_dir
+      });
+
     case 'LIST_SEQUENCES':
       return _extends({}, state, {
         sequences: action.sequences
@@ -940,6 +988,7 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.socket = undefined;
 exports.list_checkpoints = list_checkpoints;
+exports.list_epochs = list_epochs;
 exports.list_sequences = list_sequences;
 exports.get_params = get_params;
 exports.set_param = set_param;
@@ -978,6 +1027,12 @@ socket.on('res', function (data) {
         sequences: data.res
       });
       break;
+    case 'list_epochs':
+      _store.store.dispatch({
+        type: 'LIST_EPOCHS',
+        sequences: data.res
+      });
+      break;
     default:
       break;
   }
@@ -991,8 +1046,8 @@ socket.on('frame', function (data) {
   img.onload = function () {
     URL.revokeObjectURL(url);
     var player = document.querySelector('.player canvas');
-    player.getContext('2d');
-    player.drawImage(img, 0, 0, player.width, player.height);
+    var ctx = player.getContext('2d');
+    ctx.drawImage(img, 0, 0, player.width, player.height);
     // player.innerHTML = ''
     // player.appendChild(img)
   };
@@ -1004,6 +1059,11 @@ function list_checkpoints() {
     cmd: 'list_checkpoints'
   });
 }
+function list_epochs() {
+  socket.emit('cmd', {
+    cmd: 'list_epochs'
+  });
+}
 function list_sequences() {
   socket.emit('cmd', {
     cmd: 'list_sequences'