Add FFT problem class (idaholab#401)

dschwen · Jan 20, 2020 · e5f9480 · e5f9480
1 parent 67a82f0
commit e5f9480
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 0 deletions.
diff --git a/include/problems/FFTProblem.h b/include/problems/FFTProblem.h
@@ -0,0 +1,25 @@
+/**********************************************************************/
+/*                     DO NOT MODIFY THIS HEADER                      */
+/* MAGPIE - Mesoscale Atomistic Glue Program for Integrated Execution */
+/*                                                                    */
+/*            Copyright 2017 Battelle Energy Alliance, LLC            */
+/*                        ALL RIGHTS RESERVED                         */
+/**********************************************************************/
+
+#pragma once
+
+#include "FEProblem.h"
+
+/**
+ * Enhanced FEProblem that supports FFT buffers as variables
+ */
+class FFTProblem : public FEProblem
+{
+public:
+  static InputParameters validParams();
+
+  FFTProblem(const InputParameters & parameters);
+
+protected:
+  std::vector<MooseVariable *> _fft_vars;
+};
diff --git a/src/problems/FFTProblem.C b/src/problems/FFTProblem.C
@@ -0,0 +1,24 @@
+/**********************************************************************/
+/*                     DO NOT MODIFY THIS HEADER                      */
+/* MAGPIE - Mesoscale Atomistic Glue Program for Integrated Execution */
+/*                                                                    */
+/*            Copyright 2017 Battelle Energy Alliance, LLC            */
+/*                        ALL RIGHTS RESERVED                         */
+/**********************************************************************/
+
+#include "FFTProblem.h"
+
+registerMooseObject("MagpieApp", FFTProblem);
+
+defineLegacyParams(FFTProblem);
+
+InputParameters
+FFTProblem::validParams()
+{
+  InputParameters params = FFTProblem::validParams();
+  params.addClassDescription("Enhanced FEProblem that supports FFT buffers as variables.");
+
+  return params;
+}
+
+FFTProblem::FFTProblem(const InputParameters & parameters) : FEProblem(parameters) {}