package PCI.ASQ.image;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.Map;

import PCI.ASQ.classification.FissileWithMatch;
import PCI.ASQ.demension.AMV_MVS_VSQ_2D;
import PCI.ASQ.demension.AMV_MVS_VSQ_3D;
//作者,著作权人： 罗瑶光，浏阳
public class ImagePixGroupFilter{
  //思想：罗瑶光
  //作者：罗瑶光
  //将shareholder分层处理的图片像素数据 按指定的精度和对应像素进行按像素团的大小进行特定过滤。
  public static int[][] getImagePix2DGroupFilter(int[][] inputPixMatrix
      , int RBG, int distanceScale,int max, int min){
    //像素分类
    Map<Integer, ArrayList<AMV_MVS_VSQ_2D>> map= ImagePixClassification
        .getImagePixClassificationMap(inputPixMatrix);
    //获取分类后的团簇
    Map<Double, ArrayList<AMV_MVS_VSQ_2D>> mid= FissileWithMatch
        .fissilePosition2DWithMatch(map.get(RBG), distanceScale);
    //筛选团簇，过滤团簇
    Iterator<Double> iteratorMid= mid.keySet().iterator();
    while(iteratorMid.hasNext()) {
      ArrayList<AMV_MVS_VSQ_2D> list= mid.get(iteratorMid.next());
      if(null!= list) {
        if(list.size()< min|| list.size()> max) {
          Iterator< AMV_MVS_VSQ_2D> iterator= list.iterator();
          while(iterator.hasNext()) {
            AMV_MVS_VSQ_2D position2D= iterator.next();
            inputPixMatrix[(int) position2D.getX()]
                [(int) position2D.getY()]= 0;
          }
        }
      }
    }
    return inputPixMatrix;
  }

  public static int[][][] getImagePix3DGroupFilter(int[][][] inputPixMatrix
      , int RBG, int distanceScale,int max, int min){
    //像素分类
    Map<Integer, ArrayList<AMV_MVS_VSQ_3D>> map= ImagePixClassification
        .getImagePixClassificationMap(inputPixMatrix);
    //获取分类后的团簇
    Map<Double, ArrayList<AMV_MVS_VSQ_3D>> mid= FissileWithMatch
        .fissilePosition3DWithMatch(map.get(RBG), distanceScale);
    //筛选团簇，过滤团簇
    Iterator<Double> iteratorMid= mid.keySet().iterator();
    while(iteratorMid.hasNext()) {
      ArrayList<AMV_MVS_VSQ_3D> list= mid.get(iteratorMid.next());
      if(null!= list) {
        if(list.size()< min|| list.size()> max) {
          Iterator< AMV_MVS_VSQ_3D> iterator= list.iterator();
          while(iterator.hasNext()) {
            AMV_MVS_VSQ_3D position3D= iterator.next();
            inputPixMatrix[(int) position3D.getX()][(int) position3D.getY()]
                [(int) position3D.getZ()]= 0;
          }
        }
      }
    }
    return inputPixMatrix;
  }
}