/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#include "CryLegacy_precompiled.h"
#include "XMLCPB_AttrSetTableReader.h"
#include "XMLCPB_Reader.h"

using namespace XMLCPB;

CAttrSetTableReader::CAttrSetTableReader(IGeneralMemoryHeap* pHeap)
    : m_setAddrs(NAlloc::GeneralHeapAlloc(pHeap))
    , m_numAttrs(NAlloc::GeneralHeapAlloc(pHeap))
    , m_buffer(pHeap)
{
}

//////////////////////////////////////////////////////////////////////////
// returns the num of attributes that a given AttrSet has

uint32 CAttrSetTableReader::GetNumAttrs(AttrSetID setId) const
{
    assert(setId < m_numAttrs.size());
    uint32 num = m_numAttrs[setId];
    return num;
}


//////////////////////////////////////////////////////////////////////////
// returns the header of an attr (16bits value that combines datatype+tagId)

uint16 CAttrSetTableReader::GetHeaderAttr(AttrSetID setId, uint32 indAttr) const
{
    assert(setId < m_setAddrs.size());
    assert(indAttr < m_numAttrs[setId]);

    FlatAddr flatAddr = m_setAddrs[setId];
    uint16* pHeader = (uint16*)(m_buffer.GetPointer(flatAddr));
    pHeader += indAttr;
    return *pHeader;
}

//////////////////////////////////////////////////////////////////////////

void CAttrSetTableReader::ReadFromMemory(CReader& Reader, const uint8* pData, uint32 dataSize, const SFileHeader& headerInfo, uint32& outReadLoc)
{
    assert(m_numAttrs.empty() && m_setAddrs.empty());

    m_numAttrs.resize(headerInfo.m_numAttrSets);
    m_setAddrs.resize(headerInfo.m_numAttrSets);

    uint8* pNumAttrsTable = &(m_numAttrs[0]);
    Reader.ReadDataFromMemory(pData, dataSize, pNumAttrsTable, sizeof(*pNumAttrsTable) * headerInfo.m_numAttrSets, outReadLoc);

    FlatAddr16* pSetAddrsTable = &(m_setAddrs[0]);
    Reader.ReadDataFromMemory(pData, dataSize, pSetAddrsTable, sizeof(*pSetAddrsTable) * headerInfo.m_numAttrSets, outReadLoc);

    m_buffer.ReadFromMemory(Reader, pData, dataSize, headerInfo.m_sizeAttrSets, outReadLoc);
}