GrainDataTracker.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#pragma once
 
#include "GrainTracker.h"
 
template <typename T>
class GrainDataTracker : public GrainTracker
{
public:
  GrainDataTracker(const InputParameters & parameters);
 
  const T & getData(unsigned int grain_id) const;
 
protected:
  virtual T newGrain(unsigned int new_grain_id) = 0;
 
  virtual void newGrainCreated(unsigned int new_grain_id);
 
  std::vector<T> & _grain_data;
};
 
template <typename T>
GrainDataTracker<T>::GrainDataTracker(const InputParameters & parameters)
  : GrainTracker(parameters), _grain_data(declareRestartableData<std::vector<T>>("grain_data"))
{
}
 
template <typename T>
const T &
GrainDataTracker<T>::getData(unsigned int grain_id) const
{
  mooseAssert(grain_id < _grain_data.size(), "Requested data for invalid grain index.");
  return _grain_data[grain_id];
}
 
template <typename T>
void
GrainDataTracker<T>::newGrainCreated(unsigned int new_grain_id)
{
  if (_grain_data.size() <= new_grain_id)
    _grain_data.resize(new_grain_id + 1);
 
  _grain_data[new_grain_id] = newGrain(new_grain_id);
}