legocontainer.h

Go to the documentation of this file.

#ifndef LEGOCONTAINER_H
#define LEGOCONTAINER_H
 
#include "compat.h"
#include "decomp.h"
#include "legotexture.h"
#include "legotypes.h"
#include "mxstl/stlcompat.h"
 
// Note: dependency on LegoOmni
#include "lego/legoomni/include/legotextureinfo.h"
 
#pragma warning(disable : 4237)
 
struct LegoContainerInfoComparator {
    LegoBool operator()(const char* const& p_key0, const char* const& p_key1) const
    {
        return strcmp(p_key0, p_key1) > 0;
    }
};
 
template <class T>
class LegoContainerInfo : public map<const char*, T*, LegoContainerInfoComparator> {}; // [AI]
 
template <class T>
class LegoContainer {
public:
    LegoContainer() { m_ownership = TRUE; }
 
    virtual ~LegoContainer()
    {
#ifdef COMPAT_MODE
        typename LegoContainerInfo<T>::iterator it;
#else
        LegoContainerInfo<T>::iterator it;
#endif
        for (it = m_map.begin(); it != m_map.end(); it++) {
            // DECOMP: Use of const_cast here matches ~ViewLODListManager from 96 source.
            const char* const& key = (*it).first;
            delete[] const_cast<char*>(key);
 
            if (m_ownership) {
                delete (*it).second;
            }
        }
    }
 
    void Clear()
    {
#ifdef COMPAT_MODE
        for (typename LegoContainerInfo<T>::iterator it = m_map.begin(); it != m_map.end(); it++)
#else
        for (LegoContainerInfo<T>::iterator it = m_map.begin(); it != m_map.end(); it++)
#endif
        {
            delete (*it).second;
        }
    }
 
    T* Get(const char* p_name)
    {
        T* value = NULL;
 
#ifdef COMPAT_MODE
        typename LegoContainerInfo<T>::iterator it = m_map.find(p_name);
#else
        LegoContainerInfo<T>::iterator it = m_map.find(p_name);
#endif
 
        if (it != m_map.end()) {
            value = (*it).second;
        }
 
        return value;
    }
 
    void Add(const char* p_name, T* p_value)
    {
#ifdef COMPAT_MODE
        typename LegoContainerInfo<T>::iterator it = m_map.find(p_name);
#else
        LegoContainerInfo<T>::iterator it = m_map.find(p_name);
#endif
 
        char* name;
        if (it != m_map.end()) {
            name = const_cast<char*>((*it).first);
 
            if (m_ownership) {
                delete (*it).second;
            }
        }
        else {
            name = new char[strlen(p_name) + 1];
            strcpy(name, p_name);
        }
 
        m_map[name] = p_value;
    }
 
    void SetOwnership(LegoBool p_ownership) { m_ownership = p_ownership; }
 
protected:
    LegoBool m_ownership;       
    LegoContainerInfo<T> m_map; 
};
 
typedef pair<LegoTextureInfo*, BOOL> LegoCachedTexture;
 
typedef list<LegoCachedTexture> LegoCachedTextureList;
 
class LegoTextureContainer : public LegoContainer<LegoTextureInfo> {
public:
    ~LegoTextureContainer() override;
 
    LegoTextureInfo* GetCached(LegoTextureInfo* p_textureInfo);
 
    void EraseCached(LegoTextureInfo* p_textureInfo);
 
protected:
    LegoCachedTextureList m_cached; 
};
 
#endif // LEGOCONTAINER_H