Description of Program (CS193P Assignment 2)

descriptionOfProgram.m

+ (NSString *) descriptionOfProgram:(id)program
{
  NSMutableArray *stack;
  if ([program isKindOfClass:[NSArray class]]) stack = [program mutableCopy];
  NSString *programDescription = @"";
  while (stack.count) {
    programDescription = [programDescription stringByAppendingString:[self descriptionOfTopOfStack:stack]];
    if (stack.count) {
      programDescription = [programDescription stringByAppendingString:@", "];
    }
  }
  return programDescription;
}

+ (NSString *) descriptionOfTopOfStack:(NSMutableArray *)stack
{
  NSString *description = @"";
  id topOfStack = [stack lastObject];
  if (topOfStack) [stack removeLastObject];
  if ([topOfStack isKindOfClass:[NSNumber class]]){
    description = [NSString stringWithFormat:@"%g", [topOfStack doubleValue]];
  } else if ([topOfStack isKindOfClass:[NSString class]]) {
    if ([self isOperation:topOfStack] && [self isSingleOperandOperation:topOfStack]) {
      description = [NSString stringWithFormat:@"%@(%@)", topOfStack, [self descriptionOfTopOfStack:stack]];
    } else if ([self isOperation:topOfStack] && ![self isSingleOperandOperation:topOfStack] && ![self isNoOperand:topOfStack]) {
      NSString *first = [self descriptionOfTopOfStack:stack];
      NSString *second = [self descriptionOfTopOfStack:stack];
      if ([topOfStack isEqualToString:@"+"] || [topOfStack isEqualToString:@"-"]){
        description = [NSString stringWithFormat:@"(%@%@%@)",second,topOfStack,first];
      } else description = [NSString stringWithFormat:@"%@%@%@",second,topOfStack,first];
    } else description = topOfStack;
  } 
  return description;
}

+ (BOOL) isOperation:(NSString *) stringInQuestion
{
  NSSet *operationsSet = [NSSet setWithObjects: @"+", @"-", @"*", @"/", @"sin", @"cos", @"sqrt", @"π", @"+/-", @"e", @"log", nil];
  return [operationsSet containsObject:stringInQuestion];
}
                                          
+ (BOOL) isSingleOperandOperation:(NSString *) operationInQuestion
{
  NSSet *operationSet = [NSSet setWithObjects:@"sin", @"cos", @"sqrt", @"log", nil];
  return [operationSet containsObject:operationInQuestion];
}

+ (BOOL)isNoOperand:(NSString *)operation 
{ 
  return [[NSSet setWithObjects:@"π",@"x",@"y",@"foo", nil] containsObject:operation]; 
}

// My implementation. Believe it's pretty solid with test conditions in the pdf.

+ (NSString *)descriptionOfTopOfStack:(NSMutableArray *)stack 
{
    NSString *result = @"";

    id topOfStack = [stack lastObject];

    if(topOfStack) [stack removeLastObject];

    if([topOfStack isKindOfClass:[NSNumber class]])
        result = [result stringByAppendingFormat:@"%@, ",topOfStack];
    else if([topOfStack isKindOfClass:[NSString class]]){

        NSString *operation = topOfStack;

        if ([CalculatorBrains isNoOperand:operation]) {
            result = [NSString stringWithFormat:@"%@, ", operation];
        } else if ([CalculatorBrains isSingleOperand:operation]) {
            NSString * lastDigits = [self descriptionOfTopOfStack:stack];
            result = [NSString stringWithFormat:@"%@(%@), ", operation, [lastDigits substringToIndex:lastDigits.length - 2 ]];
        } else if ([CalculatorBrains isMultiOperand:topOfStack]) {
            NSString * lastDigits = [self descriptionOfTopOfStack:stack];
            NSString * lastDigits2 = [self descriptionOfTopOfStack:stack];
            result = [NSString stringWithFormat:@"(%@ %@ %@), ",[lastDigits2 substringToIndex:lastDigits2.length - 2 ], operation, [lastDigits substringToIndex:lastDigits.length - 2 ]];
        } 
    }

    return result;
}

+ (BOOL)isNoOperand:(NSString *)operation { return [[NSArray arrayWithObjects:@"π",@"x",@"y",@"z", nil] containsObject:operation]; }
+ (BOOL)isSingleOperand:(NSString *)operation { return [[NSArray arrayWithObjects:@"sqrt",@"Sin",@"Cos",@"Tan", nil] containsObject:operation]; }
+ (BOOL)isMultiOperand:(NSString *)operation { return [[NSArray arrayWithObjects:@"+",@"-",@"*",@"/", nil] containsObject:operation]; }

Thanks for posting these. Had trouble getting the ball rolling on this one, but these helped with understanding it all very much.