package com.hms.doctorconsultdemo.ml;
 
 import android.Manifest;
 import android.content.Intent;
 import android.content.pm.PackageManager;
 import android.graphics.Bitmap;
 import android.os.Bundle;
 import android.provider.MediaStore;
 import android.util.Log;
 
 import androidx.annotation.Nullable;
 import androidx.core.app.ActivityCompat;
 
 import com.huawei.hmf.tasks.Task;
 import com.huawei.hms.mlsdk.MLAnalyzerFactory;
 import com.huawei.hms.mlsdk.common.MLException;
 import com.huawei.hms.mlsdk.common.MLFrame;
 import com.huawei.hms.mlsdk.productvisionsearch.MLProductVisionSearch;
 import com.huawei.hms.mlsdk.productvisionsearch.MLVisionSearchProduct;
 import com.huawei.hms.mlsdk.productvisionsearch.MLVisionSearchProductImage;
 import com.huawei.hms.mlsdk.productvisionsearch.cloud.MLRemoteProductVisionSearchAnalyzer;
 import com.huawei.hms.mlsdk.productvisionsearch.cloud.MLRemoteProductVisionSearchAnalyzerSetting;
 
 import java.util.ArrayList;
 import java.util.List;
 
 
 public class ScanMainActivity extends BaseActivity {
 
     private static final String TAG = ScanMainActivity.class.getName();
     private static final int CAMERA_PERMISSION_CODE = 100;
 
     MLRemoteProductVisionSearchAnalyzer analyzer;
 
     @Override
     protected void onCreate(@Nullable Bundle savedInstanceState) {
         super.onCreate(savedInstanceState);
         init();
         initializeProductVisionSearch();
     }
 
     private void init() {
         if (!(ActivityCompat.checkSelfPermission(this, Manifest.permission.CAMERA)
                 == PackageManager.PERMISSION_GRANTED)) {
             this.requestCameraPermission();
         }
         initializeProductVisionSearch();
         Intent intent = new Intent(MediaStore.ACTION_IMAGE_CAPTURE);
         startActivityForResult(intent, 101);
     }
 
     private void requestCameraPermission() {
         final String[] permissions = new String[]{Manifest.permission.CAMERA};
 
         if (!ActivityCompat.shouldShowRequestPermissionRationale(this, Manifest.permission.CAMERA)) {
             ActivityCompat.requestPermissions(this, permissions, this.CAMERA_PERMISSION_CODE);
             return;
         }
     }
 
     @Override
     protected void onActivityResult(int requestCode, int resultCode, @Nullable Intent data) {
         super.onActivityResult(requestCode, resultCode, data);
         if (requestCode == 101) {
             if (resultCode == RESULT_OK) {
                 Bitmap bitmap = (Bitmap) data.getExtras().get("data");
                 if (bitmap != null) {
                     MLFrame mlFrame = new MLFrame.Creator().setBitmap(bitmap).create();
                     mlImageDetection(mlFrame);
                 }
 
             }
 
         }
     }
 
     private void mlImageDetection(MLFrame mlFrame) {
 
         Task<List<MLProductVisionSearch>> task = analyzer.asyncAnalyseFrame(mlFrame);
         task.addOnSuccessListener(products -> {
             Log.d(TAG, "success");
             displaySuccess(products);
         })
                 .addOnFailureListener(e -> {
                     try {
                         MLException mlException = (MLException) e;
                         int errorCode = mlException.getErrCode();
                         String errorMessage = mlException.getMessage();
                     } catch (Exception error) {
                         // Handle the conversion error.
                     }
                 });
     }
 
     private void initializeProductVisionSearch() {
         MLRemoteProductVisionSearchAnalyzerSetting settings = new MLRemoteProductVisionSearchAnalyzerSetting.Factory()
                 .setLargestNumOfReturns(16)
                 .setRegion(MLRemoteProductVisionSearchAnalyzerSetting.REGION_DR_CHINA)
                 .create();
         analyzer
                 = MLAnalyzerFactory.getInstance().getRemoteProductVisionSearchAnalyzer(settings);
     }
 
     private void displaySuccess(List<MLProductVisionSearch> productVisionSearchList) {
         List<MLVisionSearchProductImage> productImageList = new ArrayList<>();
         String prodcutType = "";
         for (MLProductVisionSearch productVisionSearch : productVisionSearchList) {
             Log.d(TAG, "type: " + productVisionSearch.getType());
             prodcutType = productVisionSearch.getType();
             for (MLVisionSearchProduct product : productVisionSearch.getProductList()) {
                 productImageList.addAll(product.getImageList());
                 Log.d(TAG, "custom content: " + product.getCustomContent());
             }
         }
         StringBuffer buffer = new StringBuffer();
         for (MLVisionSearchProductImage productImage : productImageList) {
             String str = "ProductID: " + productImage.getProductId() + "\nImageID: " + productImage.getImageId() + "\nPossibility: " + productImage.getPossibility();
             buffer.append(str);
             buffer.append("\n");
         }
         Log.d(TAG, "display success: " + buffer.toString());
         ScanDataActivity.start(this, productImageList);
     }
 
 }