list.cpp

来自「一本语言类编程书籍」· C++ 代码 · 共 54 行

// List.cpp
#include "Box.h"
#include "List.h"

// Package class definitions
// Package constructor
Package::Package(Box* pNewBox):pBox(pNewBox), pNext(0){}   

// Retrieve the Box pointer
Box* Package::getBox() const { return pBox; }               

// Get next package address
Package* Package::getNext() const { return pNext; }           

// Add package to end of list
void Package::setNext(Package* pPackage) { pNext = pPackage; }

// TruckLoad class member definitions
// TruckLoad constructor
TruckLoad::TruckLoad(Box* pBox, int count) {
  pHead = pTail = pCurrent = 0;

  if((count > 0) && (pBox != 0))
    for(int i = 0 ; i<count ; i++)
      addBox(pBox+i);
}

// Retrieve the first Box
Box* TruckLoad::getFirstBox() {
  pCurrent = pHead;
  return pCurrent->getBox();
}

// Retrieve the next Box
Box* TruckLoad::getNextBox() {
  if(pCurrent)
    pCurrent = pCurrent->getNext();        // pCurrent is not null so set to next
  else                                     // pCurrent is null 
    pCurrent = pHead;                      //  so set to the first list element

  return pCurrent ? pCurrent->getBox() : 0;
}

// Add a new Box to the list
void TruckLoad::addBox(Box* pBox) {
  Package* pPackage = new Package(pBox);   // Create a Package

  if(pHead)                                // Check list is not empty
    pTail->setNext(pPackage);              // Add the new object to the tail
  else                                     // List is empty
    pHead = pPackage;                      // so new object is the head
  pTail = pPackage;                        // Store its address as tail
}