summaryrefslogtreecommitdiff
path: root/app
diff options
context:
space:
mode:
Diffstat (limited to 'app')
-rw-r--r--app/client/modules/pix2pixhd/pix2pixhd.tasks.js6
-rw-r--r--app/client/modules/pix2pixhd/views/pix2pixhd.train.js14
-rw-r--r--app/client/modules/pix2pixhd/views/sequence.editor.js8
-rw-r--r--app/relay/modules/pix2pixhd.js17
4 files changed, 31 insertions, 14 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 89184d2..5d8e084 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -75,10 +75,10 @@ export const augment_task = (dataset, opt) => dispatch => {
return actions.queue.add_task(task)
}
-export const create_dataset_task = (opt) => dispatch => {
+export const splice_task = (opt) => dispatch => {
const task = {
module: module.name,
- activity: 'create_dataset',
+ activity: 'splice',
dataset: opt.title,
folder_id: opt.folder_id,
opt: {
@@ -86,6 +86,6 @@ export const create_dataset_task = (opt) => dispatch => {
}
}
console.log(task)
- console.log('add create_dataset task')
+ console.log('add splice task')
return actions.queue.add_task(task)
} \ No newline at end of file
diff --git a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
index de32fcd..65678ae 100644
--- a/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
+++ b/app/client/modules/pix2pixhd/views/pix2pixhd.train.js
@@ -30,7 +30,7 @@ class Pix2PixHDTrain extends Component {
augment_name: '',
augment_take: 100,
augment_make: 20,
- generated: false,
+ processed: false,
}
constructor(props){
super(props)
@@ -59,13 +59,13 @@ class Pix2PixHDTrain extends Component {
|| (prevProps.pix2pixhd.data && prevState.dataset !== this.state.dataset)) {
const dataset = this.props.pix2pixhd.data.datasetLookup[this.state.dataset]
if (dataset) {
- const generated = dataset.input
+ const processed = dataset.input
.map(f => this.props.pix2pixhd.data.fileLookup[f])
.reduce((a,b) => {
- return b.generated || a
+ return b.processed || a
}, false)
- dataset.generated = generated
- this.setState({ generated })
+ dataset.processed = processed
+ this.setState({ processed })
}
}
if (prevState.dataset !== this.state.dataset) {
@@ -149,7 +149,7 @@ class Pix2PixHDTrain extends Component {
renderEditor(){
const { pix2pixhd, queue, remote } = this.props
const { checkpoint, folder_id } = pix2pixhd
- const { dataset, generated } = this.state
+ const { dataset, processed } = this.state
return (
<div>
<Group title='Sequence Editor'>
@@ -157,7 +157,7 @@ class Pix2PixHDTrain extends Component {
folder_id={folder_id}
module={pix2pixhdModule}
checkpoint={checkpoint}
- generated={generated}
+ processed={processed}
/>
</Group>
<div className='columns'>
diff --git a/app/client/modules/pix2pixhd/views/sequence.editor.js b/app/client/modules/pix2pixhd/views/sequence.editor.js
index 135dfb3..7c3dc3b 100644
--- a/app/client/modules/pix2pixhd/views/sequence.editor.js
+++ b/app/client/modules/pix2pixhd/views/sequence.editor.js
@@ -60,7 +60,7 @@ class SequenceEditor extends Component {
}
render() {
- const { app, pix2pixhd, remote, checkpoint, folder_id, generated } = this.props
+ const { app, pix2pixhd, remote, checkpoint, folder_id, processed } = this.props
const { cursor, selection, title } = this.state
const path = "sequences/" + checkpoint.name
console.log(checkpoint, pix2pixhd)
@@ -82,14 +82,14 @@ class SequenceEditor extends Component {
/>
<Button
title='Create a new dataset?'
- onClick={() => remote.create_dataset_task({ title, sequence: checkpoint.name, selection, folder_id })}
+ onClick={() => remote.splice_task({ title, sequence: checkpoint.name, selection, folder_id })}
>
Create
</Button>
</Group>
<Group title='Salt dataset'>
- {generated
- ? 'Salting is only available on generated datasets.'
+ {processed
+ ? 'Salting is only available on processed datasets.'
: 'Salting coming soon!'}
</Group>
</div>
diff --git a/app/relay/modules/pix2pixhd.js b/app/relay/modules/pix2pixhd.js
index 7d3c939..a894d3e 100644
--- a/app/relay/modules/pix2pixhd.js
+++ b/app/relay/modules/pix2pixhd.js
@@ -229,6 +229,22 @@ const render_recursive = {
]
}
}
+const splice = {
+ type: 'perl',
+ script: 'splice.pl',
+ params: (task) => {
+ console.log(selection)
+ return [
+ '--dataset', task.dataset,
+ '--sequence', task.opt.sequence,
+ '--start_frame', selection.start.i,
+ '--end_frame', selection.end.i,
+ '--module', task.module,
+ '--folder_id', task.opt.folder_id,
+ '--endpoint', process.env.API_REMOTE + '/api/file/',
+ ]
+ }
+}
function pad(num, size) {
if (isNaN(parseInt(num))) return num;
@@ -249,5 +265,6 @@ export default {
live,
render,
render_recursive,
+ splice,
}
}