-
Notifications
You must be signed in to change notification settings - Fork 168
/
PlateScanner.mm
89 lines (69 loc) · 2.64 KB
/
PlateScanner.mm
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
//
// PlateScanner.m
// RNOpenAlpr
//
// Created by Evan Rosenfeld on 2/24/17.
// Copyright © 2017 CarDash. All rights reserved.
//
#import "PlateScanner.h"
#import <openalpr/alpr.h>
static PlateScanner *scanner;
@implementation PlateScanner {
alpr::Alpr* delegate;
}
+ (instancetype)sharedInstance {
static dispatch_once_t onceToken;
dispatch_once(&onceToken, ^{
scanner = [[self class] new];
});
return scanner;
}
- (instancetype) init {
if (self = [super init]) {
delegate = new alpr::Alpr(
[@"us" UTF8String],
[[[NSBundle mainBundle] pathForResource:@"openalpr.conf" ofType:nil] UTF8String],
[[[NSBundle mainBundle] pathForResource:@"runtime_data" ofType:nil] UTF8String]
);
delegate->setTopN(1);
if (delegate->isLoaded() == false) {
NSLog(@"Error initializing OpenALPR library");
delegate = nil;
}
if (!delegate) self = nil;
}
return self;
}
- (void) setCountry:(NSString *)country {
delegate->setCountry([country UTF8String]);
}
- (void)scanImage:(cv::Mat &)colorImage
onSuccess:(onPlateScanSuccess)success
onFailure:(onPlateScanFailure)failure {
if (delegate->isLoaded() == false) {
NSError *error = [NSError errorWithDomain:@"alpr" code:-100
userInfo:[NSDictionary dictionaryWithObject:@"Error loading OpenALPR" forKey:NSLocalizedDescriptionKey]];
failure(error);
}
std::vector<alpr::AlprRegionOfInterest> regionsOfInterest;
alpr::AlprResults alprResults = delegate->recognize(colorImage.data, (int)colorImage.elemSize(), colorImage.cols, colorImage.rows, regionsOfInterest);
if (alprResults.plates.size() > 0) {
alpr::AlprPlateResult alprResult = alprResults.plates[0];
PlateResult *plate = [PlateResult new];
plate.plate = @(alprResult.bestPlate.characters.c_str());
plate.confidence = alprResult.bestPlate.overall_confidence;
plate.rows = colorImage.rows;
plate.cols = colorImage.cols;
NSMutableArray *pointsArr = [NSMutableArray array];
for (int j = 0; j < 4; j++) {
[pointsArr addObject:[NSValue valueWithCGPoint:CGPointMake(alprResult.plate_points[j].x, alprResult.plate_points[j].y)]];
}
plate.points = pointsArr;
// NSLog(@"\n\n\ncols: %d\nrows: %d\npoint0.x: %d\npoint0.y: %d", colorImage.cols, colorImage.rows, alprResult.plate_points[0].x,
// alprResult.plate_points[0].y);
success(plate);
} else {
success(nil);
}
}
@end