RailNetwork.cpp

#include "RailNetwork.h"
#include <iostream>
#include <algorithm>

void Switch::setActiveTrack(int track_id) {
    auto it = std::find(connected_tracks.begin(), connected_tracks.end(), track_id);
    if (it != connected_tracks.end()) {
        active_track = track_id;
    }
}

RailNetwork::RailNetwork() {}

void RailNetwork::addStation(int id, const std::string& name, const GPS& location, int capacity) {
    stations[id] = std::make_shared<Station>(id, name, location, capacity);
    adjacency_list[id] = std::vector<int>();
}

void RailNetwork::addTrack(int id, int station1_id, int station2_id, double distance, double speed_limit) {
    tracks[id] = std::make_shared<Track>(id, station1_id, station2_id, distance, speed_limit);
    adjacency_list[station1_id].push_back(station2_id);
    adjacency_list[station2_id].push_back(station1_id);
}

void RailNetwork::addSwitch(int id, int station_id, const std::vector<int>& track_ids) {
    auto sw = std::make_shared<Switch>(id, station_id);
    sw->connected_tracks = track_ids;
    if (!track_ids.empty()) {
        sw->active_track = track_ids[0];
    }
    switches[id] = sw;
}

std::shared_ptr<Station> RailNetwork::getStation(int id) const {
    auto it = stations.find(id);
    return (it != stations.end()) ? it->second : nullptr;
}

std::shared_ptr<Track> RailNetwork::getTrack(int id) const {
    auto it = tracks.find(id);
    return (it != tracks.end()) ? it->second : nullptr;
}

std::shared_ptr<Switch> RailNetwork::getSwitch(int id) const {
    auto it = switches.find(id);
    return (it != switches.end()) ? it->second : nullptr;
}

const std::vector<int>& RailNetwork::getAdjacentStations(int station_id) const {
    static std::vector<int> empty;
    auto it = adjacency_list.find(station_id);
    return (it != adjacency_list.end()) ? it->second : empty;
}

std::shared_ptr<Track> RailNetwork::getTrackBetween(int station1_id, int station2_id) const {
    for (const auto& pair : tracks) {
        auto track = pair.second;
        if (!track->is_active) continue;
        if ((track->station1_id == station1_id && track->station2_id == station2_id) ||
            (track->station1_id == station2_id && track->station2_id == station1_id)) {
            return track;
        }
    }
    return nullptr;
}

void RailNetwork::manualSwitchOverride(int switch_id, int track_id) {
    auto sw = getSwitch(switch_id);
    if (sw) {
        sw->setActiveTrack(track_id);
        std::cout << "[MANUAL OVERRIDE] Switch " << switch_id << " set to track " << track_id << std::endl;
    }
}

void RailNetwork::deactivateTrack(int track_id) {
    auto track = getTrack(track_id);
    if (track) {
        track->is_active = false;
        std::cout << "[TRACK DEACTIVATED] Track " << track_id << " deactivated" << std::endl;
    }
}

void RailNetwork::activateTrack(int track_id) {
    auto track = getTrack(track_id);
    if (track) {
        track->is_active = true;
        std::cout << "[TRACK ACTIVATED] Track " << track_id << " activated" << std::endl;
    }
}

void RailNetwork::buildTexasNetwork() {
    addStation(1, "Houston Central", GPS(29.7604, -95.3698), 150);
    addStation(2, "Dallas Fort Worth", GPS(32.7767, -96.7970), 200);
    addStation(3, "San Antonio", GPS(29.4241, -98.4936), 120);
    addStation(4, "Austin", GPS(30.2672, -97.7431), 130);
    addStation(5, "El Paso", GPS(31.7619, -106.4850), 100);
    addStation(6, "Corpus Christi", GPS(27.8006, -97.3964), 80);
    addStation(7, "Fort Worth", GPS(32.7555, -97.3308), 110);
    addStation(8, "Lubbock", GPS(33.5779, -101.8552), 90);
    addStation(9, "Amarillo", GPS(35.2220, -101.8313), 85);
    addStation(10, "Beaumont", GPS(30.0802, -94.1266), 75);
    
    addTrack(1, 1, 2, 362.0, 90.0);
    addTrack(2, 1, 3, 306.0, 85.0);
    addTrack(3, 1, 4, 265.0, 85.0);
    addTrack(4, 2, 7, 52.0, 80.0);
    addTrack(5, 3, 4, 129.0, 80.0);
    addTrack(6, 4, 7, 313.0, 85.0);
    addTrack(7, 1, 6, 335.0, 80.0);
    addTrack(8, 3, 6, 230.0, 75.0);
    addTrack(9, 2, 8, 490.0, 90.0);
    addTrack(10, 8, 9, 197.0, 85.0);
    addTrack(11, 5, 9, 466.0, 90.0);
    addTrack(12, 1, 10, 139.0, 75.0);
    addTrack(13, 4, 8, 572.0, 90.0);
    addTrack(14, 5, 3, 885.0, 95.0);
    addTrack(15, 7, 9, 575.0, 90.0);
    
    addSwitch(1, 1, {1, 2, 3, 7, 12});
    addSwitch(2, 2, {1, 4, 9});
    addSwitch(3, 3, {2, 5, 8, 14});
    addSwitch(4, 4, {3, 5, 6, 13});
    addSwitch(5, 7, {4, 6, 15});
    
    std::cout << "Texas Rail Network initialized successfully." << std::endl;
}

void RailNetwork::printNetworkStats() const {
    std::cout << "\n===== RAIL NETWORK STATISTICS =====" << std::endl;
    std::cout << "Total Stations: " << stations.size() << std::endl;
    std::cout << "Total Tracks: " << tracks.size() << std::endl;
    std::cout << "Total Switches: " << switches.size() << std::endl;
    std::cout << "===================================\n" << std::endl;
}