//*****************************************************************
/*
  JackTrip: A System for High-Quality Audio Network Performance
  over the Internet

  Copyright (c) 2008-2021 Juan-Pablo Caceres, Chris Chafe.
  SoundWIRE group at CCRMA, Stanford University.

  Permission is hereby granted, free of charge, to any person
  obtaining a copy of this software and associated documentation
  files (the "Software"), to deal in the Software without
  restriction, including without limitation the rights to use,
  copy, modify, merge, publish, distribute, sublicense, and/or sell
  copies of the Software, and to permit persons to whom the
  Software is furnished to do so, subject to the following
  conditions:

  The above copyright notice and this permission notice shall be
  included in all copies or substantial portions of the Software.

  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
  OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
  OTHER DEALINGS IN THE SOFTWARE.
*/
//*****************************************************************

/**
 * \file PacketHeader.h
 * \author Juan-Pablo Caceres
 * \date July 2008
 */

#ifndef __PACKETHEADER_H__
#define __PACKETHEADER_H__

#include <iostream>
//#include <tr1/memory> // for shared_ptr
#include <QObject>
#include <QString>
#include <cstring>

#include "jacktrip_globals.h"
#include "jacktrip_types.h"
class JackTrip;  // Forward Declaration

/// \brief Abstract Header Struct, Header Structs should subclass it
struct HeaderStruct {
};

/// \brief Default Header Struct
struct DefaultHeaderStruct : public HeaderStruct {
   public:
    // watch out for alignment...
    uint64_t TimeStamp;     ///< Time Stamp
    uint16_t SeqNumber;     ///< Sequence Number
    uint16_t BufferSize;    ///< Buffer Size in Samples
    uint8_t SamplingRate;   ///< Sampling Rate in JackAudioInterface::samplingRateT
    uint8_t BitResolution;  ///< Audio Bit Resolution
    uint8_t NumIncomingChannelsFromNet;  ///< Number of incoming Channels from the network
    uint8_t NumOutgoingChannelsToNet;    ///< Number of outgoing Channels to the network
};

//---------------------------------------------------------
// JamLink UDP Header:
/************************************************************************/
/* values for the UDP stream type                                       */
/* streamType is a 16-bit value at the head of each UDP stream          */
/* Its bit map is as follows:  (b15-msb)                                */
/* B15:reserved, B14:extended header, B13 Stereo, B12 not 16-bit        */
/* B11-B9: 0-48 Khz, 1-44 Khz, 2-32 Khz, 3-24 Khz,                      */
/*         4-22 Khz, 5-16 Khz, 6-11 Khz, 7-8 Khz                        */
/* B8-0: Samples in packet                                              */
/************************************************************************/
const unsigned short ETX_RSVD   = (0 << 15);
const unsigned short ETX_XTND   = (1 << 14);
const unsigned short ETX_STEREO = (1 << 13);
const unsigned short ETX_MONO   = (0 << 13);
const unsigned short ETX_16BIT  = (0 << 12);
// inline unsigned short ETX_RATE_MASK(const unsigned short a) { a&(0x7<<9); }
const unsigned short ETX_48KHZ = (0 << 9);
const unsigned short ETX_44KHZ = (1 << 9);
const unsigned short ETX_32KHZ = (2 << 9);
const unsigned short ETX_24KHZ = (3 << 9);
const unsigned short ETX_22KHZ = (4 << 9);
const unsigned short ETX_16KHZ = (5 << 9);
const unsigned short ETX_11KHZ = (6 << 9);
const unsigned short ETX_8KHZ  = (7 << 9);
// able to express up to 512 SPP
// inline unsigned short  ETX_SPP(const unsigned short a) { (a&0x01FF); }

/// \brief JamLink Header Struct
struct JamLinkHeaderStuct : public HeaderStruct {
    // watch out for alignment -- need to be on 4 byte chunks
    uint16_t Common;     ///< Common part of the header, 16 bit
    uint16_t SeqNumber;  ///< Sequence Number
    uint32_t TimeStamp;  ///< Time Stamp
};

//#######################################################################
//####################### PacketHeader ##################################
//#######################################################################
/** \brief Base class for header type. Subclass this struct to
 * create a new header.
 */
class PacketHeader : public QObject
{
    Q_OBJECT

   public:
    /// \brief The class Constructor
    PacketHeader(JackTrip* jacktrip);
    /// \brief The class Destructor
    virtual ~PacketHeader() {}

    /// \brief Return a time stamp in microseconds
    /// \return Time stamp: microseconds since midnight (0 hour), January 1, 1970
    static uint64_t usecTime();
    /// \todo Implement this using a JackTrip Method (Mediator) member instead of the
    /// reference to JackAudio
    virtual void fillHeaderCommonFromAudio() = 0;
    /// \brief Parse the packet header and take appropriate measures (like change
    /// settings, or quit the program if peer settings don't match)
    virtual void parseHeader() = 0;
    /// \brief Check that the settings in the supplied packet header match the server's
    /// settings \return True if settings match, false otherwise
    virtual bool checkPeerSettings(int8_t* full_packet) = 0;

    virtual uint64_t getPeerTimeStamp(int8_t* full_packet) const          = 0;
    virtual uint16_t getPeerSequenceNumber(int8_t* full_packet) const     = 0;
    virtual uint16_t getPeerBufferSize(int8_t* full_packet) const         = 0;
    virtual uint8_t getPeerSamplingRate(int8_t* full_packet) const        = 0;
    virtual uint8_t getPeerBitResolution(int8_t* full_packet) const       = 0;
    virtual uint8_t getPeerNumIncomingChannels(int8_t* full_packet) const = 0;
    virtual uint8_t getPeerNumOutgoingChannels(int8_t* full_packet) const = 0;

    /// \brief Increase sequence number for counter, a 16bit number
    virtual void increaseSequenceNumber() { mSeqNumber++; }
    /// \brief Returns the current sequence number
    /// \return 16bit Sequence number
    virtual uint16_t getSequenceNumber() const { return mSeqNumber; }
    /// \brief Get the header size in bytes
    virtual int getHeaderSizeInBytes() const = 0;
    virtual void putHeaderInPacketBaseClass(int8_t* full_packet,
                                            const HeaderStruct& header_struct)
    {
        std::memcpy(full_packet, reinterpret_cast<const void*>(&header_struct),
                    getHeaderSizeInBytes());
    }
    /// \brief Put the header in buffer pointed by full_packet
    /// \param full_packet Pointer to full packet (audio+header). Size must be
    /// sizeof(header part) + sizeof(audio part)
    virtual void putHeaderInPacket(int8_t* full_packet) = 0;
    void setBufferRequiresSameSettings(bool sameSettings)
    {
        mBufferRequiresSameSettings = sameSettings;
    }

   signals:
    void signalError(const QString& error_message);

   protected:
    bool mBufferRequiresSameSettings;
    JackTrip* mJackTrip;  ///< JackTrip mediator class

   private:
    uint16_t mSeqNumber;
};

//#######################################################################
//####################### DefaultHeader #################################
//#######################################################################
/** \brief Default Header
 */
class DefaultHeader : public PacketHeader
{
    Q_OBJECT

   public:
    DefaultHeader(JackTrip* jacktrip);
    virtual ~DefaultHeader() {}

    virtual void fillHeaderCommonFromAudio() override;
    virtual void parseHeader() override {}
    virtual bool checkPeerSettings(int8_t* full_packet) override;
    virtual void increaseSequenceNumber() override { mHeader.SeqNumber++; }
    virtual uint16_t getSequenceNumber() const override { return mHeader.SeqNumber; }
    virtual int getHeaderSizeInBytes() const override { return sizeof(mHeader); }
    virtual void putHeaderInPacket(int8_t* full_packet) override
    {
        putHeaderInPacketBaseClass(full_packet, mHeader);
    }
    void printHeader() const;
    uint8_t getNumIncomingChannelsFromNet() const
    {
        return mHeader.NumIncomingChannelsFromNet;
    }
    uint8_t getNumOutgoingChannelsToNet() const
    {
        return mHeader.NumOutgoingChannelsToNet;
    }
    virtual uint64_t getPeerTimeStamp(int8_t* full_packet) const override;
    virtual uint16_t getPeerSequenceNumber(int8_t* full_packet) const override;
    virtual uint16_t getPeerBufferSize(int8_t* full_packet) const override;
    virtual uint8_t getPeerSamplingRate(int8_t* full_packet) const override;
    virtual uint8_t getPeerBitResolution(int8_t* full_packet) const override;
    uint8_t getPeerNumIncomingChannels(int8_t* full_packet) const override;
    uint8_t getPeerNumOutgoingChannels(int8_t* full_packet) const override;

   private:
    DefaultHeaderStruct mHeader;  ///< Default Header Struct
};

//#######################################################################
//####################### JamLinkHeader #################################
//#######################################################################

/** \brief JamLink Header
 */
class JamLinkHeader : public PacketHeader
{
    Q_OBJECT

   public:
    JamLinkHeader(JackTrip* jacktrip);
    virtual ~JamLinkHeader() {}

    virtual void fillHeaderCommonFromAudio() override;
    virtual void parseHeader() override {}
    virtual bool checkPeerSettings(int8_t* /*full_packet*/) override { return true; }

    virtual uint64_t getPeerTimeStamp(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint16_t getPeerSequenceNumber(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint16_t getPeerBufferSize(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint8_t getPeerSamplingRate(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint8_t getPeerBitResolution(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    uint8_t getPeerNumIncomingChannels(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    uint8_t getPeerNumOutgoingChannels(int8_t* /*full_packet*/) const override
    {
        return 0;
    }

    virtual void increaseSequenceNumber() override {}
    virtual int getHeaderSizeInBytes() const override { return sizeof(mHeader); }
    virtual void putHeaderInPacket(int8_t* full_packet) override
    {
        putHeaderInPacketBaseClass(full_packet, mHeader);
    }

   private:
    JamLinkHeaderStuct mHeader;  ///< JamLink Header Struct
};

//#######################################################################
//####################### EmptyHeader #################################
//#######################################################################

/** \brief Empty Header to use with systems that don't include a header.
 */
class EmptyHeader : public PacketHeader
{
    Q_OBJECT

   public:
    EmptyHeader(JackTrip* jacktrip);
    virtual ~EmptyHeader() {}

    virtual void fillHeaderCommonFromAudio() override {}
    virtual void parseHeader() override {}
    virtual bool checkPeerSettings(int8_t* /*full_packet*/) override { return true; }
    virtual void increaseSequenceNumber() override {}
    virtual int getHeaderSizeInBytes() const override { return 0; }

    virtual uint64_t getPeerTimeStamp(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint16_t getPeerSequenceNumber(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint16_t getPeerBufferSize(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint8_t getPeerSamplingRate(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    virtual uint8_t getPeerBitResolution(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    uint8_t getPeerNumIncomingChannels(int8_t* /*full_packet*/) const override
    {
        return 0;
    }
    uint8_t getPeerNumOutgoingChannels(int8_t* /*full_packet*/) const override
    {
        return 0;
    }

    virtual void putHeaderInPacket(int8_t* /*full_packet*/) override {}
};

#endif  //__PACKETHEADER_H__