coord_group.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// This file is part of the OpenStructure project <www.openstructure.org>
//
// Copyright (C) 2008-2020 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_COORD_GROUP_HH
#define OST_COORD_GROUP_HH

/*
  Author: Ansgar Philippsen, Marco Biasini
*/

#include <vector>
#include <boost/shared_array.hpp>
#include "atom_handle.hh"
#include "coord_source.hh"

namespace geom {
  // we should really have fw headers...
  class Transform;
}

namespace ost { namespace mol {

class DLLEXPORT_OST_MOL CoordGroupHandle {
public:
  CoordGroupHandle();
  CoordGroupHandle(CoordSourcePtr source);

  EntityHandle GetEntity() const;

  //EntityGroupHandle ExtractFromTo(int start, int end, int step=1);
  //EntityGroupHandle ExtractEvery(int step, int start=0);

  uint GetAtomCount() const;

  uint GetFrameCount() const;

  float GetDelta() const;

  void SetDelta(float d);

  float GetStartTime() const;

  void SetStartTime(float t);

  void SetFramePositions(uint frame, const geom::Vec3List& clist);

  void SetFrameCellSize(uint frame, const geom::Vec3& size);

  geom::Vec3List GetFramePositions(uint frame);  
  
  void CopyFrame(uint frame);
  
  void Capture();

  void CaptureInto(int pos);

  void Capture(uint frame);
  
  //void AddFrame(const std::vector<geom::Vec3>& clist);
  void AddFrame(const geom::Vec3List& clist);
  void AddFrame(const geom::Vec3List& clist,const geom::Vec3& cell_size,const geom::Vec3& cell_angles);
  
  void AddFrames(const CoordGroupHandle& cg);
  void SetAtomPos(uint frame, AtomHandle atom, const geom::Vec3& pos);

  geom::Vec3 GetAtomPos(uint frame, AtomHandle atom) const;
  
  bool IsValid() const;

  operator bool() const;
  
  AtomHandleList GetAtomList() const;
  CoordFramePtr GetFrame(uint frame) const;
  CoordFrame GetFrame2(uint frame);
  
  CoordGroupHandle Filter(const EntityView& selected,int first=0,int last=-1,int stride=1) const;

  void ApplyTransform(const geom::Transform& tf);


private:
  void CheckValidity() const;
  CoordSourcePtr source_;
};

// factory method
// create with a reference set of atoms and zero frames
DLLEXPORT_OST_MOL CoordGroupHandle CreateCoordGroup(const AtomHandleList& atoms);

}} // ns

#endif