include <CableCompany.h>
  using namespace std;
  
  void CableCompany::addPackage(const string& packageName,
                                const bitset<kNumChannels>& channels)
  {
    // Just make a key/value pair and insert it into the packages map
    mPackages.insert(make_pair(packageName, channels));
  }
  
  void CableCompany::removePackage(const string& packageName)
  {
    // Just erase the package from the package map
    mPackages.erase(packageName);
  }
  
  void CableCompany::newCustomer(const string& name, const string& package)
    throw (out_of_range)
  {
    // Get a reference to the specified package
    MapType::const_iterator it = mPackages.find(package);
    if (it == mPackages.end()) {
      // That package doesn't exist. Throw an exception.
      throw (out_of_range("Invalid package"));
    } else {
      // Create the account with the bitset representing that package
      // Note that it refers to a name/bitset pair. The bitset is the
      // second field.
      mCustomers.insert(make_pair(name, it->second));
    }
  }
  
  void CableCompany::newCustomer(const string& name,
                                 const bitset<kNumChannels>& channels)
  {
    // Just add the customer/channels pair to the customers map
    mCustomers.insert(make_pair(name, channels));
  }
  
  void CableCompany::addChannel(const string& name, int channel)
  {
    // Find a reference to the customers
    MapType::iterator it = mCustomers.find(name);
    if (it != mCustomers.end()) {
      // We found this customer; set the channel
      // Note that it is a reference to a name/bitset pair.
      // The bitset is the second field.
      it->second.set(channel);
    }
  }
  
  void CableCompany::removeChannel(const string& name, int channel)
  {
    // Find a reference to the customers
    MapType::iterator it = mCustomers.find(name);
    if (it != mCustomers.end()) {
      // We found this customer; remove the channel
      // Note that it is a refernce to a name/bitset pair.
      // The bitset is the second field.
      it->second.reset(channel);
    }
  }
  
  void CableCompany::addPackageToCustomer(const string& name, const string& package)
  {
    // Find the package
    MapType::iterator itPack = mPackages.find(package);
    // Find the customer
    MapType::iterator itCust = mCustomers.find(name);
    if (itCust != mCustomers.end() && itPack != mPackages.end()) {
      // Only if both package and customer are found, can we do the update.
      // Or-in the package to the customers existing channels.
      // Note that it is a reference to a name/bitset pair.
      // The bitset is the second field.
      itCust->second |= itPack->second; 
    }
  }
  
  void CableCompany::deleteCustomer(const string& name)
  {
    // Remove the customer with this name
    mCustomers.erase(name);
  }
  
  bitset<kNumChannels>& CableCompany::getCustomerChannels(const string& name)
    throw (out_of_range)
  {
    // Find the customer
    MapType::iterator it = mCustomers.find(name);
    if (it != mCustomers.end()) {
      // Found it!
      // Note that it is a reference to a name/bitset pair.
      // The bitset is the second field.
      return (it->second);
    }
    // Didn't find it. Throw an exception.
    throw (out_of_range("No customer of that name"));
  }