From 09b4d1541e6e06563a40f8829048e4225ed8fcf9 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 20 Jun 2018 16:12:38 +0200
Subject: mandatory results folder

---
 app/client/modules/pix2pixhd/pix2pixhd.actions.js  | 11 +++--
 app/client/modules/pix2pixhd/pix2pixhd.tasks.js    |  6 +--
 .../modules/pix2pixhd/views/pix2pixhd.live.js      | 53 ++++++++++++++--------
 .../modules/pix2pixhd/views/pix2pixhd.results.js   | 22 ++++-----
 .../modules/pix2pixhd/views/pix2pixhd.show.js      |  1 -
 5 files changed, 56 insertions(+), 37 deletions(-)

(limited to 'app/client/modules/pix2pixhd')

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index caf6b45..917b375 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -31,15 +31,16 @@ export const load_directories = (id) => (dispatch) => {
       folders,
       files,
       unsortedFolder,
+      resultsFolder,
     } = datasetApiReport
     // console.log(datasetUsage)
 
     const sequenceDirectories = sequences.filter(s => s.dir)
-    console.log(sequenceDirectories)
+    // console.log(sequenceDirectories)
     sequenceDirectories.forEach(dir => {
       const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
       dataset.isBuilt = true
-      console.log(dir.name, dataset)
+      // console.log(dir.name, dataset)
     })
 
     datasets.filter(s => s.dir).forEach(dir => {
@@ -119,6 +120,7 @@ export const load_directories = (id) => (dispatch) => {
         sequences: sequenceDirectories,
         datasets,
         checkpoints: checkpointDirectories,
+        resultsFolder,
       },
     })
     if (id) {
@@ -137,9 +139,9 @@ export const load_directories = (id) => (dispatch) => {
 }
 
 export const load_results = (id) => (dispatch) => {
-  console.log('hi')
   const module = pix2pixhdModule.name
   util.allProgress([
+    actions.folder.index({ name: 'results' }),
     actions.file.index({ module, generated: 1 }),
     actions.socket.list_sequences({ module, dir: 'results' }),
     actions.socket.list_directory({ module, dir: 'renders' }),
@@ -147,11 +149,12 @@ export const load_results = (id) => (dispatch) => {
     console.log('pix2pixhd load progress', i, n)
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
-    const [files, results, renders] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const [folders, files, results, renders] = res //, datasets, results, output, datasetUsage, lossReport] = res
     console.log(files, results, renders)
     dispatch({
       type: types.pix2pixhd.load_results,
       results: {
+        folder: folders[0],
         files,
         results,
         renders,
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 2e7cbbd..f3c5342 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -31,21 +31,21 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
     epochs: epochs,
     opt: {
       folder_id: folder_id,
-      load_size: 264, // switch to 256 for pix2wav
     }
   }
   console.log(task)
   return actions.queue.add_task(task)
 }
 
-export const live_task = (sequence, checkpoint) => dispatch => {
+export const live_task = (sequence, checkpoint, opt) => dispatch => {
   const task = {
     module: module.name,
     activity: 'live',
     dataset: sequence,
     checkpoint,
     opt: {
-      poll_delay: 0.09,
+      ...opt,
+      poll_delay: 0.01,
     }
   }
   console.log(task)
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.live.js b/app/client/modules/pix2pixhd/views/pix2pixhd.live.js
index 3f027a1..5dd4153 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'
@@ -52,10 +52,12 @@ class Pix2PixHDLive extends Component {
   }
   start(){
     // console.log(this.props)
-    const sequence = this.props.pix2pixhd.data.sequences[0].name
-    const checkpoint = this.props.pix2pixhd.data.checkpoints[0].name
+    const sequence = this.props.live.opt.sequence_name
+    const checkpoint = this.props.live.opt.checkpoint_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: 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>
@@ -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 ac0bb74..af8d90a 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.results.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.results.js
@@ -18,20 +18,21 @@ class Pix2pixHDResults extends Component {
   render(){
     if (! this.props.pix2pixhd.results) return <Loading progress={this.props.pix2pixhd.progress} />
 
-    const { results, renders, files } = this.props.pix2pixhd.results
-console.log(results)
+    const { resultsFolder, results, renders, files } = this.props.pix2pixhd.results
+    console.log(results)
+
     return (
       <div className='app pix2pixhd'>
         <div className='heading row middle'>
           <h1>Pix2PixHD Results</h1>
         </div>
         <div class='rows params renders'>
-  
-          <h3>results</h3>
+
           <FileList
-            files={results}
+            linkFiles
+            files={files}
             orderBy='date desc'
-            fields={'name date count'}
+            fields={'name date size'}
             onClick={(file, e) => {
               e.preventDefault()
               e.stopPropagation()
@@ -39,7 +40,7 @@ console.log(results)
               this.handlePick(file)
             }}
           />
-
+  
           <h3>renders</h3>
           <FileList
             files={renders}
@@ -53,12 +54,11 @@ console.log(results)
             }}
           />
 
-          <h3>files</h3>
+          <h3>results</h3>
           <FileList
-            linkFiles
-            files={files}
+            files={results}
             orderBy='date desc'
-            fields={'name date size'}
+            fields={'name date count'}
             onClick={(file, e) => {
               e.preventDefault()
               e.stopPropagation()
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'}>
-- 
cgit v1.2.3-70-g09d2