sdcuda/cprocess.h at master · oxford-pcs/sdcuda · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
#pragma once

#include <stdio.h>
#include <list>

#include "regions.h"
#include "ccube.h"
#include "cprocess.h"

enum process_stages {
	COPY_DEVICE_DATACUBE_TO_HOST,
	COPY_HOST_DATACUBE_TO_DEVICE,
	D_CROP_DATACUBE_TO_SMALLEST_DIMENSION_SLICE,
	D_FFT,
	D_FFTSHIFT,
	D_GROW_DATACUBE_TO_LARGEST_DIMENSION_SLICE,
	D_IFFT,
	D_IFFTSHIFT,
	D_PHASE_CORRELATE,
	D_REVERT_LAST_CROP,
	D_REVERT_LAST_GROW,
	D_REVERT_LAST_RESCALE,
	D_RESCALE_DATACUBE_TO_REFERENCE_WAVELENGTH,
	D_SPAXEL_FIT_POLY_AND_SUBTRACT,
	D_SET_DATA_TO_AMPLITUDE,
	H_CROP_TO_EVEN_SQUARE,
	MAKE_DATACUBE_ON_HOST
};

class input;
class process {
public:
	process() {};
	process(input*, int);
	~process();
	void run();
	input* iinput;
	hcube* h_datacube;
	dcube* d_datacube;
	int exp_idx;
	char message_buffer[255];
	std::list<process_stages> stages;
	std::vector<rectangle> last_crop_regions;
	std::vector<rectangle> last_grow_regions;
	std::vector<rectangle> last_rescale_regions;
private:
	void copyDeviceDatacubeToHost();
	void copyHostDatacubeToDevice();
	void cropToEvenSquareOnHost();
	void cropDatacubeToSmallestDimensionSliceOnDevice();
	void fitPolyToSpaxelAndSubtractOnDevice(int);
	void fftOnDevice();
	void fftshiftOnDevice();
	void growDatacubeToLargestDimensionSliceOnDevice();
	void iFftOnDevice();
	void iFftshiftOnDevice();
	void makeDatacubeOnHost();
	void normaliseOnDevice();
	void phaseCorrelateOnDevice(int, int, int);
	void rescaleDatacubeToReferenceWavelengthOnDevice(int);
	void revertLastCrop();
	void revertLastGrow();
	void revertLastRescale();
	void setDataToAmplitude();
	void step(int, int);
};