supercell-wx/wxdata/include/scwx/awips/message.hpp

#pragma once

#include <array>
#include <cstring>
#include <execution>
#include <istream>
#include <map>
#include <string>

#ifdef WIN32
#   include <WinSock2.h>
#else
#   include <arpa/inet.h>
#endif

namespace scwx
{
namespace awips
{

class MessageImpl;

class Message
{
protected:
   explicit Message();

   Message(const Message&)            = delete;
   Message& operator=(const Message&) = delete;

   Message(Message&&) noexcept;
   Message& operator=(Message&&) noexcept;

   virtual bool ValidateMessage(std::istream& is, std::size_t bytesRead) const;

public:
   virtual ~Message();

   virtual std::size_t data_size() const = 0;

   virtual bool Parse(std::istream& is) = 0;

   static void ReadBoolean(std::istream& is, bool& value)
   {
      std::string data(4, ' ');
      is.read(reinterpret_cast<char*>(&data[0]), 4);
      value = (data.at(0) == 'T');
   }

   static void ReadChar(std::istream& is, char& value)
   {
      std::string data(4, ' ');
      is.read(reinterpret_cast<char*>(&data[0]), 4);
      value = data.at(0);
   }

   static float SwapFloat(float f)
   {
      std::uint32_t temp;
      std::memcpy(&temp, &f, sizeof(std::uint32_t));
      temp = ntohl(temp);
      std::memcpy(&f, &temp, sizeof(float));
      return f;
   }

   template<std::size_t _Size>
   static void SwapArray(std::array<float, _Size>& arr,
                         std::size_t               size = _Size)
   {
      std::transform(std::execution::par_unseq,
                     arr.begin(),
                     arr.begin() + size,
                     arr.begin(),
                     [](float f) { return SwapFloat(f); });
   }

   template<std::size_t _Size>
   static void SwapArray(std::array<std::int16_t, _Size>& arr,
                         std::size_t                      size = _Size)
   {
      std::transform(std::execution::par_unseq,
                     arr.begin(),
                     arr.begin() + size,
                     arr.begin(),
                     [](std::int16_t u) { return ntohs(u); });
   }

   template<std::size_t _Size>
   static void SwapArray(std::array<std::uint16_t, _Size>& arr,
                         std::size_t                       size = _Size)
   {
      std::transform(std::execution::par_unseq,
                     arr.begin(),
                     arr.begin() + size,
                     arr.begin(),
                     [](std::uint16_t u) { return ntohs(u); });
   }

   template<std::size_t _Size>
   static void SwapArray(std::array<std::uint32_t, _Size>& arr,
                         std::size_t                       size = _Size)
   {
      std::transform(std::execution::par_unseq,
                     arr.begin(),
                     arr.begin() + size,
                     arr.begin(),
                     [](std::uint32_t u) { return ntohl(u); });
   }

   template<typename T>
   static void SwapMap(std::map<T, float>& m)
   {
      std::for_each(std::execution::par_unseq,
                    m.begin(),
                    m.end(),
                    [](auto& p) { p.second = SwapFloat(p.second); });
   }

   static void SwapVector(std::vector<std::uint16_t>& v)
   {
      std::transform(std::execution::par_unseq,
                     v.begin(),
                     v.end(),
                     v.begin(),
                     [](std::uint16_t u) { return ntohs(u); });
   }

private:
   std::unique_ptr<MessageImpl> p;
};

} // namespace awips
} // namespace scwx