#include <iostream>

#include "BodyPartT.h"

using namespace std;

std::string BodyPartTToString(BodyPartT b){
   size_t i;

   i = static_cast<size_t>(b);
   if (i <= static_cast<size_t>(BodyPartT::NONE)) {
       return PART_NAMES[i];
   } else {
       return PART_NAMES[static_cast<size_t>(BodyPartT::NONE)];
   }
}

BodyPartT NextBodyPartT(BodyPartT b){
    size_t offset;
    offset = static_cast<size_t>(b);
    if (offset > static_cast<size_t>(BodyPartT::NONE)) {
       b = BodyPartT::NONE;
    }

    if (b != BodyPartT::NONE) {
       b = static_cast<BodyPartT>(static_cast<size_t>(b) 
                                + static_cast<size_t>(1));
    }
    return b;
}

ostream & operator << (ostream & s, const BodyPartT &  b) {
    s << BodyPartTToString(b);

    return s;
}

BodyPartT operator + (BodyPartT a, size_t b){
   size_t offset{static_cast<size_t>(a)};
   BodyPartT returnValue{BodyPartT::NONE};

   offset += b;
   if (offset <= static_cast<size_t>(BodyPartT::NONE)) {
      returnValue = static_cast<BodyPartT>(offset);    
   }

   return returnValue;
}

BodyPartT & operator +=(BodyPartT & a, size_t b) {
     a = a + b;
     return a;
}

// ++i
BodyPartT & operator++(BodyPartT & b) {
    b = b + 1;
    return b;
}

// i++
BodyPartT operator++(BodyPartT & b, int) {
    BodyPartT tmp{b};
    b = b + 1;
    return tmp;
}

BodyPartT operator -(BodyPartT b) {
    BodyPartT rv{BodyPartT::NONE};

    size_t count{static_cast<size_t>(BodyPartT::NONE)-1};
    size_t offset{static_cast<size_t>(b)};

    if (count >= offset) {
        rv = static_cast<BodyPartT>(count-offset);
    }

    return rv;
}