msrd0/render_video
1
0
Fork 0
Code Issues Pull requests Releases 7 Activity Actions

fix memory limit

Browse source
This commit is contained in:
Dominic 2023-11-02 21:33:21 +01:00
parent f4fe73bf5f
commit ca3ab6a788
Signed by: msrd0
GPG key ID: DCC8C247452E98F9
2 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/main.rs
View file

@ -19,16 +19,22 @@ use std::{
collections::BTreeSet,
fmt::Display,
fs,
io::{self, BufRead as _, Write}
io::{self, BufRead as _, Write},
sync::RwLock
};
static MEM_LIMIT: RwLock<String> = RwLock::new(String::new());
#[derive(Debug, Parser)]
struct Args {
#[clap(short = 'C', long, default_value = ".")]
directory: PathBuf,
#[clap(short = 'c', long, default_value = "23ws-malo")]
course: String
course: String,
#[clap(short, long, default_value = "2G")]
mem_limit: String
}
#[allow(non_camel_case_types, clippy::upper_case_acronyms)]
@ -146,6 +152,7 @@ fn ask_time(question: impl Display) -> Time {
fn main() {
let args = Args::parse();
*(MEM_LIMIT.write().unwrap()) = args.mem_limit;
// process arguments
let directory = args.directory.canonicalize_utf8().unwrap();

6
src/render/mod.rs
View file

@ -8,7 +8,7 @@ use crate::{
iotro::{intro, outro},
render::ffmpeg::{Ffmpeg, FfmpegInput},
time::{format_date, Time},
Project, ProjectSourceMetadata, Resolution
Project, ProjectSourceMetadata, Resolution, MEM_LIMIT
};
use anyhow::{bail, Context};
use camino::{Utf8Path as Path, Utf8PathBuf as PathBuf};
@ -21,8 +21,6 @@ use std::{
process::{Command, Stdio}
};
const MEMORY_LIMIT: &str = "2G";
const INTRO_LEN: Time = Time {
seconds: 3,
micros: 0
@ -48,7 +46,7 @@ fn cmd() -> Command {
.arg("-q")
.arg("--expand-environment=no")
.arg("-p")
.arg(format!("MemoryMax={MEMORY_LIMIT}"))
.arg(format!("MemoryMax={}", MEM_LIMIT.read().unwrap()))
.arg("--user");
// we use busybox ash for having a shell that outputs commands with -x
cmd.arg("busybox")