Функция с общими входными параметрами

Question

Я относительно новичок в C++. Я написал функцию WriteToFile, которая записывает в текстовый файл (путь задается строкой a) 2D массива (хранится в 0-строки основного порядка, х строк, у колонок):

void WriteToFile(std::string a, const float *img, const int x, const int y) { 
    FILE *pFile; 
    const char * c = a.c_str(); 
    pFile = fopen(c, "w"); 
    for (int i = 0; i < x; i++){ 
     for (int j = 0; j < y; j++) 
      fprintf(pFile, "%5.5f\t", img[i*y + j]); 
     fprintf(pFile, "\n"); 
    } 
    fclose(pFile); 
} 

Теперь я хочу эту функцию, чтобы иметь дело также с int и double массивами. Для int он просто напечатает номера, как есть, и для двойного %5.10lf должен использоваться в fprintf. Я знаю, что это абсолютно возможно. Я нашел пару похожих вещей, но не понял, как справляться с входными аргументами. Конечно, я могу написать 3 разные функции, но я хочу узнать, как написать общую функцию.

Благодаря

Answer 1

Вы можете использовать шаблон функции и некоторые вспомогательные функции, чтобы получить строку формата.

Вот рабочая программа.

#include <cstdio> 
#include <string> 

template <typename T> char const* getFormatString(); 

template <> char const* getFormatString<int>() 
{ 
    return "%d\t"; 
} 

template <> char const* getFormatString<float>() 
{ 
    return "%5.5f\t"; 
} 

template <> char const* getFormatString<double>() 
{ 
    return "%15.10lf\t"; 
} 

template <typename T> 
void WriteToFile(std::string a, const T *img, const int x, const int y) { 
    FILE *pFile; 
    const char * c = a.c_str(); 
    pFile = fopen(c, "w"); 
    for (int i = 0; i < x; i++){ 
     for (int j = 0; j < y; j++) 
      fprintf(pFile, getFormatString<T>(), img[i*y + j]); 
     fprintf(pFile, "\n"); 
    } 
    fclose(pFile); 
} 

int main() 
{ 
    int img1[] = {1, 1, 1, 1}; 
    float img2[] = {1, 1, 1, 1}; 
    double img3[] = {1, 1, 1, 1}; 
    WriteToFile("int.img", img1, 2, 2); 
    WriteToFile("float.img", img2, 2, 2); 
    WriteToFile("double.img", img3, 2, 2); 
} 

Answer 2

Идея была бы

template <typename T> 
void WriteToFile(std::string a, const T *img, const int x, const int y) { 
} 

, а затем

inline 
void write(FILE *file, double) { //write double } 

inline 
void write(FILE *file, int) { // write } 

Answer 3

Если вам нужно использовать другой формат строку для каждого типа, вы можете использовать шаблонную функцию и в строке формата в качестве параметра:

template<typename T> void WriteToFile(std::string a, const T *img, 
         const int x, const int y, std::string formatStr) { 
... 
     fprintf(pFile, formatStr.c_str(), img[i*y + j]); 
... 
} 

Answer 4

«Теперь я хочу, чтобы эта функция имела дело с int и double массивами. "

Вы можете использовать шаблонные функции для обработки различных типов массивов, а также использовать стандартную библиотеку C++ I/O

template <typename T> 
void WriteToFile(std::string a, const T *img, const int x, const int y) { 
    std::ofstream file(a); 

    for (int i = 0; i < x; i++){ 
     for (int j = 0; j < y; j++) 
      file << std::fixed << std::setw(5) << std::setprecision(5) 
       << img[i*y + j] << "\t" << std::endl; 
    } 
    file.close(); 
} 

Вы можете рассмотреть, чтобы иметь специализированные реализации для разных нужд форматирования.

Answer 5

У вас есть такие опции, как перегрузка и функции шаблона. Я просто хочу ввести способность, которая поставляется с C++ 11. Использование std::is_same.

У этого есть свои плюсы и минусы. Например, предположим, что тип входа ограничен int и float:

#include <type_traits> 

template <typename T> 
void WriteToFile(std::string a, const T *img, const int x, const int y) 
{ 
    const char *format = std::is_same<T, int>::value? "%i\t" : "%5.5f\t"; 

    ... 

     fprintf(pFile, format, img[i*y + j]); 

    ... 
} 

Обратите внимание, что этот подход к обобщению и ваш код не является максимально стандартизован.