CollectionMgr.h

Go to the documentation of this file.

#ifndef COLLECTIONMGR_H
#define COLLECTIONMGR_H

#include "charm++.h"

#include "main.h"
#include "NamdTypes.h"
#include "BOCgroup.h"
#include "PatchMap.h"
#include "ProcessorPrivate.h"
#include "CollectionMgr.decl.h"


class SlaveInitMsg : public CMessage_SlaveInitMsg
{
public:
  CkChareID master;
};

class CollectionMgr : public BOCclass
{
public:

  static CollectionMgr *Object() { 
    return CkpvAccess(CollectionMgr_instance); 
  }
  CollectionMgr(SlaveInitMsg *msg);
  ~CollectionMgr(void);

  void submitPositions(int seq, FullAtomList &a, Lattice l, int prec);
  void submitVelocities(int seq, int zero, FullAtomList &a);
  void sendDataStream(const char *);

  class CollectVectorInstance
  {
  public:

    CollectVectorInstance(void) : seq(-10) { ; }

    void free() { seq = -10; }
    int notfree() { return ( seq != -10 ); }

    void reset(int s, int p) {
      if ( s == -10 ) NAMD_bug("seq == free in CollectionMgr");
      seq = s;
      precisions = p;
      remaining = PatchMap::Object()->numHomePatches();
      aid.resize(0);
      data.resize(0);
      fdata.resize(0);
    }

    // true -> send it and delete it!
    int append(AtomIDList &a, ResizeArray<Vector> &d)
    {
      int size = a.size();
      for( int i = 0; i < size; ++i )
      {
        aid.add(a[i]);
        if ( precisions & 2 ) data.add(d[i]);
        if ( precisions & 1 ) fdata.add(d[i]);
      }
      return ( ! --remaining );
    }

    int seq;
    AtomIDList aid;
    int precisions;
    ResizeArray<Vector> data;
    ResizeArray<FloatVector> fdata;

  private:
    int remaining;

  };

  class CollectVectorSequence
  {
  public:

    CollectVectorInstance* submitData(
        int seq, AtomIDList &i, ResizeArray<Vector> &d, int prec=2)
    {
      CollectVectorInstance **c = data.begin();
      CollectVectorInstance **c_e = data.end();
      for( ; c != c_e && (*c)->seq != seq; ++c );
      if ( c == c_e )
      {
       c = data.begin();
       for( ; c != c_e && (*c)->notfree(); ++c );
       if ( c == c_e ) {
        data.add(new CollectVectorInstance);
        c = data.end() - 1;
       }
       (*c)->reset(seq,prec);
      }
      if ( (*c)->append(i,d) )
      {
        return *c;
      }
      else
      {
        return 0;
      }
    }

    ResizeArray<CollectVectorInstance*> data;

  };
private:

  CkChareID master;


  CollectVectorSequence positions;
  CollectVectorSequence velocities;

public:
  void setCollectionMaster(SlaveInitMsg *msg){ 
    master = msg->master; 
    delete msg;
  }
};

#endif

---