cudaColormap.h

Go to the documentation of this file.

/*
 * Copyright (c) 2019, NVIDIA CORPORATION. All rights reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */
 
#ifndef __CUDA_COLORMAP_H__
#define __CUDA_COLORMAP_H__
 
 
#include "cudaFilterMode.h"
#include "imageFormat.h"
 
 
enum cudaColormapType
{
        // Palettized
        COLORMAP_INFERNO,   
        COLORMAP_MAGMA,     
        COLORMAP_PARULA,    
        COLORMAP_PLASMA,    
        COLORMAP_TURBO,     
        COLORMAP_VIRIDIS,   
        // Palettized (inverted)
        COLORMAP_INFERNO_INVERTED,   
        COLORMAP_MAGMA_INVERTED,     
        COLORMAP_PARULA_INVERTED,    
        COLORMAP_PLASMA_INVERTED,    
        COLORMAP_TURBO_INVERTED,     
        COLORMAP_VIRIDIS_INVERTED,   
        // Parametric
        COLORMAP_FLOW,          
        COLORMAP_NONE,          
        COLORMAP_LINEAR,        
        COLORMAP_DEFAULT = COLORMAP_VIRIDIS
};
 
cudaColormapType cudaColormapFromStr( const char* colormap );
 
const char* cudaColormapToStr( cudaColormapType colormap );
 
template<typename T>
cudaError_t cudaColormap( float* input, T* output,
                                         size_t width, size_t height,
                                         const float2& input_range=make_float2(0,255),
                                         cudaDataFormat input_format=FORMAT_DEFAULT,
                                         cudaColormapType colormap=COLORMAP_DEFAULT,
                                         cudaStream_t stream=NULL)                                      { return cudaColormap(input, (void*)output, width, height, input_range, input_format, imageFormatFromType<T>(), colormap, stream); }
                                         
cudaError_t cudaColormap( float* input, void* output,
                                         size_t width, size_t height,
                                         const float2& input_range=make_float2(0,255),
                                         cudaDataFormat input_format=FORMAT_DEFAULT,
                                         imageFormat output_format=IMAGE_UNKNOWN,
                          cudaColormapType colormap=COLORMAP_DEFAULT,
                                         cudaStream_t stream=NULL);
 
template<typename T>
cudaError_t cudaColormap( float* input, size_t input_width, size_t input_height,
                                         T* output, size_t output_width, size_t output_height,
                                         const float2& input_range=make_float2(0,255),
                                         cudaDataFormat input_format=FORMAT_DEFAULT,
                          cudaColormapType colormap=COLORMAP_DEFAULT,
                                         cudaFilterMode filter=FILTER_LINEAR,
                                         cudaStream_t stream=NULL )                                     { return cudaColormap(input, input_width, input_height, output, output_width, output_height, input_range, input_format, imageFormatFromType<T>(), colormap, filter, stream); }
                                         
cudaError_t cudaColormap( float* input, size_t input_width, size_t input_height,
                                         void* output, size_t output_width, size_t output_height,
                                         const float2& input_range=make_float2(0,255),
                                         cudaDataFormat input_format=FORMAT_DEFAULT,
                                         imageFormat output_format=IMAGE_UNKNOWN,
                          cudaColormapType colormap=COLORMAP_DEFAULT,
                                         cudaFilterMode filter=FILTER_LINEAR,
                                         cudaStream_t stream=NULL );
 
cudaError_t cudaColormapInit();
 
cudaError_t cudaColormapFree();
 
float4* cudaColormapPalette( cudaColormapType colormap );
 
 
#endif