package com.qianwen.smartman.modules.workinghour.service;

import java.awt.Color;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.text.NumberFormat;
import java.time.Duration;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.ss.usermodel.ClientAnchor.AnchorType;
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
import org.apache.poi.xssf.usermodel.XSSFDrawing;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartUtils;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.StandardChartTheme;
import org.jfree.chart.block.BlockBorder;
import org.jfree.chart.labels.AbstractCategoryItemLabelGenerator;
import org.jfree.chart.labels.CategoryItemLabelGenerator;
import org.jfree.chart.labels.ItemLabelAnchor;
import org.jfree.chart.labels.ItemLabelPosition;
import org.jfree.chart.plot.CategoryPlot;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.chart.renderer.category.StackedBarRenderer;
import org.jfree.chart.ui.RectangleEdge;
import org.jfree.chart.ui.RectangleInsets;
import org.jfree.chart.ui.TextAnchor;
import org.jfree.data.category.CategoryDataset;
import org.jfree.data.category.DefaultCategoryDataset;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.qianwen.core.excel.util.ExcelUtil;
import com.qianwen.core.mp.base.BaseServiceImpl;
import com.qianwen.core.oss.model.BladeFile;
import com.qianwen.core.tool.utils.DateUtil;
import com.qianwen.smartman.common.enums.DefaultWcsEnum;
import com.qianwen.smartman.common.utils.DurationUtil;
import com.qianwen.smartman.modules.resource.builder.oss.OssBuilder;
import com.qianwen.smartman.modules.smis.entity.Workstation;
import com.qianwen.smartman.modules.smis.mapper.WorkstationMapper;
import com.qianwen.smartman.modules.workinghour.entity.PartWorkingHour;
import com.qianwen.smartman.modules.workinghour.entity.PartWorkingProcess;
import com.qianwen.smartman.modules.workinghour.excel.PartWorkingHourExcel;
import com.qianwen.smartman.modules.workinghour.mapper.PartWorkingHourMapper;
import com.qianwen.smartman.modules.workinghour.mapper.PartWorkingProcessMapper;

import cn.hutool.core.date.LocalDateTimeUtil;

@Service
public class PartWorkingHourExportService extends BaseServiceImpl<PartWorkingHourMapper, PartWorkingHour> {

	@Autowired
	private PartWorkingHourMapper partWorkingHourMapper;
	@Autowired
	private WorkstationMapper workstationMapper;
	@Autowired
    private PartWorkingProcessMapper partWorkingProcessMapper;
	@Autowired
	private OssBuilder ossBuilder;

	/**
	 * 导出一条数据
	 * 
	 * @param id
	 * @return
	 * @throws IOException
	 */
	@Transactional(readOnly = true)
	public BladeFile export(long id) throws Exception {
		// QueryWrapper<PartWorkingProcess> wrapper = new QueryWrapper<>();
		// wrapper.lambda().eq(PartWorkingProcess::getWorkinghourId, id);

		// xx工件在x机床
		PartWorkingHour partWorkingHour = partWorkingHourMapper.selectById(id);
		Workstation ws = workstationMapper.selectById(partWorkingHour.getWorkstationId());
		String fileName = String.format("零件%s的工时数据-%s.xlsx", partWorkingHour.getPartNo(), DateUtil.time());

		PartWorkingHourExcel excelObj = new PartWorkingHourExcel();
		excelObj.setPartNo(partWorkingHour.getPartNo());
		excelObj.setAmount(partWorkingHour.getAmount());

		LocalDateTime start = LocalDateTime.now();
		LocalDateTime end;
		Duration duration;
		if (partWorkingHour.getClampingSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getClampingSecs());
			duration = Duration.between(start, end);
			excelObj.setClampingSecs(DurationUtil.toChineseDuration(duration));
		}
		final String format = "yyyy-MM-dd:hh:mm:ss";
		if (partWorkingHour.getEndTime() != null) {
			excelObj.setEndTime(LocalDateTimeUtil.format(partWorkingHour.getEndTime(), format));
		}

		if (partWorkingHour.getFirstMeasureSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getFirstMeasureSecs());
			duration = Duration.between(start, end);
			excelObj.setFirstMeasureSecs(DurationUtil.toChineseDuration(duration));
		}

		if (partWorkingHour.getFirstWorkingSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getFirstWorkingSecs());
			duration = Duration.between(start, end);
			excelObj.setFirstWorkingSecs(DurationUtil.toChineseDuration(duration));
		}

		if (partWorkingHour.getLastRemoveSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getLastRemoveSecs());
			duration = Duration.between(start, end);
			excelObj.setLastRemoveSecs(DurationUtil.toChineseDuration(duration));
		}

		if (partWorkingHour.getOccupancySecs() != null) {
			end = start.plusSeconds(partWorkingHour.getOccupancySecs());
			duration = Duration.between(start, end);
			excelObj.setOccupancySecs(DurationUtil.toChineseDuration(duration));
		}

		excelObj.setPartNo(partWorkingHour.getPartNo());

		if (partWorkingHour.getPrepareSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getPrepareSecs());
			duration = Duration.between(start, end);
			excelObj.setPrepareSecs(DurationUtil.toChineseDuration(duration));
		}

		if (partWorkingHour.getProcessingSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getProcessingSecs());
			duration = Duration.between(start, end);
			excelObj.setProcessingSecs(DurationUtil.toChineseDuration(duration));
		}

		excelObj.setProcessNo(partWorkingHour.getProcessNo());

		if (partWorkingHour.getSingleProcessSecs() != null) {
			end = start.plusSeconds(partWorkingHour.getSingleProcessSecs());
			duration = Duration.between(start, end);
			excelObj.setSingleProcessSecs(DurationUtil.toChineseDuration(duration));
		}

		excelObj.setStartTime(LocalDateTimeUtil.format(partWorkingHour.getStartTime(), format));

		excelObj.setWorkstationName(ws.getName());

		excelObj.setVersion(partWorkingHour.getVersion());

		MultipartFile multipartFile = ExcelUtil.exportToMultipartFile(fileName, "零件工时", Arrays.asList(excelObj),
				PartWorkingHourExcel.class);

		MultipartFile multipartFile2 = writeChart(id,multipartFile);

		BladeFile bladeFile = this.ossBuilder.tempTemplate().putFile(multipartFile2.getOriginalFilename(),
				multipartFile2);
		return bladeFile;
	}

	/**
	 * 创建图表
	 * @param workinghourId
	 * @param multipartFile
	 * @return
	 * @throws Exception
	 */
	MultipartFile writeChart(long workinghourId,MultipartFile multipartFile) throws Exception {
		XSSFWorkbook workbook = new XSSFWorkbook(multipartFile.getInputStream());

		byte[] chartBytes = barChart(workinghourId);
		addPicture(workbook, chartBytes);

		DiskFileItemFactory factory = new DiskFileItemFactory();
		factory.setRepository(new File(System.getProperty("java.io.tmpdir")));

		FileItem fileItem = factory.createItem("excel", multipartFile.getContentType(), true,
				multipartFile.getOriginalFilename());

		workbook.write(fileItem.getOutputStream());
		workbook.close();

		return new CommonsMultipartFile(fileItem);

	}

	public static void addPicture(XSSFWorkbook wb, byte[] bytes) {
		XSSFSheet sheet = wb.getSheetAt(0);
		// 画图的顶级管理器，一个sheet只能获取一个（一定要注意这点）
		// HSSFPatriarch patriarch = sheet.createDrawingPatriarch();
		XSSFDrawing patriarch = sheet.createDrawingPatriarch();
		// 设置图片位置
		// 八个参数, 前四个表示图片离起始单元格和结束单元格边缘的位置，
		// 后四个表示起始和结束单元格的位置, 如下表示从第2列到第12列, 从第1行到第15行,需要注意excel起始位置是0
		// {0,0,0,0,2,1,12,15}表示从第2列到第12列，从第1行到第15行，单元格内部的边距都是0
		// {0,0,0,0,1,4,20,5}表示从第2列到第12列，从第1行到第15行，单元格内部的边距都是0
		// HSSFClientAnchor anchor = new HSSFClientAnchor(anchors[0], anchors[1],
		// anchors[2], anchors[3], (short) anchors[4], anchors[5], (short) anchors[6],
		// anchors[7]);

		XSSFClientAnchor anchor = new XSSFClientAnchor(0, 0, 0, 0, 0, 3, 30, 10); // 1列-30列；3行-10行
		anchor.setAnchorType(AnchorType.DONT_MOVE_AND_RESIZE);
		// 插入图片
		int pictureIndex = wb.addPicture(bytes, XSSFWorkbook.PICTURE_TYPE_JPEG);
		// XSSFWorkbook.PICTURE_TYPE_JPEG
		patriarch.createPicture(anchor, pictureIndex);
	}

	public byte[] barChart(long workinghourId) throws Exception {
		
		// 设置柱子显示对应的数值

		QueryWrapper<PartWorkingProcess> wrapper = new QueryWrapper<>();
    	wrapper.lambda().eq(PartWorkingProcess::getWorkinghourId, workinghourId).orderByAsc(PartWorkingProcess::getStartTime);
    	List<PartWorkingProcess> processList = partWorkingProcessMapper.selectList(wrapper);
    	
    	// 支持标签显示的所有列数据		
    	List<List<Object>> dataList = new ArrayList<>();
    	
    	Duration duration;
    	PartWorkingProcess process;
    	for(int i=0;i<processList.size();i++) {
    		//dataList.add(new ArrayList<>(Arrays.asList(133)));
    		process = processList.get(i);
    		duration = Duration.between(process.getStartTime(), process.getEndTime());
    		dataList.add(Arrays.asList(duration.getSeconds()));
    		
    		//dataTable.add(new ArrayList<>(Arrays.asList(process.getProgName(), duration.getSeconds(), process.getDeviceStatus())));
    	}
    	
    	JFreeChart chart = createStackedBarChart(workinghourId,"加工记录", dataList,processList,
				JFreeChartUtil.createChartTheme("宋体"), "秒", "");
    	StackedBarRenderer renderer = new StackedBarRenderer();
		CategoryPlot plot = chart.getCategoryPlot();
		plot.setOutlineVisible(true);
		
		
		//dataTable.add(new ArrayList<>(Arrays.asList("YZL4-1100-01-50-A-2-2", 242, "运行")));

		renderer.setDefaultItemLabelsVisible(true);
		renderer.setDefaultItemLabelGenerator(new MyCategoryItemLabelGenerator(processList));
		
		renderer.setMaximumBarWidth(0.3);

		// 柱子颜色,有几段就set几个
		Color color;
		String colorStr;
		DefaultWcsEnum wcsEnum;
		for(int i=0;i<processList.size();i++) {
			process = processList.get(i);
			//if(dataTable.get(i))
			//color = 
			wcsEnum = DefaultWcsEnum.of(process.getDeviceStatus());
			colorStr = StringUtils.removeStart(wcsEnum.getColor(), '#');
			color  = new Color(Integer.parseInt(colorStr, 16));
			renderer.setSeriesPaint(i, color);
		}
		//renderer.setSeriesPaint(0, Color.green);
		//renderer.setSeriesPaint(1, Color.yellow);

		ItemLabelPosition posi = new ItemLabelPosition(ItemLabelAnchor.OUTSIDE12, TextAnchor.HALF_ASCENT_CENTER);
		//renderer.setDefaultNegativeItemLabelPosition(posi);
		renderer.setDefaultPositiveItemLabelPosition(posi);

		int offset = 10;
		renderer.setItemLabelInsets(new RectangleInsets(0, offset, offset, offset));
		plot.setRenderer(renderer);

		java.io.ByteArrayOutputStream bos = new java.io.ByteArrayOutputStream();

		ChartUtils.writeChartAsJPEG(bos, 1.0f, chart, 4096, 400, null);

		return bos.toByteArray();
	}

	/**
	 * 创建堆叠柱状图
	 * @param workinghourId
	 * @param chartTitle
	 * @param legendNameList
	 * @param xAxisNameList
	 * @param dataList
	 * @param theme
	 * @param yAxisTitle
	 * @param xAxisTitle
	 * @return
	 * @throws Exception
	 */
	JFreeChart createStackedBarChart(long workinghourId,String chartTitle, List<List<Object>> dataList,List<PartWorkingProcess> processList,
			 StandardChartTheme theme, String yAxisTitle, String xAxisTitle)
			throws Exception {
		
		
		// 设置主题，防止中文乱码
		theme = theme == null ? JFreeChartUtil.createChartTheme("") : theme;
		ChartFactory.setChartTheme(theme);
		
		boolean showLegend =false;
		//List<String> pnamelist = processList.stream().map(p -> p.getProgName()).collect(Collectors.toList());
		
		//DefaultCategoryDataset dataset = JFreeChartUtil.createDefaultCategoryDataset(pnamelist, Arrays.asList("xname"), dataList);
		DefaultCategoryDataset dataset = createDataSet(processList);
		JFreeChart chart = ChartFactory.createStackedBarChart(chartTitle, xAxisTitle, yAxisTitle,
				dataset,
				PlotOrientation.HORIZONTAL, showLegend, true, false);
		// 设置抗锯齿，防止字体显示不清楚
		chart.setTextAntiAlias(false);
		// 对柱子进行渲染
		JFreeChartUtil.setBarRenderer(chart.getCategoryPlot(), true);
		
		//showLegend=false情况下，下列代码会引起NPE
		// 设置标注无边框
		//chart.getLegend().setFrame(new BlockBorder(Color.WHITE));
		// 标注位于上侧
		//chart.getLegend().setPosition(RectangleEdge.TOP);

		return chart;
	}
	
	DefaultCategoryDataset createDataSet(List<PartWorkingProcess> processList){
		DefaultCategoryDataset dataset = new DefaultCategoryDataset();
		PartWorkingProcess p;
		 for (int xAxisIndex = 0; xAxisIndex < processList.size(); xAxisIndex++) {
			 p = processList.get(xAxisIndex);
             //String value = rowList.get(xAxisIndex).toString();
			 String value =p.getStartTime().toString();
			 Duration duration = Duration.between(p.getStartTime(), p.getEndTime());
			 value = duration.getSeconds()+"";
             dataset.setValue(Double.parseDouble(value), xAxisIndex+"", "程序时段");
         }
		 
		 return dataset;
	}
}

//自定义标签生成
class MyCategoryItemLabelGenerator extends AbstractCategoryItemLabelGenerator implements CategoryItemLabelGenerator {
	// private final Integer category;
	private List<PartWorkingProcess> dataTable;
	
	public MyCategoryItemLabelGenerator(List<PartWorkingProcess> dataTable) {
		super("", NumberFormat.getInstance());
		this.dataTable = dataTable;
	}

	@Override
	public String generateLabel(CategoryDataset dataset, int row, int column) {
		//Number val = dataset.getValue(row, column);

		PartWorkingProcess dataRow = this.dataTable.get(row);
		// dataset.
		/*
		ALARM(1, "报警"),
	    RUNNING(2, "运行"),
	    STANDBY(3, "待机"),
	    OFFLINE(4, "离线"),
	    DEBUGGING(5, "调试");
	    */
		
		//DefaultWcsEnum wcsEnum = DefaultWcsEnum.of(dataRow.getDeviceStatus());
		//return dataRow.getProgName() + ",:" + val + "," + wcsEnum.getName();
		Duration duration = Duration.between(dataRow.getStartTime(), dataRow.getEndTime());
		return dataRow.getProgName() + "," + DurationUtil.toChineseDuration(duration);
	}

}