import java.util.ArrayList;
import java.util.IntSummaryStatistics;
import java.util.Collections;

// 画像用の変数
PImage[] img = new PImage[7];
// 出力ファイル名
String[] fName = {"ノイズ", "1移動平均1", "1移動平均4", "2_1ガウシアン4広", "2_2ガウシアン4狭", "3メディアン1", "3メディアン2"};
int w, h;

void setup() {
  size(800, 600);
  w = width/2;
  h = height/2;
  img[0] = loadImage("元.jpg");
  img[0].resize(w, h);
  for (int i=0; i<1000; i++) {
    int x=(int)(random(w));
    int y=(int)(random(h));
    img[0].pixels[x+w*y] = color(random(256), random(256), random(256));
  }
  // 1～6番をノイズ画像と同じ画像にする
  for (int i=1; i<=6; i++) {
    img[i] = img[0].get();
  }
  textFont(createFont("ＭＳ Ｐゴシック", 48));
  simpleMovingAverage(1, 1);// 移動平均フィルタ(半径1)でぼかした画像をimg[1]に保存
  simpleMovingAverage(4, 2);// 移動平均フィルタ(半径4)でぼかした画像をimg[2]に保存
  double s = Math.sqrt(2*Math.log(2)); // 隣の画素の重みが中心の画素1/2になるσの値
  gaussian(4, s*2, 3);      // ガウシアンフィルタ(半径4, σが2s)でぼかした画像をimg[3]に保存
  gaussian(4, s, 4);        // ガウシアンフィルタ(半径4, σがs)でぼかした画像をimg[4]に保存
  median(1, 5);             // メディアンフィルタ(半径1)でぼかした画像をimg[5]に保存
  median(2, 6);             // メディアンフィルタ(半径2)でぼかした画像をimg[6]に保存
  println("完了");
  image(img[0], 0, 0, width, height);
  img[0].save("data/"+ fName[0] +".jpg");
}

void draw() {
}

// 移動平均フィルタでぼかす(課題1)
void simpleMovingAverage(int radius, int n) {
  for (int j=0; j<h; j++) {
    for (int i=0; i<w; i++) {
      ArrayList<Integer> r = new ArrayList<Integer>();
      ArrayList<Integer> g = new ArrayList<Integer>();
      ArrayList<Integer> b = new ArrayList<Integer>();
      // i±radius, j±radiusの範囲の平均色を求める
      for (int y=-radius; y<=radius; y++) {
        for (int x=-radius; x<=radius; x++) {
          if (i+x>=0 && i+x<w && j+y>=0 && j+y<h) {
            color c = img[0].pixels[i+x+(j+y)*w]; // (i, j)から(x, y)ずれた位置の色
            // rにcの赤成分を追加する
            // gにcの緑成分を追加する
            // bにcの青成分を追加する
          }
        }
      }
      // ar, ag, abにr, g, bの平均値をfloat化して入れる
      float ar = 0;
      float ag = 0;
      float ab = 0;
      img[n].pixels[i+j*w] = color(ar, ag, ab);
    }
  }
  img[n].save("data/"+ fName[n] +".jpg");
}

// ガウシアンフィルタでぼかす(課題2)
void gaussian(int radius, double sigma, int n) {
}

// メディアンフィルタでぼかす(課題3)
void median(int radius, int n) {
}

void keyPressed() {
  int k = key-'0';
  if (k>=0 && k<=6) {
    background(0);
    image(img[k], 0, 0, width, height);
    fill(255, 0, 0);
    text(fName[k], 30, height-30);
    fill(255);
  }
}