GPUQMCBasisFunction.h

Go to the documentation of this file.

/*
  Copyright (c) Amos G. Anderson 2005
  Distributed under GNU general public license (GPL)
  No guarantee or warantee regarding usability or stability is expressed or implied.
  nitroamos@gmail.com
*/

#ifndef GPU_BASISFUNCTIONS
#define GPU_BASISFUNCTIONS

#ifdef QMC_GPU

#include <iostream>
#include <stdlib.h>
#include <assert.h>
#include <math.h>
#include <vector>
#include <string>
#include "GPUGlobals.h"
#include "GPUQMCFramebuffer.h"
#include "Array1D.h"
#include "Array2D.h"
#include "Stopwatch.h"
#include "QMCBasisFunction.h"

static GLuint bfParametersTexID;

class GPUQMCBasisFunction : public QMCBasisFunction, GPUGlobals
  {
    public:
    
      GPUQMCBasisFunction();
      
      GPUQMCBasisFunction(QMCBasisFunction bf, int numElectrons, int max_calcs);
      
      void operator=(GPUQMCBasisFunction &rhs);
      
      GLuint runCalculation(Array1D<Array2D<double>*> &X, int num, int start, int stop);
      
      int getNumIterations();
      
      GLuint getElectronicTexture();

      ~GPUQMCBasisFunction();
      
    private:
    
      void setUpInputs();
      
      string generateShader(int which);
      
      string generateTranslationShader(bool is12);
      
      void loadShaders();
      
      void translate();
      
      void drawPrimative(GLfloat maxs, GLfloat maxt, GLfloat sShift, GLfloat tShift);
      
      void loadElectronPositions(Array1D<Array2D<double>*> &X, int start, int stop);
      
      int mapping(int i, int j, int h, int w);
      
      void unloadData(GPUQMCFramebuffer & which, int w, int h);
      
      const static int nMats = 5;
      
      const static int psi = 0;
      const static int grx = 1;
      const static int gry = 2;
      const static int grz = 3;
      const static int lap = 4;
      
      int nRows, nCols;
      
      int allocatedRows, allocatedCols;
      
      int nElectrons, nBasisF;
      
      int fxo_deltaOE, fxo_deltaBF;
      
      int txt_deltaOE, txt_deltaBF;
      
      int elecW, elecH;
      
      int maxGaussians;
      
      int basisfunctionParamsH;
      
      static vector<CGprogram> fragProg;
      static vector<CGparameter> electronsCGP, paramsCGP;
      
      static CGprogram fxo_to_txt_CG;
      static CGparameter fxo_to_txt_CGP;
      
      GPUQMCFramebuffer basisFunctionsFB;
      GPUQMCFramebuffer outputFB;

      GLuint electronsTexID;

      GLfloat * cpuData;
  };
#endif
#endif

---