package cn.how2j.test;

import java.io.File;
import java.nio.charset.Charset;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;

import cn.hutool.core.io.FileUtil;
import cn.hutool.json.JSONArray;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;

public class Test {

    public static void main(String[] args) {
//    	makeFutureWAM();
//    	remakeFutureJsonFile();

    }

	private static void remakeFutureJsonFile() {
		File releaseFolder = new File("D:/Downloads/inspector/runnable/trend/build/download/release");
    	
    	File futureJsonFile = new File(releaseFolder,"future.json");
    	File futureWAMJsonFile = new File(releaseFolder,"future_wam.json");
    	
    	JSONArray ja = JSONUtil.readJSONArray(futureJsonFile, Charset.forName("utf-8"));
    	 
    	
    	Map m = new HashMap<>();
    	for (Object object : ja) {
			JSONObject jo = (JSONObject) object;
			String code = jo.getStr("code");
			String name = jo.getStr("name");

			String codeOnly = code.replaceAll("\\d", "");
			String nameOnly = name.replaceAll("\\d", "");			
			
			m.put(codeOnly,nameOnly);
		}
    	
    	
    	List list = new ArrayList<>();
    	Set<Map.Entry<String, String>> entrys = m.entrySet();
    	System.out.println(entrys.size());
    	for (Map.Entry<String, String> entry : entrys) {
			Map m2 = new HashMap<>();
			m2.put("code", entry.getKey());
			m2.put("name", entry.getValue());
			list.add(m2);
			
		}
    	
    	String fc = JSONUtil.toJsonPrettyStr(list);
    	System.out.println(fc);
    	System.out.println(ja.size());
    	
    	FileUtil.writeUtf8String(fc, futureWAMJsonFile);
	}

	private static void makeFutureWAM() {
		File futureFolder = new File("D:/Downloads/inspector/runnable/trend/build/download/release/future");
    	File futureWAMFolder = new File("D:/Downloads/inspector/runnable/trend/build/download/release/future_wam");
    	
    	
    	File[] files= futureFolder.listFiles();
    	Map<String,List<File>> m = new HashMap<>();
    	for (File file : files) {
			String name = FileUtil.mainName(file);
			
			if(!name.matches(".*\\d.*")) //不包含数字
				continue;
			
			String codeOnly = name.replaceAll("\\d", "");
//			System.out.println(codeOnly);
			List<File> fileList = m.get(codeOnly);
			if(null==fileList){
				fileList = new ArrayList<>();
				m.put(codeOnly, fileList);
			}
			fileList.add(file);
		}
    	
    	Set<String> codes = m.keySet();
    	for (String code : codes) {
			List<File> fileList = m.get(code);
			calcWAMByList(code,fileList,futureWAMFolder);
		}
	}

	private static void calcWAMByList(String code,List<File> files, File futureWAMFolder) {
		Map<String,List<FutureUnit>> m = new LinkedHashMap<>();
        for (File file : files) {
			JSONArray ja = JSONUtil.readJSONArray(file, Charset.forName("utf-8"));
			for (Object object : ja) {
				String date = null;
				try {
					JSONObject jo = (JSONObject) object;
					if(0 == jo.getStr("volumn").length())
						continue;
					if(0 == jo.getStr("close").length())
						continue;
					if(0 == jo.getFloat("volumn"))
						continue;
					if(0 == jo.getFloat("close"))
						continue;
					
					
					date = jo.getStr("date");
					int volumn = jo.getInt("volumn");
					float close = jo.getFloat("close");
					
//					System.out.println("volumn:"+volumn);
//					System.out.println("close:"+close);
					
					FutureUnit fu = new FutureUnit(date, volumn, close);
							
					List<FutureUnit> fus = m.get(date);
					if(null==fus){
						fus = new ArrayList<>();
						m.put(date, fus);
					}
						
					fus.add(fu);
					
				} catch (Exception e) {
					e.printStackTrace();
				}
			}
		}
        Set<String> datesSet = m.keySet();
        List<String> datesList = new ArrayList<>(datesSet);
        Collections.sort(datesList);

        List<Map<String,Object>> list = new ArrayList<>();
        for (String date : datesList) {
			List<FutureUnit> fus = m.get(date);
			float closeWAM = calcWAM(fus);
			
			if(Float.isNaN(closeWAM)){
				System.out.println(code);
			}
			
			Map<String,Object> m2 = new HashMap<>();
			
			m2.put("date",date);
			m2.put("close",closeWAM);
//			System.out.println(closeWAM);
			list.add(m2);
			
		}
        
        File f = new File(futureWAMFolder,code+".txt");
        System.out.println(f);
        String fc = JSONUtil.toJsonPrettyStr(list);
        FileUtil.writeUtf8String(fc, f);
        
        
        
        
        
        
	}

	private static float calcWAM(List<FutureUnit> fus) {
		int totalVolumn = 0;
		double closeWAM= 0;
		for (FutureUnit fu : fus) {
//			System.out.println(fu);
			totalVolumn+=fu.getVolumn();
		}

		float result = 0;

		for (FutureUnit fu : fus) {
			double rate = (0f+fu.getVolumn())/totalVolumn;
			double closeWAMEach = fu.getClose()*rate;
			closeWAM +=closeWAMEach;
			
			
			
		}
		return (float) closeWAM;
	}
	
	
static class FutureUnit {

	private String date;
	int volumn;
	float close;
	public String getDate() {
		return date;
	}
	public void setDate(String date) {
		this.date = date;
	}
	public int getVolumn() {
		return volumn;
	}
	public void setVolumn(int volumn) {
		this.volumn = volumn;
	}
	public float getClose() {
		return close;
	}
	public void setClose(float close) {
		this.close = close;
	}
	public FutureUnit(String date, int volumn, float close) {
		super();
		this.date = date;
		this.volumn = volumn;
		this.close = close;
	}
	@Override
	public String toString() {
		return "FutureUnit [date=" + date + ", volumn=" + volumn + ", close=" + close + "]";
	}
	
	
	
}
	
	
}