DotConverterTreePart.hxx

Go to the documentation of this file.

 
#pragma once
 
#include <alphabet/string/RankedSymbol.h>
 
#include <tree/ranked/RankedTree.h>
 
namespace convert {
 
class DotConverterTree {
    template < class SymbolType >
    static unsigned convertRecursive ( const ext::tree < SymbolType > & node, ext::ostream & oss, unsigned & nodeIdCounter );
 
public:
    template < class SymbolType >
    static void convertInternal ( ext::ostream & oss, const tree::RankedTree < SymbolType > & tree );
};
 
template < class SymbolType >
void DotConverterTree::convertInternal ( ext::ostream & oss, const tree::RankedTree < SymbolType > & tree ) {
    unsigned nodeIdCounter = 0;
    convertRecursive ( tree.getContent ( ), oss, nodeIdCounter );
}
 
template < class SymbolType >
unsigned DotConverterTree::convertRecursive ( const ext::tree < SymbolType > & node, ext::ostream & oss, unsigned & nodeIdCounter ) {
    unsigned id = nodeIdCounter ++;
 
    oss << id << "[label=\"" + factory::StringDataFactory::toString ( node.getData ( ) ) + "\", shape=plaintext];" << std::endl;
 
    for ( const ext::tree < SymbolType > & child : node.getChildren ( ) ) {
        size_t childId = convertRecursive ( child, oss, nodeIdCounter );
        oss << id << " -> " << childId << ";" << std::endl;
    }
 
    return id;
}
 
} /* namespace convert */