import path from 'path'

const name = 'samplernn'
const cwd = process.env.SAMPLERNN_CWD || path.join(process.env.HOME, 'code/' + name + '/')

const fetch = {
  type: 'perl',
  script: 'get.pl',
  params: (task) => {
    console.log(task)
    return [ task.opt.url ]
  }
}
const train = {
  type: 'pytorch',
  script: 'train.py',
  params: (task) => {
    return [
      '--exp', task.dataset,
      '--dataset', task.dataset,
      '--frame_sizes', '8', '2',
      '--n_rnn', '2',
      '--epoch_limit', task.epochs || 4,
      '--sample_length', task.opt.sample_length || 44100 * 5,
      '--n_samples', task.opt.n_samples || 6,
      '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False',
    ]
  },
  onComplete: publish,
}
const generate = {
  type: 'pytorch',
  script: 'generate.py',
  params: (task) => {
    return [
      '--exp', task.dataset,
      '--dataset', task.dataset,
      '--frame_sizes', '8', '2',
      '--n_rnn', '2',
      '--sample_length', task.opt.sample_length || 44100 * 5,
      '--n_samples', task.opt.n_samples || 6,
      '--keep_old_checkpoints', task.opt.keep_old_checkpoints ? 'True' : 'False',
    ]
  },
  onComplete: publish,
}
const publish = {
  type: 'perl',
  script: 'latest.pl',
  params: (task) => {
    return ['-l', task.dataset]
  }
}
const report = {
  type: 'perl',
  script: 'latest.pl',
  params: ['-v'],
  isScript: true,
}

export default {
  name, cwd,
  activities: {
    fetch, train, generate, report,
  },
}