GrainDataTracker.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://www.mooseframework.org
//*
//* 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);
}