cuArrays.h

Go to the documentation of this file.

 
// code guard
#ifndef __CUARRAYS_H
#define __CUARRAYS_H
 
// cuda dependencies
#include <cuda.h>
#include <driver_types.h>
 
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <ctime>
 
 
template <typename T>
class cuArrays{
 
public:
    int height; 
    int width;  // y, col, across, range, along the sight
    int size;   // one image size, height*width
    int countW; // number of images along width direction
    int countH; // number of images along height direction
    int count;  // countW*countH, number of images
    T* devData; // pointer to data in device (gpu) memory
    T* hostData; // pointer to data in host (cpu) memory
 
    bool is_allocated; // whether the data is allocated in device memory
    bool is_allocatedHost; // whether the data is allocated in host memory
 
    // default constructor, empty
    cuArrays() : width(0), height(0), size(0), countW(0), countH(0), count(0),
        is_allocated(0), is_allocatedHost(0),
        devData(0), hostData(0) {}
 
    // constructor for single image
    cuArrays(size_t h, size_t w) : width(w), height(h), countH(1), countW(1), count(1),
        is_allocated(0), is_allocatedHost(0),
        devData(0), hostData(0)
    {
        size = w*h;
    }
 
    // constructor for multiple images with a total count
    cuArrays(size_t h, size_t w, size_t n) : width(w), height(h), countH(1), countW(n), count(n),
        is_allocated(0), is_allocatedHost(0),
        devData(0), hostData(0)
    {
        size = w*h;
    }
 
    // constructor for multiple images with (countH, countW)
    cuArrays(size_t h, size_t w, size_t ch, size_t cw) : width(w), height(h), countW(cw), countH(ch),
        is_allocated(0), is_allocatedHost(0),
        devData(0), hostData(0)
    {
        size = w*h;
        count = countH*countW;
    }
 
    // memory allocation
    void allocate();
    void allocateHost();
    void deallocate();
    void deallocateHost();
 
    // copy data between device and host memories
    void copyToHost(cudaStream_t stream);
    void copyToDevice(cudaStream_t stream);
 
    // get the total size
    size_t getSize()
    {
        return size*count;
    }
 
    // get the total size in byte
    inline long getByteSize()
    {
        return width*height*count*sizeof(T);
    }
 
    // destructor
    ~cuArrays()
    {
        if(is_allocated)
            deallocate();
        if(is_allocatedHost)
            deallocateHost();
    }
 
    // set zeroes
    void setZero(cudaStream_t stream);
    // output when debugging
    void debuginfo(cudaStream_t stream) ;
    void debuginfo(cudaStream_t stream, float factor);
    // write to files
    void outputToFile(std::string fn, cudaStream_t stream);
    void outputHostToFile(std::string fn);
 
};
 
#endif //__CUARRAYS_H
//end of file