package test2;

import java.awt.Color;

import java.awt.image.BufferedImage;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import javax.imageio.ImageIO;

public class SegWaterDrop {

private int minD = 8;//最小字符宽度

private int maxD = 16;

private int meanD = 11;//平均字符宽度

private BufferedImage sourceImage;

public SegWaterDrop() {

try {

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

drop(sourceImage);

} catch (IOException e) {

// TODO Auto-generated catch block

e.printStackTrace();

}

}

private void drop(BufferedImage sourceImage){

int width = sourceImage.getWidth();

int height = sourceImage.getHeight();

//在x轴的投影

int[] histData = new int[width];

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

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

if (isBlack(sourceImage.getRGB(x, y))) {

histData[x]++;

}

}

}

ArrayList<Integer> extrems = Extremum.getMinExtrem(histData);

Point[] startRoute = new Point[height];

Point[] endRoute = null;

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

startRoute[y] = new Point(0, y);

}

int num = (int)Math.round((double)width/meanD);//字符的个数

int lastP = 0; //上一次分割的位置

int curSplit = 1;//分割点的个数，小于等于 num - 1;

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

if (curSplit > num - 1) {

break;

}

//判断两个分割点之间的距离是否合法

int curP = extrems.get(i);

int dBetween = curP - lastP + 1;

if (dBetween - lastP < minD || dBetween > maxD) {

continue;

}

//判断当前分割点与末尾结束点的位置是否合法

int dAll = width - curP + 1;

if (dAll < minD*(num - curSplit) || dAll > maxD*(num - curSplit)) {

continue;

}

endRoute = getEndRoute(new Point(curP, 0), height);

doSplit(startRoute, endRoute, "4_"+curSplit+".jpg");

startRoute = endRoute;

curSplit ++;

}

endRoute = new Point[height];

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

endRoute[y] = new Point(width - 1, y);

}

doSplit(startRoute, endRoute, "4_"+curSplit+".jpg");

System.out.println("=================");

System.out.println(width+","+height);

}

private Point[] getEndRoute(Point startP, int height){

//获得分割的路径

Point[] endRoute = new Point[height];

Point curP = new Point(startP.x, startP.y);

Point lastP = curP;

endRoute[0] = curP;

while(curP.y < height - 1){

int maxW = 0;

int sum = 0;

int nextX = curP.x;

int nextY = curP.y;

int[][] orderArr = { {5, 0, 4},

{1, 2, 3}};

for(int y = curP.y, i = 0; y <= curP.y + 1 && i < 2; y++, i++){

for(int x = curP.x - 1, j = 0; x <= curP.x + 1 && j < 3; x++, j++){

if (orderArr[i][j] == 0) {

continue;

}

int curW = getPixelValue(sourceImage.getRGB(x, y)) * (6 - orderArr[i][j]);

sum += curW;

if (curW > maxW) {

maxW = curW;

nextX = x;

nextY = y;

}

}

}

if (lastP.x == nextX && lastP.y == nextY) {//如果出现重复运动

lastP = curP;

if (nextX < curP.x) {//向左重复

curP = new Point(curP.x - 1, curP.y + 1);

}else{//向右重复

curP = new Point(curP.x, curP.y + 1);

}

}else{

lastP = curP;

if (sum == 0 || sum == 15) {

curP = new Point(curP.x, curP.y + 1);

}else{

curP = new Point(nextX, nextY);

}

}

endRoute[curP.y] = curP;

}

return endRoute;

}

private void doSplit(Point[] starts, Point[] ends, String filename){

int left = starts[0].x;

int top = starts[0].y;

int right = ends[0].x;

int bottom = ends[0].y;

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

left = Math.min(starts[i].x, left);

top = Math.min(starts[i].y, top);

right = Math.max(ends[i].x, right);

bottom = Math.max(ends[i].y, bottom);

}

int width = right - left + 1;

int height = bottom - top + 1;

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

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

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

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

}

}

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

Point start = starts[i];

Point end = ends[i];

for (int x = start.x; x < end.x; x++) {

if (isBlack(sourceImage.getRGB(x, i))) {

System.out.println((x - left) + ", " + (start.y - top));

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

}

}

}

System.out.println("-----------------------");

try {

ImageIO.write(image, "JPG", new File("img2/" + filename));

// ImageIO.write(sourceImage.getSubimage(left, top, width, height), "JPG", new File("img2/" + filename));

} catch (Exception 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 int getPixelValue(int rgb){

if (isBlack(rgb)) {

return 0;

}else{

return 1;

}

}

public static void main(String[] args){

SegWaterDrop model = new SegWaterDrop();

}

}