/*
  The following program demonstrates the use of a menu, switch, and functions.
*/

void f1(int count);
void f2(void);
char doMenu(void);

main()
{
  char ch;

  ch = doMenu();

  while(ch != 'Q' && ch != 'q')
  {
    switch(ch)
    {
      case 'A':
      case 'a':
        f1(3);
        break;
      case 'B':
      case 'b':
        f2();
        break;
      default:
        printf("Unknown menu option\n");
    }

    ch = doMenu();
  }
}

char doMenu()
{
  char opt;

  printf("\nA -- Select option 1\n");
  printf("B -- Select option 2\n");
  printf("Q -- Quit\n");

  printf("Enter option: ");
  scanf("%c", &opt);

  while(opt != 'A' && opt != 'a' && opt != 'B' && opt != 'b' && opt != 'Q' && opt != 'q')
  {
    printf("Enter option: ");
    scanf("%c", &opt);
  }

  return opt;
}

void f1(int count)
{
  int i;

  for(i=0; i<count; i++)
  {
    printf("Hello\n");
  }
}

void f2()
{
 printf("World\n");
}