oop

3 files changed, 75 insertions, 45 deletions

diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.live.js b/app/client/modules/pix2pixhd/views/pix2pixhd.live.js
index 3f027a1..0bdcefe 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.live.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.live.js
@@ -4,7 +4,7 @@ import { connect } from 'react-redux'
 
 import {
   ParamGroup, Param, Player,
-  Slider, Select, Button, Loading
+  Slider, Select, TextInput, Button, Loading
 } from '../../../common/'
 
 import { startRecording, stopRecording, saveFrame, toggleFPS } from '../../../live/player'
@@ -51,11 +51,13 @@ class Pix2PixHDLive extends Component {
     this.props.actions.live.seek(frame)
   }
   start(){
-    // console.log(this.props)
-    const sequence = this.props.pix2pixhd.data.sequences[0].name
-    const checkpoint = this.props.pix2pixhd.data.checkpoints[0].name
+    console.log(this.props.opt)
+    const sequence = this.props.opt.sequence_name || this.props.pix2pixhd.data.sequences[0].name
+    const checkpoint = this.props.opt.checkpoint_name || this.props.pix2pixhd.data.checkpoints[0].name
     console.log('starting up!', sequence, checkpoint)
-    this.props.actions.tasks.live_task(sequence, checkpoint)
+    this.props.actions.tasks.live_task(sequence, checkpoint, {
+      folder_id: this.props.pix2pixhd.data.resultsFolder.id,
+    })
   }
   interrupt(){
     this.props.actions.queue.stop_task('gpu')
@@ -121,21 +123,10 @@ class Pix2PixHDLive extends Component {
               />
               <Slider live
                 name='frame_delay'
-                min={0.05} max={2.0} type='float'
+                min={0.0} max={2.0} type='float'
               />
               {this.renderRestartButton()}
-              <Button
-                title={
-                  this.props.opt.savingVideo
-                  ? 'Saving video...'
-                  : this.props.opt.recording
-                  ? 'Recording (' + timeInSeconds(this.props.opt.recordFrames/25) +')'
-                  : 'Record video'
-                }
-                onClick={this.toggleRecording}
-              >
-                {this.props.opt.savingVideo ? 'Saving' : this.props.opt.recording ? 'Recording' : 'Record'}
-              </Button>
+              {this.renderRecordButton()}
               <Button
                 title={'Save frame'}
                 onClick={saveFrame}
@@ -143,12 +134,20 @@ class Pix2PixHDLive extends Component {
                 Save
               </Button>
               <ParamGroup
-                title='Render on server'
+                title='Record video'
                 name='store_b'
                 onToggle={(value) => {
                   // now storing frames on server...
                 }}
               >
+                <TextInput
+                  title='Video name'
+                  name='final_tag'
+                  value={this.props.opt.final_tag}
+                  onSave={value => {
+                    this.props.actions.live.set_param('final_tag', value)
+                  }}
+                />
               </ParamGroup>
 
               <p class='last_message'>{this.props.last_message}</p>
@@ -197,7 +196,7 @@ class Pix2PixHDLive extends Component {
             >
               <Slider live
                 name='sequence_frac'
-                min={0.0} max={0.5} type='float'
+                min={0.0} max={1.0} type='float'
               />
               <Slider live
                 name='process_frac'
@@ -264,6 +263,7 @@ class Pix2PixHDLive extends Component {
     )
   }
   renderRestartButton(){
+    console.log(this.props.runner.gpu)
     if (this.props.runner.gpu.status === 'IDLE') {
       return (
         <Button
@@ -307,6 +307,23 @@ class Pix2PixHDLive extends Component {
       </div>
     )
   }
+  renderRecordButton(){
+    return null
+    // return (
+    //   <Button
+    //     title={
+    //       this.props.opt.savingVideo
+    //       ? 'Saving video...'
+    //       : this.props.opt.recording
+    //       ? 'Recording (' + timeInSeconds(this.props.opt.recordFrames/25) +')'
+    //       : 'Record video'
+    //     }
+    //     onClick={this.toggleRecording}
+    //   >
+    //     {this.props.opt.savingVideo ? 'Saving' : this.props.opt.recording ? 'Recording' : 'Record'}
+    //   </Button>
+    // )
+  }
 }
 function timeInSeconds(n){
   return n.toFixed(1) + ' s.'
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.results.js b/app/client/modules/pix2pixhd/views/pix2pixhd.results.js
index dcbbfad..aa4d6af 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.results.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.results.js
@@ -13,33 +13,13 @@ import { FileList, FileRow } from '../../../common/fileList.component'
 class Pix2pixHDResults extends Component {
   constructor(props){
     super()
-    // if (!props.pix2pixhd.data) props.actions.load_directories()
+    if (!props.pix2pixhd.results) props.actions.load_results()
   }
   render(){
-    if (this.props.pix2pixhd.loading) return <Loading progress={this.props.pix2pixhd.progress} />
-    // const { folderLookup, fileLookup, datasetLookup } = this.props.samplernn.data
+    if (! this.props.pix2pixhd.results) return <Loading progress={this.props.pix2pixhd.progress} />
 
-    // console.log(bestRenders.map(r => r.epoch))
-    // const path = folder.name === 'unsorted'
-    //   ? "/samplernn/import/"
-    //   : "/samplernn/datasets/" + folder.id + "/"
-    // return (
-    //   <div className='col bestRenders'>
-    //     <h3><Link to={path}>{folder.name}</Link></h3>
-    //     <FileList
-    //       linkFiles
-    //       files={bestRenders}
-    //       orderBy='date desc'
-    //       fields={'name date epoch size'}
-    //       onClick={(file, e) => {
-    //         e.preventDefault()
-    //         e.stopPropagation()
-    //         console.log('picked a file', file)
-    //         this.handlePick(file)
-    //       }}
-    //     />
-    //   </div>
-    // )
+    const { resultsFolder, results, renders, files } = this.props.pix2pixhd.results
+    console.log(resultsFolder, results)
 
     return (
       <div className='app pix2pixhd'>
@@ -47,6 +27,40 @@ class Pix2pixHDResults extends Component {
           <h1>Pix2PixHD Results</h1>
         </div>
         <div class='rows params renders'>
+
+          <FileList
+            linkFiles
+            files={files}
+            orderBy='date desc'
+            fields={'name date size'}
+          />
+  
+          <h3>renders</h3>
+          <FileList
+            files={renders}
+            orderBy='date desc'
+            fields={'name date size'}
+            onClick={(file, e) => {
+              e.preventDefault()
+              e.stopPropagation()
+              console.log('picked a result', file)
+              this.handlePick(file)
+            }}
+          />
+
+          <h3>results</h3>
+          <FileList
+            files={results}
+            orderBy='date desc'
+            fields={'name date count'}
+            onClick={(file, e) => {
+              e.preventDefault()
+              e.stopPropagation()
+              console.log('picked a result', file)
+              this.handlePick(file)
+            }}
+          />
+
         </div>
       </div>
     )
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.show.js b/app/client/modules/pix2pixhd/views/pix2pixhd.show.js
index 5777ac0..d58ee80 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.show.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.show.js
@@ -86,7 +86,6 @@ class Pix2PixHDShow extends Component {
     const input = pix2pixhd.data.fileLookup[dataset.input[0]]
     if (! input) return null
     if (input.name && input.name.match(/(gif|jpe?g|png)$/i)) return null
-    console.log(dataset)
     return (
       <div>
         <div class={'actions'}>