bioem.h 1.56 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#ifndef BIOEM_H
#define BIOEM_H
#include <fstream>
#include <iostream>
#include <complex>

#include "defs.h"
#include "bioem.h"
#include "model.h"
#include "map.h"
#include "param.h"

class bioem
{
public:
16
17
18
19
20
21
22
23
24
	bioem();
	virtual ~bioem();

	int configure(int ac, char* av[]);
	int precalculate(); // Is it better to pass directly the input File names?
	int dopreCalCrossCorrelation(int iRefMap, int iRefMapLocal);
	int run();
	int doProjections(int iMap);
	int createConvolutedProjectionMap(int iOreint,int iMap, mycomplex_t* lproj,bioem_map& Mapconv,mycomplex_t* localmultFFT,myfloat_t& sumC,myfloat_t& sumsquareC);
25
26


27
	virtual int compareRefMaps(int iProjectionOut, int iConv, const bioem_map& conv_map, const int startMap = 0);
28
29
30
31
32
33
34
35
36
	int compareRefMaps2(int iProjectionOut, int iConv,mycomplex_t* localmultFFT,myfloat_t sumC,myfloat_t sumsquareC);

	int createProjection(int iMap, mycomplex_t* map);
	int calcross_cor(bioem_map& localmap,myfloat_t& sum,myfloat_t& sumsquare);
	int calProb(int iRefMap,int iOrient, int iConv,myfloat_t sumC,myfloat_t sumsquareC);
	int calculateCCFFT(int iMap, int iConv, mycomplex_t* localConvFFT,mycomplex_t* localCCT,mycomplex_t* lCC);

	bioem_Probability* pProb;
	bioem_crossCor* crossCor;
37
38
39
40
41
42

protected:
	virtual int deviceInit();
	virtual int deviceStartRun();
	virtual int deviceFinishRun();

43
44
45
46
47
48
	bioem_param param;
	bioem_model Model;
	bioem_RefMap RefMap;

	int nReferenceMaps;			//Maps in memory at a time
	int nReferenceMapsTotal;	//Maps in total
49

50
51
	int nProjectionMaps;		//Maps in memory at a time
	int nProjectionMapsTotal;	//Maps in total
52

53
	int FFTAlgo;
54
55
56
};

#endif