res_tracker.cc

Go to the documentation of this file.

#include "res_tracker.h"
 
#include "recorder.h"
#include "cyc_limits.h"
 
namespace cyclus {
 
ResTracker::ResTracker(Context* ctx, Resource* r)
    : tracked_(true), res_(r), ctx_(ctx), parent1_(0), parent2_(0) {}
 
void ResTracker::DontTrack() {
  tracked_ = false;
}
 
void ResTracker::Create(Agent* creator) {
  if (!tracked_) {
    return;
  }
 
  parent1_ = 0;
  parent2_ = 0;
  bool bumpId = false;
  Record(bumpId);
  ctx_->NewDatum("ResCreators")
      ->AddVal("ResourceId", res_->state_id())
      ->AddVal("AgentId", creator->id())
      ->Record();
}
 
void ResTracker::Modify() {
  if (!tracked_) {
    return;
  }
 
  parent1_ = res_->state_id();
  parent2_ = 0;
  Record();
}
 
void ResTracker::Extract(ResTracker* removed) {
  if (!tracked_) {
    return;
  }
 
  if (res_->quantity() > eps_rsrc()) {
    parent1_ = res_->state_id();
    parent2_ = 0;
  }
 
  // removed parent must be set before the resource is recorded, otherwise
  // removed ends up with parent1 of the other child (which is this resource
  // after being bumped)
  removed->parent1_ = res_->state_id();
  removed->parent2_ = 0;
  removed->tracked_ = tracked_;
  removed->Record();
 
  if (res_->quantity() > eps_rsrc()) {
    Record();
  }
}
 
void ResTracker::Absorb(ResTracker* absorbed) {
  if (!tracked_) {
    return;
  }
 
  parent1_ = res_->state_id();
  parent2_ = absorbed->res_->state_id();
  Record();
}
 
void ResTracker::Package(ResTracker* parent) {
  if (!tracked_) {
    return;
  }
  parent2_ = 0;
  tracked_ = tracked_;
  package_name_ = res_->package_name();
 
  if (parent != NULL) {
    parent1_ = parent->res_->state_id();
 
    // Resource was just created, with packaging info, and assigned a state id.
    // Do not need to bump again
    bool bumpId = false;
    Record(bumpId);
  } else {
    // Resource was not just created. It is being re-packaged. It needs to be
    // bumped to get a new state id.
    parent1_ = res_->state_id();
    Record();
  }
}
 
void ResTracker::Record(bool bumpId) {
  if (bumpId) {
    res_->BumpStateId();
  }
  ctx_->NewDatum("Resources")
      ->AddVal("ResourceId", res_->state_id())
      ->AddVal("ObjId", res_->obj_id())
      ->AddVal("Type", res_->type())
      ->AddVal("TimeCreated", ctx_->time())
      ->AddVal("Quantity", res_->quantity())
      ->AddVal("Units", res_->units())
      ->AddVal("UnitValue", res_->UnitValue())
      ->AddVal("QualId", res_->qual_id())
      ->AddVal("PackageName", res_->package_name())
      ->AddVal("Parent1", parent1_)
      ->AddVal("Parent2", parent2_)
      ->Record();
  res_->Record(ctx_);
}
 
}  // namespace cyclus