Add mosaic process & time recorder

Author: y310

Classes/OpenCVTestViewController.h

@@ -1,5 +1,6 @@
 #import <UIKit/UIKit.h>
 #import <AudioToolbox/AudioToolbox.h>
+#import "TimeRecorder.h"
 
 @interface UIProgressIndicator : UIActivityIndicatorView {
 }
@@ -49,12 +50,14 @@ typedef enum {
 	OpenCVTestViewControllerActionSheetAction actionSheetAction;
 	UIProgressHUD *progressHUD;
 	SystemSoundID alertSoundID;
+	TimeRecorder *timeRecorder;
 }
 
 - (IBAction)loadImage:(id)sender;
 - (IBAction)saveImage:(id)sender;
 - (IBAction)edgeDetect:(id)sender;
 - (IBAction)faceDetect:(id)sender;
+- (IBAction)mosaic:(id)sender;
 
 @property (nonatomic, retain) UIImageView *imageView;
 @end

Classes/OpenCVTestViewController.m

@@ -7,6 +7,7 @@ @implementation OpenCVTestViewController
 
 - (void)dealloc {
 	AudioServicesDisposeSystemSoundID(alertSoundID);
+	[timeRecorder release];
 	[imageView dealloc];
 	[super dealloc];
 }
@@ -36,7 +37,7 @@ - (IplImage *)CreateIplImageFromUIImage:(UIImage *)image {
 
 // NOTE You should convert color mode as RGB before passing to this function
 - (UIImage *)UIImageFromIplImage:(IplImage *)image {
-	NSLog(@"IplImage (%d, %d) %d bits by %d channels, %d bytes/row %s", image->width, image->height, image->depth, image->nChannels, image->widthStep, image->channelSeq);
+	//NSLog(@"IplImage (%d, %d) %d bits by %d channels, %d bytes/row %s", image->width, image->height, image->depth, image->nChannels, image->widthStep, image->channelSeq);
 
 	CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
 	NSData *data = [NSData dataWithBytes:image->imageData length:image->imageSize];
@@ -168,6 +169,70 @@ - (void) opencvFaceDetect:(UIImage *)overlayImage  {
 	}
 }
 
+- (void)opencvMosaic:(NSNumber*)mosaicSize {
+	NSAutoreleasePool * pool = [[NSAutoreleasePool alloc] init];
+	if (imageView.image) {
+		[timeRecorder start];
+		int mSize = [mosaicSize intValue];
+		IplImage *img = [self CreateIplImageFromUIImage:imageView.image];
+		IplImage *dst = cvCreateImage(cvGetSize(img), img->depth, img->nChannels);
+		unsigned int r, g, b;
+		int p, pp, c;
+		
+		for (int i = 0; i < img->height; i += mSize) {
+			for (int j = 0; j < img->width; j += mSize) {
+				r = b = g = 0;
+				p = (i * img->width + j) * 3;
+				c = 0;
+				for (int k = 0; k < mSize; k++) {
+					if (i + k < img->height) {
+						for (int l = 0; l < mSize; l++) {
+							if (j + l < img->width) {
+								pp = p + (k * img->width + l) * 3;
+								b += (unsigned char)img->imageData[pp];
+								g += (unsigned char)img->imageData[pp + 1];
+								r += (unsigned char)img->imageData[pp + 2];
+								c++;
+							}
+						}
+					}
+				}
+				r /= c;
+				g /= c;
+				b /= c;
+				
+				for (int k = 0; k < mSize; k++) {
+					if (i + k < img->height) {
+						for (int l = 0; l < mSize; l++) {
+							if (j + l < img->width) {
+								pp = p + (k * img->width + l) * 3;
+								dst->imageData[pp] = (char)r;
+								dst->imageData[pp + 1] = (char)g;
+								dst->imageData[pp + 2] = (char)b;
+							}
+						}
+					}
+				}
+			}
+		}
+		cvReleaseImage(&img);
+		imageView.image = [self UIImageFromIplImage:dst];
+		cvReleaseImage(&dst);
+		double time = [timeRecorder end];
+		NSLog(@"mosaic: %lfmsec", time);
+		if (mSize < 128) {
+			NSNumber *doubleSize = [[NSNumber alloc] initWithInt:mSize * 2]; 
+			[self performSelectorInBackground:@selector(opencvMosaic:) withObject:doubleSize];
+			[doubleSize release];
+		}
+		else {
+			NSLog(@"average: %lfmsec", [timeRecorder average]);
+		}
+
+	}
+	[self hideProgressIndicator];
+	[pool release];
+}
 
 #pragma mark -
 #pragma mark IBAction
@@ -213,6 +278,15 @@ - (IBAction)faceDetect:(id)sender {
 	}
 }
 
+- (IBAction)mosaic:(id)sender {
+	cvSetErrMode(CV_ErrModeParent);
+	[self showProgressIndicator:@"Mosaic"];
+	NSNumber *mosaicSize = [[NSNumber alloc] initWithInt:2];
+	[timeRecorder reset];
+	[self performSelectorInBackground:@selector(opencvMosaic:) withObject:mosaicSize];
+	[mosaicSize release];
+}
+
 #pragma mark -
 #pragma mark UIViewControllerDelegate
 
@@ -223,6 +297,7 @@ - (void)viewDidLoad {
 
 	NSURL *url = [NSURL fileURLWithPath:[[NSBundle mainBundle] pathForResource:@"Tink" ofType:@"aiff"] isDirectory:NO];
 	AudioServicesCreateSystemSoundID((CFURLRef)url, &alertSoundID);
+	timeRecorder = [[TimeRecorder alloc] init];
 }
 
 - (BOOL)shouldAutorotateToInterfaceOrientation:(UIInterfaceOrientation)interfaceOrientation {

Classes/TimeRecorder.h

@@ -0,0 +1,20 @@
+//
+//  TimeRecorder.h
+//  OpenCVTest
+//
+//  Created by mito on 10/01/24.
+//  Copyright 2010 __MyCompanyName__. All rights reserved.
+//
+
+#import <Foundation/Foundation.h>
+
+
+@interface TimeRecorder : NSObject {
+	NSDate *mStartTime;
+	NSMutableArray *mTimeArray;
+}
+-(void) start;
+-(double) end;
+-(void) reset;
+-(double) average;
+@end

Classes/TimeRecorder.m

@@ -0,0 +1,49 @@
+//
+//  TimeRecorder.m
+//  OpenCVTest
+//
+//  Created by mito on 10/01/24.
+//  Copyright 2010 __MyCompanyName__. All rights reserved.
+//
+
+#import "TimeRecorder.h"
+
+
+@implementation TimeRecorder
+- (void) dealloc {
+	[mTimeArray release];
+	[super dealloc];
+}
+
+- (id)init {
+	if ([super init]) {
+		mTimeArray = [[NSMutableArray alloc] init];
+	}
+	return self;
+}
+
+-(void) start {
+	mStartTime = [NSDate date];
+}
+
+-(double) end {
+	double time = [[NSDate date] timeIntervalSinceDate:mStartTime];
+	NSNumber *interval = [[NSNumber alloc] initWithDouble:time];
+	[mTimeArray addObject:interval];
+	[interval release];
+	return time;
+}
+
+-(void) reset {
+	[mTimeArray removeAllObjects];
+}
+
+-(double) average {
+	double total = 0;
+	for (int i = 0; i < [mTimeArray count]; i++) {
+		NSNumber *time = [mTimeArray objectAtIndex:i];
+		total += [time doubleValue];
+	}
+	return total / [mTimeArray count];
+}
+@end