main.c

#include <stdio.h>
#include "memory.h"

int main() {
    int total;
    printf("Enter total memory size: ");
    scanf("%d", &total);
    initializeMemory(total);

    int choice;
    char prog;
    int size;

    while (1) {
        printf("\n1. Allocate\n2. Free\n3. Defragment\n4. Display\n5. Exit\n> ");
        scanf("%d", &choice);

        switch (choice) {
            case 1:
                printf("Enter program name (A-Z): ");
                scanf(" %c", &prog);
                printf("Enter size: ");
                scanf("%d", &size);
                allocate(prog, size);
                break;

            case 2:
                printf("Enter program name to free: ");
                scanf(" %c", &prog);
                freeProgram(prog);
                break;

            case 3:
                defragmentation();
                break;

            case 4:
                displayDisk();
                break;

            case 5:
                printf("Exiting...\n");
                return 0;

            default:
                printf("Invalid choice!\n");
        }
    }
}