/* -*- c++ -*- */
/* Copyright 2016 Free Software Foundation, Inc.
 *
 * This file is part of GNU Radio
 *
 * GNU Radio is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3, or (at your option)
 * any later version.
 *
 * GNU Radio is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GNU Radio; see the file COPYING.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street,
 * Boston, MA 02110-1301, USA.
 */

#ifndef INCLUDED_DIGITAL_HEADER_FORMAT_OFDM_H
#define INCLUDED_DIGITAL_HEADER_FORMAT_OFDM_H

#include <pmt/pmt.h>
#include <gnuradio/digital/api.h>
#include <gnuradio/digital/header_format_crc.h>
#include <boost/crc.hpp>

namespace gr {
  namespace digital {

    /*!
     * \brief Header formatter that includes the payload length,
     * packet number, and a CRC check on the header.
     * \ingroup packet_operators_blk
     *
     * \details
     *
     * Child class of header_format_base. This version's header
     * format looks like:
     *
     * \li length (12 bits): length of the payload
     * \li number (12 bits): packet number
     * \li CRC8 (8 bits): A CRC8 check on the header contents
     *
     * Instead of duplicating the payload length, we only add it once
     * and use the CRC8 to make sure it's correctly received.
     *
     * \verbatim
         |  0 -- 11 | 12 -- 23 | 24 -- 31 |
         |    len   | pkt len  |   CRC8   |
       \endverbatim
     *
     * Reimplements packet_header_default in the style of the
     * header_format_base.
     */
    class DIGITAL_API header_format_ofdm
      : public header_format_crc
    {
     public:
      header_format_ofdm(const std::vector<std::vector<int> > &occupied_carriers,
                            int n_syms,
                            const std::string &len_key_name="packet_len",
                            const std::string &frame_key_name="frame_len",
                            const std::string &num_key_name="packet_num",
                            int bits_per_header_sym=1,
                            int bits_per_payload_sym=1,
                            bool scramble_header=false);
      virtual ~header_format_ofdm();

      /*!
       * \brief Encodes the header information in the given tags into
       * bits and places them into \p out.
       *
       * \details
       * Uses the following header format:
       *  - Bits 0-11: The packet length (what was stored in the tag with key \p len_tag_key)
       *  - Bits 12-23: The header number (counts up everytime this function is called)
       *  - Bit 24-31: 8-Bit CRC
       */
      virtual bool format(int nbytes_in,
                          const unsigned char *input,
                          pmt::pmt_t &output,
                          pmt::pmt_t &info);

      virtual bool parse(int nbits_in,
                         const unsigned char *input,
                         std::vector<pmt::pmt_t> &info,
                         int &nbits_processed);

      /*!
       * Returns the length of the formatted header in bits.
       */
      virtual size_t header_nbits() const;

      /*!
       * Factory to create an async packet header formatter; returns
       * an sptr to the object.
       */
      static sptr make(const std::vector<std::vector<int> > &occupied_carriers,
                       int n_syms,
                       const std::string &len_key_name="packet_len",
                       const std::string &frame_key_name="frame_len",
                       const std::string &num_key_name="packet_num",
                       int bits_per_header_sym=1,
                       int bits_per_payload_sym=1,
                       bool scramble_header=false);

    protected:
      pmt::pmt_t d_frame_key_name; //!< Tag key of the additional frame length tag
      const std::vector<std::vector<int> > d_occupied_carriers; //!< Which carriers/symbols carry data
      int d_syms_per_set; //!< Helper variable: Total number of elements in d_occupied_carriers
      int d_bits_per_payload_sym;
      std::vector<uint8_t> d_scramble_mask; //!< Bits are xor'd with this before tx'ing
      size_t d_header_len;

      /*! Get info from the header; return payload length and package
       *  rest of data in d_info dictionary.
       */
      virtual int header_payload();
    };

  } // namespace digital
} // namespace gr

#endif /* INCLUDED_DIGITAL_HEADER_FORMAT_OFDM_H */