231 lines
12 KiB
Java
231 lines
12 KiB
Java
|
package com.google.android.gms.internal.firebase_ml;
|
||
|
|
|
||
|
|
import android.graphics.Rect;
|
||
|
|
import com.google.android.gms.common.internal.GmsLogger;
|
||
|
|
import com.google.android.gms.common.internal.Preconditions;
|
||
|
|
import com.google.android.gms.maps.model.BitmapDescriptorFactory;
|
||
|
|
import com.google.firebase.ml.vision.text.FirebaseVisionText;
|
||
|
|
import com.google.firebase.ml.vision.text.RecognizedLanguage;
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.HashSet;
|
||
|
|
import java.util.Iterator;
|
||
|
|
import java.util.List;
|
||
|
|
|
||
|
|
/* loaded from: classes2.dex */
|
||
|
|
public final class zzql {
|
||
|
|
private static final GmsLogger zzaoz = new GmsLogger("TextAnnotationConverter", "");
|
||
|
|
|
||
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
||
|
|
public static FirebaseVisionText zzb(zzjs zzjsVar, float f) {
|
||
|
|
Iterator<zzjl> it;
|
||
|
|
boolean z;
|
||
|
|
Iterator<zziu> it2;
|
||
|
|
Iterator<zzjl> it3;
|
||
|
|
boolean z2;
|
||
|
|
Iterator<zzjm> it4;
|
||
|
|
Iterator<zziu> it5;
|
||
|
|
String sb;
|
||
|
|
FirebaseVisionText.Element element;
|
||
|
|
Preconditions.checkNotNull(zzjsVar, "The input TextAnnotation can not be null");
|
||
|
|
if (zzjsVar.getPages().size() <= 0) {
|
||
|
|
zzaoz.d("TextAnnotationConverter", "Text Annotation is null, return null");
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
boolean z3 = true;
|
||
|
|
if (zzjsVar.getPages().size() > 1) {
|
||
|
|
zzaoz.d("TextAnnotationConverter", "Text Annotation has more than one page, which should not happen");
|
||
|
|
}
|
||
|
|
ArrayList arrayList = new ArrayList();
|
||
|
|
Iterator<zzjl> it6 = zzjsVar.getPages().iterator();
|
||
|
|
FirebaseVisionText.TextBlock textBlock = null;
|
||
|
|
while (it6.hasNext()) {
|
||
|
|
Iterator<zziu> it7 = it6.next().getBlocks().iterator();
|
||
|
|
while (it7.hasNext()) {
|
||
|
|
zziu next = it7.next();
|
||
|
|
Preconditions.checkNotNull(next, "Input block can not be null");
|
||
|
|
ArrayList arrayList2 = new ArrayList();
|
||
|
|
if (next.getParagraphs() == null) {
|
||
|
|
it = it6;
|
||
|
|
z = z3;
|
||
|
|
it2 = it7;
|
||
|
|
} else {
|
||
|
|
Iterator<zzjm> it8 = next.getParagraphs().iterator();
|
||
|
|
while (it8.hasNext()) {
|
||
|
|
zzjm next2 = it8.next();
|
||
|
|
if (next2 != null) {
|
||
|
|
Preconditions.checkNotNull(next2, "Input Paragraph can not be null");
|
||
|
|
ArrayList arrayList3 = new ArrayList();
|
||
|
|
ArrayList arrayList4 = new ArrayList();
|
||
|
|
HashSet<RecognizedLanguage> hashSet = new HashSet();
|
||
|
|
StringBuilder sb2 = new StringBuilder();
|
||
|
|
int i = 0;
|
||
|
|
float f2 = BitmapDescriptorFactory.HUE_RED;
|
||
|
|
while (i < next2.getWords().size()) {
|
||
|
|
zzjx zzjxVar = next2.getWords().get(i);
|
||
|
|
if (zzjxVar != null) {
|
||
|
|
Preconditions.checkNotNull(zzjxVar, "Input Word can not be null");
|
||
|
|
Rect zza = zzpm.zza(zzjxVar.zzhq(), f);
|
||
|
|
it3 = it6;
|
||
|
|
List<RecognizedLanguage> zze = zze(zzjxVar.zzhr());
|
||
|
|
Preconditions.checkNotNull(zzjxVar, "Input Word can not be null");
|
||
|
|
String str = "";
|
||
|
|
if (zzjxVar.getSymbols() == null) {
|
||
|
|
it4 = it8;
|
||
|
|
sb = "";
|
||
|
|
} else {
|
||
|
|
StringBuilder sb3 = new StringBuilder();
|
||
|
|
Iterator<zzjr> it9 = zzjxVar.getSymbols().iterator();
|
||
|
|
while (it9.hasNext()) {
|
||
|
|
sb3.append(it9.next().getText());
|
||
|
|
it8 = it8;
|
||
|
|
}
|
||
|
|
it4 = it8;
|
||
|
|
sb = sb3.toString();
|
||
|
|
}
|
||
|
|
if (sb.isEmpty()) {
|
||
|
|
it5 = it7;
|
||
|
|
element = null;
|
||
|
|
} else {
|
||
|
|
it5 = it7;
|
||
|
|
element = new FirebaseVisionText.Element(sb, zza, zze, zzjxVar.getConfidence());
|
||
|
|
}
|
||
|
|
if (element != null) {
|
||
|
|
arrayList4.add(element);
|
||
|
|
float zza2 = f2 + zzpm.zza(element.getConfidence());
|
||
|
|
hashSet.addAll(element.getRecognizedLanguages());
|
||
|
|
sb2.append(element.getText());
|
||
|
|
Preconditions.checkNotNull(zzjxVar, "Input word can not be null");
|
||
|
|
String zza3 = zza(zzjxVar);
|
||
|
|
if (zza3 != null) {
|
||
|
|
if (zza3.equals("SPACE") || zza3.equals("SURE_SPACE")) {
|
||
|
|
str = " ";
|
||
|
|
} else if (zza3.equals("HYPHEN")) {
|
||
|
|
str = "-";
|
||
|
|
}
|
||
|
|
}
|
||
|
|
sb2.append(str);
|
||
|
|
Preconditions.checkNotNull(zzjxVar, "Input word can not be null");
|
||
|
|
String zza4 = zza(zzjxVar);
|
||
|
|
if (zza4 == null || !(zza4.equals("EOL_SURE_SPACE") || zza4.equals("LINE_BREAK") || zza4.equals("HYPHEN"))) {
|
||
|
|
z2 = true;
|
||
|
|
if (i != next2.getWords().size() - 1) {
|
||
|
|
f2 = zza2;
|
||
|
|
i++;
|
||
|
|
z3 = z2;
|
||
|
|
it6 = it3;
|
||
|
|
it8 = it4;
|
||
|
|
it7 = it5;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
z2 = true;
|
||
|
|
}
|
||
|
|
Preconditions.checkNotNull(arrayList4, "Input elements can not be null");
|
||
|
|
int size = arrayList4.size();
|
||
|
|
int i2 = 0;
|
||
|
|
Rect rect = null;
|
||
|
|
while (i2 < size) {
|
||
|
|
Object obj = arrayList4.get(i2);
|
||
|
|
i2++;
|
||
|
|
FirebaseVisionText.Element element2 = (FirebaseVisionText.Element) obj;
|
||
|
|
if (element2.getBoundingBox() != null) {
|
||
|
|
if (rect == null) {
|
||
|
|
rect = new Rect();
|
||
|
|
}
|
||
|
|
Rect rect2 = rect;
|
||
|
|
rect2.union(element2.getBoundingBox());
|
||
|
|
rect = rect2;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
String sb4 = sb2.toString();
|
||
|
|
ArrayList arrayList5 = new ArrayList();
|
||
|
|
for (RecognizedLanguage recognizedLanguage : hashSet) {
|
||
|
|
if (recognizedLanguage != null && recognizedLanguage.getLanguageCode() != null && !recognizedLanguage.getLanguageCode().isEmpty()) {
|
||
|
|
arrayList5.add(recognizedLanguage);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
arrayList3.add(new FirebaseVisionText.Line(sb4, rect, arrayList5, arrayList4, Float.compare(zza2, BitmapDescriptorFactory.HUE_RED) > 0 ? Float.valueOf(zza2 / arrayList4.size()) : null));
|
||
|
|
ArrayList arrayList6 = new ArrayList();
|
||
|
|
hashSet.clear();
|
||
|
|
arrayList4 = arrayList6;
|
||
|
|
sb2 = new StringBuilder();
|
||
|
|
f2 = 0.0f;
|
||
|
|
i++;
|
||
|
|
z3 = z2;
|
||
|
|
it6 = it3;
|
||
|
|
it8 = it4;
|
||
|
|
it7 = it5;
|
||
|
|
} else {
|
||
|
|
z2 = true;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
it3 = it6;
|
||
|
|
z2 = z3;
|
||
|
|
it4 = it8;
|
||
|
|
it5 = it7;
|
||
|
|
}
|
||
|
|
i++;
|
||
|
|
z3 = z2;
|
||
|
|
it6 = it3;
|
||
|
|
it8 = it4;
|
||
|
|
it7 = it5;
|
||
|
|
}
|
||
|
|
arrayList2.addAll(arrayList3);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
it = it6;
|
||
|
|
z = z3;
|
||
|
|
it2 = it7;
|
||
|
|
if (arrayList2.isEmpty()) {
|
||
|
|
textBlock = null;
|
||
|
|
} else {
|
||
|
|
StringBuilder sb5 = new StringBuilder();
|
||
|
|
int size2 = arrayList2.size();
|
||
|
|
int i3 = 0;
|
||
|
|
while (i3 < size2) {
|
||
|
|
Object obj2 = arrayList2.get(i3);
|
||
|
|
i3++;
|
||
|
|
sb5.append(((FirebaseVisionText.Line) obj2).getText());
|
||
|
|
sb5.append("\n");
|
||
|
|
}
|
||
|
|
textBlock = new FirebaseVisionText.TextBlock(sb5.toString(), zzpm.zza(next.zzhq(), f), zze(next.zzhr()), arrayList2, next.getConfidence());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (textBlock != null) {
|
||
|
|
arrayList.add(textBlock);
|
||
|
|
}
|
||
|
|
textBlock = null;
|
||
|
|
z3 = z;
|
||
|
|
it6 = it;
|
||
|
|
it7 = it2;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return new FirebaseVisionText(zzjsVar.getText(), arrayList);
|
||
|
|
}
|
||
|
|
|
||
|
|
public static List<RecognizedLanguage> zze(zzjt zzjtVar) {
|
||
|
|
ArrayList arrayList = new ArrayList();
|
||
|
|
if (zzjtVar != null && zzjtVar.zzia() != null) {
|
||
|
|
Iterator<zziz> it = zzjtVar.zzia().iterator();
|
||
|
|
while (it.hasNext()) {
|
||
|
|
RecognizedLanguage zza = RecognizedLanguage.zza(it.next());
|
||
|
|
if (zza != null) {
|
||
|
|
arrayList.add(zza);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return arrayList;
|
||
|
|
}
|
||
|
|
|
||
|
|
private static String zza(zzjx zzjxVar) {
|
||
|
|
Preconditions.checkNotNull(zzjxVar, "Input Word can not be null");
|
||
|
|
if (zzjxVar.getSymbols() == null || zzjxVar.getSymbols().size() <= 0) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
zzjr zzjrVar = zzjxVar.getSymbols().get(zzjxVar.getSymbols().size() - 1);
|
||
|
|
if (zzjrVar.zzhr() == null || zzjrVar.zzhr().zzhz() == null) {
|
||
|
|
return null;
|
||
|
|
}
|
||
|
|
return zzjxVar.getSymbols().get(zzjxVar.getSymbols().size() - 1).zzhr().zzhz().getType();
|
||
|
|
}
|
||
|
|
}
|