101 lines
2.4 KiB
C
101 lines
2.4 KiB
C
|
/*
|
||
|
|
* cuArrays.h
|
||
|
|
* Header file for declaring a group of images
|
||
|
|
*
|
||
|
|
* Lijun Zhu
|
||
|
|
* Seismo Lab, Caltech
|
||
|
|
* V1.0 11/29/2016
|
||
|
|
*/
|
||
|
|
|
||
|
|
#ifndef __CUIMAGES_H
|
||
|
|
#define __CUIMAGES_H
|
||
|
|
|
||
|
|
#include <cuda.h>
|
||
|
|
#include <driver_types.h>
|
||
|
|
|
||
|
|
#include <iostream>
|
||
|
|
#include <fstream>
|
||
|
|
#include <cstdlib>
|
||
|
|
#include <ctime>
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
template <typename T>
|
||
|
|
class cuArrays{
|
||
|
|
public:
|
||
|
|
int height; // x, row, down, length, azimuth, along the track
|
||
|
|
int width; // y, col, across, range, along the sight
|
||
|
|
int size; // chip size, heigh*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;
|
||
|
|
|
||
|
|
bool is_allocated;
|
||
|
|
bool is_allocatedHost;
|
||
|
|
|
||
|
|
cuArrays() : width(0), height(0), size(0), countW(0), countH(0), count(0),
|
||
|
|
is_allocated(0), is_allocatedHost(0),
|
||
|
|
devData(0), hostData(0) {}
|
||
|
|
|
||
|
|
// 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;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
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;
|
||
|
|
}
|
||
|
|
|
||
|
|
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;
|
||
|
|
}
|
||
|
|
|
||
|
|
void allocate();
|
||
|
|
void allocateHost();
|
||
|
|
void deallocate();
|
||
|
|
void deallocateHost();
|
||
|
|
|
||
|
|
void copyToHost(cudaStream_t stream);
|
||
|
|
void copyToDevice(cudaStream_t stream);
|
||
|
|
|
||
|
|
size_t getSize()
|
||
|
|
{
|
||
|
|
return size*count;
|
||
|
|
}
|
||
|
|
|
||
|
|
long getByteSize()
|
||
|
|
{
|
||
|
|
return width*height*count*sizeof(T);
|
||
|
|
}
|
||
|
|
|
||
|
|
~cuArrays()
|
||
|
|
{
|
||
|
|
if(is_allocated)
|
||
|
|
deallocate();
|
||
|
|
if(is_allocatedHost)
|
||
|
|
deallocateHost();
|
||
|
|
}
|
||
|
|
|
||
|
|
void setZero(cudaStream_t stream);
|
||
|
|
void debuginfo(cudaStream_t stream) ;
|
||
|
|
void debuginfo(cudaStream_t stream, float factor);
|
||
|
|
void outputToFile(std::string fn, cudaStream_t stream);
|
||
|
|
void outputHostToFile(std::string fn);
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif //__CUIMAGES_H
|