added offset to increment node

2024-07-28 23:04:28 -07:00
parent dbecf07c5c
commit 59c0fbfc57
2 changed files with 129 additions and 9 deletions
@@ -15,29 +15,148 @@ This package contains a collection of custom nodes for ComfyUI, designed to enha
 8. [ResizeByImage](#resizebyimage)
 9. [IncrementEveryN](#incrementeveryn)

-...
+## TextFileLineIterator

-[Previous content remains unchanged]
+This node iterates through lines in a text file.

-...
+### Parameters:
+- `file_path` (STRING): Path to the text file.
+- `line_index` (INT, default: 0): Index of the line to retrieve.
+
+### Output:
+- (STRING): The selected line from the text file.
+
+## ImageDirIterator
+
+This node iterates through images in a directory or multiple directories, with various sorting and selection options.
+
+### Parameters:
+- `directory_path` (STRING): Path to the directory containing images.
+- `glob_patterns` (STRING, default: "**/*.png, **/*.jpg"): Comma-separated list of glob patterns for selecting files.
+- `image_index` (INT, default: 0): Starting index for image selection.
+- `sort_by` (["date_modified", "name", "size", "random"]): Method to sort the images.
+- `sort_order` (["ascending", "descending"]): Order of sorting.
+- `batch_size` (INT, default: 1, min: 1, max: 64): Number of images to select at once.
+- `increment_by_batch` (BOOLEAN, default: False): Whether to increment the index by batch size.
+- `randomize_final_list` (BOOLEAN, default: False): Whether to randomize the final list of images.
+
+### Output:
+- (IMAGE): List of selected images.
+- (STRING): List of filenames for the selected images.
+
+## VidDirIterator
+
+This node iterates through video files in a directory or multiple directories, with various sorting and selection options.
+
+### Parameters:
+- `directory_path` (STRING): Path to the directory containing video files.
+- `glob_patterns` (STRING, default: "**/*.mp4, **/*.mov"): Comma-separated list of glob patterns for selecting files.
+- `video_index` (INT, default: 0): Index of the video to select.
+- `sort_by` (["date_modified", "name", "size", "random"]): Method to sort the videos.
+- `sort_order` (["ascending", "descending"]): Order of sorting.
+- `randomize_final_list` (BOOLEAN, default: False): Whether to randomize the final list of videos. This will allow full randomization across all specified folders and files. Otherwise, it will randomize within each folder before moving on to the next.
+
+### Output:
+- (STRING): Path to the selected video file.
+
+## Modelscopet2v
+
+This node generates video frames from text using the Modelscope text-to-video model.
+
+### Parameters:
+- `prompt` (STRING): Text prompt for video generation.
+- `negative_prompt` (STRING, default: None): Negative prompt for generation.
+- `model_path` (STRING, default: "cerspense/zeroscope_v2_576w"): Path to the model.
+- `num_inference_steps` (INT, default: 25): Number of inference steps.
+- `guidance_scale` (FLOAT, default: 9.0): Guidance scale for generation.
+- `seed` (INT, default: 42): Seed for random generation.
+- `width` (INT, default: 576): Width of the generated video.
+- `height` (INT, default: 320): Height of the generated video.
+- `num_frames` (INT, default: 24): Number of frames to generate.
+
+### Output:
+- (IMAGE): Generated video frames.
+
+## Modelscopev2v
+
+This node transforms video frames using the Modelscope video-to-video model.
+
+### Parameters:
+- `video_frames` (IMAGE): Input video frames.
+- `prompt` (STRING): Text prompt for video transformation.
+- `negative_prompt` (STRING, default: None): Negative prompt for transformation.
+- `model_path` (STRING, default: "cerspense/zeroscope_v2_XL"): Path to the model.
+- `strength` (FLOAT, default: 0.70): Strength of the transformation.
+- `num_inference_steps` (INT, default: 25): Number of inference steps.
+- `guidance_scale` (FLOAT, default: 8.50): Guidance scale for transformation.
+- `seed` (INT, default: 42): Seed for random generation.
+- `enable_forward_chunking` (BOOLEAN, default: False): Enable forward chunking for memory optimization.
+- `enable_vae_slicing` (BOOLEAN, default: True): Enable VAE slicing for memory optimization.
+
+### Output:
+- (IMAGE): Transformed video frames.
+
+## SplitImageChannels
+
+This node splits an image into its red, green, and blue channels.
+
+### Parameters:
+- `image` (IMAGE): Input image to split.
+
+### Output:
+- (IMAGE): Red channel as a grayscale image.
+- (IMAGE): Green channel as a grayscale image.
+- (IMAGE): Blue channel as a grayscale image.
+
+## RemapRange
+
+This node remaps a value from one range to another.
+
+### Parameters:
+- `value` (FLOAT, default: 0.0, range: -10000.0 to 10000.0): Input value to remap.
+- `input_min` (FLOAT, default: 0.0): Minimum of the input range.
+- `input_max` (FLOAT, default: 1.0): Maximum of the input range.
+- `output_min` (FLOAT, default: 0.0): Minimum of the output range.
+- `output_max` (FLOAT, default: 1.0): Maximum of the output range.
+- `clamp` (BOOLEAN, default: False): Whether to clamp the output to the output range.
+
+### Output:
+- (FLOAT): Remapped value.
+
+## ResizeByImage
+
+This node resizes an input image to match the dimensions of a reference image.
+
+### Parameters:
+- `reference_image` (IMAGE): Image to use as a reference for resizing.
+- `input_image` (IMAGE, optional): Image to resize. If not provided, a black image is created.
+
+### Output:
+- (IMAGE): Resized image matching the dimensions of the reference image.

 ## IncrementEveryN

-This node takes an input integer and outputs a new integer that increments every N steps of the input.
+This node takes an input integer and outputs a new integer that increments every N steps of the input, with an optional offset.

 ### Parameters:
 - `input_value` (INT, default: 0, min: 0): The input integer value that is incrementing.
 - `step_size` (INT, default: 1, min: 1): The number of steps in the input required to increment the output by 1.
+- `offset` (INT, default: 0): An integer value added to the final output to offset the result.

 ### Output:
 - (INT): The incrementing output value.

 ### Behavior:
-This node divides the input value by the step size and returns the integer result. For example:
- If `step_size` is 6:
+This node divides the input value by the step size, adds the offset, and returns the integer result. For example:
+- If `step_size` is 6 and `offset` is 0:
  - Input values 0-5 will output 0
  - Input values 6-11 will output 1
  - Input values 12-17 will output 2
  - And so on...
+- If `step_size` is 6 and `offset` is 10:
+  - Input values 0-5 will output 10
+  - Input values 6-11 will output 11
+  - Input values 12-17 will output 12
+  - And so on...

-This node is useful for creating slower-changing values from rapidly incrementing inputs, which can be helpful in various animation and procedural generation scenarios.
+This node is useful for creating slower-changing values from rapidly incrementing inputs, which can be helpful in various animation and procedural generation scenarios. The offset parameter allows for further customization of the output range.
@@ -407,6 +407,7 @@ class IncrementEveryN:
            "required": {
                "input_value": ("INT", {"default": 0, "min": 0, "step": 1}),
                "step_size": ("INT", {"default": 1, "min": 1, "step": 1}),
+                "offset": ("INT", {"default": 0, "step": 1}),
            }
        }

@@ -414,8 +415,8 @@ class IncrementEveryN:
    FUNCTION = "increment_every_n"
    CATEGORY = "cspnodes"

-    def increment_every_n(self, input_value, step_size):
-        output_value = input_value // step_size
+    def increment_every_n(self, input_value, step_size, offset):
+        output_value = (input_value // step_size) + offset
        return (output_value,)