partialChargeProcessor.h

Go to the documentation of this file.

// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
//
 
#ifndef BALL_QSAR_PARTIALCHARGEPROCESSOR_H
#define BALL_QSAR_PARTIALCHARGEPROCESSOR_H
 
#include <BALL/KERNEL/PTE.h>
#include <BALL/DATATYPE/hashSet.h>
 
#include <vector>
#include <utility>
 
namespace BALL
{
  class AtomContainer;
  class Atom;

  class BALL_EXPORT PartialChargeProcessor
    : public UnaryProcessor<AtomContainer>
  {
    public:
 
    BALL_CREATE(PartialChargeProcessor)
 
    

    PartialChargeProcessor();
  
    PartialChargeProcessor(const PartialChargeProcessor& pc);

    virtual ~PartialChargeProcessor();


    PartialChargeProcessor& operator = (const PartialChargeProcessor& pc);
    

      void calculatePEOE(AtomContainer& ac);
  
    void setDataFolder(const char* folder);

    virtual Processor::Result operator () (AtomContainer& ac);
 
    private:
 
    /*_ @name Accessors
    */
 
    std::pair<std::vector<float>, std::vector<float> > readIonizationEnergies_();
 
    
    /*_ Helper function to get the ionization energy for an element
    */
    float getIonizationEnergy_(Element::AtomicNumber atomic_number, Size charge);
 
 
    void readElectronAffinities_(std::vector<float>& electron_affinities);
 
    /*_ Helper funtion to get the electron affinity of an element
    */
    float getElectronAffinity_(Element::AtomicNumber atomic_number, Size charge);
 
    /*_ Helper function to get the PEOE a,b and c factor
    */
    void getabcFactors_(Atom* atom, float& a, float& b, float& c);
 
    HashSet<Element::AtomicNumber> warned_elements_;
    
    String data_folder_;
  };
} // namespace BALL
 
#endif // BALL_QSAR_RINGFINDER_H