LaserProfile.h

Go to the documentation of this file.

#ifndef OPAL_LASERPROFILE_HH
#define OPAL_LASERPROFILE_HH
// ------------------------------------------------------------------------
// $RCSfile: LaserProfile.h,v $
// ------------------------------------------------------------------------
// $Revision: 1.3.4.1 $
// ------------------------------------------------------------------------
// Copyright: see Copyright.readme
// ------------------------------------------------------------------------
//
// Class: LaserProfile
//
// ------------------------------------------------------------------------
 
#include "OPALTypes.h"
 
#include <gsl/gsl_histogram2d.h>
#include <gsl/gsl_rng.h>
#include <string>
#include "hdf5.h"
 
class LaserProfile {
public:
    LaserProfile(
        const std::string& fileName, const std::string& imageName, double intensityCut,
        short flags);
 
    ~LaserProfile();
 
    void getXY(double& x, double& y);
 
    enum { FLIPX = 1, FLIPY = 2, ROTATE90 = 4, ROTATE180 = 8, ROTATE270 = 16 };
 
private:
    unsigned short* readFile(const std::string& fileName, const std::string& imageName);
    unsigned short* readPGMFile(const std::string& fileName);
    unsigned short* readHDF5File(const std::string& fileName, const std::string& imageName);
    void flipX(unsigned short* image);
    void flipY(unsigned short* image);
    void swapXY(unsigned short* image);
    void filterSpikes(unsigned short* image);
    void normalizeProfileData(double intensityCut, unsigned short* image);
    void computeProfileStatistics(unsigned short* image);
    void fillHistrogram(unsigned short* image);
    void setupRNG();
    void printInfo();
 
    void saveData(const std::string& fname, unsigned short* image);
    void saveHistogram();
    void sampleDist();
 
    unsigned short getProfileMax(unsigned short* image);
 
    hsize_t sizeX_m, sizeY_m;
    gsl_histogram2d* hist2d_m;
    gsl_rng* rng_m;
    gsl_histogram2d_pdf* pdf_m;
 
    Vector_t<double, 3> centerMass_m;
    Vector_t<double, 3> standardDeviation_m;
};
#endif