source: GTP/trunk/Lib/Illum/IBRBillboardCloudTrees/OGRE/src/LBBCKMeansClusterSerializer.cpp @ 745

#include "LBBCKMeansClusterSerializer.h"

namespace LBBC {

KMeansClusterSerializer::KMeansClusterSerializer() {
}

KMeansClusterSerializer::~KMeansClusterSerializer() {
}

void KMeansClusterSerializer::readBillboardCloud(TiXmlDocument *document, bool groupedBillboards)
{
        TiXmlNode *billboardCloudNode = document->FirstChild("billboardCloud");
        TiXmlNode *billboardGroupsNode = billboardCloudNode->FirstChild("billboardGroups");

        unsigned int numBillboardGroups = Ogre::StringConverter::parseInt(billboardGroupsNode->ToElement()->Attribute("count"));
        
        for (unsigned int iBillboardGroup = 0; iBillboardGroup < numBillboardGroups; iBillboardGroup++)
        {
                TiXmlNode *billboardGroupNode = billboardGroupsNode->FirstChild("billboardGroup");

                unsigned int numBillboards = Ogre::StringConverter::parseInt(billboardGroupNode->ToElement()->Attribute("count"));

                if (!groupedBillboards)
                {
                        for (TiXmlNode *billboardNode = billboardGroupNode->FirstChild("billboard"); billboardNode; billboardNode = billboardNode->NextSibling() )
                        {
                                BBC::BillboardPtr billboard = BBC::BillboardPtr( new BBC::Billboard() );
                                BBC::BillboardClusterDataPtr billboardClusterData = BBC::BillboardClusterDataPtr( new BillboardKMeansClusterData() );
                                mBillboardCloud->addBillboard(billboard);
                                billboard->setBillboardClusterData(billboardClusterData);
                                billboard->getBillboardClusterData()->readBillboardClusterData(billboardNode, -1);
                        }
                }
                else
                {
                        BBC::BillboardGroupPtr billboardGroup = BBC::BillboardGroupPtr( new BBC::BillboardGroup() );
                        mBillboardCloud->addBillboardGroup(billboardGroup);

                        for (TiXmlNode *billboardNode = billboardGroupNode->FirstChild("billboard"); billboardNode; billboardNode = billboardNode->NextSibling() )
                        {
                                BBC::BillboardPtr billboard = BBC::BillboardPtr( new BBC::Billboard() );
                                BBC::BillboardClusterDataPtr billboardClusterData = BBC::BillboardClusterDataPtr( new BillboardKMeansClusterData() );
                                mBillboardCloud->addBillboard(billboard);
                                billboard->setBillboardClusterData(billboardClusterData);
                                billboard->getBillboardClusterData()->readBillboardClusterData(billboardNode, iBillboardGroup);
                                billboardGroup->addBillboardHandle(billboard->getBillboardHandle());
                        }
                }
        }
}

void KMeansClusterSerializer::writeBillboardCloud(TiXmlDocument *document, bool groupedBillboards)
{
        TiXmlNode *billboardCloudNode = document->InsertEndChild(TiXmlElement("billboardCloud"))->ToElement();

        TiXmlNode *billboardGroupsNode = billboardCloudNode->InsertEndChild(TiXmlElement("billboardGroups"))->ToElement();      

        if (!groupedBillboards)
        {
                billboardGroupsNode->ToElement()->SetAttribute("count",Ogre::StringConverter::toString(1));
                TiXmlNode *billboardGroupNode = billboardGroupsNode->InsertEndChild(TiXmlElement("billboardGroup"))->ToElement();

                unsigned int numSerializedBillboards = 0;
                unsigned int numSerializedEntities = 0;
                for (unsigned int iBillboard = 0; iBillboard < mBillboardCloud->getNumBillboards(); iBillboard++)
                {
                        BBC::BillboardPtr billboard = mBillboardCloud->getBillboard(iBillboard);
                        BBC::EntityClusterPtr entityCluster = billboard->getBillboardClusterData()->getEntityCluster();

                        // If the Billboard has Entities...
                        if (entityCluster->getNumEntitiesClusterData() > 0)
                        {
                                BillboardKMeansClusterData *billboardKMeansClusterData = (BillboardKMeansClusterData*)billboard->getBillboardClusterData().get();
                                billboardKMeansClusterData->writeBillboardClusterData(billboardGroupNode, -1);  
                                numSerializedBillboards++;
                                numSerializedEntities = numSerializedEntities + entityCluster->getNumEntitiesClusterData();
                        }
                }
                billboardGroupNode->ToElement()->SetAttribute("count",Ogre::StringConverter::toString(numSerializedBillboards));
                Ogre::LogManager::getSingleton().logMessage("Num.Serialized Billboards:" + Ogre::StringConverter::toString(numSerializedBillboards) + " with " + Ogre::StringConverter::toString(numSerializedEntities) + " Entities");
        }
        else
        {
                billboardGroupsNode->ToElement()->SetAttribute("count",Ogre::StringConverter::toString(mBillboardCloud->getNumBillboardGroups()));

                unsigned int numSerializedBillboards = 0;
                unsigned int numSerializedEntities = 0;

                for (unsigned int iBillboardGroup = 0; iBillboardGroup < mBillboardCloud->getNumBillboardGroups(); iBillboardGroup++)
                {
                        TiXmlNode *billboardGroupNode = billboardGroupsNode->InsertEndChild(TiXmlElement("billboardGroup"))->ToElement();

                        for (unsigned int iBillboard = 0; iBillboard < mBillboardCloud->getBillboardGroup(iBillboardGroup)->getNumBillboards(); iBillboard++)
                        {
                                BBC::BillboardPtr billboard = mBillboardCloud->getBillboard(mBillboardCloud->getBillboardGroup(iBillboardGroup)->getBillboardHandle(iBillboard));
                                BBC::EntityClusterPtr entityCluster = billboard->getBillboardClusterData()->getEntityCluster();

                                // If the Billboard has Entities...
                                if (entityCluster->getNumEntitiesClusterData() > 0)
                                {
                                        BillboardKMeansClusterData *billboardKMeansClusterData = (BillboardKMeansClusterData*)billboard->getBillboardClusterData().get();
                                        billboardKMeansClusterData->writeBillboardClusterData(billboardGroupNode, iBillboardGroup);     
                                        numSerializedBillboards++;
                                        numSerializedEntities = numSerializedEntities + entityCluster->getNumEntitiesClusterData();
                                }
                        }
                        billboardGroupNode->ToElement()->SetAttribute("count",Ogre::StringConverter::toString(mBillboardCloud->getBillboardGroup(iBillboardGroup)->getNumBillboards()));
                }

                Ogre::LogManager::getSingleton().logMessage("Num.Serialized Billboards:" + Ogre::StringConverter::toString(numSerializedBillboards) + " with " + Ogre::StringConverter::toString(numSerializedEntities) + " Entities");
        }
}

}