ballAndStickModel.h

Go to the documentation of this file.

// -*- Mode: C++; tab-width: 2; -*-
// vi: set ts=2:
//
// $Id: ballAndStickModel.h,v 1.17.18.1 2007/03/25 21:26:05 oliver Exp $
//

#ifndef BALL_VIEW_MODELS_BALLANDSTICKMODEL_H
#define BALL_VIEW_MODELS_BALLANDSTICKMODEL_H

#ifndef BALL_VIEW_MODELS_ATOMBONDMODELBASEPROCESSOR_H
# include <BALL/VIEW/MODELS/atomBondModelBaseProcessor.h>
#endif

#ifndef BALL_KERNEL_ATOM_H
 #include <BALL/KERNEL/atom.h>
#endif

namespace BALL
{
  class Atom;

  namespace VIEW
  {
    class BALL_VIEW_EXPORT AddBallAndStickModel: public AtomBondModelBaseProcessor
    {
      public:

      BALL_CREATE(AddBallAndStickModel)

      

      AddBallAndStickModel();

      AddBallAndStickModel(const AddBallAndStickModel& add_ball_and_stick_model);

      virtual ~AddBallAndStickModel();

      virtual void clear();



      void set(const AddBallAndStickModel& add_ball_and_stick_model);

      const AddBallAndStickModel& operator = (const AddBallAndStickModel& processor);



      void setBallRadius(const float radius)
        throw(Exception::OutOfRange);

      float getBallRadius() const;

      void setStickRadius(const float radius)
        throw(Exception::OutOfRange);

      float getStickRadius() const;

      void enableBallAndStickModel();

      void enableStickModel();



      virtual Processor::Result operator() (Composite& composite);



      bool isBallAndStickModel() const;

      bool isStickModel() const;

      bool dashedBondsEnabled() const
        { return dashed_bonds_;}

      void enableDashedBonds(bool state)
        { dashed_bonds_ = state;}



      virtual void dump(std::ostream& s = std::cout, Size depth = 0) const;


      virtual bool createGeometricObjects();
    
      protected:

      void visualiseBond_(const Bond& bond);

      virtual void visualiseRings_();

      inline void renderStandardBond_(const Bond& bond);

      inline void renderMultipleBond_(const Bond& bond, Vector3 normal, Vector3 dir);

      inline void renderDashedBond_(const Bond& bond, Vector3 n)
        throw(Exception::DivisionByZero);

      void collectRingBonds_();

      Vector3 getSP2Plane_(const Atom& atom, const Bond& bond, const Vector3& dir) const;

      private:

      float ball_radius_;
      float stick_radius_;
      float special_radius_;
    
      bool ball_and_stick_;
      bool dashed_bonds_;

      vector<vector<Bond*> > ring_bonds_;
    };

# ifndef BALL_NO_INLINE_FUNCTIONS
#   include <BALL/VIEW/MODELS/ballAndStickModel.iC>
# endif

  } // namespace VIEW
} // namespace BALL

#endif // BALL_VIEW_MODELS_BALLANDSTICKMODEL_H