Java圖像拼接處理

/**
	 * 圖片拼接
	 * @param files  要拼接的文件列表
	 * @param type   1 橫向拼接， 2 縱向拼接
	 * @return
	 */
	public static InputStream merge(String[] files, int type) {

		int len = files.length;
		if (len < 1) {
			log.info("圖片數量小於1");
			return null;
		}
		
		File[] src = new File[len];
		BufferedImage[] images = new BufferedImage[len];
		int[][] ImageArrays = new int[len][];
		for (int i = 0; i < len; i++) {
			try {
				src[i] = new File(files[i]);
				images[i] = ImageIO.read(src[i]);
			} catch (Exception e) {
				e.printStackTrace();
				return null;
			}
			int width = images[i].getWidth();
			int height = images[i].getHeight();
			ImageArrays[i] = new int[width * height];// 從圖片中讀取RGB
			ImageArrays[i] = images[i].getRGB(0, 0, width, height, ImageArrays[i], 0, width);
		}

		int newHeight = 0;
		int newWidth = 0;
		for (int i = 0; i < images.length; i++) {
			//橫向
			if(type == 1){
				newHeight = newHeight > images[i].getHeight() ? newHeight : images[i].getHeight();
				newWidth += images[i].getWidth();
			}else if(type == 2){//縱向
				newWidth = newWidth > images[i].getWidth() ? newWidth : images[i].getWidth();
				newHeight += images[i].getHeight();
			}
		}
		
		log.info("拼接後圖像寬度："+newWidth);
		log.info("拼接後圖像高度："+newHeight);
		if (type == 1 && newWidth < 1) {
			log.info("拼接後圖像寬度小於1");
			return null;
		}
		if (type == 2 && newHeight < 1) {
			log.info("拼接後圖像高度小於1");
			return null;
		}

		//生成新圖片
		try {
			BufferedImage ImageNew = new BufferedImage(newWidth, newHeight, BufferedImage.TYPE_INT_RGB);
			int height_i = 0;
			int width_i = 0;
			for (int i = 0; i < images.length; i++) {
				if(type == 1){
					ImageNew.setRGB(width_i, 0, images[i].getWidth(), newHeight, ImageArrays[i], 0, images[i].getWidth());
					width_i += images[i].getWidth();
				}else if(type == 2){
					ImageNew.setRGB(0, height_i, newWidth, images[i].getHeight(), ImageArrays[i], 0, newWidth);
					height_i += images[i].getHeight();
				}
			}
//			File outFile = new File("d://123.jpg");
//			ImageIO.write(ImageNew, "jpg", outFile);// 寫圖片
			
			ByteArrayOutputStream out = new ByteArrayOutputStream();
			ImageIO.write(ImageNew, "jpg", out);// 圖片寫入到輸出流中
			
			ByteArrayInputStream in = new ByteArrayInputStream(out.toByteArray());
			return in;
			
		} catch (Exception e) {
			e.printStackTrace();
			return null;
		}
	}