train-ticket-system/TripInfo.h at main · Netreee/train-ticket-system · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#ifndef TRIPINFO_H_
#define TRIPINFO_H_

#include "Utils.h"
#include "StationManager.h"

namespace trainsys {
    extern StationManager *stationManager;

    struct TripInfo {
        TrainID trainID;
        StationID departureStation;
        StationID arrivalStation;
        int ticketNumber;
        int duration;
        int price;
        Date date;

        TripInfo() = default;

        TripInfo(const TrainID &trainID, const StationID &departureStation,
                 const StationID &arrivalStation, int ticketNumber, int duration, int price, const Date &date) {
            this->trainID = trainID;
            this->departureStation = departureStation;
            this->arrivalStation = arrivalStation;
            this->ticketNumber = ticketNumber;
            this->duration = duration;
            this->price = price;
            this->date = date;
        }

        bool operator ==(const TripInfo &rhs) const {
            return trainID == rhs.trainID && departureStation == rhs.departureStation &&
                   arrivalStation == rhs.arrivalStation && ticketNumber == rhs.ticketNumber &&
                   duration == rhs.duration && price == rhs.price && date == rhs.date;
        }

        bool operator !=(const TripInfo &rhs) const {
            return !(*this == rhs);
        }

        bool operator <(const TripInfo &rhs) const {
            if (trainID != rhs.trainID) {
                return trainID < rhs.trainID;
            } else if (departureStation != rhs.departureStation) {
                return departureStation < rhs.departureStation;
            } else if (arrivalStation != rhs.arrivalStation) {
                return arrivalStation < rhs.arrivalStation;
            } else if (date != rhs.date) {
                return date < rhs.date;
            } else if (ticketNumber != rhs.ticketNumber) {
                return ticketNumber < rhs.ticketNumber;
            } else if (price != rhs.price) {
                return price < rhs.price;
            } else {
                return duration < rhs.duration;
            }
        }

        friend std::ostream &operator <<(std::ostream &os, const TripInfo &tripInfo) {
            os << "TripInfo:" << std::endl;
            os << "trainID: " << tripInfo.trainID << std::endl;
            os << "departureStation: " << stationManager->getStationName(tripInfo.departureStation) << std::endl;
            os << "arrivalStation: " << stationManager->getStationName(tripInfo.arrivalStation) << std::endl;
            os << "duration: " << tripInfo.duration << std::endl;
            os << "price: " << tripInfo.price << std::endl;
            os << "date: " << tripInfo.date << std::endl;
            return os;
        }
    };
}

#endif