Planificador de procesos: FCFS, SJF, SRTF y Round Robin

1import sys
2
3NOMBRES = {"FCFS": "FCFS (primero en llegar, primero en ser servido)", "SJF": "SJF (el más corto primero, sin expulsión)",
4           "SRTF": "SRTF (menor tiempo restante, con expulsión)", "RR": "Round Robin"}
5
6
7def coma(x):
8    return f"{x:.2f}".replace(".", ",")
9
10
11def simular(algoritmo, quantum, procesos):
12    """Devuelve el diagrama de Gantt como lista de [nombre, inicio, fin] y el instante en que termina cada proceso."""
13    restante = {p["nombre"]: p["rafaga"] for p in procesos}
14    orden = {p["nombre"]: i for i, p in enumerate(procesos)}
15    pendientes = sorted(procesos, key=lambda p: (p["llegada"], orden[p["nombre"]]))
16    listos, gantt, fin = [], [], {}
17    t = 0
18
19    def admitir(hasta):
20        """Pasa a la cola de listos los procesos que han llegado hasta el instante «hasta»."""
21        while pendientes and pendientes[0]["llegada"] <= hasta:
22            listos.append(pendientes.pop(0))
23
24    def ejecutar(p, duracion):
25        nonlocal t
26        if gantt and gantt[-1][0] == p["nombre"] and gantt[-1][2] == t:
27            gantt[-1][2] += duracion          # el mismo proceso sigue: se alarga el tramo
28        else:
29            gantt.append([p["nombre"], t, t + duracion])
30        t += duracion
31        restante[p["nombre"]] -= duracion
32
33    admitir(t)
34    while listos or pendientes:
35        if not listos:                        # CPU ociosa hasta la siguiente llegada
36            gantt.append(["--", t, pendientes[0]["llegada"]])
37            t = pendientes[0]["llegada"]
38            admitir(t)
39            continue
40        if algoritmo == "FCFS":
41            p = listos.pop(0)
42            ejecutar(p, restante[p["nombre"]])
43        elif algoritmo == "SJF":
44            p = min(listos, key=lambda q: (q["rafaga"], q["llegada"], orden[q["nombre"]]))
45            listos.remove(p)
46            ejecutar(p, restante[p["nombre"]])
47        elif algoritmo == "SRTF":
48            p = min(listos, key=lambda q: (restante[q["nombre"]], q["llegada"], orden[q["nombre"]]))
49            # Se ejecuta hasta que acabe o hasta la próxima llegada, que puede expulsarlo
50            hasta = t + restante[p["nombre"]]
51            if pendientes:
52                hasta = min(hasta, pendientes[0]["llegada"])
53            ejecutar(p, hasta - t)
54            if restante[p["nombre"]] > 0:
55                admitir(t)
56                continue
57            listos.remove(p)
58        else:  # RR
59            p = listos.pop(0)
60            ejecutar(p, min(quantum, restante[p["nombre"]]))
61            admitir(t)                        # los que llegan entran en la cola antes que el expulsado
62            if restante[p["nombre"]] > 0:
63                listos.append(p)
64                continue
65        fin[p["nombre"]] = t
66        admitir(t)
67    return gantt, fin
68
69
70def main():
71    lineas = sys.stdin.read().split("\n")
72    cabecera = lineas[0].split()
73    algoritmo = cabecera[0].upper() if cabecera else ""
74    quantum = 0
75    if algoritmo == "RR":
76        if len(cabecera) != 2 or not cabecera[1].isdigit() or int(cabecera[1]) == 0:
77            print("Algoritmo no válido")
78            return
79        quantum = int(cabecera[1])
80    elif algoritmo not in NOMBRES or len(cabecera) != 1:
81        print("Algoritmo no válido")
82        return
83
84    procesos = []
85    for num, linea in enumerate(lineas[1:], start=2):
86        partes = linea.split()
87        if not partes:
88            continue
89        if len(partes) != 3 or not partes[1].isdigit() or not partes[2].isdigit() or int(partes[2]) == 0 \
90                or any(p["nombre"] == partes[0] for p in procesos):
91            print(f"Aviso: línea {num} ignorada")
92            continue
93        procesos.append({"nombre": partes[0], "llegada": int(partes[1]), "rafaga": int(partes[2])})
94    if not procesos:
95        print("No hay procesos")
96        return
97
98    gantt, fin = simular(algoritmo, quantum, procesos)
99    print("Algoritmo: " + NOMBRES[algoritmo] + (f" con quantum {quantum}" if algoritmo == "RR" else ""))
100    print("Diagrama: " + " ".join(f"| {n} {a}-{b}" for n, a, b in gantt) + " |")
101    print(f"{'Proceso':<8}{'Llegada':>8}{'Ráfaga':>8}{'Fin':>6}{'Retorno':>9}{'Espera':>8}")
102    retornos, esperas = [], []
103    for p in procesos:
104        retorno = fin[p["nombre"]] - p["llegada"]
105        espera = retorno - p["rafaga"]
106        retornos.append(retorno)
107        esperas.append(espera)
108        print(f"{p['nombre']:<8}{p['llegada']:>8}{p['rafaga']:>8}{fin[p['nombre']]:>6}{retorno:>9}{espera:>8}")
109    print(f"Media de retorno: {coma(sum(retornos) / len(procesos))} · Media de espera: {coma(sum(esperas) / len(procesos))}")
110    ocupada = sum(b - a for n, a, b in gantt if n != "--")
111    print(f"Uso de la CPU: {coma(100 * ocupada / gantt[-1][2])} %")
112
113
114main()