BeadFinder Namespace Reference

Classes
struct	Config
struct	Position

Functions
std::vector< Position >	Find (ImageData *img, float *sample, Config *cfg)
std::vector< Position >	Find (uint8_t *img, int pitch, int w, int h, int smpCornerX, int smpCornerY, Config *cfg)

Function Documentation

Find() [1/2]

std::vector< Position > BeadFinder::Find	(	ImageData *	img,
		float *	sample,
		Config *	cfg
	)

Definition at line 168 of file BeadFinder.cpp.

{
    typedef std::complex<float> pixelc_t;

    // Compute nearest power of two dimension
    int w=NextPowerOf2(img->w);
    int h=NextPowerOf2(img->h);

    // Convert to std::complex and copy the image in there (subtract mean first)
    float mean = img->mean();
    pixelc_t* cimg = new pixelc_t[w*h];
    std::fill(cimg, cimg+w*h, pixelc_t());
    for (int y=0;y<img->h;y++) {
        for (int x=0;x<img->w;x++) {
            cimg[y*w+x] = img->data[y*img->w+x]-mean;
        }
    }

    // Inplace 2D fft
    FFT2D(cimg,w,h,false);

    // Create an image to put the sample in with size [w,h] as well
    pixelc_t* smpimg = new pixelc_t[w*h];
    std::fill(smpimg, smpimg+w*h, pixelc_t());
    float smpmean = ImageData(sample,cfg->roi,cfg->roi).mean();
    for (int y=0;y<cfg->roi;y++) 
        for (int x=0;x<cfg->roi;x++) 
            smpimg[ ((y-cfg->roi)&(h-1)) * w + ( (x-cfg->roi)&(w-1))] = sample[cfg->roi*y+x]-smpmean;

//  ComplexToJPEGFile("smpimg.jpg", smpimg, w,h);
    FFT2D(smpimg,w,h,false);

    for (int i=0;i<w*h;i++)
        cimg[i]*=smpimg[i];
    FFT2D(cimg,w,h,true);

    float maxVal = 0.0f;
    for (int i=0;i<w*h;i++)
        maxVal=std::max(maxVal, cimg[i].real());

    std::vector<Position> pts;
    for (int y=0;y<img->h;y++) {
        for (int x=0;x<img->w;x++) {
            if (cimg[y*w+x].real()>maxVal*cfg->similarity )
                pts.push_back ( SearchArea(cimg, w, h, x,y, cfg->MinPixelDistance()) );
        }
    }
    //ComplexToJPEGFile("result.jpg", cimg, w,h);
    delete[] smpimg;
    delete[] cimg;

    RecenterAndFilter(img, pts, cfg);

    return pts;
}

Find() [2/2]

std::vector< Position > BeadFinder::Find	(	uint8_t *	img,
		int	pitch,
		int	w,
		int	h,
		int	smpCornerX,
		int	smpCornerY,
		BeadFinder::Config *	cfg
	)

Definition at line 224 of file BeadFinder.cpp.

{
    ImageData fimg = ImageData::alloc(w,h);

    for (int y=0;y<h;y++)
        for (int x=0;x<w;x++) 
            fimg.at(x,y) = img[y*pitch+x];

    float *fsmp = new float[cfg->roi*cfg->roi];
    for (int y=0;y<cfg->roi;y++) {
        for (int x=0;x<cfg->roi;x++) {
            fsmp [y*cfg->roi+x] = fimg [ (y+smpCornerY) * h + x + smpCornerX ];
        }
    }

    std::vector<Position> beads = Find(&fimg, fsmp, cfg);

    delete[] fsmp;
    fimg.free();
    return beads;
}