902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
|
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
|
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
+
+
-
-
+
-
+
-
-
-
-
-
-
+
+
-
+
+
-
-
-
-
+
+
+
+
-
+
-
+
+
+
+
-
+
|
data: SolverData,
cnt_descent: usize,
cnt_null: usize,
cnt_evals: usize,
weighter_state: W,
master_state: Option<MasterProcessState<M>>,
}
#[derive(Debug, Error)]
#[non_exhaustive]
pub enum SolverStateError {
#[error("Master problem creation failed")]
BuildMaster,
#[error("Restoring the master state failed")]
RestoreMasterState,
#[error("Getting the master state failed")]
GetMasterState,
#[error("Restoring the weighter state failed")]
RestoreWeighterState,
#[error("Getting the weighter state failed")]
GetWeighterState,
}
impl<P, T, W, M> Saveable for Solver<P, T, W, M>
where
P: FirstOrderProblem,
T: Terminator<SolverData> + Default,
W: Weighter<SolverData> + Default,
W: Weighter<SolverData> + Saveable + Default,
M: MasterBuilder<P::Minorant>,
M::MasterProblem: Saveable,
{
type State =
SyncSolverState<W::State, <M::MasterProblem as MasterProblem<<P as FirstOrderProblem>::Minorant>>::State>;
type Err = Error<
type Err = SolverStateError;
<P as FirstOrderProblem>::Err,
<<M as MasterBuilder<<P as FirstOrderProblem>::Minorant>>::MasterProblem as MasterProblem<
<P as FirstOrderProblem>::Minorant,
>>::Err,
>;
fn set_state(&mut self, state: Self::State) -> Result<(), P, M> {
fn set_state(&mut self, state: Self::State) -> std::result::Result<(), Self::Err> {
self.data = state.data;
self.cnt_descent = state.cnt_descent;
self.cnt_null = state.cnt_null;
self.cnt_evals = state.cnt_evals;
self.weighter
self.weighter.set_state(state.weighter_state);
.set_state(state.weighter_state)
.map_err(|_| SolverStateError::RestoreWeighterState)?;
let (tx, rx) = channel();
self.client_tx = Some(tx.clone());
self.client_rx = Some(rx);
self.master_proc = if let Some(s) = state.master_state {
let master = self
.master
.build_with_state(s.master_state)
.map_err(Error::BuildMaster)?;
let mut master = self.master.build().map_err(|_| SolverStateError::BuildMaster)?;
master
.set_state(s.master_state)
.map_err(|_| SolverStateError::RestoreMasterState)?;
Some(MasterProcess::start_with_center_idx(
master,
tx,
&mut self.threadpool,
s.center_idx,
))
} else {
None
};
Ok(())
}
/// Return the current solver state to the given writer.
fn get_state(&self) -> Result<Self::State, P, M> {
fn get_state(&self) -> std::result::Result<Self::State, Self::Err> {
Ok(SyncSolverState {
data: self.data.clone(),
cnt_descent: self.cnt_descent,
cnt_null: self.cnt_null,
cnt_evals: self.cnt_evals,
weighter_state: self.weighter.get_state(),
weighter_state: self
.weighter
.get_state()
.map_err(|_| SolverStateError::GetWeighterState)?,
master_state: if let Some(m) = self.master_proc.as_ref() {
Some(m.get_state()?)
Some(m.get_state().map_err(|_| SolverStateError::GetMasterState)?)
} else {
None
},
})
}
}
|