!7

sclass VidFile {

  File file;

  double length = -1;

  transient float[] profile;

  transient L<DoubleRange> ranges;

  S id() { ret md5FromFilePathSizeAndDate(file); }

  File audioFile() { ret prepareCacheProgramFile("preview-" + id() + ".wav"); }

}

module AJCBatch > DynObjectTable<VidFile> {

  S outputFile;

  int volumeThresholdPercent = 15;

  S minSilence = "0.2", leadIn = "0.2", leadOut = "0.2";

  transient double profileSamplingInterval = 0.05; // 50 ms

  transient int audioWindowSize = iround(profileSamplingInterval*16000);

  S originalLength, cutLength;

  transient ReliableSingleThread rstScan = dm_rst(this, r scan);

  visualize {

    JComponent sup = super.visualize();

    ret centerAndSouthWithMargins(

      jHandleMultiFileDrop(vf<L<File>> onFileDrop,

        withCenteredTitle("Input videos to jump-cut & concatenate (you can drag&drop files here):",

        withRightAlignedButtons(sup,

          tableDependentButton(table, "Move up", r moveUp),

          tableDependentButton(table, "Move down", r moveDown),

          tableDependentButton(table, "Remove selected", r removeSelected)

        ))),

      vstackWithSpacing(

        centeredLine(withLabel("Total input duration:", dm_boldLabel('originalLength))),

        dm_ajc_parametersSection(),

        centeredLine(withLabel("Total output duration (est.):", dm_boldLabel('cutLength))),

        withLabel("Output video:", filePathInputWithBrowseButton(dm_textField('outputFile))),

        rightAlignedLine(fontSizePlus(3, jbutton("Make video", rThread makeVideo))),

        ffmpegVersionPanel()

      ));

  }

  void onFileDrop(L<File> files) enter {

    print("Have file drop");

    Set<File> haveFiles = collectAsSet file(data());

    addAll(map(listMinusSet(files, haveFiles), f -> nu VidFile(file := f)));

    rstScan.trigger();

  }

  start {

    itemToMap = func(VidFile v) -> Map {

      litorderedmap(

        "Video" := fileName(v.file),

        "Folder" := dirPath(v.file),

        "Duration" := !fileExists(v.file) ? "File not found"

          : v.length < 0 ? "[calculating]" : formatMinuteAndSeconds(iceil(v.length))

      )

    };

    rstScan.trigger();

    dm_watchFields(allNonStaticNonTransientFields(AJCParameters), r recut);

  }

  void recut {

    for (VidFile v : clonedList())

      v.ranges = null;

    rstScan.trigger();

  }

  void scan {

    bool change;

    double lenIn = 0, lenOut = 0;

    for (VidFile v : clonedList()) pcall {

      if (v.ranges == null || v.length < 0 && fileExists(v.file)) {

        File audio = v.audioFile();

        if (fileLength(audio) == 0) {

          print("Extracting audio from " + v.file);

          ffmpeg_toMonoAudio_16k(v.file, audio); // TODO: temp file for safety?

          v.profile = null;

        }

        if (v.profile == null)

          v.profile = decodeWAVToMonoSamples_floatVolumeProfile(audio, audioWindowSize);

        v.length = l(v.profile)*profileSamplingInterval;

        v.ranges = ajc_findSpeechPartsFromVolumeProfile(v.profile,

          shallowCloneToClass AJCParameters(module()));

        set change;

      }

      lenIn += v.length;

      lenOut += totalLengthOfDoubleRanges(v.ranges);

    }

    setField(originalLength := formatMinuteAndSeconds(iceil(lenIn));

    setField(cutLength := formatMinuteAndSeconds(iceil(lenOut));

    if (change) fireDataChanged();

  }

  void makeVideo {

    temp dm_tempDisableAllButtons();

    while (rstScan.running()) sleep(10);

    if (emptyAfterTrim(outputFile)) ret with infoBox("Need output file path");

    File out = newFile(trim(outputFile));

    L<VidFile> l = clonedList();

    if (empty(l)) ret with infoBox("No input files");

    if (fileExists(out) && !confirmOKCancel("Overwrite " + fileName(out) + "?")) ret;

    L<File> files = collect file(l);

    LL<DoubleRange> ranges = collect ranges(l);

    temp tempInfoBox_noHide("Splicing video...");

    backtickToConsole(ffmpegCmd() + " -y " + ffmpeg_argsForSplice_multipleInputVideos(files, out, ranges));

    if (fileExists(out))

      infoBox("Done splicing video!"  + fileInfo(out));

    else

      infoBox("Something went wrong...");

  }

  void moveUp {

    L<Int> indices = selectedIndices();

    int i = first(indices), j = last(indices)+1;

    if (i <= 0) ret;

    VidFile v = data.get(i-1);

    remove(v);

    add(j-1, v);

    new L<Int> sel;

    for (int k = i; k < j; k++) sel.add(k-1);

    selectTableRows(table, asIntArray(sel));

  }

  void moveDown {

    L<Int> indices = selectedIndices();

    int i = first(indices), j = last(indices)+1;

    if (j >= l(data)) ret;

    VidFile v = data.get(j);

    remove(v);

    add(i, v);

    new L<Int> sel;

    for (int k = i; k < j; k++) sel.add(k+1);

    selectTableRows(table, asIntArray(sel));

  }

}