xiaoyan159/OneMapQS
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

编译工程

Browse Source
This commit is contained in:
squallzhjch 2023-03-22 10:27:37 +08:00
parent f69d668557
commit 69dcc5e8ea
1 changed files with 221 additions and 221 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

442
collect-library/src/main/java/com/navinfo/collect/library/data/handler/DataCameraHandler.kt
View File

@ -1,223 +1,223 @@
package com.navinfo.collect.library.data.handler //package com.navinfo.collect.library.data.handler
//
import android.content.Context //import android.content.Context
import android.graphics.Bitmap //import android.graphics.Bitmap
import android.graphics.BitmapFactory //import android.graphics.BitmapFactory
import android.media.ExifInterface //import android.media.ExifInterface
import android.util.Log //import android.util.Log
import com.baidu.ai.edge.ui.util.ImageUtil //import com.baidu.ai.edge.ui.util.ImageUtil
import com.baidu.ai.edge.ui.view.model.OcrViewResultModel //import com.baidu.ai.edge.ui.view.model.OcrViewResultModel
import com.navinfo.collect.FlutterBaseActivity //import com.navinfo.collect.FlutterBaseActivity
import com.navinfo.collect.library.data.dao.impl.MapLifeDataBase //import com.navinfo.collect.library.data.dao.impl.MapLifeDataBase
import com.navinfo.ocr.OCRManager //import com.navinfo.ocr.OCRManager
import kotlinx.coroutines.Dispatchers //import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.launch //import kotlinx.coroutines.launch
import kotlinx.coroutines.withContext //import kotlinx.coroutines.withContext
import java.io.* //import java.io.*
import java.util.* //import java.util.*
//
interface OnOCRBatchListener { //interface OnOCRBatchListener {
fun onProgress(total: Int, current: Int) // fun onProgress(total: Int, current: Int)
suspend fun onResult(list: List<Map<String, Any>>) // suspend fun onResult(list: List<Map<String, Any>>)
} //}
//
open class DataCameraHandler( //open class DataCameraHandler(
context: Context, // context: Context,
activity: FlutterBaseActivity, // activity: FlutterBaseActivity,
dataBase: MapLifeDataBase // dataBase: MapLifeDataBase
) : //) :
BaseDataHandler(context, dataBase) { // BaseDataHandler(context, dataBase) {
//
private val mActivity: FlutterBaseActivity = activity // private val mActivity: FlutterBaseActivity = activity
//
init { // init {
OCRManager.instance.init(activity) // OCRManager.instance.init(activity)
} // }
//
fun openCamera() { // fun openCamera() {
OCRManager.instance.openCamera(mActivity) // OCRManager.instance.openCamera(mActivity)
} // }
//
/** // /**
* 批量OCR识别 // * 批量OCR识别
*/ // */
fun ocrBatch(filePath: String, listener: OnOCRBatchListener) { // fun ocrBatch(filePath: String, listener: OnOCRBatchListener) {
mActivity.launch(Dispatchers.IO) { // mActivity.launch(Dispatchers.IO) {
Log.e("jingo", "OCRManager 线程开始 ${Thread.currentThread().name}") // Log.e("jingo", "OCRManager 线程开始 ${Thread.currentThread().name}")
val file = File(filePath) // val file = File(filePath)
val resultList = mutableListOf<Map<String, Any>>() // val resultList = mutableListOf<Map<String, Any>>()
if (file.isDirectory) { // if (file.isDirectory) {
val fileList = file.listFiles() // val fileList = file.listFiles()
val bitmapList = mutableListOf<String>() // val bitmapList = mutableListOf<String>()
for (item in fileList!!) { // for (item in fileList!!) {
if (item.isFile) { // if (item.isFile) {
if (checkIsImageFile(item.name)) { // if (checkIsImageFile(item.name)) {
bitmapList.add(item.path) // bitmapList.add(item.path)
} // }
} // }
} // }
val bfw: BufferedWriter // val bfw: BufferedWriter
val csvFile = File("$filePath/ocr.csv") // val csvFile = File("$filePath/ocr.csv")
val out: FileOutputStream // val out: FileOutputStream
val osw: OutputStreamWriter // val osw: OutputStreamWriter
try { // try {
out = FileOutputStream(csvFile) // out = FileOutputStream(csvFile)
//用excel打开，中文会乱码，所以用GBK编译 // //用excel打开，中文会乱码，所以用GBK编译
osw = OutputStreamWriter(out, "GBK") // osw = OutputStreamWriter(out, "GBK")
bfw = BufferedWriter(osw) // bfw = BufferedWriter(osw)
//第一行表头数据 // //第一行表头数据
bfw.write("图片路径和名称,") // bfw.write("图片路径和名称,")
bfw.write("识别结果序号,") // bfw.write("识别结果序号,")
bfw.write("识别结果内容,") // bfw.write("识别结果内容,")
bfw.write("置信度,") // bfw.write("置信度,")
bfw.write("识别面积,") // bfw.write("识别面积,")
bfw.write("图片大小,") // bfw.write("图片大小,")
bfw.write("识别区域,") // bfw.write("识别区域,")
//写好表头后换行 // //写好表头后换行
bfw.newLine() // bfw.newLine()
for (i in 0 until bitmapList.size) { // for (i in 0 until bitmapList.size) {
val path = bitmapList[i] // val path = bitmapList[i]
//
val bitmap: Bitmap? = readFile(path) // val bitmap: Bitmap? = readFile(path)
var exif = ExifInterface(path) // var exif = ExifInterface(path)
val exifRotation = exif.getAttributeInt( // val exifRotation = exif.getAttributeInt(
ExifInterface.TAG_ORIENTATION, // ExifInterface.TAG_ORIENTATION,
ExifInterface.ORIENTATION_NORMAL // ExifInterface.ORIENTATION_NORMAL
) // )
val rotation = ImageUtil.exifToDegrees(exifRotation) // val rotation = ImageUtil.exifToDegrees(exifRotation)
val rotateBitmap = ImageUtil.createRotateBitmap(bitmap, rotation) // val rotateBitmap = ImageUtil.createRotateBitmap(bitmap, rotation)
val list = OCRManager.instance.ocr(rotateBitmap) // val list = OCRManager.instance.ocr(rotateBitmap)
//
//
// val list = ocrBitmap(path) //// val list = ocrBitmap(path)
if (list != null) { // if (list != null) {
for (o in list) { // for (o in list) {
bfw.write("$path,") // bfw.write("$path,")
bfw.write("${o.index},") // bfw.write("${o.index},")
bfw.write("${o.name},") // bfw.write("${o.name},")
bfw.write("${o.confidence.toString()},") // bfw.write("${o.confidence.toString()},")
val pointList = o.bounds // val pointList = o.bounds
bfw.write("${(pointList[3].y - pointList[0].y) * (pointList[2].x - pointList[0].x)},") // bfw.write("${(pointList[3].y - pointList[0].y) * (pointList[2].x - pointList[0].x)},")
bfw.write("${rotateBitmap.width} ${rotateBitmap.height},") // bfw.write("${rotateBitmap.width} ${rotateBitmap.height},")
bfw.write("${o.bounds[0].x} ${o.bounds[0].y};${o.bounds[1].x} ${o.bounds[1].y};${o.bounds[2].x} ${o.bounds[2].y};${o.bounds[3].x} ${o.bounds[3].y},") // bfw.write("${o.bounds[0].x} ${o.bounds[0].y};${o.bounds[1].x} ${o.bounds[1].y};${o.bounds[2].x} ${o.bounds[2].y};${o.bounds[3].x} ${o.bounds[3].y},")
bfw.newLine() // bfw.newLine()
} // }
bfw.newLine() // bfw.newLine()
withContext(Dispatchers.Main) { // withContext(Dispatchers.Main) {
listener.onProgress(bitmapList.size, i) // listener.onProgress(bitmapList.size, i)
} // }
val m1 = mutableMapOf<String, Any>(); // val m1 = mutableMapOf<String, Any>();
m1["data"] = list; // m1["data"] = list;
m1["width"] = rotateBitmap.width // m1["width"] = rotateBitmap.width
m1["height"] = rotateBitmap.height // m1["height"] = rotateBitmap.height
m1["path"] = path // m1["path"] = path
resultList.add(m1) // resultList.add(m1)
} // }
rotateBitmap.recycle() // rotateBitmap.recycle()
} // }
//
//将缓存数据写入文件 // //将缓存数据写入文件
bfw.flush() // bfw.flush()
//释放缓存 // //释放缓存
bfw.close() // bfw.close()
osw.close() // osw.close()
out.close() // out.close()
} catch (e: Throwable) { // } catch (e: Throwable) {
//
} // }
//将缓存数据写入文件 // //将缓存数据写入文件
//
withContext(Dispatchers.Main) { // withContext(Dispatchers.Main) {
Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}") // Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}")
listener.onResult(resultList) // listener.onResult(resultList)
} // }
} else if (file.isFile && checkIsImageFile(file.name)) { // } else if (file.isFile && checkIsImageFile(file.name)) {
val list = ocrBitmap(filePath) // val list = ocrBitmap(filePath)
if (list != null) { // if (list != null) {
withContext(Dispatchers.Main) { // withContext(Dispatchers.Main) {
Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}") // Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}")
listener.onProgress(1, 1) // listener.onProgress(1, 1)
} // }
val m = mutableMapOf<String, List<OcrViewResultModel>>() // val m = mutableMapOf<String, List<OcrViewResultModel>>()
m[file.name] = list // m[file.name] = list
resultList.add(m) // resultList.add(m)
} // }
withContext(Dispatchers.Main) { // withContext(Dispatchers.Main) {
Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}") // Log.e("jingo", "OCRManager 线程名称2 ${Thread.currentThread().name}")
listener.onResult(resultList) // listener.onResult(resultList)
} // }
}
}
}
private fun ocrBitmap(path: String): List<OcrViewResultModel>? {
try {
val bitmap: Bitmap? = readFile(path)
var exif = ExifInterface(path)
val exifRotation = exif.getAttributeInt(
ExifInterface.TAG_ORIENTATION,
ExifInterface.ORIENTATION_NORMAL
)
val rotation = ImageUtil.exifToDegrees(exifRotation)
val rotateBitmap = ImageUtil.createRotateBitmap(bitmap, rotation)
val res = OCRManager.instance.ocr(rotateBitmap)
rotateBitmap.recycle()
return res
} catch (e: IOException) {
Log.e("jingo", "图像识别，获取图像信息失败 ${e.printStackTrace()}")
}
return null
}
/**
* 检查是不是bitmap文件
*/
private fun checkIsImageFile(fName: String): Boolean {
val isImageFile: Boolean
// 获取扩展名
val fileEnd = fName.substring(
fName.lastIndexOf(".") + 1,
fName.length
).lowercase(Locale.getDefault())
isImageFile =
fileEnd == "jpg" || fileEnd == "png" || fileEnd == "webp" || fileEnd == "jpeg" || fileEnd == "bmp"
return isImageFile
}
/**
* 读取bitmap文件
*/
private fun readFile(path: String): Bitmap? {
var stream: FileInputStream? = null
try {
stream = FileInputStream(path)
return BitmapFactory.decodeStream(stream)
} catch (e: FileNotFoundException) {
e.printStackTrace()
} finally {
if (stream != null) {
try {
stream.close()
} catch (e: IOException) {
e.printStackTrace()
}
}
}
return null
}
// private fun onOcrBitmap(
// bitmap, confidence,
// object: OcrListener {
// override fun onResult(models: List<BasePolygonResultModel>) {
// if (models == null) {
// listener.onResult(null)
// return
// } // }
// ocrResultModelCache = models //
// listener.onResult(models)
// } // }
// }) // }
//
} // private fun ocrBitmap(path: String): List<OcrViewResultModel>? {
// try {
// val bitmap: Bitmap? = readFile(path)
// var exif = ExifInterface(path)
// val exifRotation = exif.getAttributeInt(
// ExifInterface.TAG_ORIENTATION,
// ExifInterface.ORIENTATION_NORMAL
// )
// val rotation = ImageUtil.exifToDegrees(exifRotation)
// val rotateBitmap = ImageUtil.createRotateBitmap(bitmap, rotation)
// val res = OCRManager.instance.ocr(rotateBitmap)
// rotateBitmap.recycle()
// return res
// } catch (e: IOException) {
// Log.e("jingo", "图像识别，获取图像信息失败 ${e.printStackTrace()}")
// }
// return null
// }
//
// /**
// * 检查是不是bitmap文件
// */
// private fun checkIsImageFile(fName: String): Boolean {
// val isImageFile: Boolean
// // 获取扩展名
// val fileEnd = fName.substring(
// fName.lastIndexOf(".") + 1,
// fName.length
// ).lowercase(Locale.getDefault())
// isImageFile =
// fileEnd == "jpg" || fileEnd == "png" || fileEnd == "webp" || fileEnd == "jpeg" || fileEnd == "bmp"
// return isImageFile
// }
//
// /**
// * 读取bitmap文件
// */
// private fun readFile(path: String): Bitmap? {
// var stream: FileInputStream? = null
// try {
// stream = FileInputStream(path)
// return BitmapFactory.decodeStream(stream)
// } catch (e: FileNotFoundException) {
// e.printStackTrace()
// } finally {
// if (stream != null) {
// try {
// stream.close()
// } catch (e: IOException) {
// e.printStackTrace()
// }
// }
// }
// return null
// }
//
//// private fun onOcrBitmap(
//// bitmap, confidence,
//// object: OcrListener {
//// override fun onResult(models: List<BasePolygonResultModel>) {
//// if (models == null) {
//// listener.onResult(null)
//// return
//// }
//// ocrResultModelCache = models
//// listener.onResult(models)
//// }
//// })
//
//}