XmlFileWriter.h

Go to the documentation of this file.

 
#ifndef PANDORA_XML_FILE_WRITER_H
#define PANDORA_XML_FILE_WRITER_H 1
 
#include "Pandora/Pandora.h"
 
#include "Objects/CartesianVector.h"
#include "Objects/TrackState.h"
 
#include "Persistency/FileWriter.h"
 
#include "Xml/tinyxml.h"
 
namespace pandora
{
 
class XmlFileWriter : public FileWriter
{
public:
    XmlFileWriter(const pandora::Pandora &pandora, const std::string &fileName, const FileMode fileMode = APPEND);
 
    ~XmlFileWriter();
 
    template<typename T>
    StatusCode WriteVariable(const std::string &xmlKey, const T &t);
 
private:
    StatusCode WriteHeader(const ContainerId containerId);
    StatusCode WriteFooter();
    StatusCode WriteSubDetector(const SubDetector *const pSubDetector);
    StatusCode WriteLArTPC(const LArTPC *const pLArTPC);
    StatusCode WriteDetectorGap(const DetectorGap *const pDetectorGap);
    StatusCode WriteCaloHit(const CaloHit *const pCaloHit);
    StatusCode WriteTrack(const Track *const pTrack);
    StatusCode WriteMCParticle(const MCParticle *const pMCParticle);
    StatusCode WriteRelationship(const RelationshipId relationshipId, const void *address1, const void *address2, const float weight);
 
    TiXmlDocument      *m_pXmlDocument;         
    TiXmlElement       *m_pContainerXmlElement; 
    TiXmlElement       *m_pCurrentXmlElement;   
};
 
//------------------------------------------------------------------------------------------------------------------------------------------
 
template<typename T>
inline StatusCode XmlFileWriter::WriteVariable(const std::string &xmlKey, const T &t)
{
    if (!m_pCurrentXmlElement)
        return STATUS_CODE_FAILURE;
 
    TiXmlElement *const pTiXmlElement = new TiXmlElement(xmlKey);
    pTiXmlElement->LinkEndChild(new TiXmlText(TypeToStringPrecision(t)));
    m_pCurrentXmlElement->LinkEndChild(pTiXmlElement);
 
    return STATUS_CODE_SUCCESS;
}
 
template<>
inline StatusCode XmlFileWriter::WriteVariable(const std::string &xmlKey, const CartesianVector &t)
{
    return this->WriteVariable(xmlKey, TypeToStringPrecision(t.GetX()) + " " + TypeToStringPrecision(t.GetY()) + " " + TypeToStringPrecision(t.GetZ()));
}
 
template<>
inline StatusCode XmlFileWriter::WriteVariable(const std::string &xmlKey, const TrackState &t)
{
    return this->WriteVariable(xmlKey, TypeToStringPrecision(t.GetPosition().GetX()) + " " + TypeToStringPrecision(t.GetPosition().GetY()) + " " + TypeToStringPrecision(t.GetPosition().GetZ()) +
        " " + TypeToStringPrecision(t.GetMomentum().GetX()) + " " + TypeToStringPrecision(t.GetMomentum().GetY()) + " " + TypeToStringPrecision(t.GetMomentum().GetZ()));
}
 
} // namespace pandora
 
#endif // #ifndef PANDORA_XML_FILE_WRITER_H