labelsTranslator.h

Go to the documentation of this file.

 
#ifndef SI_FOLD_LABELSTRANSLATOR_H
#define SI_FOLD_LABELSTRANSLATOR_H
 
 
#include <map>
#include <string>
#include <si/graph/cgraph.h>
 
class Tree;
 
namespace aims
{
  class SelectionSet;
}
 
 
namespace sigraph
{

  class FoldLabelsTranslator : public std::map<std::string, std::string>
  {
  public:
    FoldLabelsTranslator() : std::map<std::string, std::string>() {}
    FoldLabelsTranslator( const std::string & filename ) 
      : std::map<std::string, std::string>()
    { readLabels( filename ); }
    FoldLabelsTranslator(  const MGraph & mg, 
                           const std::string & filename = "" ) 
      : std::map<std::string, std::string>()
    { makeFromModel( mg, filename ); }
    ~FoldLabelsTranslator() {}

    void readLabels( const std::string & filename );
    void makeFromModel( const MGraph & mg, const std::string & filename );
    void translate( CGraph & gr, const std::string & ilabel="name", 
                    const std::string & olabel="name", 
                    const std::string & altilabel = "" ) const;
    std::string lookupLabel( std::string label ) const;
 
  protected:
    bool readJFM95( const std::string & filename );
    bool readHierarchy( const std::string & filename );
    bool readSelection( const std::string & filename );
    bool makeFromModelJFM95( const MGraph & mg, const std::string & filename );
    bool makeFromModelHierarchy( const MGraph & mg, 
                                 const std::string & filename );
    /* bool makeFromModelSelection( const MGraph & mg, 
       const std::string & filename ); */
 
  private:
    void insertTree( const Tree & t );
    void insertTree( const Tree & t, const std::set<std::string> & mnames );
    void insertSelection( const aims::SelectionSet & );
  };
 
}
 
#endif