NICE_VisLearning/cbaselib/ClassNames.h

/** 
* @file ClassNames.h
* @brief simple interface for class name confusion
* @author Erik Rodner
* @date 02/08/2008

*/
#ifndef CLASSNAMESINCLUDE
#define CLASSNAMESINCLUDE

#include "core/image/ImageT.h"
#include "core/vector/VectorT.h"
#include "core/vector/MatrixT.h"
 
#include <string>
#include <map>

#include "core/basics/Config.h"
#include "core/basics/Persistent.h"


namespace OBJREC {

/** simple interface for class name confusion */
class ClassNames : public NICE::Persistent
{

    protected:

		std::map<std::string, std::string> tbl_code_text;
		std::map<std::string, std::string> tbl_text_code;
		std::map<int, std::string> tbl_classno_code;
		std::map<std::string, int> tbl_code_classno;
		std::map<long, int> tbl_color_classno;
		std::map<int, long> tbl_classno_color;

		int maxClassNo;

    public:
  
	/** simple constructor */
	ClassNames();
    
	/** copy constructor */
	ClassNames ( const ClassNames & cn );
  
	/** create sub selection */
	ClassNames ( const ClassNames & cn, 
		     const std::string & classselection );

	/** simple destructor */
	virtual ~ClassNames();
    
	/** get the readable class number of a class number */
	std::string text ( int classno ) const;

	/** get the class code (e.g. ascii code) of a class number */
	std::string code ( int classno ) const;

	/** get the class number corresponding to the code */
	int classno ( std::string code ) const;
	
	/** get the class number corresponding to the name */
	int classnoFromText ( std::string text ) const;

	/** number of classes */
	int numClasses () const;
	void getSelection ( const std::string & classselection,
			    std::set<int> & classnos ) const;

	/** add new class information 
	 * @param classno class number
	 * @param code class code (such as ascii number or just the name)
	 * @param text class name as readable ascii representation
	 */
	void addClass ( int classno, const std::string & code, 
	    const std::string & text );

	bool readFromConfig ( const NICE::Config & datasetconf, 
		    const std::string & classselection );

	/** is there any class with this class number */
	bool existsClassno ( int classno ) const;
	/** is there any class with this class code */
	bool existsClassCode ( const std::string & classcode ) const;

	int getMaxClassno () const;

	void getRGBColor ( int classno, int & r, int & g, int & b ) const;
	
	void getClassnoFromColor ( int & classno, int r, int g, int b ) const;

	/** colorize a labeled image using color information given in the class 
	 *  information file */
	void labelToRGB ( const NICE::Image & img, NICE::ColorImage & rgb ) const;

	int getBackgroundClass () const;

	/** restore classnames in the format <classtext> <classcode> <classno> */ 
	void restore (std::istream & is, int format = 0);

	/** store classnames in the format <classtext> <classcode> <classno> */ 
	void store (std::ostream & os, int format = 0) const;

	/** clear class information */
    void clear ();
};


} // namespace

#endif