Photon noise arguments (#764)

* Add photon-noise arguments

Width and height, use if you are changing the resolution of the input.

* Update chunk.rs

Readjustment and removed println

* cargo fmt

Author: superyu1337

av1an-core/src/chunk.rs

@@ -22,6 +22,7 @@ pub struct Chunk {
   pub passes: u8,
   pub video_params: Vec<String>,
   pub encoder: Encoder,
+  pub noise_size: (Option<u32>, Option<u32>),
   // do not break compatibility with output produced by older versions of av1an
   /// Optional target quality CQ level
   #[serde(rename = "per_shot_target_quality_cq")]
@@ -57,7 +58,13 @@ impl Chunk {
       let grain_table = Path::new(&self.temp).join(format!("iso{iso_setting}-grain.tbl"));
       if !grain_table.exists() {
         debug!("Generating grain table at ISO {}", iso_setting);
-        let (width, height) = self.input.resolution()?;
+        let (mut width, mut height) = self.input.resolution()?;
+        if self.noise_size.0.is_some() {
+          width = self.noise_size.0.unwrap();
+        }
+        if self.noise_size.1.is_some() {
+          height = self.noise_size.1.unwrap();
+        }
         let transfer_function = self
           .input
           .transfer_function_params_adjusted(&self.video_params)?;
@@ -102,6 +109,7 @@ mod tests {
       passes: 1,
       video_params: vec![],
       encoder: Encoder::x264,
+      noise_size: (None, None),
     };
     assert_eq!("00001", ch.name());
   }
@@ -120,6 +128,7 @@ mod tests {
       passes: 1,
       video_params: vec![],
       encoder: Encoder::x264,
+      noise_size: (None, None),
     };
     assert_eq!("10000", ch.name());
   }
@@ -139,6 +148,7 @@ mod tests {
       passes: 1,
       video_params: vec![],
       encoder: Encoder::x264,
+      noise_size: (None, None),
     };
     assert_eq!("d/encode/00001.ivf", ch.output());
   }

av1an-core/src/context.rs

@@ -870,6 +870,7 @@ impl Av1anContext {
       ),
       passes: self.args.passes,
       encoder: self.args.encoder,
+      noise_size: self.args.photon_noise_size,
       tq_cq: None,
     };
     chunk.apply_photon_noise_args(
@@ -922,6 +923,7 @@ impl Av1anContext {
       ),
       passes: self.args.passes,
       encoder: self.args.encoder,
+      noise_size: self.args.photon_noise_size,
       tq_cq: None,
     };
     chunk.apply_photon_noise_args(
@@ -1104,6 +1106,7 @@ impl Av1anContext {
       ),
       passes: self.args.passes,
       encoder: self.args.encoder,
+      noise_size: self.args.photon_noise_size,
       tq_cq: None,
     };
     chunk.apply_photon_noise_args(

av1an-core/src/scenes.rs

@@ -269,6 +269,7 @@ fn get_test_args() -> Av1anContext {
     encoder: Encoder::aom,
     extra_splits_len: Some(100),
     photon_noise: Some(10),
+    photon_noise_size: (None, None),
     chroma_noise: false,
     sc_pix_format: None,
     keep: false,

av1an-core/src/settings.rs

@@ -55,6 +55,7 @@ pub struct EncodeArgs {
   pub workers: usize,
   pub set_thread_affinity: Option<usize>,
   pub photon_noise: Option<u8>,
+  pub photon_noise_size: (Option<u32>, Option<u32>), // Width and Height
   pub chroma_noise: bool,
   pub zones: Option<PathBuf>,
 

av1an/src/main.rs

@@ -355,6 +355,14 @@ pub struct CliOpts {
   #[clap(long, help_heading = "Encoding")]
   pub photon_noise: Option<u8>,
 
+  /// Manually set the width for the photon noise table.
+  #[clap(long, help_heading = "Encoding")]
+  pub photon_noise_width: Option<u32>,
+
+  /// Manually set the height for the photon noise table.
+  #[clap(long, help_heading = "Encoding")]
+  pub photon_noise_height: Option<u32>,
+
   /// Adds chroma grain synthesis to the grain table generated by `--photon-noise`. (Default: false)
   #[clap(long, help_heading = "Encoding", requires = "photon_noise")]
   pub chroma_noise: bool,
@@ -667,6 +675,7 @@ pub fn parse_cli(args: CliOpts) -> anyhow::Result<Vec<EncodeArgs>> {
       photon_noise: args
         .photon_noise
         .and_then(|arg| if arg == 0 { None } else { Some(arg) }),
+      photon_noise_size: (args.photon_noise_width, args.photon_noise_height),
       chroma_noise: args.chroma_noise,
       sc_pix_format: args.sc_pix_format,
       keep: args.keep,