//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
//=============================================================================//

#ifndef SPRITETRAIL_H
#define SPRITETRAIL_H
#ifdef _WIN32
#pragma once
#endif

#include "Sprite.h"

#if defined(CLIENT_DLL)
#define CSpriteTrail C_SpriteTrail
#endif

//-----------------------------------------------------------------------------
// Sprite trail
//-----------------------------------------------------------------------------
struct TrailPoint_t {
    DECLARE_SIMPLE_DATADESC();

    Vector m_vecScreenPos;
    float m_flDieTime;
    float m_flTexCoord;
    float m_flWidthVariance;
};

class CSpriteTrail : public CSprite {
    DECLARE_CLASS(CSpriteTrail, CSprite);
    DECLARE_DATADESC();
    DECLARE_NETWORKCLASS();
    DECLARE_PREDICTABLE();

   public:
    CSpriteTrail(void);

    // Sets parameters of the sprite trail
    void SetLifeTime(float time);
    void SetStartWidth(float flStartWidth);
    void SetEndWidth(float flEndWidth);
    void SetStartWidthVariance(float flStartWidthVariance);
    void SetTextureResolution(float flTexelsPerInch);
    void SetMinFadeLength(float flMinFadeLength);
    void SetSkybox(const Vector &vecSkyboxOrigin, float flSkyboxScale);

    // Is the trail in the skybox?
    bool IsInSkybox() const;
    void Spawn(void);
    void Precache(void);
    void SetTransmit(bool bTransmit = true) {
        m_bDrawForMoveParent = bTransmit;
    }

#if defined(CLIENT_DLL)
    // Client only code
    virtual int DrawModel(int flags);
    virtual const Vector &GetRenderOrigin(void);
    virtual const QAngle &GetRenderAngles(void);

    // On data update
    virtual void OnPreDataChanged(DataUpdateType_t updateType);
    virtual void OnDataChanged(DataUpdateType_t updateType);
    virtual void GetRenderBounds(Vector &mins, Vector &maxs);
    virtual void ClientThink();

    virtual bool ValidateEntityAttachedToPlayer(bool &bShouldRetry);

#else
    // Server only code

    virtual int ShouldTransmit(const CCheckTransmitInfo *pInfo);
    static CSpriteTrail *SpriteTrailCreate(const char *pSpriteName,
                                           const Vector &origin, bool animate);

#endif

   private:
#if defined(CLIENT_DLL)
    enum {
        // NOTE: # of points max must be a power of two!
        MAX_SPRITE_TRAIL_POINTS = 256,
        MAX_SPRITE_TRAIL_MASK = MAX_SPRITE_TRAIL_POINTS - 1,
    };

    TrailPoint_t *GetTrailPoint(int n);
    void UpdateTrail(void);
    void ComputeScreenPosition(Vector *pScreenPos);
    void ConvertSkybox();
    void UpdateBoundingBox(void);

    TrailPoint_t m_vecSteps[MAX_SPRITE_TRAIL_POINTS];
    int m_nFirstStep;
    int m_nStepCount;
    float m_flUpdateTime;
    Vector m_vecPrevSkyboxOrigin;
    float m_flPrevSkyboxScale;
    Vector m_vecRenderMins;
    Vector m_vecRenderMaxs;
#endif

    CNetworkVar(
        float,
        m_flLifeTime);  // Amount of time before a new trail segment fades away
    CNetworkVar(float, m_flStartWidth);          // The starting scale
    CNetworkVar(float, m_flEndWidth);            // The ending scale
    CNetworkVar(float, m_flStartWidthVariance);  // The starting scale
    CNetworkVar(float, m_flTextureRes);  // Texture resolution along the trail
    CNetworkVar(float, m_flMinFadeLength);  // The end of the trail must fade
                                            // out for this many units
    CNetworkVector(m_vecSkyboxOrigin);      // What's our skybox origin?
    CNetworkVar(float, m_flSkyboxScale);    // What's our skybox scale?

    string_t m_iszSpriteName;
    bool m_bAnimate;
    bool m_bDrawForMoveParent;

#if defined(CLIENT_DLL)
   public:
    void SetUpdateTime(float setTo) { m_flUpdateTime = setTo; }
#endif
};

#endif  // SPRITETRAIL_H