package train;

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.FilenameFilter;

import java.io.IOException;

import java.nio.file.Files;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.LinkedList;

import javax.imageio.ImageIO;

import model.Point;

import model.SubImage;

public class SegCfg {

/**

* cfg 分割出来的结果

*/

private ArrayList<BufferedImage> cfgList;

public SegCfg(){

this.init();

}

private static void test(){

try {

SegCfg model = new SegCfg();

File file = new File("1_gray/10.jpg");

BufferedImage img = ImageIO.read(file);

ArrayList<BufferedImage> list = model.cfs(img);

for(int j=0; j<list.size(); j++){

BufferedImage subImg = list.get(j);

String prex = file.getName().split("\\.")[0];

String filename = "2_cfs/" + prex + "-" + j + ".jpg";

ImageIO.write(subImg, "JPG", new File(filename));

}

} catch (Exception e) {

e.printStackTrace();

}

}

/**

* 读取灰度图片，进行切割

*/

private static void run(){

File dir = new File("1_gray/");

//只列出jpg

File[] files = dir.listFiles(new FilenameFilter() {

public boolean isJpg(String file){

if (file.toLowerCase().endsWith(".jpg")){

return true;

}else{

return false;

}

}

@Override

public boolean accept(File dir, String name) {

// TODO Auto-generated method stub

return isJpg(name);

}

});

SegCfg model = new SegCfg();

for (int i=0; i<files.length; i++) {

try {

File file = files[i];

BufferedImage img = ImageIO.read(file);

ArrayList<BufferedImage> list = model.cfs(img);

for(int j=0; j<list.size(); j++){

BufferedImage subImg = list.get(j);

String prex = file.getName().split("\\.")[0];

String filename = "2_cfs/" + prex + "-" + j + ".jpg";

ImageIO.write(subImg, "JPG", new File(filename));

}

} catch (IOException e) {

e.printStackTrace();

}

}

}

private void init(){

cfgList = new ArrayList<BufferedImage>();

}

/**

* cfs进行分割,返回分割后的数组

* @param sourceImage

* @return

*/

public ArrayList<BufferedImage> cfs(BufferedImage sourceImage){

this.cfgList.clear();

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 - 1) ? width - 1 : tmp.x + 1;

int endY = (tmp.y + 1 > height - 1) ? height - 1 : 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;

System.out.println(key);

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);

return this.cfgList;

}

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);

}

}

private 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){

run();

}

}