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 "Algorithms/Vektor.h"
 
#include <gsl/gsl_rng.h>
#include <gsl/gsl_histogram2d.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 centerMass_m;
    Vector_t standardDeviation_m;
};
#endif