1 files changed, 8 insertions, 2 deletions
diff --git a/src/dabOutput/edi/AFPacket.h b/src/dabOutput/edi/AFPacket.h
index 1fd0d16..0d41878 100644
--- a/src/dabOutput/edi/AFPacket.h
+++ b/src/dabOutput/edi/AFPacket.h
@@ -30,8 +30,12 @@
 #include "config.h"
 #include <vector>
 #include <stdint.h>
+#include "TagItems.h"
+#include "TagPacket.h"
 #define PACKED __attribute__ ((packed))
 
+#define EDI_AFPACKET_PROTOCOLTYPE_TAGITEMS ('T')
+
 // ETSI TS 102 821, 6.1 AF packet structure
 struct AFHeader
 {
@@ -47,15 +51,17 @@ struct AFHeader
 class AFPacket
 {
     public:
-        AFPacket();
+        AFPacket(char protocolType) : protocol_type(protocolType) {};
 
-        std::vector<uint8_t> Assemble(char protocol_type, std::vector<uint8_t> payload);
+        std::vector<uint8_t> Assemble(TagPacket tag_packet);
 
     private:
         static const bool have_crc = true;
 
         AFHeader header;
         uint16_t seq; //counter that overflows at 0xFFFF
+
+        char protocol_type;
 };
 
 #endif