chamomind/jws-frontend
chamomind/jws-frontend
7
0
Fork 0
Code Issues Projects Releases Packages 1 Wiki Activity Actions

refactor: update ocr

Browse source
This commit is contained in:
Methapon Metanipat 2024-10-02 16:50:51 +07:00
parent 1ea507583d
commit 5798c6f56b
2 changed files with 22 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

31
src/utils/mrz.ts
View file

@ -139,6 +139,13 @@ function mrzCleanResult(obj: Record<string, string>) {
return obj; return obj;
} }
function mrzFieldExtract(expression: RegExp, line: string) {
if (expression.test(line)) {
return expression.exec(line)?.groups || {};
}
return {};
}
export function checkSum(data: string) { export function checkSum(data: string) {
if (!/[0-9A-Z<]/.test(data)) return null; if (!/[0-9A-Z<]/.test(data)) return null;
@ -159,11 +166,9 @@ export function checkSum(data: string) {
export function parseType1(mrz: MRZ) { export function parseType1(mrz: MRZ) {
const result: Record<string, string> = {}; const result: Record<string, string> = {};
mrz.zone.forEach((line, i) => { mrz.zone.forEach((line, i) =>
if (MRZ_TD_1[i].test(line)) { Object.assign(result, mrzFieldExtract(MRZ_TD_1[i], line)),
Object.assign(result, MRZ_TD_1[i].exec(line)?.groups); );
}
});
return { mrz, result: mrzCleanResult(result) }; return { mrz, result: mrzCleanResult(result) };
} }
@ -171,11 +176,9 @@ export function parseType1(mrz: MRZ) {
export function parseType2(mrz: MRZ) { export function parseType2(mrz: MRZ) {
const result: Record<string, string> = {}; const result: Record<string, string> = {};
mrz.zone.forEach((line, i) => { mrz.zone.forEach((line, i) =>
if (MRZ_TD_2[i].test(line)) { Object.assign(result, mrzFieldExtract(MRZ_TD_2[i], line)),
Object.assign(result, MRZ_TD_2[i].exec(line)?.groups); );
}
});
return { mrz, result: mrzCleanResult(result) }; return { mrz, result: mrzCleanResult(result) };
} }
@ -183,11 +186,9 @@ export function parseType2(mrz: MRZ) {
export function parseType3(mrz: MRZ) { export function parseType3(mrz: MRZ) {
const result: Record<string, string> = {}; const result: Record<string, string> = {};
mrz.zone.forEach((line, i) => { mrz.zone.forEach((line, i) =>
if (MRZ_TD_3[i].test(line)) { Object.assign(result, mrzFieldExtract(MRZ_TD_3[i], line)),
Object.assign(result, MRZ_TD_3[i].exec(line)?.groups); );
}
});
return { mrz, result: mrzCleanResult(result) }; return { mrz, result: mrzCleanResult(result) };
} }

12
src/utils/ocr.ts
View file

@ -26,13 +26,13 @@ export function parseResultMRZ(result: RecognizeResult) {
.split(/[\s\r\n]+/) .split(/[\s\r\n]+/)
.filter((v) => /[A-Z0-9<]{30,}/.test(v)); .filter((v) => /[A-Z0-9<]{30,}/.test(v));
if (zone.length === 3 && zone[0].length === 30) { if (zone.length === 2 && zone[0].length >= 44) {
return parseMRZ({ type: 'TD1', zone });
} else if (zone.length === 2 && zone[0].length === 36) {
return parseMRZ({ type: 'TD2', zone });
} else if (zone.length === 2 && zone[0].length === 44) {
return parseMRZ({ type: 'TD3', zone }); return parseMRZ({ type: 'TD3', zone });
} else if (zone.length === 2 && zone[0].length >= 36) {
return parseMRZ({ type: 'TD2', zone });
} else if (zone.length === 3 && zone[0].length >= 30) {
return parseMRZ({ type: 'TD1', zone });
} else { } else {
return null; return { mrz: { type: 'UNKNOWN', zone }, result: null };
} }
} }