cuCorrNormalizer.h

/*
 * @file cuNormalizer.h
 * @brief normalize the correlation surface
 *
 * cuNormalizeProcessor is an abstract class for processors to normalize the correlation surface.
 * It has different implementations wrt different image sizes.
 * cuNormalizeFixed<64/128/.../1024> use a shared memory accelerated algorithm, which are limited by the number of cuda threads in a block.
 * cuNormalizeSAT uses the sum area table based algorithm, which applies to any size (used for >1024).
 * cuNormalizer is a wrapper class which determines which processor to use.
 */
 
#ifndef __CUNORMALIZER_H
#define __CUNORMALIZER_H
 
#include "cuArrays.h"
#include "cudaUtil.h"

class cuNormalizeProcessor {
public:
    // default constructor and destructor
    cuNormalizeProcessor() {}
    ~cuNormalizeProcessor() {}
    // execute interface
    virtual void execute(cuArrays<float> * correlation, cuArrays<float> *reference, cuArrays<float> *secondary, cudaStream_t stream) = 0;
};
 
class cuNormalizer {
private:
    cuNormalizeProcessor *processor;
public:
     // disable the default constructor
     cuNormalizer() = delete;
     // constructor with the secondary dimension
     cuNormalizer(int secondaryNX, int secondaryNY, int count);
     // destructor
     ~cuNormalizer();
     // execute correlation surface normalization
     void execute(cuArrays<float> *correlation, cuArrays<float> *reference, cuArrays<float> *secondary,
        cudaStream_t stream);
};
 
 
template<int Size>
class cuNormalizeFixed : public cuNormalizeProcessor
{
public:
    void execute(cuArrays<float> * correlation, cuArrays<float> *reference, cuArrays<float> *search, cudaStream_t stream) override;
};
 
class cuNormalizeSAT : public cuNormalizeProcessor
{
private:
    cuArrays<float> *referenceSum2;
    cuArrays<float> *secondarySAT;
    cuArrays<float> *secondarySAT2;
 
public:
    cuNormalizeSAT(int secondaryNX, int secondaryNY, int count);
    ~cuNormalizeSAT();
    void execute(cuArrays<float> * correlation, cuArrays<float> *reference, cuArrays<float> *search, cudaStream_t stream) override;
};
 
#endif
// end of file