华为文档识别

 

package com.huawei.mlkit.intermediate;

import android.content.Intent;
import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.os.Bundle;
import android.view.View;
import android.widget.Button;
import android.widget.ImageView;
import android.widget.TextView;
import android.widget.Toast;

import androidx.appcompat.app.AppCompatActivity;

import com.huawei.alliance.ml.ImageSelectedAct;
import com.huawei.hmf.tasks.OnFailureListener;
import com.huawei.hmf.tasks.OnSuccessListener;
import com.huawei.hmf.tasks.Task;
import com.huawei.hms.mlsdk.MLAnalyzerFactory;
import com.huawei.hms.mlsdk.common.MLFrame;
import com.huawei.hms.mlsdk.text.MLLocalTextSetting;
import com.huawei.hms.mlsdk.text.MLText;
import com.huawei.hms.mlsdk.text.MLTextAnalyzer;

import java.util.List;
/**
 * 主Activity，请不要删除，否则影响考试成绩
 */
public class MainActivity extends AppCompatActivity {

    // 选择的图片回调 requestCode - 请不要删除，否则影响考试成绩
    private static final int REQUEST_IMG = 1000;
    // 选择的图片 - 请不要删除，否则影响考试成绩
    private ImageView mIvSelected;
    // 矫正的图片 - 请不要删除，否则影响考试成绩
    private ImageView mIvSkewResult;
    // 选择图片实体类 - 请不要删除，否则影响考试成绩
    private int mImageId;
    //选择图片按钮
    private Button mBtnImagerSelect;
    // 识别文本的内容 - 请不要删除，否则影响考试成绩
    private TextView mTvResult;
    // 图片矫正成功保存的Bitmap - 请不要删除，否则影响考试成绩
    private Bitmap mSkewBitmap;


    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        initView();
    }

    /**
     * 初始化View - 请不要删除，否则影响考试成绩
     */
    private void initView() {
        // 选择的图片 - 请不要删除，否则影响考试成绩
        mIvSelected = findViewById(R.id.selectedImg);
        // 矫正图片 - 请不要删除，否则影响考试成绩
        mIvSkewResult = findViewById(R.id.documentResultImg);
        // 文档识别的内容 - 请不要删除，否则影响考试成绩
        mTvResult = findViewById(R.id.textAnalyzerResult);
        // TODO 获取id为btn_imageSelect的Button控件mBtnImagerSelect
       mBtnImagerSelect = findViewById(R.id.btn_imageSelect);
        if (null != mBtnImagerSelect){
            mBtnImagerSelect.setOnClickListener(new View.OnClickListener() {
                @Override
                public void onClick(View v) {
                    // 实现选择图片功能 - 请不要删除，否则影响考试成绩
                    startActivityForResult(new Intent(MainActivity.this, ImageSelectedAct.class), REQUEST_IMG);
                }
            });
		}

        findViewById(R.id.btn_textAnalyzer).setOnClickListener(new View.OnClickListener() {
            @Override
            public void onClick(View v) {
                // 实现文本识别功能 - 请不要删除，否则影响考试成绩
                onTextAsync();
            }
        });
    }

    /**
     * 展示选择图片 - 请不要删除，否则影响考试成绩
     */
    private void createImageBySelectedImageId() {
        if (mImageId == 0) {
            return;
        }
        mIvSelected.setImageResource(mImageId);
        mSkewBitmap = BitmapFactory.decodeResource(getResources(), mImageId);
    }

    /**
     * 实现文本识别功能 - 请不要删除，否则影响考试成绩
     */
    private void onTextAsync() {
        if (mSkewBitmap == null) {
			Toast.makeText(this, "请选择图片", Toast.LENGTH_SHORT).show();
            return;
        }
        // TODO 根据文档矫正生成的 mSkewBitmap 对象,创建 MLFrame对象
        MLFrame frame = MLFrame.fromBitmap(mSkewBitmap);
        // TODO 根据MLLocalTextSetting对象创建文本识别 MLTextAnalyzer 对象
        MLLocalTextSetting setting = new MLLocalTextSetting.Factory()
                .setOCRMode(MLLocalTextSetting.OCR_DETECT_MODE)
                // 设置识别语种。
                .setLanguage("zh")
                .create();
        MLTextAnalyzer analyzer = MLAnalyzerFactory.getInstance().getLocalTextAnalyzer(setting);
        // TODO 调用异步识别方法 asyncAnalyseFrame
        Task<MLText> task = analyzer.asyncAnalyseFrame(frame);
        // TODO 在回调接口监听onSuccess中 获取文本识别对象MLText，并调用 initResult(text.getBlocks()) 方法展示识别的文字信息
        task.addOnSuccessListener(new OnSuccessListener<MLText>() {
            @Override
            public void onSuccess(MLText text) {
                // 识别成功处理。
                initResult(text.getBlocks());
            }
        }).addOnFailureListener(new OnFailureListener() {
            @Override
            public void onFailure(Exception e) {
                // 识别失败处理。
            }
        });

    }

    /**
     * 展示识别图片中文字- 请不要删除，否则影响考试成绩
     */
    private void initResult(List<MLText.Block> blocks) {
        StringBuilder result = new StringBuilder();
        for (MLText.Block block : blocks) {
            for (MLText.TextLine line : block.getContents()) {
                String value = line.getStringValue();
                result.append(value).append("\n");
            }
        }
        this.mTvResult.setText(result.toString());
    }

    @Override
    protected void onActivityResult(int requestCode, int resultCode, Intent data) {
        super.onActivityResult(requestCode, resultCode, data);
        // 请不要删除以及修改，否则影响考试成绩
        if (requestCode == REQUEST_IMG) {
            if (data != null && data.getExtras() != null) {
                Bundle bundle = data.getExtras();
                if (bundle.containsKey("imageId")) {
                    mImageId = bundle.getInt("imageId");
                    createImageBySelectedImageId();
                }
            }
        }
    }
}