Add support for new augmentations in rocAL - Group 4

rocAL/include/api/rocal_api_augmentation.h

@@ -599,6 +599,78 @@ extern "C" RocalTensor ROCAL_API_CALL rocalSnPNoiseFixed(RocalContext context, R
                                                          RocalTensorLayout output_layout = ROCAL_NONE,
                                                          RocalTensorOutputType output_datatype = ROCAL_UINT8);
 
+/*! \brief Applies gaussian noise effect on images.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] mean specifies the mean value for the Gaussian noise distribution
+ * \param [in] stddev specifies the standard deviation for the Gaussian noise distribution
+ * \param [in] seed seed value for the random number generator
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalGaussianNoise(RocalContext context, RocalTensor input,
+                                                         bool is_output,
+                                                         RocalFloatParam mean = NULL, RocalFloatParam stddev = NULL,
+                                                         int seed = 0,
+                                                         RocalTensorLayout output_layout = ROCAL_NONE,
+                                                         RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies gaussian noise on images with fixed parameters.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] mean specifies the mean value for the Gaussian noise distribution
+ * \param [in] stddev specifies the standard deviation for the Gaussian noise distribution
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] seed seed value for the random number generator
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalGaussianNoiseFixed(RocalContext context, RocalTensor input,
+                                                              bool is_output,
+                                                              float mean = 0.0, float stddev = 1.0, int seed = 0,
+                                                              RocalTensorLayout output_layout = ROCAL_NONE,
+                                                              RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies shot noise effect on images.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] noise_factor specifies the noise intensity factor for the shot (Poisson) noise
+ * \param [in] seed seed value for the random number generator
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalShotNoise(RocalContext context, RocalTensor input,
+                                                     bool is_output,
+                                                     RocalFloatParam noise_factor = NULL,
+                                                     int seed = 0,
+                                                     RocalTensorLayout output_layout = ROCAL_NONE,
+                                                     RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies shot noise on images with fixed parameters.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] noise_factor specifies the noise intensity factor for the shot (Poisson) noise
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] seed seed value for the random number generator
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalShotNoiseFixed(RocalContext context, RocalTensor input,
+                                                          float noise_factor,
+                                                          bool is_output, int seed = 0,
+                                                          RocalTensorLayout output_layout = ROCAL_NONE,
+                                                          RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
 /*! \brief Applies snow effect on images.
  * \ingroup group_rocal_augmentations
  * \param [in] context Rocal context
@@ -686,6 +758,42 @@ extern "C" RocalTensor ROCAL_API_CALL rocalRainFixed(RocalContext context, Rocal
                                                      RocalTensorLayout output_layout = ROCAL_NONE,
                                                      RocalTensorOutputType output_datatype = ROCAL_UINT8);
 
+/*! \brief Applies spatter effect on images.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] red red channel value of the spatter color
+ * \param [in] green green channel value of the spatter color
+ * \param [in] blue blue channel value of the spatter color
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalSpatter(RocalContext context, RocalTensor input,
+                                                   bool is_output,
+                                                   RocalIntParam red = NULL, RocalIntParam green = NULL, RocalIntParam blue = NULL,
+                                                   RocalTensorLayout output_layout = ROCAL_NONE,
+                                                   RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies spatter effect on images with fixed parameters.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] red red channel value of the spatter color
+ * \param [in] green green channel value of the spatter color
+ * \param [in] blue blue channel value of the spatter color
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalSpatterFixed(RocalContext context, RocalTensor input,
+                                                        uint8_t red, uint8_t green, uint8_t blue,
+                                                        bool is_output,
+                                                        RocalTensorLayout output_layout = ROCAL_NONE,
+                                                        RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
 /*! \brief Adjusts the color temperature in images.
  * \ingroup group_rocal_augmentations
  * \param [in] context Rocal context
@@ -717,6 +825,47 @@ extern "C" RocalTensor ROCAL_API_CALL rocalColorTempFixed(RocalContext context,
                                                           RocalTensorLayout output_layout = ROCAL_NONE,
                                                           RocalTensorOutputType output_datatype = ROCAL_UINT8);
 
+/*! \brief Applies color jitter augmentation on images.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] brightness brightness adjustment factor
+ * \param [in] contrast contrast adjustment factor
+ * \param [in] hue hue adjustment value in degrees
+ * \param [in] saturation saturation adjustment factor
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalColorJitter(RocalContext context, RocalTensor input,
+                                                       bool is_output,
+                                                       RocalFloatParam brightness = NULL,
+                                                       RocalFloatParam contrast = NULL,
+                                                       RocalFloatParam hue = NULL,
+                                                       RocalFloatParam saturation = NULL,
+                                                       RocalTensorLayout output_layout = ROCAL_NONE,
+                                                       RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies color jitter augmentation on images with fixed parameters.
+ * \ingroup group_rocal_augmentations
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] brightness brightness adjustment factor
+ * \param [in] contrast contrast adjustment factor
+ * \param [in] hue hue adjustment value in degrees
+ * \param [in] saturation saturation adjustment factor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalColorJitterFixed(RocalContext context, RocalTensor input,
+                                                             float brightness, float contrast, float hue, float saturation,
+                                                             bool is_output,
+                                                             RocalTensorLayout output_layout = ROCAL_NONE,
+                                                             RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
 /*! \brief Applies fog effect on images.
  * \ingroup group_rocal_augmentations
  * \param [in] context Rocal context
@@ -1748,4 +1897,65 @@ extern "C" RocalTensor ROCAL_API_CALL rocalCropAndPatch(RocalContext context,
                                                         RocalTensorLayout output_layout = ROCAL_NONE,
                                                         RocalTensorOutputType output_datatype = ROCAL_UINT8);
 
+/*! \brief Computes the natural logarithm of input element-wise and returns the output
+ * \param [in] p_context Rocal context
+ * \param [in] p_input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalLog(RocalContext p_context,
+                                                RocalTensor p_input,
+                                                bool is_output);
+
+/*! \brief Applies water effect on images.
+ * \ingroup group_rocal_augmentations
+ * \note Accepts U8 and RGB24 input.
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] amplitude_x amplitude of the water effect in the x direction
+ * \param [in] amplitude_y amplitude of the water effect in the y direction
+ * \param [in] frequency_x frequency of the water effect in the x direction
+ * \param [in] frequency_y frequency of the water effect in the y direction
+ * \param [in] phase_x phase of the water effect in the x direction
+ * \param [in] phase_y phase of the water effect in the y direction
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalWater(RocalContext context, RocalTensor input,
+                                                 bool is_output,
+                                                 RocalFloatParam amplitude_x = NULL,
+                                                 RocalFloatParam amplitude_y = NULL,
+                                                 RocalFloatParam frequency_x = NULL,
+                                                 RocalFloatParam frequency_y = NULL,
+                                                 RocalFloatParam phase_x = NULL,
+                                                 RocalFloatParam phase_y = NULL,
+                                                 RocalTensorLayout output_layout = ROCAL_NONE,
+                                                 RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
+/*! \brief Applies water effect on images with fixed parameters.
+ * \ingroup group_rocal_augmentations
+ * \note Accepts U8 and RGB24 input.
+ * \param [in] context Rocal context
+ * \param [in] input Input Rocal tensor
+ * \param [in] amplitude_x amplitude of the water effect in the x direction
+ * \param [in] amplitude_y amplitude of the water effect in the y direction
+ * \param [in] frequency_x frequency of the water effect in the x direction
+ * \param [in] frequency_y frequency of the water effect in the y direction
+ * \param [in] phase_x phase of the water effect in the x direction
+ * \param [in] phase_y phase of the water effect in the y direction
+ * \param [in] is_output is the output tensor part of the graph output
+ * \param [in] output_layout the layout of the output tensor
+ * \param [in] output_datatype the data type of the output tensor
+ * \return RocalTensor
+ */
+extern "C" RocalTensor ROCAL_API_CALL rocalWaterFixed(RocalContext context, RocalTensor input,
+                                                      float amplitude_x, float amplitude_y,
+                                                      float frequency_x, float frequency_y,
+                                                      float phase_x, float phase_y,
+                                                      bool is_output,
+                                                      RocalTensorLayout output_layout = ROCAL_NONE,
+                                                      RocalTensorOutputType output_datatype = ROCAL_UINT8);
+
 #endif  // MIVISIONX_ROCAL_API_AUGMENTATION_H

rocAL/include/augmentations/arithmetic_augmentations/node_log.h

@@ -0,0 +1,35 @@
+/*
+Copyright (c) 2026 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+#pragma once
+
+#include "pipeline/node.h"
+
+class LogNode : public Node {
+   public:
+    LogNode(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs);
+    LogNode() = delete;
+
+   protected:
+    void create_node() override;
+    void update_node() override {};
+};

rocAL/include/augmentations/augmentations_nodes.h

@@ -27,8 +27,12 @@ THE SOFTWARE.
 #include "augmentations/color_augmentations/node_vignette.h"
 #include "augmentations/effects_augmentations/node_jitter.h"
 #include "augmentations/effects_augmentations/node_snp_noise.h"
+#include "augmentations/effects_augmentations/node_gaussian_noise.h"
+#include "augmentations/effects_augmentations/node_shot_noise.h"
 #include "augmentations/effects_augmentations/node_snow.h"
 #include "augmentations/effects_augmentations/node_rain.h"
+#include "augmentations/effects_augmentations/node_spatter.h"
+#include "augmentations/effects_augmentations/node_water.h"
 #include "augmentations/color_augmentations/node_color_temperature.h"
 #include "augmentations/effects_augmentations/node_fog.h"
 #include "augmentations/effects_augmentations/node_pixelate.h"
@@ -44,6 +48,7 @@ THE SOFTWARE.
 #include "augmentations/geometry_augmentations/node_resize.h"
 #include "augmentations/geometry_augmentations/node_rotate.h"
 #include "augmentations/color_augmentations/node_color_twist.h"
+#include "augmentations/color_augmentations/node_color_jitter.h"
 #include "augmentations/color_augmentations/node_hue.h"
 #include "augmentations/color_augmentations/node_saturation.h"
 #include "augmentations/geometry_augmentations/node_crop_mirror_normalize.h"
@@ -86,3 +91,4 @@ THE SOFTWARE.
 #include "augmentations/geometry_augmentations/node_crop_and_patch.h"
 #include "augmentations/geometry_augmentations/node_remap.h"
 #include "augmentations/geometry_augmentations/node_ricap.h"
+#include "augmentations/arithmetic_augmentations/node_log.h"

rocAL/include/augmentations/color_augmentations/node_color_jitter.h

@@ -0,0 +1,50 @@
+/*
+Copyright (c) 2026 Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+#pragma once
+
+#include "pipeline/graph.h"
+#include "pipeline/node.h"
+#include "parameters/parameter_factory.h"
+#include "parameters/parameter_vx.h"
+
+class ColorJitterNode : public Node {
+   public:
+    ColorJitterNode(const std::vector<Tensor *> &inputs, const std::vector<Tensor *> &outputs);
+    ColorJitterNode() = delete;
+    void init(float brightness, float contrast, float hue, float saturation);
+    void init(FloatParam *brightness_param, FloatParam *contrast_param, FloatParam *hue_param, FloatParam *saturation_param);
+
+   protected:
+    void create_node() override;
+    void update_node() override;
+
+   private:
+    ParameterVX<float> _brightness;
+    ParameterVX<float> _contrast;
+    ParameterVX<float> _hue;
+    ParameterVX<float> _saturation;
+    constexpr static float BRIGHTNESS_RANGE[2] = {0.0f, 20.0f};
+    constexpr static float CONTRAST_RANGE[2] = {0.0f, 255.0f};
+    constexpr static float HUE_RANGE[2] = {0.0f, 359.0f};
+    constexpr static float SATURATION_RANGE[2] = {0.0f, 10.0f};
+};