feat: add OverrideInitData

Author: AayushSabharwal

src/SciMLBase.jl

@@ -654,6 +654,7 @@ Internal. Used for signifying the AD context comes from a Tracker.jl context.
 struct TrackerOriginator <: ADOriginator end
 
 include("utils.jl")
+include("initialization.jl")
 include("function_wrappers.jl")
 include("scimlfunctions.jl")
 include("alg_traits.jl")

src/initialization.jl

@@ -0,0 +1,26 @@
+"""
+    $(TYPEDEF)
+
+A collection of all the data required for `OverrideInit`.
+"""
+struct OverrideInitData{IProb, UIProb, IProbMap, IProbPmap}
+    """
+    The `AbstractNonlinearProblem` to solve for initialization.
+    """
+    initializeprob::IProb
+    """
+    A function which takes `(initializeprob, prob)` and updates
+    the parameters of the former with their values in the latter.
+    """
+    update_initializeprob!::UIProb
+    """
+    A function which takes the solution of `initializeprob` and returns
+    the state vector of the original problem.
+    """
+    initializeprobmap::IProbMap
+    """
+    A function which takes the solution of `initializeprob` and returns
+    the parameter object of the original problem.
+    """
+    initializeprobpmap::IProbPmap
+end