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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
//
// RotatedRect.m
//
// Created by Giles Payne on 2019/12/26.
//
#import "RotatedRect.h"
#import "Point2f.h"
#import "Size2f.h"
#import "Rect2f.h"
#include <math.h>
@implementation RotatedRect {
cv::RotatedRect native;
}
- (cv::RotatedRect&)nativeRef {
native.center.x = self.center.x;
native.center.y = self.center.y;
native.size.width = self.size.width;
native.size.height = self.size.height;
native.angle = self.angle;
return native;
}
- (instancetype)init {
return [self initWithCenter:[Point2f new] size:[Size2f new] angle:0.0];
}
- (instancetype)initWithCenter:(Point2f*)center size:(Size2f*)size angle:(double)angle {
self = [super init];
if (self) {
self.center = center;
self.size = size;
self.angle = angle;
}
return self;
}
- (instancetype)initWithVals:(NSArray<NSNumber*>*)vals {
self = [self init];
if (self) {
[self set:vals];
}
return self;
}
+ (instancetype)fromNative:(cv::RotatedRect&)rotatedRect {
return [[RotatedRect alloc] initWithCenter:[Point2f fromNative:rotatedRect.center] size:[Size2f fromNative:rotatedRect.size] angle:rotatedRect.angle];
}
- (void)set:(NSArray<NSNumber*>*)vals {
self.center.x = (vals != nil && vals.count > 0) ? vals[0].floatValue : 0.0;
self.center.y = (vals != nil && vals.count > 1) ? vals[1].floatValue : 0.0;
self.size.width = (vals != nil && vals.count > 2) ? vals[2].floatValue : 0.0;
self.size.height = (vals != nil && vals.count > 3) ? vals[3].floatValue : 0.0;
self.angle = (vals != nil && vals.count > 4) ? vals[4].doubleValue : 0.0;
}
- (NSArray<Point2f*>*)points {
double angleRadians = self.angle * M_PI / 180.0;
double b = cos(angleRadians) * 0.5;
double a = sin(angleRadians) * 0.5f;
Point2f* p0 = [[Point2f alloc] initWithX:self.center.x - a * self.size.height - b * self.size.width y:self.center.y + b * self.size.height - a * self.size.width];
Point2f* p1 = [[Point2f alloc] initWithX:self.center.x + a * self.size.height - b * self.size.width y:self.center.y - b * self.size.height - a * self.size.width];
Point2f* p2 = [[Point2f alloc] initWithX:2 * self.center.x - p0.x y:2 * self.center.y - p0.y];
Point2f* p3 = [[Point2f alloc] initWithX:2 * self.center.x - p1.x y:2 * self.center.y - p1.y];
return [NSArray arrayWithObjects:p0, p1, p2, p3, nil];
}
- (Rect2f*)boundingRect {
NSArray<Point2f*>* pts = [self points];
Rect2f* rect = [[Rect2f alloc] initWithX:(int)floor(MIN(MIN(MIN(pts[0].x, pts[1].x), pts[2].x), pts[3].x)) y:(int)floor(MIN(MIN(MIN(pts[0].y, pts[1].y), pts[2].y), pts[3].y)) width:(int)ceil(MAX(MAX(MAX(pts[0].x, pts[1].x), pts[2].x), pts[3].x)) height:(int)ceil(MAX(MAX(MAX(pts[0].y, pts[1].y), pts[2].y), pts[3].y))];
rect.width -= rect.x - 1;
rect.height -= rect.y - 1;
return rect;
}
- (RotatedRect*)clone {
return [[RotatedRect alloc] initWithCenter:[self.center clone] size:[self.size clone] angle:self.angle];
}
- (BOOL)isEqual:(id)other {
if (other == self) {
return YES;
} else if (![other isKindOfClass:[RotatedRect class]]) {
return NO;
} else {
RotatedRect* rect = (RotatedRect*)other;
return [self.center isEqual:rect.center] && [self.size isEqual:rect.size] && self.angle == rect.angle;
}
}
#define FLOAT_TO_BITS(x) ((Cv32suf){ .f = x }).i
#define DOUBLE_TO_BITS(x) ((Cv64suf){ .f = x }).i
- (NSUInteger)hash {
int prime = 31;
uint32_t result = 1;
result = prime * result + FLOAT_TO_BITS(self.center.x);
result = prime * result + FLOAT_TO_BITS(self.center.y);
result = prime * result + FLOAT_TO_BITS(self.size.width);
result = prime * result + FLOAT_TO_BITS(self.size.height);
int64_t temp = DOUBLE_TO_BITS(self.angle);
result = prime * result + (int32_t) (temp ^ (temp >> 32));
return result;
}
- (NSString*)description {
return [NSString stringWithFormat:@"RotatedRect {%@,%@,%lf}", self.center.description, self.size.description, self.angle];
}
@end
|
