PeakFinder.h

Go to the documentation of this file.

//
// Class PeakFinder
//   Find peaks of radial profile.
//   It computes a histogram based on the radial distribution of the particle
//   bunch. After that all peaks of the histogram are searched.
//   The radii are written in ASCII format to a file.
//   This class is used for the cyclotron probe element.
//
// Copyright (c) 2017 - 2021, Matthias Frey, Jochem Snuverink, Paul Scherrer Institut, Villigen PSI,
// Switzerland All rights reserved
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#ifndef PEAKFINDER_H
#define PEAKFINDER_H
 
#include "OPALTypes.h"
 
#include <fstream>
#include <list>
#include <string>
#include <vector>
 
class PeakFinder {
public:
    using container_t = std::vector<double>;
 
public:
    PeakFinder() = delete;
 
    PeakFinder(std::string elem, double min, double max, double binwidth, bool singlemode);

    void addParticle(const Vector_t<double, 3>& R);

    void evaluate(const int& turn);
 
    void save();
 
private:
    // compute global histogram, involves some inter-node communication
    void createHistogram_m();
 
    /***************
     * Output file *
     ***************/
    void open_m();
    void append_m();
    void close_m();
    void saveASCII_m();
 
    void computeCentroid_m();
 
private:
    container_t radius_m;
    container_t globHist_m;

    std::string fileName_m;

    std::string hist_m;

    std::ofstream os_m;

    std::ofstream hos_m;

    std::string outputName_m;
 
    // Histogram details
    unsigned int nBins_m;
    double binWidth_m;
    double min_m, max_m;
 
    int turn_m;
    double peakRadius_m;
    int registered_m;
    std::list<double> peaks_m;
    bool singlemode_m;
    bool first_m;
    bool finished_m;
};
 
#endif