CDegrain_cuda.h

//CDegrain.h CUDA ATTEMPT
#ifndef CDEGRAIN_H
#define CDEGRAIN_H

#include "avs_types.h"
#include "avisynth.h"

class CDegrain : public GenericVideoFilter
{
    int radius;
    int nt;
    int thr;
    int blksize;
    PClip dClip;
    int bit_depth;
    void CDegrain_Y(PVideoFrame& src, PVideoFrame (&den)[15], PVideoFrame& dst, int nt, int thr, int radius, int blksize);
    void CDegrain_YUV420(PVideoFrame& src, PVideoFrame (&den)[15], PVideoFrame& dst, int nt, int thr, int radius, int blksize);
    void CDegrain_YUV422(PVideoFrame& src, PVideoFrame (&den)[15], PVideoFrame& dst, int nt, int thr, int radius, int blksize);
    void CDegrain_YUV444(PVideoFrame& src, PVideoFrame (&den)[15], PVideoFrame& dst, int nt, int thr, int radius, int blksize);
public:
    CDegrain(PClip _child, int _radius, int _nt, int _thr, int _blksize, PClip _dClip, int _bit_depth, IScriptEnvironment* env);
    PVideoFrame __stdcall GetFrame(int n, IScriptEnvironment* env);
};

#endif