summaryrefslogtreecommitdiff
path: root/app/client/modules/samplernn/samplernn.show.js
blob: 64ccd348dec710d41620440eab9eca7ae111a85e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import { h, Component } from 'preact'
import { bindActionCreators } from 'redux'
import { connect } from 'react-redux'
import * as util from '../../util'

import * as samplernnActions from './samplernn.actions'
import * as audioPlayerActions from '../../common/audioPlayer/audioPlayer.actions'

import Loading from '../../common/loading.component'
import DatasetForm from '../../dataset/dataset.form'
import NewDatasetForm from '../../dataset/dataset.new'
import { FileList, FileRow } from '../../common/fileList.component'

import SampleRNNDatasets from './samplernn.datasets'

const samplernnModule = {
  name: 'samplernn',
  datatype: 'audio',
}

class SampleRNNShow extends Component {
  render(){
    const { samplernn, match, history } = this.props
    const { folderLookup } = (samplernn.data || {})
    const folder = (folderLookup || {})[samplernn.folder_id] || {}
    return (
      <div className='app'>
        <div class='heading'>
          <h1>{folder ? folder.name : <Loading />}</h1>
        </div>
        {folder && folder.name && folder.name !== 'unsorted' &&
          <DatasetForm
            title='Add Files'
            module={samplernnModule}
            folder={folder}
            canUpload canAddURL
          />
        }
        <SampleRNNDatasets
          id={this.props.match.params.id || localStorage.getItem('samplernn.last_id')}
          folder={folder}
          history={history}
          onPickFile={(file, e) => {
            e.preventDefault()
            e.stopPropagation()
            console.log('picked a file', file)
            this.handlePick(file)
          }}
          datasetActions={(dataset) => this.datasetActions(dataset)}
        />
      </div>
    )
  }
  datasetActions(dataset){
    const actions = this.props.actions.samplernn
    return (
      <div>
        <div class={'actions'}>
          <span class='link' onClick={() => actions.train_task(dataset, 1)}>train</span>
          <span class='link' onClick={() => actions.train_task(dataset, 2)}>2x</span>
          <span class='link' onClick={() => actions.train_task(dataset, 4)}>4x</span>
          <span class='link' onClick={() => actions.train_task(dataset, 6)}>6x</span>
          <span class='link' onClick={() => actions.train_task(dataset, 18)}>18x</span>
        </div>
        {dataset.isBuilt
        ? <div class='subtext'>
            {'fetched '}
            <span class='link' onClick={() => actions.clear_cache_task(dataset)}>rm</span>
          </div>
        : <div class='subtext'>
            <span class='link' onClick={() => actions.fetch_task(dataset)}>fetch</span>
          </div>
        }
      </div>
    )
  }
  handlePick(file){
    this.props.audioPlayer.play(file)
  }
}

const mapStateToProps = state => ({
  samplernn: state.module.samplernn,
  dataset: state.dataset,
})

const mapDispatchToProps = (dispatch, ownProps) => ({
  actions: bindActionCreators(samplernnActions, dispatch),
  audioPlayer: bindActionCreators(audioPlayerActions, dispatch),
})

export default connect(mapStateToProps, mapDispatchToProps)(SampleRNNShow)