潘钊 / bsth_control · Files

package com.bsth.service.realcontrol.buffer;

import java.util.ArrayList;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.bsth.entity.realcontrol.ScheduleRealInfo;
import com.google.common.collect.ArrayListMultimap;
import com.google.common.collect.LinkedListMultimap;

/**
 * 
 * @ClassName: ScheduleRealInfoBuffer 
 * @Description: TODO(当天实际排班缓存) 
 * @author PanZhao
 * @date 2016年6月14日 上午9:40:38 
 *
 */
public class ScheduleBuffer {
	
	static Logger logger = LoggerFactory.getLogger(ScheduleBuffer.class);
	
	/**
	 * 当天实际排班   K 线路编码 V 按时间升序的班次
	 */
	public static ArrayListMultimap<String, ScheduleRealInfo> schedulListMap;
	
	/**
	 * K: 车辆自编号  V: 未完成班次链表
	 */
	/*public static Map<String, LinkedList<ScheduleRealInfo>> vehLinkedMap;*/
	
	/**
	 * K: 车辆自编号  V: 班次链表
	 */
	public static ArrayListMultimap<String, ScheduleRealInfo> vehSchListMap;
	/*public static Map<String, LinkedList<ScheduleRealInfo>> vehSchLinkedMap;*/
	
	
	/**
	 * K: 车辆自编号  V: 已执行班次链表
	 */
	/*public static LinkedListMultimap<String, ScheduleRealInfo> finishLinkedMap;*/
	
	
	/**
	 * 主键 和 排班映射
	 */
	public static Map<Long, ScheduleRealInfo> pkSchedulMap;
	
	/**
	 * 需要持久化的排班
	 */
	public static LinkedList<ScheduleRealInfo> persistentList;
	
	/** K: 线路编码_上下行  V: 班次平均间隔时间 */
	public static Map<String , Long> schSpaceMap;
	
	/**
	 * 线路是否托管  K：线路编码  V：0 托管  1 非托管
	 */
	public static Map<Integer, Integer> trustMap; 
	
	static ScheduleComparator scheduleComparator = new ScheduleComparator();
	
	static{
		clear();
	}
	
	public static void clear(){
		schedulListMap = ArrayListMultimap.create();
		pkSchedulMap = new HashMap<>();
		persistentList = new LinkedList<>();
		/*vehLinkedMap = new HashMap<>();
		finishLinkedMap = LinkedListMultimap.create();*/
		vehSchListMap = ArrayListMultimap.create();
		trustMap = new HashMap<>();
	}
	
	public static int init(List<ScheduleRealInfo> list){
		
		try{
			//计算时间戳
			for(ScheduleRealInfo schedul : list){
				schedul.syncTime();
			}
			
			//发车顺序号排序
			Collections.sort(list, scheduleComparator);
			
			String zbh;
			for(ScheduleRealInfo schedul : list){
				schedulListMap.put(schedul.getXlBm(), schedul);
				pkSchedulMap.put(schedul.getId(), schedul);
				
				//初始化车辆和班次列表对照
				zbh = schedul.getClZbh();
				vehSchListMap.put(zbh, schedul);
				/*if(!vehLinkedMap.containsKey(zbh))
					vehLinkedMap.put(zbh, new LinkedList<ScheduleRealInfo>());
				
				vehLinkedMap.get(zbh).add(schedul);*/
			}
			
			//计算起点应到时间
			Set<String> codes = schedulListMap.keySet();
			for(String code : codes)
				calcArrDateQd(code);
			
			//计算班次平均间隔
			calcAvgSpace();
		}catch(Exception e){
			logger.error("缓存排班数据失败...", e);
			return -1;
		}
		return 0;
	}
	
	//计算平均间隔
	public static void calcAvgSpace(){
		schSpaceMap = new HashMap<>();
		Set<String> set = schedulListMap.keySet();
		
		List<ScheduleRealInfo> list = null
				,upList = null
				,downList =  null;
		for(String lineCode : set){
			list = schedulListMap.get(lineCode);
			Long sumUp = 0L, sumDown = 0L;
			
			upList = new ArrayList<>();
			downList = new ArrayList<>();
			//按上下行分组
			for(ScheduleRealInfo sch : list){
				if(sch.getXlDir().equals("0"))
					upList.add(sch);
				else
					downList.add(sch);
			}
			for(int i = 1; i < upList.size(); i ++){
				sumUp += (upList.get(i).getFcsjT() - upList.get(i - 1).getFcsjT());
			}
			
			for(int i = 1; i < downList.size(); i ++){
				sumDown += (downList.get(i).getFcsjT() - downList.get(i - 1).getFcsjT());
			}
			
			schSpaceMap.put(lineCode + "_0", sumUp / upList.size());
			schSpaceMap.put(lineCode + "_1", sumDown / downList.size());
		}
	}
	
	public static int put(ScheduleRealInfo sch){
		schedulListMap.put(sch.getXlBm(), sch);
		
		pkSchedulMap.put(sch.getId(), sch);
		String zbh = sch.getClZbh();
		vehSchListMap.put(zbh, sch);
		
		//List<ScheduleRealInfo> list = vehSchListMap.get(sch.getClZbh());
		
		/*if(!vehLinkedMap.containsKey(zbh))
			vehLinkedMap.put(zbh, new LinkedList<ScheduleRealInfo>());
		vehLinkedMap.get(zbh).add(sch);*/
		
		//重新排序
		Collections.sort(schedulListMap.get(sch.getXlBm()), scheduleComparator);
		
		//重新计算应到时间
		//calcArrDateQd(vehLinkedMap.get(zbh));
		calcArrDateQd(vehSchListMap.get(zbh));
		return 0;
	}
	
	public static class ScheduleComparator implements Comparator<ScheduleRealInfo>{

		@Override
		public int compare(ScheduleRealInfo o1, ScheduleRealInfo o2) {
			//return (int) (o1.getFcsjT() - o2.getFcsjT());
			return o1.getFcno() - o2.getFcno();
		}
	}
	
	/**
	 * 
	 * @Title: finishSch 
	 * @Description: TODO(完成一个班次) 
	 * @param @param sch
	 * @return ScheduleRealInfo  返回 下一个未执行的班次
	 * @throws
	 */
	public static ScheduleRealInfo finishSch(ScheduleRealInfo sch){
		//LinkedList<ScheduleRealInfo> list = vehLinkedMap.get(sch.getClZbh());
		//状态修改为已执行
		sch.setStatus(2);
		persistentList.add(sch);
		
		//下一个班次
		ScheduleRealInfo next = getNext(sch);
		next.setQdzArrDateSJ(sch.getZdsjActual());
		/*ScheduleRealInfo temp;
		int len = list.size();
		for(int i = 0; i < len; i ++){
			temp = list.poll();
			if(temp.equals(sch)){
				//加入已执行
				finishLinkedMap.put(sch.getClZbh(), sch);
				break;
			}
		}
		ScheduleRealInfo next = list.getFirst();*/
		//next.setQdzArrDateSJ(sch.getZdsjActual());
		return next;
	}
	
	public static ScheduleRealInfo getNext(ScheduleRealInfo sch){
		List<ScheduleRealInfo> list = vehSchListMap.get(sch.getClZbh());
		
		boolean flag = false;
		ScheduleRealInfo next = null;
		for(ScheduleRealInfo temp : list){
			if(temp.getId() == sch.getId()){
				flag = true;
				continue;
			}
			//忽略烂班
			if(temp.getStatus() == -1)
				continue;
			
			if(flag){
				next = temp;
				break;
			}
		}
		
		
		/*int size = list.size();
		for(int i = 0; i < size; i ++){
			current = list.get(i);
			if(current.getId() == sch.getId()
					&& i < (size - 1)){
				next = list.get(i + 1);
				break;
			}
		}*/
		return next;
	}
	
	/**
	 * 
	 * @Title: getFinishSchNo 
	 * @Description: TODO(获取车辆已完成的班次数) 
	 * @param @param nbbm
	 * @throws
	 */
	public static int getFinishSchNo(String nbbm){
		List<ScheduleRealInfo> list = vehSchListMap.get(nbbm);
		int no = 0;
		for(ScheduleRealInfo sch : list){
			if(sch.getStatus() == 2)
				no ++;
		}
		return no;
		//return finishLinkedMap.get(nbbm) == null ? 0 : finishLinkedMap.get(nbbm).size();
	}
	
	/**
	 * 
	 * @Title: findCurrent 
	 * @Description: TODO(获取车辆当前正在执行的班次) 
	 * @param @param nbbm
	 * @throws
	 */
	public static ScheduleRealInfo findCurrent(String nbbm){
		List<ScheduleRealInfo> list = ScheduleBuffer.vehSchListMap.get(nbbm);
		
		int size = list.size();
		ScheduleRealInfo sch;
		//最后一个状态为 正在执行的班次
		for(int i = size - 1; i > 0; i --){
			sch = list.get(i);
			if(sch.getStatus() == 1)
				return sch;
		}
		return null;
	}
	
	/**
	 * 
	 * @Title: findNextList 
	 * @Description: TODO(获取后续班次) 
	 * @param @param sch
	 * @throws
	 */
	public static List<ScheduleRealInfo> findNextList(ScheduleRealInfo sch){
		List<ScheduleRealInfo> list = vehSchListMap.get(sch.getClZbh()),
					rs = null;
		
		for(ScheduleRealInfo temp : list){
			if(temp.getId() == sch.getId())
				rs = new ArrayList<>();
			
			if(rs != null)
				rs.add(temp);
		}
		return rs == null?new ArrayList<ScheduleRealInfo>():rs;
	}
	
	public static void calcArrDateQd(List<ScheduleRealInfo> list){
		//排序
		Collections.sort(list, scheduleComparator);
		int len = list.size();
		if(len == 0)
			return;
		
		ScheduleRealInfo prve = list.get(0), curr;
		for(int i = 1; i < len; i ++){
			curr = list.get(i);
			if(prve.getZdzName().equals(curr.getQdzName()))
				curr.setQdzArrDateJH(prve.getZdsj());
			
			prve = curr;
		}
	}
	
	/**
	 * 
	 * @Title: calcArrDateQd 
	 * @Description: TODO(计算计划起点时间) 
	 * @throws
	 */
	public static void calcArrDateQd(String lineCode){
		List<ScheduleRealInfo> list = schedulListMap.get(lineCode);
		//按车辆分组
		ArrayListMultimap<String, ScheduleRealInfo> map = ArrayListMultimap.create();
		for(ScheduleRealInfo sch : list){
			map.put(sch.getClZbh(), sch);
		}
		
		//链接班次
		Set<String> set = map.keySet();
		for(String k : set){
			calcArrDateQd(map.get(k));
		}
	}
	
	public static ScheduleRealInfo findOne(Long id){
		return pkSchedulMap.get(id);
	}
	
	public static List<ScheduleRealInfo> findByLineAndUpDown(Integer lineCode, Integer upDown){
		List<ScheduleRealInfo> list = schedulListMap.get(String.valueOf(lineCode))
				,subList = new ArrayList<>();
		//按走向过滤
		for(ScheduleRealInfo sch : list){
			if(Integer.parseInt(sch.getXlDir()) == upDown){
				subList.add(sch);
			}
		}
		return subList;
	}
}