import path from 'path'
import moment from 'moment'

const name = 'morph'
const cwd = process.env.MORPH_CWD || path.join(process.cwd(), '..', 'pytorch-sepconv')

const morph = {
  type: 'pytorch',
  script: 'test.pl',
  params: (task) => {
    let params = [
      '--first', path.join(cwd, '..', task.opt.dataset_module, 'sequences', task.opt.a)
      '--second', path.join(cwd, '..', task.opt.dataset_module, 'sequences', task.opt.b)
      '--a-offset', task.opt.a_offset,
      '--b-offset', task.opt.b_offset,
      '--steps', task.opt.steps,
      '--dilate', task.opt.dilate,
      '--video-out', ['morph', task.opt.mode, task.opt.dataset, moment().format('%YYYY%MM%DD_%hh%mm')].join('_') + '.mp4'
    ]
    if (task.opt.mode === 'average') {
      params.push('--average-videos')
      params.push('True')
    } else {
      params.push('--mix-videos')
      params.push('True')
    }
    return params
  }

  },
  listen: (task, line, i) => {
    if ( (parseInt(line) % 10) === 0) {
      return { type: 'epoch', task, epoch: (i/10)|0 }
    }
    return null
  }
}

export default {
  name, cwd,
  activities: {
    morph,
  }
}