pocket.h

Go to the documentation of this file.

// -*- C++ -*-
// The POCKET class
 
#ifndef _POCKET_H_
#define _POCKET_H_
 
#include <set>
#include <unordered_map>
#include "ball.h"
#include "internal.h"


struct ball_order {
  bool operator() (const ball_t* a, const ball_t* b) const {
    return (a->color < b->color) ||
      ((a->color == b->color) && (a->uniq < b->uniq));
  };
};
 
typedef typename std::set<ball_t*,ball_order>::const_iterator ball_it;
typedef typename std::set<ball_t*,ball_order>::const_reverse_iterator ball_rev_it;


class pocket_t : public std::set<ball_t*,ball_order> {
 
private:

  void clean (void) {
    for (ball_t *b : *this) delete b;
    clear();
  };
 
public:
 
  // SERIALIZATION
  size_t bytesize (void) const {
    return sizeof(size_t) + size()*(ball_t::bytesize);
  };

  friend raw_t* operator>> (const pocket_t &p, raw_t *o) {
    size_t psize = p.size();
    memcpy(o,&psize,sizeof(size_t)); o += sizeof(size_t);
    for (ball_t *i : p)
      o = (*i >> o);
    return o;
  };

  friend raw_t* operator>> (raw_t *o, pocket_t &p) {
    p.clean();
    size_t psize;
    memcpy(&psize,o,sizeof(size_t)); o += sizeof(size_t);
    for (size_t i = 0; i < psize; i++) {
      ball_t *b = new ball_t();
      o = (o >> *b);
      p.insert(b);
    }
    return o;
  };
 
protected:
  void repair_holder (node_t* p) {
    for (ball_t *b : *this) {
      b->holder() = p;
    }
  };
 
public:
  void repair_owners (const std::unordered_map<name_t,node_t*>& node_name,
                      std::unordered_map<name_t,ball_t*> *ball_name) {
    for (ball_t *b : *this) {
      if (b->is(green)) {
        b->owner() = node_name.at(b->uniq);
        (*ball_name)[b->uniq] = b;
      }
    }
  };
 
public:

  ~pocket_t (void) {
    clean();
  };
 
public:

  bool holds (ball_t *b) const {
    ball_it i = find(b);
    return (i != end());
  };

  bool holds (color_t c) const {
    bool result = false;
    for (ball_it i = begin(); !result && i != end(); i++) {
      result = ((*i)->color == c);
    }
    return result;
  };

  ball_t* first_ball (void) const {
    return *cbegin();
  };

  ball_t* last_ball (void) const {
    return *crbegin();
  };

  ball_t* ball (const color_t c) const {
    ball_t *a = 0;
    for (ball_t *b : *this) {
      if (b->color == c) a = b;
    }
    assert(a != 0);
    return a;
  };

  string_t yaml (string_t tab = "") const;
  SEXP structure (void) const;
};
 
#endif