testutils.cpp File Reference

#include <time.h>
#include <string>
#include <stdio.h>
#include <iostream>
#include <fstream>
#include <sstream>
#include "testutils.h"

Go to the source code of this file.

Functions
float	distance (vector2f a, vector2f b)
bool	DirExists (const std::string &dirName_in)
int	NumFilesInDir (const std::string &dirName_in)
int	NumJpgInDir (const std::string &dirName_in)
ImageData	CropImage (ImageData img, int x, int y, int w, int h)
ImageData	ResizeImage (ImageData img, int factor)
ImageData	AddImages (ImageData img1, ImageData img2, vector2f displacement)
ImageData	GaussMask (ImageData img, float sigma)
ImageData	SkewImage (ImageData img, float fact)
void	GetOuterEdges (float *out, int size, ImageData img)
float	BackgroundMedian (ImageData img)
float	BackgroundStdDev (ImageData img)
float	BackgroundRMS (ImageData img)

Function Documentation

AddImages()

ImageData AddImages	(	ImageData	img1,
		ImageData	img2,
		vector2f	displacement
	)

Definition at line 164 of file testutils.cpp.

{
    ImageData addedImg = ImageData::alloc(img1.w,img1.h);

    for(int x_i=0;x_i<img1.w;x_i++){
        for(int y_i=0;y_i<img2.h;y_i++){
            if(x_i-displacement.x > 0 && x_i-displacement.x < img1.w && y_i-displacement.y > 0 && y_i-displacement.y < img1.h) {
                addedImg.at(x_i,y_i) = ( img1.at(x_i,y_i) + img2.at(x_i-displacement.x,y_i-displacement.y) )/2;
            } else {
                addedImg.at(x_i,y_i) = img1.at(x_i,y_i);
            }
        }
    }
    return addedImg;
}

BackgroundMedian()

float BackgroundMedian

(

ImageData

img

)

Definition at line 235 of file testutils.cpp.

                                     {
    int size = img.w * 2 + img.h * 2 - 4;
    float* outeredge = new float[size];
    GetOuterEdges(outeredge,size,img);
    std::sort(outeredge,outeredge+size);
    float median;
    if(size % 2 == 0)
        median = (outeredge[(int)(size/2-1)] + outeredge[(int)(size/2+1)])/2;
    else
        median = outeredge[size/2];
    delete[] outeredge;
    return median;
}

BackgroundRMS()

float BackgroundRMS

(

ImageData

img

)

Definition at line 258 of file testutils.cpp.

                                  {
    int size = img.w * 2 + img.h * 2 - 4;
    float* outeredge = new float[size];
    GetOuterEdges(outeredge,size,img);
    float sqsum = 0.0f;
    for(int ii = 0; ii < size; ii++){
        sqsum += outeredge[ii]*outeredge[ii];
    }
    delete[] outeredge;
    return sqrt(1/(float)size*sqsum);
}

BackgroundStdDev()

float BackgroundStdDev

(

ImageData

img

)

Definition at line 249 of file testutils.cpp.

                                     {
    int size = img.w * 2 + img.h * 2 - 4;
    float* outeredge = new float[size];
    GetOuterEdges(outeredge,size,img);
    float stddev = ComputeStdDev(outeredge,size);
    delete[] outeredge;
    return stddev;
}

CropImage()

ImageData CropImage	(	ImageData	img,
		int	x,
		int	y,
		int	w,
		int	h
	)

Definition at line 132 of file testutils.cpp.

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

    if( x < 0 || y < 0 || x + w > img.w || y + h > img.h){
        return img;
    }

    for(int x_i = x; x_i < x+w; x_i++){
        for(int y_i = y; y_i < y+h; y_i++){
            croppedImg.at(x_i-x,y_i-y) = img.at(x_i,y_i);
        }
    }
    return croppedImg;
}

DirExists()

bool DirExists

(

const std::string &

dirName_in

)

Definition at line 12 of file testutils.cpp.

{
  DWORD ftyp = GetFileAttributesA(dirName_in.c_str());
  if (ftyp == INVALID_FILE_ATTRIBUTES)
    return false;  //something is wrong with your path!

  if (ftyp & FILE_ATTRIBUTE_DIRECTORY)
    return true;   // this is a directory!

  return false;    // this is not a directory!
}

distance()

float distance	(	vector2f	a,
		vector2f	b
	)

Definition at line 10 of file testutils.cpp.

{ return distance(a.x-b.x,a.y-b.y); }

GaussMask()

ImageData GaussMask	(	ImageData	img,
		float	sigma
	)

Definition at line 180 of file testutils.cpp.

{
    ImageData gaussImg = ImageData::alloc(img.w,img.h);
    vector2f centre = vector2f(img.w/2,img.h/2);
    for(int x_i=0;x_i<img.w;x_i++){
        for(int y_i=0;y_i<img.h;y_i++){
            float gaussfact = expf(- (x_i-centre.x)*(x_i-centre.x) / (2*sigma*sigma) - (y_i-centre.y)*(y_i-centre.y) / (2*sigma*sigma));
            gaussImg.at(x_i,y_i) = img.at(x_i,y_i)*gaussfact;
        }
    }

    return gaussImg;
}

GetOuterEdges()

void GetOuterEdges	(	float *	out,
		int	size,
		ImageData	img
	)

Definition at line 212 of file testutils.cpp.

                                                      {
    int x,y=0;
    for(int ii = 0; ii < size; ii++){
        if(ii < img.w){ // Top
            x = ii;
            y = 0;
        }
        else if(ii < img.w + img.h - 1){ // Right
            x = img.w-1;
            y = ii-(img.w-1);
        }
        else if(ii < img.w * 2 + img.h - 2){ // Bottom
            y = img.h-1;
            x = ii-(img.h+img.w-1);
        }
        else{ // Left
            x = 0;
            y = ii-(img.h+img.w*2-3);
        }
        out[ii] = img.at(x,y);
    }
}

NumFilesInDir()

int NumFilesInDir

(

const std::string &

dirName_in

)

Definition at line 24 of file testutils.cpp.

{
    WIN32_FIND_DATA FindFileData;
    HANDLE hFind;

    std::string dirName = dirName_in + "*";
    hFind = FindFirstFile(dirName.c_str(),&FindFileData);
    if (hFind == INVALID_HANDLE_VALUE) {
        printf ("FindFirstFile failed (%d)\n", GetLastError());
        return 0;
    } 
    int out = -1;
    while(FindNextFile(hFind,&FindFileData)){
        out++;
    }
    return out;
}

NumJpgInDir()

int NumJpgInDir

(

const std::string &

dirName_in

)

Definition at line 42 of file testutils.cpp.

{
    WIN32_FIND_DATA FindFileData;
    HANDLE hFind;

    std::string dirName = dirName_in + "*";
    hFind = FindFirstFile(dirName.c_str(),&FindFileData);
    if (hFind == INVALID_HANDLE_VALUE) {
        printf ("FindFirstFile failed (%d)\n", GetLastError());
        return 0;
    } 
    int out = -1;
    while(FindNextFile(hFind,&FindFileData)){
        std::string name = FindFileData.cFileName;
        if(name.find(".jpg")!=name.npos) {
            out++;
        }
    }
    return out;
}

ResizeImage()

ImageData ResizeImage	(	ImageData	img,
		int	factor
	)

Definition at line 148 of file testutils.cpp.

{
    ImageData resizedImg = ImageData::alloc(img.w*factor,img.h*factor);
    
    for(int x_i=0;x_i<img.w;x_i++){
        for(int y_i=0;y_i<img.h;y_i++){
            for(int x_fact=0;x_fact<factor;x_fact++){
                for(int y_fact=0;y_fact<factor;y_fact++){
                    resizedImg.at(x_i*factor+x_fact,y_i*factor+y_fact) = img.at(x_i,y_i);
                }
            }
        }
    }
    return resizedImg;
}

SkewImage()

ImageData SkewImage	(	ImageData	img,
		float	fact
	)

Definition at line 194 of file testutils.cpp.

{
    ImageData skewImg = ImageData::alloc(img.w,img.h);
    vector2f centre = vector2f(img.w/2,img.h/2);
    float median  = BackgroundMedian(img);
    float stddev  = BackgroundStdDev(img);
    float maxskew = median*stddev;
    for(int x_i=0;x_i<img.w;x_i++){
        for(int y_i=0;y_i<img.h;y_i++){
            int diagonalOffset = (y_i - x_i*img.h/img.w);
            float skew = fact*((float)diagonalOffset/img.h)*maxskew;
            skewImg.at(x_i,y_i) = img.at(x_i,y_i)+skew;
        }
    }

    return skewImg;
}