FlightPeriodStruct.cpp

Go to the documentation of this file.

// //////////////////////////////////////////////////////////////////////
// Import section
// //////////////////////////////////////////////////////////////////////
// STL
#include <cassert>
#include <sstream>
// StdAir
#include <stdair/basic/BasConst_Period_BOM.hpp>
#include <stdair/service/Logger.hpp>
// AirTSP
#include <airtsp/AIRTSP_Types.hpp>
#include <airtsp/bom/FlightPeriodStruct.hpp>

namespace AIRTSP {

  // ////////////////////////////////////////////////////////////////////
  FlightPeriodStruct::FlightPeriodStruct ()
    : _dateRange (stdair::BOOST_DEFAULT_DATE_PERIOD),
      _dow (stdair::DEFAULT_DOW_STRING),
      _legAlreadyDefined (false), _itSeconds (0) {
  }

  // ////////////////////////////////////////////////////////////////////
  stdair::Date_T FlightPeriodStruct::getDate() const {
    return stdair::Date_T (_itYear, _itMonth, _itDay);
  }

  // ////////////////////////////////////////////////////////////////////
  stdair::Duration_T FlightPeriodStruct::getTime() const {
    return boost::posix_time::hours (_itHours)
      + boost::posix_time::minutes (_itMinutes)
      + boost::posix_time::seconds (_itSeconds);
  }
  
  // ////////////////////////////////////////////////////////////////////
  const std::string FlightPeriodStruct::describe() const {
    std::ostringstream ostr;
    ostr << _airlineCode << _flightNumber << ", " << _dateRange
         << " - " << _dow << std::endl;
      
    for (LegStructList_T::const_iterator itLeg = _legList.begin();
         itLeg != _legList.end(); ++itLeg) {
      const LegStruct& lLeg = *itLeg;
      ostr << lLeg.describe();
    }

    for (SegmentStructList_T::const_iterator itSegment = _segmentList.begin();
         itSegment != _segmentList.end(); ++itSegment) {
      const SegmentStruct& lSegment = *itSegment;
      ostr << lSegment.describe();
    }

    //ostr << "[Debug] - Staging Leg: ";
    //ostr << _itLeg.describe();
    //ostr << "[Debug] - Staging Cabin: ";
    //ostr << _itCabin.describe();

    return ostr.str();
  }

  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::addAirport (const stdair::AirportCode_T& iAirport) {
    AirportList_T::const_iterator itAirport = _airportList.find (iAirport);
    if (itAirport == _airportList.end()) {
      // Add the airport code to the airport set
      const bool insertSuccessful = _airportList.insert (iAirport).second;

      if (insertSuccessful == false) {
        // TODO: throw an exception
      }
          
      // Add the airport code to the airport vector
      _airportOrderedList.push_back (iAirport);
    }
  }

  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::buildSegments () {
    // The list of airports encompasses all the airports on which
    // the flight takes off or lands. Moreover, that list is
    // time-ordered: the first airport is the initial departure of
    // the flight, and the last airport is the eventual point of
    // rest of the flight.
    // Be l the size of the ordered list of airports.
    // We want to generate all the segment combinations from the legs
    // and, hence, from all the possible (time-ordered) airport pairs.
    // Thus, we both iterator on i=0...l-1 and j=i+1...l
    assert (_airportOrderedList.size() >= 2);

    _segmentList.clear();
    for (AirportOrderedList_T::const_iterator itAirport_i =
           _airportOrderedList.begin();
         itAirport_i != _airportOrderedList.end()-1; ++itAirport_i) {
      for (AirportOrderedList_T::const_iterator itAirport_j = itAirport_i + 1;
           itAirport_j != _airportOrderedList.end(); ++itAirport_j) {
        SegmentStruct lSegmentStruct;
        lSegmentStruct._boardingPoint = *itAirport_i;
        lSegmentStruct._offPoint = *itAirport_j;
          
        _segmentList.push_back (lSegmentStruct);
      }
    }

    // Clear the lists of airports, so that it is ready for the next flight
    _airportList.clear();
    _airportOrderedList.clear();
  }
      
  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::
  addSegmentCabin (const SegmentStruct& iSegment,
                   const SegmentCabinStruct& iCabin) {
    // Retrieve the Segment structure corresponding to the (boarding, off) point
    // pair.
    SegmentStructList_T::iterator itSegment = _segmentList.begin();
    for ( ; itSegment != _segmentList.end(); ++itSegment) {
      const SegmentStruct& lSegment = *itSegment;

      const stdair::AirportCode_T& lBoardingPoint = iSegment._boardingPoint;
      const stdair::AirportCode_T& lOffPoint = iSegment._offPoint;
      if (lSegment._boardingPoint == lBoardingPoint
          && lSegment._offPoint == lOffPoint) {
        break;
      }
    }

    if (itSegment == _segmentList.end()) {
      std::ostringstream oStr;
      oStr << "Within the schedule input file, there is a flight, for which "
           << "the airports of segments and those of the legs "
           << "do not correspond";
      STDAIR_LOG_ERROR (oStr.str());
      throw SegmentDateNotFoundException (oStr.str());
    }

    // Add the Cabin structure to the Segment Cabin structure.
    assert (itSegment != _segmentList.end());
    SegmentStruct& lSegment = *itSegment;
    lSegment._cabinList.push_back (iCabin);
  }
    
  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::
  addSegmentCabin (const SegmentCabinStruct& iCabin) {
    // Iterate on all the Segment structures (as they get the same cabin
    // definitions)
    for (SegmentStructList_T::iterator itSegment = _segmentList.begin();
         itSegment != _segmentList.end(); ++itSegment) {
      SegmentStruct& lSegment = *itSegment;

      lSegment._cabinList.push_back (iCabin);
    }
  }

  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::
  addFareFamily (const SegmentStruct& iSegment,
                 const SegmentCabinStruct& iCabin,
                 const FareFamilyStruct& iFareFamily) {
    // Retrieve the Segment structure corresponding to the (boarding, off) point
    // pair.
    SegmentStructList_T::iterator itSegment = _segmentList.begin();
    for ( ; itSegment != _segmentList.end(); ++itSegment) {
      const SegmentStruct& lSegment = *itSegment;

      const stdair::AirportCode_T& lBoardingPoint = iSegment._boardingPoint;
      const stdair::AirportCode_T& lOffPoint = iSegment._offPoint;
      if (lSegment._boardingPoint == lBoardingPoint
          && lSegment._offPoint == lOffPoint) {
        break;
      }
    }

    if (itSegment == _segmentList.end()) {
      std::ostringstream oStr;
      oStr << "Within the schedule input file, there is a flight, for which "
           << "the airports of segments and those of the legs "
           << "do not correspond";
      STDAIR_LOG_ERROR (oStr.str());
      throw SegmentDateNotFoundException (oStr.str());
    }

    // Add the Cabin structure to the Segment Cabin structure.
    assert (itSegment != _segmentList.end());
    SegmentStruct& lSegment = *itSegment;

    // Retrieve the Segment cabin structure given the cabin code
    SegmentCabinStructList_T::iterator itCabin = lSegment._cabinList.begin();
    for ( ; itCabin != lSegment._cabinList.end(); ++itCabin) {
      const SegmentCabinStruct& lCabin = *itCabin;

      const stdair::CabinCode_T& lCabinCode = lCabin._cabinCode;
      if (iCabin._cabinCode == lCabinCode) {
        break;
      }
    }

    if (itCabin == lSegment._cabinList.end()) {
      std::ostringstream oStr;
      oStr << "Within the schedule input file, there is a flight "
           << "for which the cabin code does not exist.";
      STDAIR_LOG_ERROR (oStr.str());
      throw SegmentDateNotFoundException (oStr.str());
    }

    // Add the Cabin structure to the Segment Cabin structure.
    assert (itCabin != lSegment._cabinList.end());
    SegmentCabinStruct& lCabin = *itCabin;
    lCabin._fareFamilies.push_back(iFareFamily);
  }
    
  // ////////////////////////////////////////////////////////////////////
  void FlightPeriodStruct::
  addFareFamily (const SegmentCabinStruct& iCabin,
                 const FareFamilyStruct& iFareFamily) {
    // Iterate on all the Segment structures (as they get the same cabin
    // definitions)
      
    for (SegmentStructList_T::iterator itSegment = _segmentList.begin();
         itSegment != _segmentList.end(); ++itSegment) {
      SegmentStruct& lSegment = *itSegment;

      // Retrieve the Segment cabin structure given the cabin code
      SegmentCabinStructList_T::iterator itCabin = lSegment._cabinList.begin();
      for ( ; itCabin != lSegment._cabinList.end(); ++itCabin) {
        const SegmentCabinStruct& lCabin = *itCabin;

        const stdair::CabinCode_T& lCabinCode = lCabin._cabinCode;
        if (iCabin._cabinCode == lCabinCode) {
          break;
        }
      }

      if (itCabin == lSegment._cabinList.end()) {
        std::ostringstream oStr;
        oStr << "Within the schedule input file, there is a flight "
             << "for which the cabin code does not exist.";
        STDAIR_LOG_ERROR (oStr.str());
        throw SegmentDateNotFoundException (oStr.str());
      }

      // Add the Cabin structure to the Segment Cabin structure.
      assert (itCabin != lSegment._cabinList.end());
      SegmentCabinStruct& lCabin = *itCabin;
      lCabin._fareFamilies.push_back(iFareFamily);
    }
  }

}