sutra_tscreen.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_TSCREEN_H_
#define _SUTRA_TSCREEN_H_
 
#include <sutra_phase.h>
#include <sutra_utils.h>
 
class SutraTurbuScreen {
 public:
  int device;              // The device #
  SutraPhase *d_tscreen;  // The phase screen
  CarmaObj<float>
      *d_tscreen_o;  // Additional space of the same size as the phase screen
  CarmaObj<float> *d_mat_a;                 // A matrix for extrusion
  CarmaObj<float> *d_mat_b;                 // B matrix for extrusion
  CarmaObj<unsigned int> *d_istencilx;  // stencil for column extrusion
  CarmaObj<unsigned int> *d_istencily;  // stencil for row extrusion
  CarmaObj<float> *d_z;                 // tmp array for extrusion process
  CarmaObj<float> *d_noise;             // tmp array containing random numbers
  CarmaObj<float> *d_ytmp;  // contains the extrude update (row or column)
  long screen_size;          // size of phase screens
  float r0;                  // layer r0 (pixel units)
  float amplitude;           // amplitude for extrusion (r0^-5/6)
  float altitude;
  float windspeed;
  float winddir;
  float deltax;  // number of rows to extrude per iteration
  float deltay;  // number of lines to extrude per iteration
  // internal
  float accumx;
  float accumy;
  cudaChannelFormatDesc
      channel_desc;  // Channel descriptor for texture memory access
 
  CarmaObj<cuFloatComplex>
      *d_tscreen_c;  // Additional space for von karman screen generation
  float norm_vk;
  bool vk_on;
  CarmaContext *current_context;
 
 public:
  SutraTurbuScreen(CarmaContext *context, long size, long size2, float amplitude,
                float altitude, float windspeed, float winddir, float deltax,
                float deltay, int device);
  // SutraTurbuScreen(const SutraTurbuScreen &tscreen);
  ~SutraTurbuScreen();
 
  int init_screen(float *h_A, float *h_B, unsigned int *h_istencilx,
                  unsigned int *h_istencily, int seed);
  int extrude(int dir);
  int init_vk(int seed, int pupd);
  int generate_vk(float l0, int nalias);
  int refresh_screen();
  int set_seed(int seed);
  int set_deltax(float deltax);
  int set_deltay(float deltay);
  int set_istencilx(unsigned int* istencil);
  int set_istencily(unsigned int* istencil);
};
 
int gene_vonkarman(cuFloatComplex *d_odata, float *d_idata, float k0,
                   int nalias, int nx, int ny, int block_size);
int norm_pscreen(float *d_odata, float *d_idata, int nx, int ny,
                 float norm_fact, CarmaDevice *device);
 
#endif  // _SUTRA_TSCREEN_H_