!7

module VideoSplicer > DynPrintLog {
  S inputFile, outputFile;
  S timestamps;
  S originalLength, cutLength;
  S ffmpegOptions = "", videoCodec = "";
  
  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);
  
  transient int timeFieldMinWidth = 40;
  transient ImageSurface isPreview;
  transient JAmplitudeGraph graph;
  transient float[] profile;
  
  visual jHandleFileDrop(voidfunc(File f) { setField(inputFile := f2s(f)) },
    northCenterAndSouthWithMargins(
      vstackWithSpacing(
        withLabel("Input video:", filePathInputWithBrowseButton(dm_textField inputFile(), onChoose := rThread loadVideo)),
        withLabel("Output video:", filePathInputWithBrowseButton(dm_textField outputFile())),
        centeredLine(
          jbutton("Load video", rThread loadVideo),
          jbutton("Find jump cuts", rThread autoJumpCuts),
          jhspacer(10),
          jbutton("Make video", rThread makeVideo)),
        jCenteredSection("Jump Cut Parameters", centeredLineWithSpacing(20,
          withLabelLeftAndRight("Volume threshold:", dm_spinner volumeThresholdPercent(0, 100), "%"),
          withLabelLeftAndRight("Shortest silence to cut:", jMinWidth(timeFieldMinWidth, dm_centeredTextField minSilence()), "s"),
          jline(
            withLabel("Lead-in/out:", jMinWidth(timeFieldMinWidth, dm_centeredTextField leadIn())),
            jlabel("/"),
            withLabelToTheRight(jMinWidth(timeFieldMinWidth, dm_centeredTextField leadOut()), "s"))
        ))),
      jhsplit(0.2,
        centerAndSouthWithMargin(
          jLiveValueSection(dm_calculatedLiveValue(S, () -> "Timestamps (" + countLines(timestamps) + ")"), dm_textArea("timestamps")),
          vstack(
            rightAlignedLine(withLabel("Total in:", dm_label("originalLength"))),
            rightAlignedLine(withLabel("Total out (est.):", dm_label("cutLength"))))
        ),
        northAndCenterWithMargin(
          jsection("Audio", jMinHeight(50, graph =
            setForeground(Color.red, swingNu(JAmplitudeGraph)))),
          hgridWithSpacing(jsection("Preview", jscroll_center(isPreview = jImageSurface())),
          super))),
      jvstackWithSpacing(
        ffmpegVersionPanel(),
        hgridWithSpacing(dm_textFieldWithLabel ffmpegOptions(), dm_textFieldWithLabel videoCodec()))));
    
  void loadVideo enter {
    temp dm_tempDisableAllButtons();
    File f = newFile(inputFile);
    if (!fileExists(f)) ret with infoBox("File not found: " + f2s(f));
    S id = md5(f2s(f));
    
    File previewFile = prepareCacheProgramFile("preview-" + id + ".jpg");
    File audioFile = prepareCacheProgramFile("preview-" + id + ".wav");
    if (!fileExists(previewFile)) {
      print("Getting preview image...");
      ffmpeg_getSingleFrame(f, previewFile, 0.0);
      print("Done");
    } else print("Have preview image");
    isPreview.setImageAndZoomToDisplay(loadImage2(previewFile));

    if (!fileExists(audioFile)) {
      print("Extracting audio...");
      ffmpeg_toMonoAudio_16k(f, audioFile);
      print("Done - " + fileInfo(audioFile));
    } else print("Have audio");
    
    print("Getting volume profile...");
    profile = decodeWAVToMonoSamples_floatVolumeProfile(audioFile, audioWindowSize);
    print("Have volume profile (" + nEntries(l(profile)) + ")");
    setField(originalLength := formatMinuteAndSeconds(iceil(l(profile)*profileSamplingInterval)));
    //printStruct(takeFirstOfFloatArray(100, profile));
    
    graph.setValues(profile);
  }
  
  void makeVideo enter {
    temp dm_tempDisableAllButtons();
    if (empty(outputFile)) ret with infoBox("Need output file path");
    File in = newFile(inputFile), out = newFile(outputFile);
    if (empty(timestamps)) autoJumpCuts();
    L<DoubleRange> ranges = parseTimestampRanges(timestamps);
    if (empty(ranges)) ret with infoBox("No timestamps");
    print(+ranges);
    if (fileExists(out) && !confirmOKCancel("Overwrite " + fileName(out) + "?")) ret;
    temp tempInfoBox_noHide("Splicing video...");
    backtickToConsole(ffmpegCmd() + " -y " + ffmpegOptions + " " + ffmpeg_argsForSplice_usingFile(in, out, ranges,
      beforeOut := empty(videoCodec) ? "" : "-vcodec " + videoCodec));
    if (fileExists(out))
      infoBox("Done splicing video!"  + fileInfo(out));
    else
      infoBox("Something went wrong...");
  }
  
  void autoJumpCuts enter {
    if (profile == null) loadVideo();
    if (profile == null) ret;
    L<DoubleRange> ranges = audio_findSpeechPartsFromVolumeProfile(profile, profileSamplingInterval,
      volumeThreshold := volumeThresholdPercent/100.0,
      minSilenceDuration := parseDouble(minSilence),
      pre := parseDouble(leadIn),
      post := parseDouble(leadOut));
    setField(timestamps := formatTimestampRanges(ranges));
    infoBox("Created " + nSlices(ranges) + ", ready to make video");
  }
  
  start {
    shouldKeepTempFiles = () -> true;
    dm_watchField timestamps(r {
      pcall-silent {
        double len = totalLengthOfDoubleRanges(parseTimestampRanges(timestamps));
        setField(cutLength := formatMinuteAndSeconds(iceil(len)));
      }
    });
    
    dm_watchField inputFile(r {
      setField(outputFile := f2s(appendToBaseName(newFile(inputFile), "-cut")))
    });
  }
}