package test2;

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.LinkedList;

import javax.imageio.ImageIO;

public class SegCfg {

/**

* cfg 分割出来的结果

*/

private ArrayList<BufferedImage> cfgList;

public SegCfg(){

this.init();

try {

BufferedImage sourceImage = ImageIO.read(new File("img2/1_1.jpg"));

cfs(sourceImage);

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

}

private void init(){

cfgList = new ArrayList<BufferedImage>();

}

public void cfs(BufferedImage sourceImage){

int width = sourceImage.getWidth();

int height = sourceImage.getHeight();

ArrayList<SubImage> subImgList = new ArrayList<SubImage>(); //保存子图像

HashMap<String, Boolean> trackMap = new HashMap<String, Boolean>(); //已经访问过的点

for (int x = 0; x < width; x++) {

for (int y = 0; y < height; y++) {

int rgb = sourceImage.getRGB(x, y);

String key = x + "-" + y;

//如果不是黑色，或者已经被访问过，则跳过cfg

if (!isBlack(rgb) || trackMap.containsKey(key)) {

continue;

}

//如果黑色，且没有访问，则以此点开始进行连通域探索

SubImage subImage = new SubImage();//保存当前字符块的坐标点

LinkedList<Point> queue = new LinkedList<Point>();//保存当前字符块的访问队列

queue.offer(new Point(x, y, true));

trackMap.put(key, true);

subImage.pixelList.add(new Point(x, y, true));

subImage.left = x;

subImage.top = y;

subImage.right = x;

subImage.bottom = y;

while(queue.size() != 0){

Point tmp = queue.poll();

//搜寻目标的八个方向

int startX = (tmp.x - 1 < 0) ? 0 : tmp.x-1;

int startY = (tmp.y - 1 < 0) ? 0 : tmp.y-1;

int endX = (tmp.x + 1 > width) ? width : tmp.x+1;

int endY = (tmp.y + 1 > height) ? height : tmp.y+1;

for (int tx = startX; tx <= endX; tx++) {

for (int ty = startY; ty <= endY; ty++) {

// if (tx == tmp.x && ty == tmp.y) {

// continue;

// }

key = tx + "-" + ty;

if (isBlack(sourceImage.getRGB(tx, ty)) && !trackMap.containsKey(key)) {

queue.offer(new Point(tx, ty, true));

trackMap.put(key, true);

subImage.pixelList.add(new Point(tx, ty, true)); //加入到路径中

//更新边界区域

subImage.left = Math.min(subImage.left, tx);

subImage.top = Math.min(subImage.top, ty);

subImage.right = Math.max(subImage.right, tx);

subImage.bottom = Math.max(subImage.bottom, ty);

}

}

}

}//end of while

subImage.width = subImage.right - subImage.left + 1;

subImage.height = subImage.bottom - subImage.top + 1;

subImgList.add(subImage);

}

}

System.out.println();

cfsToImage(subImgList);

}

private void cfsToImage(ArrayList<SubImage> subImgList){

for (int i = 0; i < subImgList.size(); i++) {

SubImage subImage = subImgList.get(i);

BufferedImage image = new BufferedImage(subImage.width, subImage.height, BufferedImage.TYPE_BYTE_BINARY);

for (int x = 0; x < subImage.width; x++) {

for (int y = 0; y < subImage.height; y++) {

image.setRGB(x, y, new Color(255,255,255).getRGB());

}

}

ArrayList<Point> pixeList = subImage.pixelList;

for (Point point : pixeList) {

// System.out.println("(" + (point.x - subImage.left) + "," + (point.y - subImage.top) + ")");

image.setRGB(point.x - subImage.left, point.y - subImage.top, new Color(0, 0, 0).getRGB());

}

//将切割的中间图片加入到cfgList中

this.cfgList.add(image);

try {

ImageIO.write(image, "JPG", new File("img2/2_"+i+".jpg"));

} catch (IOException e) {

e.printStackTrace();

}

}

}

public boolean isBlack(int rgb) {

Color color = new Color(rgb);

if (color.getRed() + color.getGreen() + color.getBlue() <= 300) {

return true;

}

return false;

}

public static void main(String[] args){

SegCfg model = new SegCfg();

}

}