sutra_perfectCoronagraph.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//  This file is part of COMPASS <https://anr-compass.github.io/compass/>
//
//  Copyright (C) 2011-2023 COMPASS Team <https://github.com/ANR-COMPASS>
//  All rights reserved.
 
// -----------------------------------------------------------------------------
 
 
#ifndef _SUTRA_PERFECT_CORONAGRAPH_H_
#define _SUTRA_PERFECT_CORONAGRAPH_H_
 
#include <sutra_coronagraph.h>
#include <map>
 
class SutraPerfectCoronagraph : public SutraCoronagraph {
    public:
        std::map<std::string, std::vector<CarmaObj<cuFloatComplex>*>> AA;
        std::map<std::string, std::vector<CarmaObj<cuFloatComplex>*>> BB;
        std::map<std::string, std::vector<float>> norm;
 
        CarmaObj<cuFloatComplex> *tmp_mft;
 
    public:
        SutraPerfectCoronagraph(CarmaContext *context, SutraSource *d_source,int im_dimx, 
                                int im_dimy, float *wavelength, int nWavelength, int device);
        ~SutraPerfectCoronagraph()=default;
        int compute_image(bool accumulate);
        int compute_psf(bool accumulate);
        int set_mft(cuFloatComplex *A, cuFloatComplex *B, float* norm, std::string mftType);
 
    private:
        int _compute_image(bool psf, bool remove_coro, bool accumulate);
};
 
#endif //_SUTRA_PERFECT_CORONAGRAPH_H_