54 lines
1.4 KiB
C
54 lines
1.4 KiB
C
#include "a_station.h"
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#define MAX_INPUT 100
|
|
|
|
void test_station(struct station* station) {
|
|
char station_name[MAX_INPUT];
|
|
int capacity;
|
|
|
|
while (1) {
|
|
printf("Enter station name and capacity (or 'exit' to stop):\n");
|
|
printf("Station name: ");
|
|
if (!fgets(station_name, MAX_INPUT, stdin)) break;
|
|
|
|
// Remove trailing newline
|
|
station_name[strcspn(station_name, "\n")] = '\0';
|
|
|
|
if (strcmp(station_name, "exit") == 0) break;
|
|
|
|
printf("Capacity: ");
|
|
if (scanf("%d", &capacity) != 1) {
|
|
printf("Invalid input. Try again.\n");
|
|
while (getchar() != '\n'); // Clear input buffer
|
|
continue;
|
|
}
|
|
|
|
// Clear input buffer
|
|
while (getchar() != '\n');
|
|
|
|
add_target_capacity(station, station_name, capacity);
|
|
|
|
int track = select_track(station, station_name);
|
|
printf("Track for %s: %d\n", station_name, track);
|
|
|
|
int current_capacity = get_target_capacity(station, station_name);
|
|
printf("Current capacity to %s: %d\n\n", station_name, current_capacity);
|
|
}
|
|
|
|
printf("Final Statistics:\n");
|
|
printf("Total number of targets: %d\n", count_targets(station));
|
|
printf("Total capacity: %d\n", count_capacity(station));
|
|
}
|
|
|
|
int main() {
|
|
struct station* station = create_station();
|
|
test_station(station);
|
|
destroy_station(station);
|
|
|
|
return 0;
|
|
}
|
|
|