Можно ли взять тип в качестве аргумента в функции?

Question

Я пытаюсь написать функцию для класса базы данных, которая в основном представляет собой оболочку вокруг hash_map объектов (например, фигур), индексированных идентификационными номерами, которые будут искать идентификатор и передавать его соответствующему типу указателя.

например. Я хотел бы быть в состоянии сделать что-то вроде этого:

Circle* shapeToLookup = NULL; 
int idNum = 12; 
database.lookup(idNum, circleToLookup); 
if(circleToLookup != NULL) 
{ 
    // Do stuff with the circle. 
} 

и есть база данных о типе аргумента. Есть ли способ сделать это без перегрузки функции (lookup(int, Circle*), lookup(int, Rect*), ad nauseum)? Можете ли вы объявить такую ​​функцию, как lookup(int, Shape*), и знать, какой тип она дана?

Спасибо!

Answer 1

Вы можете сделать это с помощью шаблона.

Редактировать: новая реализация основана на дополнительной информации. Если mymap является std::map<int, Shape*>:

template <typename T> 
void lookup(int idNum, T* &ptr) { 
    auto it = mymap.find(idNum); 
    if (it == mymap.end()) { 
     ptr = 0; 
    } else { 
     ptr = dynamic_cast<T*>(*it); // Shape must have a virtual member function 
    } 
} 

Или вы можете предпочесть:

template <typename T> 
T* lookup(int idNum) { 
    auto it = mymap.find(idNum); 
    if (it == mymap.end()) { 
     return 0; 
    } 
    return dynamic_cast<T*>(*it); 
} 

Затем вызовите его, как Circle *circle = database.lookup<Circle>(123);

Очевидно полиморфных контейнеров целая куча удовольствия сами по себе, но я буду предположим, что у вас это отсортировано. Там может быть shared_ptr где-то, что я забыл.

старой реализации, когда я думал, что DB может хранить копии POD:

template <typename T> 
void lookup(int idNum, T* &ptr) { 
    void *theresult = // something based on idNum 

    // some check needed here that theresult really is the right type. 
    // how you do this depends on the database, but suppose that 
    // the database gives us some integer "type" which indicates the type 
    if (type != type_constant<T>::value) { 
     ptr = 0; 
    } else { 
     ptr = static_cast<T*>(theresult); 
    } 
} 

type_constant является примером "черт типа" вы реализуете это нравится:

template <typename T> 
struct type_constant {}; 

template <> 
struct type_constant<Circle> { 
    static const int value = 1; 
}; 

template <> 
struct type_constant<Rectangle> { 
    static const int value = 2; 
}; 

// etc... 

Answer 2

Другие объяснили, как передать тип функции (используя шаблоны функций). Я хотел бы дать другую точку зрения:

Это может быть еще лучше ввести новую виртуальную функцию Shape, а затем переместить Do stuff with the Circle части в переопределение этой виртуальной функции в Cricle классе.

Таким образом, вы удаляете необходимость знать тип. Вы просто получаете объект Shape из своей базы данных, а затем вызываете функцию doStuff() - и она делает правильную вещь в зависимости от фактического типа Shape. Хороший вариант использования для виртуальной функции. :-)

Конечно, это может быть более или менее просто, в зависимости от того, что на самом деле делает Do stuff.

Answer 3

template <T> 
Database::lookup(int idNum, T TobjectToLookup) 
{ 
    // inside here, T is the type of the object passed in/ 
}