package com.basic.analy.utils;
import com.basic.analy.service.FaceSdkTool;
import lombok.Data;
import org.apache.log4j.Logger;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.concurrent.RecursiveTask;
@Data
public class SearchImgThreadUtil extends RecursiveTask<Integer> {
    private static final int THRESHOLD = 200;
    private List<Map> faceArray;
    private byte[] imageData;
    private int likePer;
    private int start;
    private int end;
    public static List<Map> resultList = new ArrayList<>();
    private Logger logger = Logger.getLogger(EsThreadUtils.class);

    public SearchImgThreadUtil(byte[] imageData,List<Map> faceArray,int likePer,int start,int end){
        this.imageData = imageData;
        this.faceArray =faceArray;
        this.likePer = likePer;
        this.start = start;
        this.end = end;
    }
    public Integer compute(){

        boolean canCompute = faceArray.size() <= THRESHOLD;
        if (canCompute){
            logger.info("进入likePer:"+likePer);
            for (int i = 0,item = faceArray.size(); i <item; i++){

                if((byte[])faceArray.get(i).get("feature")!=null&&!"".equals((byte[])faceArray.get(i).get("feature"))){
                    // 业务处理
                    Float num= new FaceSdkTool().compareFeature(imageData, (byte[])faceArray.get(i).get("feature"));
                    if(Math.round(num*100)<likePer){ // 相似度小于阈值数据丢弃
                        continue;
                    }
                    logger.info("相似值为"+likePer);
                    faceArray.get(i).put("likePer",Math.round(num*100));
                    resultList.add(faceArray.get(i));
                }
            }
        }else {
            int middle = (end + start) / 2;
            SearchImgThreadUtil subtask1 = new SearchImgThreadUtil(imageData,faceArray.subList(start, end),likePer,start, middle);
            SearchImgThreadUtil subtask2 = new SearchImgThreadUtil(imageData,faceArray.subList(start, end),likePer,middle, end);
            invokeAll(subtask1, subtask2);
            subtask1.join();
            subtask2.join();
        }
        return faceArray.size();
    }
//    public static void main(String[] args) {
//
//    }
//
//    public byte[] getImageData() {
//        return imageData;
//    }
//
//    public void setImageData(byte[] imageData) {
//        this.imageData = imageData;
//    }
//
//    public int getLikePer() {
//        return likePer;
//    }
//    public void setLikePer(int likePer) {
//        this.likePer = likePer;
//    }
//
//    public List<Map> getFaceArray() {
//        return faceArray;
//    }
//    public void setFaceArray(List<Map> faceArray) {
//        this.faceArray = faceArray;
//    }
//    public static List<Map<String, Object>> getCurrentList() {
//        return currentList;
//    }
//    public static void setCurrentList(List<Map<String, Object>> currentList) {
//        SearchImgThreadUtil.currentList = currentList;
//    }
}