play frames

2 files changed, 43 insertions, 9 deletions

diff --git a/app/client/modules/pix2pix/index.js b/app/client/modules/pix2pix/index.js
index 076fef9..f3a4050 100644
--- a/app/client/modules/pix2pix/index.js
+++ b/app/client/modules/pix2pix/index.js
@@ -12,8 +12,8 @@ function router () {
   return (
     <section>
       <Route exact path='/pix2pix/new/' component={Pix2PixNew} />
-      <Route exact path='/pix2pix/datasets/' component={Pix2PixShow} />
-      <Route exact path='/pix2pix/datasets/:id/' component={Pix2PixShow} />
+      <Route exact path='/pix2pix/sequences/' component={Pix2PixShow} />
+      <Route exact path='/pix2pix/sequences/:id/' component={Pix2PixShow} />
       <Route exact path='/pix2pix/live/' component={Pix2PixLive} />
     </section>
   )
@@ -22,7 +22,7 @@ function router () {
 function links(){
   return (
     <span>
-      <span><Link to="/pix2pix/datasets/">datasets</Link></span>
+      <span><Link to="/pix2pix/sequences/">sequences</Link></span>
       <span>train</span>
       <span>process</span>
       <span><Link to="/pix2pix/live/">live</Link></span>
diff --git a/app/client/modules/pix2pix/views/pix2pix.live.js b/app/client/modules/pix2pix/views/pix2pix.live.js
index eebb364..76b6727 100644
--- a/app/client/modules/pix2pix/views/pix2pix.live.js
+++ b/app/client/modules/pix2pix/views/pix2pix.live.js
@@ -44,6 +44,12 @@ class Pix2PixLive extends Component {
     const frame = Math.floor(percentage * (parseInt(this.props.frame.sequence_len) || 1) + 1)
     this.props.actions.seek(frame)
   }
+  start(){
+    //
+  }
+  kill(){
+    //
+  }
   togglePlaying(){
     if (this.props.opt.processing) {
       this.props.actions.pause()
@@ -99,12 +105,7 @@ class Pix2PixLive extends Component {
                 value={(this.props.frame.sequence_i || 0) / (this.props.frame.sequence_len || 1)}
                 onChange={this.seek}
               />
-              <Button
-                title={'Processing: ' + (this.props.opt.processing ? 'yes' : 'no')}
-                onClick={this.togglePlaying}
-              >
-                {this.props.opt.processing ? 'Pause' : 'Restart'}
-              </Button>
+              {this.renderRestartButton()}
               <Button
                 title={
                   this.props.opt.savingVideo
@@ -234,6 +235,38 @@ class Pix2PixLive extends Component {
       </div>
     )
   }
+  renderRestartButton(){
+    if (this.props.runner.gpu.status === 'IDLE') {
+      return (
+        <Button
+          title={'GPU Idle'}
+          onClick={this.start}
+        >Start</Button>
+      )
+    }
+    if (this.props.runner.gpu.task.module !== 'pix2pix') {
+      return (
+        <Button
+          title={'GPU Busy'}
+          onClick={this.kill}
+        >Kill</Button>
+      )
+    }
+    if (! this.props.opt.processing) {
+      return (
+        <Button
+          title={'Not processing'}
+          onClick={this.togglePlaying}
+        >Restart</Button>
+      )
+    }
+    return (
+      <Button
+        title={'Processing'}
+        onClick={this.togglePlaying}
+      >Restart</Button>
+    )
+  }
 }
 function timeInSeconds(n){
   return (n / 10).toFixed(1) + ' s.'
@@ -244,6 +277,7 @@ const mapStateToProps = state => ({
   checkpoints: state.live.checkpoints,
   epochs: state.live.epochs,
   sequences: state.live.sequences,
+  runner: state.system.runner,
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({