From 8ef08b72cc5c224b4fb46cfae124755bae53aec8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tom=C3=A1=C5=A1=20Vl=C4=8Dek?= <tomas.vlcek@student.tuke.sk>
Date: Wed, 22 Apr 2026 23:58:23 +0000
Subject: [PATCH] Aktualizovat du6/a_station.c

---
 du6/a_station.c | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/du6/a_station.c b/du6/a_station.c
index 99ef39a..1fd94e1 100644
--- a/du6/a_station.c
+++ b/du6/a_station.c
@@ -52,5 +52,14 @@ int count_targets(struct station* station){
 }
 
 int count_capacity(struct station* station){
+    int count = 0;
+    for (int i = 0; i < station->track_count, i++)
+    {
+        struct car* currentTracks = station->tracks[i];
+        while (currentTracks->next != NULL)
+        {
+            count += currentTracks->capacity;
+        }
+    }
     return 0;
 }
\ No newline at end of file