#include "box.cpp"
#include <ctime>

#define SECONDS_DIFFERENCE 2082844800

class Box_tkhd {
  public:
    Box_tkhd( );
    ~Box_tkhd();
    Box * GetBox();
    void SetCreationTime( uint32_t TimeStamp = 0 );
    void SetModificationTime( uint32_t TimeStamp = 0 );
    void SetDurationTime( uint32_t TimeUnits = 0 );
    void SetWidth( uint16_t Width = 0 );
    void SetHeight( uint16_t Height = 0 );
    void SetFlags( bool Bit0 = true, bool Bit1 = true, bool Bit2 = true );
    void SetVersion( uint32_t Version = 0 );
    void SetTrackID( uint32_t TrackID = 0 );
  private:
    void SetReserved();
    void SetDefaults();
    Box * Container;
    
    uint32_t CurrentFlags;
    uint32_t CurrentVersion;
};//Box_ftyp Class

Box_tkhd::Box_tkhd( ) {
  Container = new Box( 0x746B6864 );
  CurrentVersion = 0;
  CurrentFlags = 0;
}

Box_tkhd::~Box_tkhd() {
  delete Container;
}

Box * Box_tkhd::GetBox() {
  return Container;
}

void Box_tkhd::SetCreationTime( uint32_t TimeStamp ) {
  uint32_t CreationTime;
  if(!TimeStamp) {
    CreationTime = time(NULL) + SECONDS_DIFFERENCE;
  } else {
    CreationTime = TimeStamp;
  }
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(CreationTime),4);
}

void Box_tkhd::SetModificationTime( uint32_t TimeStamp ) {
  uint32_t ModificationTime;
  if(!TimeStamp) {
    ModificationTime = time(NULL) + SECONDS_DIFFERENCE;
  } else {
    ModificationTime = TimeStamp;
  }
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(ModificationTime),8);
}

void Box_tkhd::SetDurationTime( uint32_t TimeUnits ) {
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(TimeUnits),16);
}

void Box_tkhd::SetReserved() {
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0x40000000),68);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),64);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),60);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),56);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0x10000),52);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),48);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),44);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),40);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0x10000),36);
  Container->SetPayload((uint32_t)4,Box::uint16_to_uint8(0),34);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),28);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),24);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),20);
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(0),12);
}

void Box_tkhd::SetVersion( uint32_t Version ) {
  if ( Version >= 2 ) { return; }
  CurrentVersion = Version;
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8((CurrentVersion<<24)&(CurrentFlags)));
}

void Box_tkhd::SetFlags( bool Bit0, bool Bit1, bool Bit2 ) {
  CurrentFlags = (( Bit0 ? 0x80 : 0 ) + ( Bit1 ? 0x40 : 0 ) + ( Bit2 ? 0x20 : 0 )) << 16 ;
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8((CurrentVersion<<24)&(CurrentFlags)));
}

void Box_tkhd::SetTrackID( uint32_t TrackID ) {
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(TrackID),12);
}

void Box_tkhd::SetWidth( uint16_t Width ) {
  uint32_t ResultWidth = ( Width << 16 );
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(ResultWidth),72);
}

void Box_tkhd::SetHeight( uint16_t Height ) {
  uint32_t ResultHeight = ( Height << 16 );
  Container->SetPayload((uint32_t)4,Box::uint32_to_uint8(ResultHeight),76);
}

void Box_tkhd::SetDefaults() {
  SetHeight();
  SetWidth();
  SetCreationTime();
  SetModificationTime();
  SetDurationTime();
  SetFlags();
  SetVersion();
  SetTrackID();
}