From 76c64dd94d25cd896c75bbe9d000aed7dfc33653 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 4 Jun 2018 22:32:57 +0200
Subject: messages from tasks

---
 app/client/modules/samplernn/samplernn.tasks.js | 37 +++++++++++++++++++++++++
 app/client/socket/socket.task.js                |  6 ++++
 2 files changed, 43 insertions(+)
 create mode 100644 app/client/modules/samplernn/samplernn.tasks.js

(limited to 'app/client')

diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
new file mode 100644
index 0000000..fa41fe8
--- /dev/null
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -0,0 +1,37 @@
+import uuidv1 from 'uuid/v1'
+
+import socket from '../../socket'
+import types from '../../types'
+
+import actions from '../../actions'
+
+export const train_task = (dataset, epochs=1) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'train',
+    dataset: dataset,
+    epochs: epochs,
+    opt: {
+      sample_length: 44100 * 5,
+      n_samples: 6,
+      keep_old_checkpoints: false,
+    }
+  }
+  return actions.queue.add_task(task)
+}
+export const fetch_task = (dataset) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'fetch',
+    dataset: dataset,
+  }
+  return actions.queue.add_task(task)
+}
+export const clear_cache_task = (dataset) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'clear_cache',
+    dataset: dataset,
+  }
+  return actions.queue.add_task(task)
+}
\ No newline at end of file
diff --git a/app/client/socket/socket.task.js b/app/client/socket/socket.task.js
index 2f3c65a..3a78c6c 100644
--- a/app/client/socket/socket.task.js
+++ b/app/client/socket/socket.task.js
@@ -40,6 +40,12 @@ socket.on('task_res', (data) => {
       break
     case 'set_priority':
       break
+    case 'progress':
+      dispatch({ type: types.task.progress, task: data.task })
+      break
+    case 'epoch':
+      dispatch({ type: types.task.progress, task: data.task })
+      break
     case 'error':
       return console.log('task error', data)
     default:
-- 
cgit v1.2.3-70-g09d2