KernelIterator now indicates when it is switching channels and frames

src/c/Cuda/KernelIterator.cu

@@ -50,9 +50,11 @@ __device__ KernelIterator& KernelIterator::operator++()
 			if(++iterator.dims.z>kernelEndIdx.z)
 			{
 				iterator.dims.z = kernelEndIdx.z;
+				isChannelEnd = true;
 				if(++iterator.chan>numChans)
 				{
 					iterator.chan = 0;
+					isFrameEnd = true;
 					if(++iterator.frame>numFrames)
 					{
 						iterator.frame = 0;

src/c/Cuda/KernelIterator.cuh

@@ -10,8 +10,14 @@ public:
 	__device__ __host__ ~KernelIterator();
 
 	__device__ KernelIterator& operator++();
-	__device__ bool end() { return isEnd; }
+	__device__ bool end() const { return isEnd; }
+	__device__ bool channelEnd() const { return isChannelEnd; }
+	__device__ bool frameEnd() const { return isFrameEnd; }
+
+	__device__ void nextChannel() { isChannelEnd = false; }
+	__device__ void nextFrame() { isFrameEnd = false; }
 	__device__ void reset();
+
 	__device__ Vec<float> getImageCoordinate() const;
 	__device__ unsigned int getChannel() const;
 	__device__ unsigned int getFrame() const;
@@ -38,6 +44,8 @@ private:
 	// This indicates the current position
 	ImageDimensions iterator;
 
+	bool isChannelEnd;
+	bool isFrameEnd;
 	bool isEnd;
 };