chain_impl.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// This file is part of the OpenStructure project <www.openstructure.org>
//
// Copyright (C) 2008-2011 by the OpenStructure authors
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the Free
// Software Foundation; either version 3.0 of the License, or (at your option)
// any later version.
// This library 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 Lesser General Public License for more
// details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this library; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//------------------------------------------------------------------------------
#ifndef OST_CHAIN_IMPL_HH
#define OST_CHAIN_IMPL_HH

#include <boost/enable_shared_from_this.hpp>

#include <ost/mol/module_config.hh>
#include <ost/geom/geom.hh>

#include <ost/mol/residue_prop.hh>
#include <ost/mol/chain_type.hh>
#include <ost/mol/impl/chain_impl_fw.hh>
#include <ost/mol/impl/residue_impl_fw.hh>
#include <ost/mol/impl/entity_impl_fw.hh>
#include <ost/mol/entity_visitor_fw.hh>
#include <ost/mol/impl/atom_impl_fw.hh>

#include <ost/generic_property.hh>

#include <ost/mol/sec_structure.hh>

namespace ost { namespace mol {namespace impl {

class ChainImpl: public GenericPropContainerImpl,
                 public boost::enable_shared_from_this<ChainImpl>
{
public:
  ChainImpl(const EntityImplPtr& e, const String& name);

  void SetName(const String& new_name);
  String GetName() const;

  void SetType(const ChainType type)
  {
    type_ = type;
  }

  ChainType GetType() const
  {
    return type_;
  }
  
  bool IsPolymer() const
  {
    return type_==CHAINTYPE_POLY || this->IsPolypeptide() || 
          this->IsPolynucleotide() || this->IsPolysaccharide();
  }
  bool IsPolysaccharide() const
  {
    return type_==CHAINTYPE_POLY_SAC_D || type_==CHAINTYPE_POLY_SAC_L;
  }
  bool IsPolypeptide() const
  {
    return type_==CHAINTYPE_POLY_PEPTIDE_D || type_==CHAINTYPE_POLY_PEPTIDE_L;
  }
  bool IsPolynucleotide() const
  {
    return type_==CHAINTYPE_POLY_DN || type_==CHAINTYPE_POLY_RN || 
           type_==CHAINTYPE_POLY_DN_RN;
  }

  void SetDescription(const String desc)
  {
    description_ = desc;
  }

  String GetDescription() const
  {
    return description_;
  }

  ResidueImplPtr AppendResidue(const ResidueImplPtr& res);
  
  ResidueImplPtr InsertResidueBefore(int index, const ResNum& n, 
                                     const ResidueKey& k);
  ResidueImplPtr InsertResidueAfter(int index, const ResNum& n, 
                                    const ResidueKey& k);
  ResidueImplPtr AppendResidue(const ResidueKey& k);

  ResidueImplPtr AppendResidue(const ResidueKey& k, const ResNum& n);
  
  // next residue, not necessarily in sequence
  ResidueImplPtr GetPrev(const ResidueImplPtr& r) const;
  
  ResidueImplPtr GetNext(const ResidueImplPtr& r) const;

  const ResidueImplList& GetResidueList() const;
  bool InSequence() const;
  
  ResidueImplList& GetResidueList() {
    return residue_list_;
  }
  
  Real GetMass() const;
  geom::Vec3 GetCenterOfMass() const;
  geom::Vec3 GetCenterOfAtoms() const;
  
  geom::AlignedCuboid GetBounds() const;
  
  
  ResidueImplPtr FindResidue(const ResNum& number) const;
  
  AtomImplPtr FindAtom(const ResNum& number, 
                       const String& atom_name) const;  
                     
  int GetResidueCount() const;
  
  int GetAtomCount() const;

  int GetBondCount() const;
      
  void Apply(EntityVisitor& v);

  EntityImplPtr GetEntity() const;
  
  void DeleteResidue(const ResNum& number);
  
  void DeleteResidue(const ResidueImplPtr& residue);
  
  void DeleteAllResidues();

  void ReorderResidues();

  void RenumberAllResidues(int start, bool keep_spacing);
  
  int GetIndex(const ResidueImplPtr& res) const;
  void AssignSecondaryStructure(SecStructure ss,
                                const ResNum& start,
                                const ResNum& end); 
  int GetIndexForResNum(const ResNum& number) const;
  void SetInSequence(int index);

private:
  int GetIndexForResNumInSequence(const ResNum& number) const;
  void UpdateShifts();
  typedef struct {
    int start;
    int shift;
  } Shift;
  std::list<Shift> shifts_;
  EntityImplW      ent_;
  String           name_;
  ResidueImplList  residue_list_;
  bool             in_sequence_;
  ChainType        type_;
  String           description_; 
};

}}} // ns

#endif