#import "Anonymization.h"

#import "AnonymizationTagsPopUpButton.h"

#import "DCMTagDictionary.h"

#import "DCMAttributeTag.h"

#import "N2CustomTitledPopUpButtonCell.h"

#import "AnonymizationCustomTagPanelController.h"

#import "AnonymizationViewController.h"

#import "AnonymizationSavePanelController.h"

#import "DicomFile.h"

#import <OsiriX/DCMObject.h>

#import <OsiriX/DCMAttribute.h>

#import "N2Debug.h"

@implementation AnonymizationTagsPopUpButton

@synthesize selectedTag;

NSInteger CompareDCMAttributeTagNames(id lsp, id rsp, void* context) {

return [[lsp name] caseInsensitiveCompare: [rsp name]];

}

NSInteger CompareDCMAttributeTagStringValues(id lsp, id rsp, void* context) {

return [[lsp stringValue] caseInsensitiveCompare:[rsp stringValue]];

}

+(NSMenu*)tagsMenu {

return [self tagsMenuWithTarget:NULL action:NULL];

}

+ (NSArray*) tagsForFile: (NSString*) dicomFile

{

if([DicomFile isDICOMFile: dicomFile])

{

DCMObject *dcmObject = [DCMObject objectWithContentsOfFile: dicomFile decodingPixelData:NO];

NSArray *sortedKeys = [[[dcmObject attributes] allKeys] sortedArrayUsingSelector:@selector(caseInsensitiveCompare:)];

NSMutableArray *tags = [NSMutableArray arrayWithCapacity: [sortedKeys count]];

for( NSString *key in sortedKeys)

{

DCMAttribute *attr = [[dcmObject attributes] objectForKey: key];

if( attr)

[tags addObject: attr];

}

return tags;

}

return nil;

}

+(NSMenu*)tagsMenuWithTarget:(id)obj action:(SEL)action {

NSMenu* tagsMenu = [[[NSMenu alloc] initWithTitle:@"DCM Annotation Tags"] autorelease];

NSArray *tagsOfFile = [self tagsForFile: [Anonymization templateDicomFile]];

if( tagsOfFile)

{

NSMenu* tagsOfTheDICOMFile = [[[NSMenu alloc] initWithTitle:@""] autorelease];

NSMenuItem* tagsOfTheDICOMFileMenuItem = [[[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"File(s) tags", NULL) action:NULL keyEquivalent:@""] autorelease];

[tagsOfTheDICOMFileMenuItem setSubmenu:tagsOfTheDICOMFile];

[tagsMenu addItem:tagsOfTheDICOMFileMenuItem];

for (DCMAttribute* tag in tagsOfFile)

{

@try

{

NSString* valDescription = @"";

if( [tag valueLength] < 100)

{

for( id v in [tag values])

valDescription = [valDescription stringByAppendingFormat:@" %@", [v description]];

}

NSString *description;

if( [valDescription length])

description = [NSString stringWithFormat:@"%@ - %@ -%@", [tag attrTag].stringValue, [tag attrTag].name, valDescription];

else

description = [NSString stringWithFormat:@"%@ - %@", [tag attrTag].stringValue, [tag attrTag].name];

NSMenuItem* item = [[[NSMenuItem alloc] initWithTitle: description action:action keyEquivalent:@""] autorelease];

item.representedObject = [tag attrTag];

item.target = obj;

[tagsOfTheDICOMFile addItem:item];

}

@catch (NSException * e)

{

N2LogExceptionWithStackTrace(e);

}

}

}

NSMutableArray* dcmTags = [NSMutableArray arrayWithCapacity:[[DCMTagDictionary sharedTagDictionary] count]];

for (NSString* dcmTagsKey in [[DCMTagDictionary sharedTagDictionary] allKeys])

[dcmTags addObject:[DCMAttributeTag tagWithTagString:dcmTagsKey]];

NSMenu* tagsSortedByNameMenu = [[[NSMenu alloc] initWithTitle:@""] autorelease];

NSMenuItem* tagsSortedByNameMenuItem = [[[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Sorted by Name", NULL) action:NULL keyEquivalent:@""] autorelease];

[tagsSortedByNameMenuItem setSubmenu:tagsSortedByNameMenu];

[tagsMenu addItem:tagsSortedByNameMenuItem];

for (DCMAttributeTag* tag in [dcmTags sortedArrayUsingFunction:CompareDCMAttributeTagNames context:NULL]) {

NSMenuItem* item = [[[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"%@ - %@", tag.name, tag.stringValue] action:action keyEquivalent:@""] autorelease];

item.representedObject = tag;

item.target = obj;

[tagsSortedByNameMenu addItem:item];

}

NSMenu* tagsSortedByStringValueMenu = [[[NSMenu alloc] initWithTitle:@""] autorelease];

NSMenuItem* tagsSortedByStringValueMenuItem = [[[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Sorted by Value", NULL) action:NULL keyEquivalent:@""] autorelease];

[tagsSortedByStringValueMenuItem setSubmenu:tagsSortedByStringValueMenu];

[tagsMenu addItem:tagsSortedByStringValueMenuItem];

for (DCMAttributeTag* tag in [dcmTags sortedArrayUsingFunction:CompareDCMAttributeTagStringValues context:NULL])

{

NSMenuItem* item = [[[NSMenuItem alloc] initWithTitle:[NSString stringWithFormat:@"%@ - %@", tag.stringValue, tag.name] action:action keyEquivalent:@""] autorelease];

item.representedObject = tag;

item.target = obj;

[tagsSortedByStringValueMenu addItem:item];

}

return tagsMenu;

}

-(id)initWithFrame:(NSRect)frame {

self = [super initWithFrame:frame];

self.cell = [[[N2CustomTitledPopUpButtonCell alloc] init] autorelease];

self.autoenablesItems = NO;

self.menu = [AnonymizationTagsPopUpButton tagsMenuWithTarget:self action:@selector(tagsMenuItemSelectedAction:)];

NSMenuItem* extraItem = [[[NSMenuItem alloc] initWithTitle:NSLocalizedString(@"Custom...", @"Title of menu item allowing the user to specify a custom anonymization dicom tag") action:@selector(customMenuItemAction:) keyEquivalent:@""] autorelease];

extraItem.target = self;

[self.menu addItem:extraItem];

self.selectedTag = NULL;

return self;

}

-(void)customMenuItemAction:(id)sender {

AnonymizationCustomTagPanelController* panelController = [[AnonymizationCustomTagPanelController alloc] init];

[panelController setAttributeTag:self.selectedTag];

[NSApp beginSheet:panelController.window modalForWindow:self.window modalDelegate:self didEndSelector:@selector(addCustomTagPanelDidEnd:returnCode:contextInfo:) contextInfo:panelController];

[panelController.window orderFront:self];

}

-(void)addCustomTagPanelDidEnd:(NSPanel*)panel returnCode:(NSInteger)returnCode contextInfo:(void*)contextInfo {

AnonymizationCustomTagPanelController* panelController = (id)contextInfo;

if (returnCode == NSRunStoppedResponse) {

[self setSelectedTag:[panelController attributeTag]];

}

[panel close];

[panelController release];

}

-(void)tagsMenuItemSelectedAction:(NSMenuItem*)menuItem {

[self setSelectedTag:menuItem.representedObject];

}

-(void)setSelectedTag:(DCMAttributeTag*)tag {

[selectedTag release];

selectedTag = [tag retain];

[self didChangeValueForKey:@"title"];

for (NSMenuItem* item in self.itemArray) {

item.state = NSOffState;

if (item.hasSubmenu)

for (NSMenuItem* subitem in item.submenu.itemArray)

subitem.state = tag && [subitem.representedObject isEqual:tag];

}

[self.cell setDisplayedTitle: selectedTag? (selectedTag.name? [NSString stringWithFormat:@"%@ - %@", selectedTag.name, selectedTag.stringValue] : selectedTag.stringValue) : NSLocalizedString(@"Select a DICOM tag...", NULL) ];

[self setNeedsDisplay:YES];

[self didChangeValueForKey:@"selectedTag"];

}

- (BOOL)validateMenuItem:(NSMenuItem *)menuItem

{

AnonymizationSavePanelController *s = [[self window] windowController];

AnonymizationViewController *v = [s anonymizationViewController];

BOOL found = NO;

int mg = [[menuItem representedObject] group], me = [[menuItem representedObject] element];

for( DCMAttributeTag * t in [v tags])

{

if( t.group == mg && t.element == me)

{

found = YES;

break;

}

}

if( found)

[menuItem setState: NSOnState];

else

[menuItem setState: NSOffState];

return YES;

}