noise function

Image noise(Image image, num sigma, { NoiseType type = NoiseType.gaussian, Random? random, })

Add random noise to pixel values. sigma determines how strong the effect should be. type should be one of the following: NoiseType.gaussian, NoiseType.uniform, NoiseType.salt_pepper, NoiseType.poisson, or NoiseType.rice.

Implementation

Image noise(Image image, num sigma,
    {NoiseType type = NoiseType.gaussian, Random? random}) {
  random ??= Random();

  var nsigma = sigma;
  var m = 0;
  var M = 0;

  if (nsigma == 0.0 && type != NoiseType.poisson) {
    return image;
  }

  if (nsigma < 0.0 || type == NoiseType.salt_pepper) {
    final mM = minMax(image);
    m = mM[0];
    M = mM[1];
  }

  if (nsigma < 0.0) {
    nsigma = -nsigma * (M - m) / 100.0;
  }

  final len = image.length;
  switch (type) {
    case NoiseType.gaussian:
      for (var i = 0; i < len; ++i) {
        final c = image[i];
        final r = (getRed(c) + nsigma * grand(random)).toInt();
        final g = (getGreen(c) + nsigma * grand(random)).toInt();
        final b = (getBlue(c) + nsigma * grand(random)).toInt();
        final a = getAlpha(c);
        image[i] = getColor(r, g, b, a);
      }
      break;
    case NoiseType.uniform:
      for (var i = 0; i < len; ++i) {
        final c = image[i];
        final r = (getRed(c) + nsigma * crand(random)).toInt();
        final g = (getGreen(c) + nsigma * crand(random)).toInt();
        final b = (getBlue(c) + nsigma * crand(random)).toInt();
        final a = getAlpha(c);
        image[i] = getColor(r, g, b, a);
      }
      break;
    case NoiseType.salt_pepper:
      if (nsigma < 0) {
        nsigma = -nsigma;
      }
      if (M == m) {
        m = 0;
        M = 255;
      }
      for (var i = 0; i < len; ++i) {
        final c = image[i];
        if (random.nextDouble() * 100.0 < nsigma) {
          final r = (random.nextDouble() < 0.5 ? M : m);
          final g = (random.nextDouble() < 0.5 ? M : m);
          final b = (random.nextDouble() < 0.5 ? M : m);
          final a = getAlpha(c);
          image[i] = getColor(r, g, b, a);
        }
      }
      break;
    case NoiseType.poisson:
      for (var i = 0; i < len; ++i) {
        final c = image[i];
        final r = prand(random, getRed(c).toDouble());
        final g = prand(random, getGreen(c).toDouble());
        final b = prand(random, getBlue(c).toDouble());
        final a = getAlpha(c);
        image[i] = getColor(r, g, b, a);
      }
      break;
    case NoiseType.rice:
      final num sqrt2 = sqrt(2.0);
      for (var i = 0; i < len; ++i) {
        final c = image[i];

        var val0 = getRed(c) / sqrt2;
        var re = (val0 + nsigma * grand(random));
        var im = (val0 + nsigma * grand(random));
        var val = sqrt(re * re + im * im);
        final r = val.toInt();

        val0 = getGreen(c) / sqrt2;
        re = (val0 + nsigma * grand(random));
        im = (val0 + nsigma * grand(random));
        val = sqrt(re * re + im * im);
        final g = val.toInt();

        val0 = getBlue(c) / sqrt2;
        re = (val0 + nsigma * grand(random));
        im = (val0 + nsigma * grand(random));
        val = sqrt(re * re + im * im);
        final b = val.toInt();

        final a = getAlpha(c);
        image[i] = getColor(r, g, b, a);
      }
      break;
  }

  return image;
}