1 files changed, 137 insertions, 0 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
new file mode 100644
index 0000000..1ee4886
--- /dev/null
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -0,0 +1,137 @@
+import uuidv1 from 'uuid/v1'
+
+import socket from '../../socket'
+import types from '../../types'
+
+import * as datasetLoader from '../../dataset/dataset.loader'
+
+import actions from '../../actions'
+
+import util from '../../util'
+
+import pix2pixhdModule from './pix2pixhd.module'
+
+export const load_directories = (id) => (dispatch) => {
+  const module = pix2pixhdModule.name
+  util.allProgress([
+    datasetLoader.load(module),
+    actions.socket.list_directory({ module, dir: 'sequences' }),
+    actions.socket.list_directory({ module, dir: 'datasets' }),
+    actions.socket.list_directory({ module, dir: 'checkpoints' }),
+    // actions.socket.disk_usage({ module, dir: 'datasets' }),
+  ], (percent, i, n) => {
+    console.log('pix2pixhd load progress', i, n)
+    dispatch({ type: types.app.load_progress, progress: { i, n }})
+  }).then(res => {
+    const [datasetApiReport, sequences, datasets, checkpoints] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const {
+      folderLookup,
+      fileLookup,
+      datasetLookup,
+      folders,
+      files,
+      unsortedFolder,
+    } = datasetApiReport
+    // console.log(datasetUsage)
+
+    const sequenceDirectories = sequences.filter(s => s.dir)
+    console.log(sequenceDirectories)
+    sequenceDirectories.forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.isBuilt = true
+      console.log(dir.name, dataset)
+    })
+
+    datasets.filter(s => s.dir).forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasDataset = true
+    })
+
+    const checkpointDirectories = checkpoints.filter(s => s.dir)
+    checkpointDirectories.forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasCheckpoints = true
+      dataset.checkpoints = [dir]
+    })
+
+    // console.log(res)
+
+    // flatDatasets.forEach(file => {
+    //   file.uuid = uuidv1()
+    //   fileLookup[file.uuid] = file
+    //   const name = file.name.split('.')[0]
+    //   const dataset = datasetLoader.getDataset(module, datasetLookup, name, unsortedFolder, file.date)
+    //   file.persisted = false
+    //   dataset.input.push(file.uuid)
+    // })
+
+    // // exp:coccokit_3-frame_sizes:8,2-n_rnn:2-dataset:coccokit_3
+    // const checkpoints = results.filter(s => s.dir).map(s => {
+    //   const checkpoint = s.name
+    //     .split('-')
+    //     .map(s => s.split(':'))
+    //     .filter(b => b.length && b[1])
+    //     .reduce((a,b) => (a[b[0]] = b[1]) && a, {})
+    //   checkpoint.name = checkpoint.name || checkpoint.dataset || checkpoint.exp
+    //   checkpoint.date = s.date
+    //   checkpoint.dir = s
+    //   checkpoint.persisted = false
+    //   const dataset = datasetLoader.getDataset(module, datasetLookup, checkpoint.name, unsortedFolder, checkpoint.date)
+    //   const loss = lossReport[checkpoint.name]
+    //   if (loss) {
+    //     dataset.epoch = checkpoint.epoch = loss.length
+    //     checkpoint.training_loss = loss
+    //   }
+    //   dataset.checkpoints.push(checkpoint)
+    //   return checkpoint
+    // })
+
+    // output.map(file => {
+    //   file.uuid = uuidv1()
+    //   fileLookup[file.uuid] = file
+    //   const pair = file.name.split('.')[0].split('-')
+    //   const dataset = datasetLoader.getDataset(module, datasetLookup, pair[0], unsortedFolder, file.date)
+    //   file.persisted = false
+    //   file.epoch = parseInt(file.epoch || pair[1].replace(/^\D+/, '')) || 0
+    //   dataset.epoch = Math.max(file.epoch, dataset.epoch || 0)
+    //   // here check if the file exists in dataset, if so just check that it's persisted
+    //   const found = dataset.output.some(file_id => {
+    //     // if (f.name === 
+    //     if (fileLookup[file_id].name === file.name) {
+    //       fileLookup[file_id].persisted = true
+    //       return true
+    //     }
+    //     return false
+    //   })
+    //   if (! found) {
+    //     dataset.output.push(file.uuid)
+    //   }
+    // })
+
+    dispatch({
+      type: types.dataset.load,
+      data: {
+        module,
+        folderLookup,
+        fileLookup,
+        datasetLookup,
+        folders, files,
+        sequences: sequenceDirectories,
+        datasets,
+        checkpoints: checkpointDirectories,
+      },
+    })
+    if (id) {
+      console.log('folder id', id)
+      dispatch({
+        type: types.dataset.set_folder,
+        data: {
+          folder_id: id,
+          module
+        },
+      })
+    }
+  }).catch(e => {
+    console.error(e)
+  })
+}