#ifndef XGBOOST_FMAP_H
#define XGBOOST_FMAP_H
/*!
 * \file xgboost_fmap.h
 * \brief helper class that holds the feature names and interpretations
 * \author Tianqi Chen: tianqi.tchen@gmail.com
 */
#include <vector>
#include <string>
#include <cstring>
#include "xgboost_utils.h"

namespace xgboost{
    namespace utils{
        /*! \brief helper class that holds the feature names and interpretations */
        class FeatMap{
        public:
            enum Type{
                kIndicator  = 0,
                kQuantitive = 1
            };
        public:
            /*! \brief load feature map from text format */
            inline void LoadText( const char *fname ){
                FILE *fi = utils::FopenCheck( fname, "r" );
                this->LoadText( fi );
                fclose( fi );
            }
            /*! \brief load feature map from text format */
            inline void LoadText( FILE *fi ){
                int fid;
                char fname[256], ftype[256];
                while( fscanf( fi, "%d%s%s", &fid, fname, ftype ) == 3 ){
                    utils::Assert( fid == (int)names_.size(), "invalid fmap format" );
                    names_.push_back( std::string(fname) );
                    types_.push_back( GetType( ftype ) );
                }
            }
            /*! \brief number of known features */
            size_t size( void ) const{
                return names_.size();
            }
            /*! \brief return name of specific feature */
            const char* name( size_t idx ) const{
                utils::Assert( idx < names_.size(), "utils::FMap::name feature index exceed bound" );
                return names_[ idx ].c_str();
            }
            /*! \brief return type of specific feature */
            const Type& type( size_t idx ) const{
                utils::Assert( idx < names_.size(), "utils::FMap::name feature index exceed bound" );
                return types_[ idx ];
            }
        private:
            inline static Type GetType( const char *tname ){
                if( !strcmp( "i", tname ) ) return kIndicator;        
                if( !strcmp( "q", tname ) ) return kQuantitive;
                utils::Error("unknown feature type, use i for indicator and q for quantity");
                return kIndicator;
            }
        private:
            /*! \brief name of the feature */
            std::vector<std::string> names_;
            /*! \brief type of the feature */
            std::vector<Type>        types_;
        };
    }; // namespace utils
}; // namespace xgboost
#endif // XGBOOST_FMAP_H