#pragma once

#include "RetroTypes.hpp"
#include "CToken.hpp"
#include "CAnimPOIData.hpp"
#include "zeus/CVector3f.hpp"

namespace urde {
class IObjectStore;

class CFBStreamedCompression {
  friend class CFBStreamedAnimReader;
  friend class CFBStreamedAnimReaderTotals;
  friend class CFBStreamedPairOfTotals;

public:
  struct Header {
    u32 unk0;
    float duration;
    float interval;
    u32 rootBoneId;
    u32 looping;
    u32 rotDiv;
    float translationMult;
    u32 boneChannelCount;
    u32 unk3;

    void read(CInputStream& in) {
      /* unk0 */
      unk0 = in.readUint32Big();
      /* duration */
      duration = in.readFloatBig();
      /* interval */
      interval = in.readFloatBig();
      /* rootBoneId */
      rootBoneId = in.readUint32Big();
      /* looping */
      looping = in.readUint32Big();
      /* rotDiv */
      rotDiv = in.readUint32Big();
      /* translationMult */
      translationMult = in.readFloatBig();
      /* boneChannelCount */
      boneChannelCount = in.readUint32Big();
      /* unk3 */
      unk3 = in.readUint32Big();
    }
  };

private:
  bool m_pc;
  u32 x0_scratchSize;
  CAssetId x4_evnt;
  TLockedToken<CAnimPOIData> x8_evntToken;
  std::unique_ptr<u32[]> xc_rotsAndOffs;
  float x10_averageVelocity;
  zeus::CVector3f x14_rootOffset;

  u8* ReadBoneChannelDescriptors(u8* out, CInputStream& in);
  u32 ComputeBitstreamWords(const u8* chans);
  std::unique_ptr<u32[]> GetRotationsAndOffsets(u32 words, CInputStream& in);
  float CalculateAverageVelocity(const u8* chans);

public:
  CFBStreamedCompression(CInputStream& in, IObjectStore& objStore, bool pc);
  const Header& MainHeader() const { return *reinterpret_cast<const Header*>(xc_rotsAndOffs.get()); }
  const u32* GetTimes() const;
  const u8* GetPerChannelHeaders() const;
  const u8* GetBitstreamPointer() const;
  bool IsLooping() const { return MainHeader().looping; }
  CCharAnimTime GetAnimationDuration() const { return MainHeader().duration; }
  float GetAverageVelocity() const { return x10_averageVelocity; }
  const zeus::CVector3f& GetRootOffset() const { return x14_rootOffset; }
  bool HasPOIData() const { return x8_evntToken; }
  const std::vector<CBoolPOINode>& GetBoolPOIStream() const { return x8_evntToken->GetBoolPOIStream(); }
  const std::vector<CInt32POINode>& GetInt32POIStream() const { return x8_evntToken->GetInt32POIStream(); }
  const std::vector<CParticlePOINode>& GetParticlePOIStream() const { return x8_evntToken->GetParticlePOIStream(); }
  const std::vector<CSoundPOINode>& GetSoundPOIStream() const { return x8_evntToken->GetSoundPOIStream(); }
};

} // namespace urde