cas-offinder.h

#include "oclkernels.h"

#include <vector>
#include <string>
#include <cstring>
#include <fstream>
#include <algorithm>

#ifdef __APPLE__
#  include <OpenCL/cl.h>
#else
#  include <CL/cl.h>
#endif

#ifndef MIN
#define MIN(a,b) ( ((a)<(b))?(a):(b) )
#endif

using namespace std;

static cl_platform_id platforms[MAX_PLATFORM_NUM];
static cl_uint platform_cnt;

class Cas_OFFinder {
private:
    cl_device_type m_devtype;

	vector<cl_command_queue> m_queues;
	vector<cl_context> m_contexts;
	vector<cl_ulong> MAX_ALLOC_MEMORY; // on device, in bytes
	vector<size_t> MAX_LOCAL_SIZE;

	unsigned long long m_chrdatasize;
	vector<string> m_chrnames;
	vector<string> m_compares;
	vector<cl_ushort> m_thresholds;
	vector<unsigned long long> m_chrpos;
	string m_chrdata;
	cl_char* m_pattern;

	cl_uint m_threshold;
	cl_uint m_patternlen;
	cl_uint m_entrycount;

	vector<cl_kernel> m_finderkernels;
	vector<cl_kernel> m_comparerkernels;

	vector<cl_mem> m_chrdatabufs;
	vector<cl_mem> m_patternbufs;
	vector<cl_mem> m_patternflagbufs;
	vector<cl_mem> m_flagbufs;
	vector<cl_mem> m_locibufs;
	vector<cl_mem> m_entrycountbufs;
	vector<cl_mem> m_comparebufs;
	vector<cl_mem> m_compareflagbufs;


	vector<cl_mem> m_mmlocibufs;
	vector<cl_mem> m_mmcountbufs;
	vector<cl_mem> m_directionbufs;

	vector <cl_uint> m_locicnts;
	vector <cl_uint *> m_locis;
	vector <cl_ushort *> m_mmcounts;
	vector <cl_char *> m_flags;
	vector <cl_char *> m_directions;
	vector <cl_uint *> m_mmlocis;

	vector<size_t> m_dicesizes;
	unsigned long long m_totalanalyzedsize;
	unsigned long long m_lasttotalanalyzedsize;
	vector<unsigned long long> m_worksizes;
	cl_uint m_devnum;
	unsigned int m_activedevnum;
	unsigned long long m_lastloci;

	unsigned long long m_linenum;
	unsigned long long m_filenum;
	size_t m_totalcompcount;

	void set_complementary_sequence(cl_char* seq, size_t seqlen);
	void set_seq_flags(int* seq_flags, const cl_char* seq, size_t seqlen);
	void initOpenCL(vector<int> dev_ids);

public:
	vector<string> chrnames;
	string chrdata;
	vector<unsigned long long> chrpos;

	string chrdir;

	Cas_OFFinder(cl_device_type devtype, string devarg);
	~Cas_OFFinder();

	void setChrData();

	bool loadNextChunk();
	void findPattern();
	void releaseLociinfo();

	void indicate_mismatches(cl_char* seq, cl_char* comp);

	void compareAll(const char* outfilename);
	void readInputFile(const char* inputfile);

	static void print_usage();
	static void init_platforms();
};