obint.h

//
// obint.h
//
// Copyright (C) 1996 Limit Point Systems, Inc.
//
// Author: Curtis Janssen <cljanss@limitpt.com>
// Maintainer: LPS
//
// This file is part of the SC Toolkit.
//
// The SC Toolkit is free software; you can redistribute it and/or modify
// it under the terms of the GNU Library General Public License as published by
// the Free Software Foundation; either version 2, or (at your option)
// any later version.
//
// The SC Toolkit is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Library General Public License for more details.
//
// You should have received a copy of the GNU Library General Public License
// along with the SC Toolkit; see the file COPYING.LIB.  If not, write to
// the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
//
// The U.S. Government is granted a limited license as per AL 91-7.
//
 
#ifndef _chemistry_qc_basis_obint_h
#define _chemistry_qc_basis_obint_h
 
#include <util/ref/ref.h>
#include <util/state/state.h>
#include <util/container/stdarray.h>
#include <math/scmat/matrix.h>
#include <math/scmat/elemop.h>
 
#include <chemistry/qc/basis/gaussbas.h>
#include <chemistry/qc/basis/dercent.h>
#include <chemistry/qc/basis/intparams.h>
 
namespace sc {
 
class Integral;
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyInt : public RefCount {
  protected:
    // this is who created me
    Integral *integral_;
 
    Ref<GaussianBasisSet> bs1_;
    Ref<GaussianBasisSet> bs2_;
 
    double *buffer_;
 
    OneBodyInt(Integral *integral,
               const Ref<GaussianBasisSet>&b1,
               const Ref<GaussianBasisSet>&b2 = 0);
 
  public:
    virtual ~OneBodyInt();
  
    int nbasis() const;
 
    int nbasis1() const;
    int nbasis2() const;
 
    int nshell() const;
 
    int nshell1() const;
    int nshell2() const;
 
    Ref<GaussianBasisSet> basis(size_t c) const;
 
    Ref<GaussianBasisSet> basis1() const;
    Ref<GaussianBasisSet> basis2() const;
 
    const double * buffer() const;
    
    virtual void compute_shell(int,int) = 0;
 
    std::pair<const double *, std::array<unsigned long, 2> > compute_shell_array(int,int);
 
    virtual void reinitialize();
 
    virtual bool cloneable() const;
 
    virtual Ref<OneBodyInt> clone();
 
    Integral *integral() const { return integral_; }
};
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyOneCenterInt : public RefCount {
  protected:
    // this is who created me
    Integral *integral_;
 
    Ref<GaussianBasisSet> bs1_;
 
    double *buffer_;
 
    OneBodyOneCenterInt(Integral *integral,
               const Ref<GaussianBasisSet>&b1);
 
  public:
    virtual ~OneBodyOneCenterInt();
  
    int nbasis() const;
 
    int nbasis1() const;
 
    int nshell() const;
 
    int nshell1() const;
 
    Ref<GaussianBasisSet> basis(size_t c) const;
 
    Ref<GaussianBasisSet> basis1() const;
 
    const double * buffer() const;
    
    virtual void compute_shell(int) = 0;
 
    virtual void reinitialize();
 
    virtual bool cloneable() const;
 
    virtual Ref<OneBodyOneCenterInt> clone();
 
    Integral *integral() const { return integral_; }
};
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyOneCenterWrapper : public OneBodyOneCenterInt {
    Ref<OneBodyInt> ob_;
    int jsh_;
  public:
    OneBodyOneCenterWrapper(const Ref<OneBodyInt>& ob,
                            int sh2 = 0);
    void compute_shell(int);
};
 
// //////////////////////////////////////////////////////////////////////////
 
class ShellPairIter {
  private:
    const double * buf;
    double scale_;
 
    int e12;
 
    int index;
    
    int ioffset;
    int joffset;
 
    int iend;
    int jend;
 
    int icur;
    int jcur;
    
  public:
    ShellPairIter();
    ~ShellPairIter();
 
    void init(const double * buffer, int ishell, int jshell,
              int ioff, int joff, int nfunci, int nfuncj, int redund=0,
              double scale=1.0);
 
    void start() { icur=jcur=index=0; }
    int ready() const { return (icur < iend); }
 
    void next() {
      if (jcur < ((e12)?(icur):((jend)-1))) {
        index++;
        jcur++;
        return;
      }
 
      jcur=0;
      icur++;
 
      index = icur*jend;
    }
 
    int current_i() const { return icur; }
    int current_j() const { return jcur; }
 
    int i() const { return icur+ioffset; }
    int j() const { return jcur+joffset; }
 
    int nint() const { return iend*jend; }
    
    double val() const { return buf[index]*scale_; }
};
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyIntIter : public RefCount {
  protected:
    Ref<OneBodyInt> obi; // help me obi wan
    ShellPairIter spi;
    
    int redund;
    
    int istart;
    int jstart;
    
    int iend;
    int jend;
 
    int icur;
    int jcur;
 
    int ij;
    
  public:
    OneBodyIntIter();
    OneBodyIntIter(const Ref<OneBodyInt>&);
    virtual ~OneBodyIntIter();
    
    virtual void start(int ist=0, int jst=0, int ien=0, int jen=0);
    virtual void next();
 
    int ready() const { return (icur < iend); }
 
    int ishell() const { return icur; }
    int jshell() const { return jcur; }
 
    int ijshell() const { return ij; }
 
    int redundant() const { return redund; }
    void set_redundant(int i) { redund=i; }
    
    virtual double scale() const;
 
    Ref<OneBodyInt> one_body_int() { return obi; }
 
    ShellPairIter& current_pair();
 
    virtual bool cloneable() const;
    virtual Ref<OneBodyIntIter> clone();
};
 
 
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyIntOp: public SCElementOp {
  protected:
    Ref<OneBodyIntIter> iter;
 
  public:
    OneBodyIntOp(const Ref<OneBodyInt>&);
    OneBodyIntOp(const Ref<OneBodyIntIter>&);
    virtual ~OneBodyIntOp();
  
    void process(SCMatrixBlockIter&);
    void process_spec_rect(SCMatrixRectBlock*);
    void process_spec_ltri(SCMatrixLTriBlock*);
    void process_spec_rectsub(SCMatrixRectSubBlock*);
    void process_spec_ltrisub(SCMatrixLTriSubBlock*);
 
    bool cloneable() const;
    Ref<SCElementOp> clone();
 
    int has_side_effects();
};
 
class OneBody3IntOp: public SCElementOp3 {
  private:
    Ref<OneBodyIntIter> iter;
 
  public:
    OneBody3IntOp(const Ref<OneBodyInt>&b);
    OneBody3IntOp(const Ref<OneBodyIntIter>&);
    virtual ~OneBody3IntOp();
  
    void process(SCMatrixBlockIter&,
                 SCMatrixBlockIter&,
                 SCMatrixBlockIter&);
    void process_spec_rect(SCMatrixRectBlock*,
                           SCMatrixRectBlock*,
                           SCMatrixRectBlock*);
    void process_spec_ltri(SCMatrixLTriBlock*,
                           SCMatrixLTriBlock*,
                           SCMatrixLTriBlock*);
 
    int has_side_effects();
    int has_side_effects_in_arg1();
    int has_side_effects_in_arg2();
 
};
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyDerivInt : public RefCount {
  protected:
    // this is who created me
    Integral *integral_;
 
    Ref<GaussianBasisSet> bs1;
    Ref<GaussianBasisSet> bs2;
 
    double *buffer_;
 
  public:
    OneBodyDerivInt(Integral *, const Ref<GaussianBasisSet>&b);
    OneBodyDerivInt(Integral *,
                    const Ref<GaussianBasisSet>&b1,
                    const Ref<GaussianBasisSet>&b2);
    virtual ~OneBodyDerivInt();
  
    int nbasis() const;
    int nbasis1() const;
    int nbasis2() const;
 
    int nshell() const;
    int nshell1() const;
    int nshell2() const;
 
    Ref<GaussianBasisSet> basis() const;
    Ref<GaussianBasisSet> basis1() const;
    Ref<GaussianBasisSet> basis2() const;
 
    const double * buffer() const;
    
    virtual void compute_shell(int ish, int jsh, int center) = 0;
 
    virtual void reinitialize();
 
    Integral *integral() const { return integral_; }
};
 
// //////////////////////////////////////////////////////////////////////////
 
class OneBodyOneCenterDerivInt : public RefCount {
  protected:
    // this is who created me
    Integral *integral_;
 
    Ref<GaussianBasisSet> bs1;
 
    double *buffer_;
 
  public:
    OneBodyOneCenterDerivInt(Integral *, const Ref<GaussianBasisSet>&b);
    virtual ~OneBodyOneCenterDerivInt();
  
    int nbasis() const;
    int nbasis1() const;
 
    int nshell() const;
    int nshell1() const;
 
    Ref<GaussianBasisSet> basis() const;
    Ref<GaussianBasisSet> basis1() const;
 
    const double * buffer() const;
    
    virtual void compute_shell(int ish, int center) = 0;
 
    virtual void reinitialize();
 
    Integral *integral() const { return integral_; }
};
 
}
 
#endif
 
// Local Variables:
// mode: c++
// c-file-style: "ETS"
// End: