比较VTK数据类型和基本数据类型

Question

VTK具有用于基本类型（浮点型，整数型，双精度型等）的typedef，并且为每种类型分配一个整数。 它们在此处指定。

例如， vtkDataArray的函数GetDataType()返回一个整数，该整数对应于一种类型。 我想将该整数与基本数据类型（浮点型，整数型，双精度型）进行比较。

有没有办法轻松地做到这一点？

我对此的用法是一个模板类，其参数T是标量。 我想检查数据集的标量点数据是否具有与T相同的数据类型。

现在，我要做的是类型大小比较：

vtkDataArray *scalars = image->GetPointData()->GetScalars();
if(scalars->GetDataTypeSize() != sizeof(T))
{
  std::cerr<<"Incompatible types"<<std::endl;
}

但显然， float和int均为4，因此它实际上并没有工作。

有任何想法吗 ？

Answer 1

从几个地方收集了信息之后，我得出结论，如果没有在两个类型列表之间自己创建映射，就无法编写信息。

因此，这是我发现的最优雅的方法：

我使用了norisknofun地图的想法，但我将其反转了。 我没有使用std::type_index()因为看来您可以直接从typeid()的结果中获取hash_code。 我将此映射放在将基本类型转换为VTK类型的函数中，因为它不仅可以用于比较，还可以用于其他目的（请参阅我的其他文章 ）。

#include <vtkType.h>

int GetVTKType(std::size_t hash_code)
{
    static std::map<std::size_t, long> typeMap;
    if(typeMap.empty())
    {
        typeMap[typeid(void).hash_code()]               = VTK_VOID;
        typeMap[typeid(char).hash_code()]               = VTK_CHAR;
        typeMap[typeid(signed char).hash_code()]        = VTK_SIGNED_CHAR;
        typeMap[typeid(unsigned char).hash_code()]      = VTK_UNSIGNED_CHAR;
        typeMap[typeid(short).hash_code()]              = VTK_SHORT;
        typeMap[typeid(unsigned short).hash_code()]     = VTK_UNSIGNED_SHORT;
        typeMap[typeid(int).hash_code()]                = VTK_INT;
        typeMap[typeid(unsigned int).hash_code()]       = VTK_UNSIGNED_INT;
        typeMap[typeid(long).hash_code()]               = VTK_LONG;
        typeMap[typeid(unsigned long).hash_code()]      = VTK_UNSIGNED_LONG;
        typeMap[typeid(float).hash_code()]              = VTK_FLOAT;
        typeMap[typeid(double).hash_code()]             = VTK_DOUBLE;
        typeMap[typeid(std::string).hash_code()]        = VTK_STRING;
        typeMap[typeid(long long).hash_code()]          = VTK_LONG_LONG;
        typeMap[typeid(unsigned long long).hash_code()] = VTK_UNSIGNED_LONG_LONG;
        typeMap[typeid(int64_t).hash_code()]            = VTK___INT64;
        typeMap[typeid(uint64_t).hash_code()]           = VTK_UNSIGNED___INT64;
    }
    return typeMap[hash_code];
}

因此，为了比较vtk数据类型和基本类型（我的模板参数T），我这样做：

vtkDataArray *scalars = image->GetPointData()->GetScalars();
if(scalars->GetDataType() != GetVTKType(typeid(T).hash_code()))
{
  std::cerr<<"Incompatible types"<<std::endl;
}

或者，如果我想像norisknofun那样做一个不错的比较函数，可以这样做：

template < class T > 
bool is_same(long vtkType)
{
  return vtkType != GetVTKType(typeid(T).hash_code())
}

// somewhere.cpp
if(!is_same<T>(scalars->GetDataType()))
{
  std::cerr<<"Incompatible types"<<std::endl;
}

Answer 2

我想您必须详细说明自己的映射。 您的编译器必须符合c ++ 11并激活RTTI，但现在大多数现代编译器都支持此功能。

我不知道什么是'bit'，'id_type'和'opaque'类型的C ++等价物...

#include <iostream>
#include <typeinfo>
#include <typeindex>
#include <cstdint>
#include <string>
#include <map>

// copy from vtk header, use #include <vtkType.h> instead
#define VTK_VOID            0
#define VTK_BIT             1
#define VTK_CHAR            2
#define VTK_SIGNED_CHAR    15
#define VTK_UNSIGNED_CHAR   3
#define VTK_SHORT           4
#define VTK_UNSIGNED_SHORT  5
#define VTK_INT             6
#define VTK_UNSIGNED_INT    7
#define VTK_LONG            8
#define VTK_UNSIGNED_LONG   9
#define VTK_FLOAT          10
#define VTK_DOUBLE         11
#define VTK_ID_TYPE        12
#define VTK_STRING         13
#define VTK_OPAQUE         14
#define VTK_LONG_LONG          16
#define VTK_UNSIGNED_LONG_LONG 17
#define VTK___INT64            18
#define VTK_UNSIGNED___INT64   19

// vtktypes
    typedef long vtktypes ;

    // standard c++ types
    typedef std::size_t   mytypes ;
    typedef std::map< vtktypes, mytypes> map_t;

template < class T > 
bool is_same(vtktypes x)
{
    static std::map< vtktypes, mytypes> _map;
    if(_map.empty())
    {
        _map[VTK_VOID           ]   = std::type_index(typeid(void)).hash_code();
        //_map[VTK_BIT              ]   = std::type_index(typeid(void)).hash_code();
        _map[VTK_CHAR           ]   = std::type_index(typeid(char)).hash_code();
        _map[VTK_SIGNED_CHAR    ]   = std::type_index(typeid(signed char)).hash_code();
        _map[VTK_UNSIGNED_CHAR  ]   = std::type_index(typeid(unsigned char)).hash_code();
        _map[VTK_SHORT          ]   = std::type_index(typeid(short)).hash_code();
        _map[VTK_UNSIGNED_SHORT ]   = std::type_index(typeid(unsigned short)).hash_code();
        _map[VTK_INT            ]   = std::type_index(typeid(int)).hash_code();
        _map[VTK_UNSIGNED_INT   ]   = std::type_index(typeid(unsigned int)).hash_code();
        _map[VTK_LONG           ]   = std::type_index(typeid(long)).hash_code();
        _map[VTK_UNSIGNED_LONG  ]   = std::type_index(typeid(unsigned long)).hash_code();
        _map[VTK_FLOAT          ]   = std::type_index(typeid(float)).hash_code();

        _map[VTK_DOUBLE         ]   = std::type_index(typeid(double)).hash_code();
        //_map[VTK_ID_TYPE          ]   = type_index(typeid()).hash_code();
        _map[VTK_STRING         ]   = std::type_index(typeid(std::string)).hash_code();
        //_map[VTK_OPAQUE           ]   = type_index(typeid(void)).hash_code();
        _map[VTK_LONG_LONG         ]= std::type_index(typeid(long long)).hash_code();
        _map[VTK_UNSIGNED_LONG_LONG]= std::type_index(typeid(unsigned long long)).hash_code();
        _map[VTK___INT64           ]= std::type_index(typeid(int64_t)).hash_code();
        _map[VTK_UNSIGNED___INT64  ]= std::type_index(typeid(uint64_t)).hash_code();
    }

    map_t::iterator it = _map.find(x);
    return (it != _map.end()) && it->second == std::type_index(typeid(T)).hash_code();
}

int main()
{
    std::cout << "is same ? " << is_same<char>(2) << std::endl ;
    std::cout << "is same ? " << is_same<std::string>(13) << std::endl ;    

    return 0;
}

就您而言，您可以这样使用它：

vtkDataArray *scalars = image->GetPointData()->GetScalars();
if(!is_same<T>(scalars->GetDataTypeSize()))
{
  std::cerr<<"Incompatible types"<<std::endl;
}