package it.acxent.face.callable;

import it.acxent.face.FotoFace;
import it.acxent.face.api.FaceRecognitionApi;
import it.acxent.face.api.FaceRecognitionApiResult;
import it.acxent.face.api.TrainingImage;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.Callable;
import org.json.JSONArray;
import org.json.JSONObject;

public class TrainingSendTask implements Callable<FaceRecognitionApiResult> {
  private final FaceRecognitionApi frApi;
  
  private final long faceRecognizerId;
  
  private final List<TrainingImage> imageFilenames;
  
  public TrainingSendTask(FaceRecognitionApi frApi, long faceRecognizerId, List<TrainingImage> imageFilenames) {
    this.frApi = frApi;
    this.faceRecognizerId = faceRecognizerId;
    this.imageFilenames = imageFilenames;
  }
  
  public FaceRecognitionApiResult call() throws Exception {
    FaceRecognitionApiResult resF = this.frApi.__trainingSendImages(this.faceRecognizerId, this.imageFilenames, false);
    handleApiResponse(resF);
    if (resF.isOk() && resF.getResult() != null) {
      JSONObject jres = (JSONObject)resF.getResult();
      if (jres.has("noImg")) {
        List<TrainingImage> missingImages = new ArrayList<>();
        JSONArray jNoimgA = jres.getJSONArray("noImg");
        for (int k = 0; k < jNoimgA.length(); k++) {
          JSONObject jNoImgRow = jNoimgA.getJSONObject(k);
          FotoFace ff = new FotoFace(this.frApi.getApFull());
          ff.findByMd5(jNoImgRow.getString("md5"));
          if (ff.getId_fotoFace() > 0L)
            missingImages.add(new TrainingImage(ff.getFullFaceImgPath(), ff.getLabel(), ff.getMd5())); 
        } 
        if (!missingImages.isEmpty()) {
          FaceRecognitionApiResult missingResF = this.frApi.__trainingSendImages(this.faceRecognizerId, missingImages, true);
          handleApiResponse(missingResF);
        } 
      } 
    } 
    return resF;
  }
  
  private void handleApiResponse(FaceRecognitionApiResult resF) {
    if (resF.isOk() && resF.getResult() != null) {
      JSONObject jres = (JSONObject)resF.getResult();
      if (jres.has("labels"))
        FotoFace.updateLabels(this.frApi.getApFull(), jres); 
    } 
  }
}