package jp.gr.java_conf.coskx.testtessaract4;

import android.content.res.AssetManager;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.os.Bundle;
import android.util.Log;
import android.view.View;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.TextView;

import androidx.activity.EdgeToEdge;
import androidx.appcompat.app.AppCompatActivity;
import androidx.core.graphics.Insets;
import androidx.core.view.ViewCompat;
import androidx.core.view.WindowInsetsCompat;

import com.googlecode.tesseract.android.TessBaseAPI;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.Arrays;


public class MainActivity extends AppCompatActivity implements View.OnClickListener {
    TessBaseAPI tess;
    final String TAG = "MainActivity";
    private Button button1;
    private TextView textview1;
    private Bitmap bitmap1;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        EdgeToEdge.enable(this);
        setContentView(R.layout.activity_main);
        ViewCompat.setOnApplyWindowInsetsListener(findViewById(R.id.main), (v, insets) -> {
            Insets systemBars = insets.getInsets(WindowInsetsCompat.Type.systemBars());
            v.setPadding(systemBars.left, systemBars.top, systemBars.right, systemBars.bottom);
            return insets;
        });
        Log.i(TAG,"onCreate()");

        //Layout
        button1 = findViewById(R.id.button1);
        button1.setOnClickListener(this);
        textview1 = findViewById(R.id.textview1);
        bitmap1 = BitmapFactory.decodeResource(getResources(), R.drawable.sample_image);
        ImageView imageview1 = findViewById(R.id.imageview1);
        imageview1.setImageBitmap(bitmap1); //テスト用imageを表示

        // TessBaseAPIインスタンスの作成
        tess = new TessBaseAPI();
        Log.i(TAG,"onCreate() new TessBaseAPI done");
        String tessversion = tess.getVersion();
        Log.i(TAG,"onCreate() TessBaseAPI version = " + tessversion);

        //Assetsからアプリ固有の内部storageへのファイルコピー
        setReadyTrainedFile();
        Log.i(TAG,"onCreate() setReadyTrainedFile() done");

        //tesseractAPIに学習ファイル(traineddata)を与えて初期化します
        initTessWithTrainedFile();
        Log.i(TAG,"onCreate() initTessWithTrainedFile() done");
    }

    @Override
    protected  void onResume() {
        super.onResume();
        Log.i(TAG,"onResume()");
        textview1.setText("Hello, world!");
    }

    @Override
    public void onPause() {
        super.onPause();
        Log.i(TAG,"onPause()");
    }
    @Override
    public void onDestroy() {
        super.onDestroy();
        tess.recycle();
    }

    public void onClick(View view) {
        if (view == button1) {
            Log.i(TAG,"onClick() view == button1");
            String recognizedText= "no data";

            //initTessWithTrainedFile();
            //Log.i(TAG,"onClick() initTessWithTrainedFile() done");

            tess.setImage(bitmap1);
            recognizedText = tess.getUTF8Text();
            textview1.setText(recognizedText);
            Log.i(TAG,"onClick() recognizedText = [" + recognizedText +"]");
        }
    }

    public void setReadyTrainedFile() {

        AssetManager assetManager = getAssets();

        /* 確認用
        String[] fileList = null;
        try {
            fileList = assetManager.list("tessdata");
            Log.i(TAG,"setReadyTrainedFile() fileList = " + Arrays.toString(fileList));
        } catch (IOException e) {
            e.printStackTrace();
        }
        */

        String owndatapath = this.getFilesDir() + "/"; //app specific internal storage path
        Log.i(TAG,"setReadyTrainedFile() owndatapath = " + owndatapath);
        //owndatapath = /data/user/0/jp.gr.java_conf.coskx.testtessaract4/files/

        //traineddataファイルをコピーするためのディレクトリを作成
        String[] paths = new String[] { owndatapath, owndatapath + "tessdata/" };

        for (String path : paths) {
            Log.i(TAG,"setReadyTrainedFile() path = " + path);
            File dir = new File(path);
            if (!dir.exists()) {
                if (!dir.mkdirs()) {
                    Log.v(TAG, "ERROR: on Creation of directory [" + path + "]");
                    return;
                } else {
                    Log.v(TAG, "Created directory [" + path + "]");
                }
            }

        }
        //traineddataファイルをコピー　（eng.traineddataと7seg.traineddata）
        copyTessTrainedFile("eng", assetManager, owndatapath);
        copyTessTrainedFile("7seg", assetManager, owndatapath);
    }

    public void initTessWithTrainedFile() {

        String owndatapath = this.getFilesDir() + "/"; //app specific internal storage path

        /*　ファイルが思った場所にあるかどうか確認用です
        String ldpass1 = owndatapath + "tessdata/eng.traineddata";
        if ((new File(ldpass1)).exists()) {
            Log.i(TAG,"initTessWithTrainedFile() exist [" + ldpass1 + "]");
        } else {
            Log.i(TAG,"initTessWithTrainedFile() NOT exist [" + ldpass1 + "]");
        }
        String ldpass2 = owndatapath + "tessdata/7seg.traineddata";
        if ((new File(ldpass2)).exists()) {
            Log.i(TAG,"initTessWithTrainedFile() exist [" + ldpass2 + "]");
        } else {
            Log.i(TAG,"initTessWithTrainedFile() NOT exist [" + ldpass2 + "]");
        }
        */

        String tessdataPath = owndatapath; //フォルダ名"tessdata"はinit()内で追加される
        Log.i(TAG,"initTessWithTrainedFile() tessdataPath = " + tessdataPath);
        // 初期化
        // "eng+7seg"はeng.traineddataと7seg.traineddataの2つを指示しています
        if (!tess.init(tessdataPath, "eng+7seg")) {
            Log.i(TAG,"initTessWithTrainedFile() tess.init failed");
            tess.recycle();
            return;
        }
        tess.setPageSegMode(TessBaseAPI.PageSegMode.PSM_SINGLE_BLOCK);
        //ホワイトリスト指示
        tess.setVariable(TessBaseAPI.VAR_CHAR_WHITELIST, "-01:23456789.");

    }

    //assets/tessdataにあるtraineddataファイルをアプリ固有の内部ストレージにコピーする
    //ただし，すでにアプリ固有の内部ストレージ内に当該ファイルが存在する場合はコピーしない
    private void copyTessTrainedFile(String lang, AssetManager assetManager, String owndatapath) {
        if (!(new File(owndatapath + "tessdata/" + lang + ".traineddata")).exists()) {
            try {

                //AssetManager assetManager = getAssets();
                InputStream in = assetManager.open("tessdata/" + lang + ".traineddata");
                //GZIPInputStream gin = new GZIPInputStream(in);
                OutputStream out = new FileOutputStream(owndatapath
                        + "tessdata/" + lang + ".traineddata");

                // Transfer bytes from in to out
                byte[] buf = new byte[1024];
                int len;
                //while ((lenf = gin.read(buff)) > 0) {
                while ((len = in.read(buf)) > 0) {
                    out.write(buf, 0, len);
                }
                in.close();
                //gin.close();
                out.close();

                Log.v(TAG, "Copied " + lang + " traineddata");
            } catch (IOException e) {
                Log.e(TAG, "Was unable to copy " + lang + " traineddata " + e.toString());
            }
        } else {
            Log.i(TAG, " already the file exists, quit copying " + lang + " traineddata ");
        }
    }
}