// ABBYY® Mobile Capture © 2019 ABBYY Production LLC.
// ABBYY is a registered trademark or a trademark of ABBYY Software Ltd.

package com.abbyy.mobile.rtr.javascript.data;

import android.app.Application;
import android.graphics.Bitmap;

import com.abbyy.mobile.rtr.Engine;
import com.abbyy.mobile.rtr.IDataCaptureCoreAPI;
import com.abbyy.mobile.rtr.IDataCaptureCoreAPI.Warning;
import com.abbyy.mobile.rtr.javascript.JSCallback;
import com.abbyy.mobile.rtr.javascript.SharedEngine;
import com.abbyy.mobile.rtr.javascript.utils.ImageUri;
import com.abbyy.mobile.rtr.javascript.utils.SettingsParserUtils;

import org.json.JSONException;
import org.json.JSONObject;

import java.util.LinkedHashSet;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicReference;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;

public class DataCapture {

	private static final String ERROR_TAG = "DataCapture";

	private DataCapture()
	{
		// Utility
	}

	public static void extractDataSync(
		@NonNull final Application application,
		@Nullable final JSONObject jsonSettings,
		@NonNull final JSCallback callback
	)
	{
		final Settings settings;
		final Bitmap image;
		try {
			String licenseFilename = SettingsParserUtils.parseLicenseFilename( jsonSettings );
			if( !SharedEngine.initializeEngineIfNeeded( application, licenseFilename, callback ) ) {
				return;
			}

			SettingsParserUtils.checkForNullSettings( jsonSettings );
			settings = SettingsParser.parse( jsonSettings );
			image = ImageUri.fromUri( application, SettingsParserUtils.parseImageUri( jsonSettings ) );
		} catch( Exception exception ) {
			callback.onError( ERROR_TAG, "Parse error: " + exception.getMessage(), exception );
			return;
		}

		Engine engine = SharedEngine.get();
		try( final IDataCaptureCoreAPI api = engine.createDataCaptureCoreAPI() ) {
			IDataCaptureCoreAPI.DataCaptureSettings textRecognitionSettings = api.getDataCaptureSettings();
			textRecognitionSettings.setProfile( settings.profile );
			textRecognitionSettings.setTextOrientationDetectionEnabled( settings.isTextOrientationDetectionEnabled );
			textRecognitionSettings.setRecognitionLanguage( settings.languages );

			extractDataOnImage( image, api, callback );
		} catch( Exception exception ) {
			callback.onError( ERROR_TAG, "Error: " + exception.getMessage(), exception );
		}
	}

	private static void extractDataOnImage( Bitmap image, IDataCaptureCoreAPI configuredApi, final JSCallback callback )
	{
		final Set<Warning> warnings = new LinkedHashSet<>();
		// Use atomic wrappers to change final variables content
		final AtomicBoolean isRejected = new AtomicBoolean( false );
		final AtomicReference<Integer> orientationAtomicReference = new AtomicReference<>( 0 );
		IDataCaptureCoreAPI.DataField[] dataFields = configuredApi.extractDataFromImage( image,
			new IDataCaptureCoreAPI.Callback() {

				@Override
				public boolean onProgress( int progress, Warning warning )
				{
					if( warning != null ) {
						warnings.add( warning );
					}
					return false;
				}

				@Override
				public void onTextOrientationDetected( int orientation )
				{
					orientationAtomicReference.set( orientation );
				}

				@Override
				public void onError( Exception e )
				{
					isRejected.set( true );
					callback.onError( ERROR_TAG, "exception during recognition: " + e.getMessage(), e );
				}
			}
		);

		if( isRejected.get() ) {
			return;
		}

		if( dataFields == null ) {
			callback.onError( ERROR_TAG, "Recognition error: null text blocks", null );
			return;
		}
		JSONObject result;
		try {
			result = DataCaptureResult.getResult(
				dataFields,
				warnings,
				orientationAtomicReference.get()
			);
		} catch( JSONException e ) {
			callback.onError( ERROR_TAG, "Result JSON error", e );
			return;
		}
		callback.onSuccess( result );
	}

}