97 lines
3.2 KiB
Plaintext
97 lines
3.2 KiB
Plaintext
//
|
|
// KeyPoint.m
|
|
//
|
|
// Created by Giles Payne on 2019/12/25.
|
|
//
|
|
|
|
#import "KeyPoint.h"
|
|
#import "Point2f.h"
|
|
|
|
@implementation KeyPoint {
|
|
cv::KeyPoint native;
|
|
}
|
|
|
|
- (cv::KeyPoint&)nativeRef {
|
|
native.pt.x = self.pt.x;
|
|
native.pt.y = self.pt.y;
|
|
native.size = self.size;
|
|
native.angle = self.angle;
|
|
native.response = self.response;
|
|
native.octave = self.octave;
|
|
native.class_id = self.classId;
|
|
return native;
|
|
}
|
|
|
|
- (instancetype)init {
|
|
return [self initWithX:0 y:0 size:0];
|
|
}
|
|
|
|
- (instancetype)initWithX:(float)x y:(float)y size:(float)size angle:(float)angle response:(float)response octave:(int)octave classId:(int)classId {
|
|
self = [super init];
|
|
if (self != nil) {
|
|
self.pt = [[Point2f alloc] initWithX:x y:y];
|
|
self.size = size;
|
|
self.angle = angle;
|
|
self.response = response;
|
|
self.octave = octave;
|
|
self.classId = classId;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithX:(float)x y:(float)y size:(float)size angle:(float)angle response:(float)response octave:(int)octave {
|
|
return [self initWithX:x y:y size:size angle:angle response:response octave:octave classId:-1];
|
|
}
|
|
|
|
- (instancetype)initWithX:(float)x y:(float)y size:(float)size angle:(float)angle response:(float)response {
|
|
return [self initWithX:x y:y size:size angle:angle response:response octave:0];
|
|
}
|
|
|
|
- (instancetype)initWithX:(float)x y:(float)y size:(float)size angle:(float)angle {
|
|
return [self initWithX:x y:y size:size angle:angle response:0];
|
|
}
|
|
|
|
- (instancetype)initWithX:(float)x y:(float)y size:(float)size {
|
|
return [self initWithX:x y:y size:size angle:-1];
|
|
}
|
|
|
|
+ (instancetype)fromNative:(cv::KeyPoint&)keyPoint {
|
|
return [[KeyPoint alloc] initWithX:keyPoint.pt.x y:keyPoint.pt.y size:keyPoint.size angle:keyPoint.angle response:keyPoint.response octave:keyPoint.octave classId:keyPoint.class_id];
|
|
}
|
|
|
|
- (KeyPoint*)clone {
|
|
return [[KeyPoint alloc] initWithX:self.pt.x y:self.pt.y size:self.size angle:self.angle response:self.response octave:self.octave classId:self.classId];
|
|
}
|
|
|
|
- (BOOL)isEqual:(id)other {
|
|
if (other == self) {
|
|
return YES;
|
|
} else if (![other isKindOfClass:[KeyPoint class]]) {
|
|
return NO;
|
|
} else {
|
|
KeyPoint* keyPoint = (KeyPoint*)other;
|
|
return [self.pt isEqual:keyPoint.pt] && self.size == keyPoint.size && self.angle == keyPoint.angle && self.response == keyPoint.response && self.octave == keyPoint.octave && self.classId == keyPoint.classId;
|
|
}
|
|
}
|
|
|
|
#define FLOAT_TO_BITS(x) ((Cv32suf){ .f = x }).i
|
|
|
|
- (NSUInteger)hash {
|
|
int prime = 31;
|
|
uint32_t result = 1;
|
|
result = prime * result + FLOAT_TO_BITS(self.pt.x);
|
|
result = prime * result + FLOAT_TO_BITS(self.pt.y);
|
|
result = prime * result + FLOAT_TO_BITS(self.size);
|
|
result = prime * result + FLOAT_TO_BITS(self.angle);
|
|
result = prime * result + FLOAT_TO_BITS(self.response);
|
|
result = prime * result + self.octave;
|
|
result = prime * result + self.classId;
|
|
return result;
|
|
}
|
|
|
|
- (NSString*)description {
|
|
return [NSString stringWithFormat:@"KeyPoint { pt: %@, size: %f, angle: %f, response: %f, octave: %d, classId: %d}", self.pt.description, self.size, self.angle, self.response, self.octave, self.classId];
|
|
}
|
|
|
|
@end
|