95 lines
2.8 KiB
Rust
95 lines
2.8 KiB
Rust
use crate::pus::PusReceiver;
|
|
use satrs::pool::{StoreAddr, StoreError};
|
|
use satrs::pus::{EcssTcAndToken, MpscTmAsVecSender};
|
|
use satrs::spacepackets::ecss::PusPacket;
|
|
use satrs::{
|
|
pus::ReceivesEcssPusTc,
|
|
spacepackets::{ecss::tc::PusTcReader, SpHeader},
|
|
tmtc::ReceivesCcsdsTc,
|
|
};
|
|
use std::sync::mpsc::{self, SendError, Sender, TryRecvError};
|
|
use thiserror::Error;
|
|
|
|
#[derive(Debug, Clone, PartialEq, Eq, Error)]
|
|
pub enum MpscStoreAndSendError {
|
|
#[error("Store error: {0}")]
|
|
Store(#[from] StoreError),
|
|
#[error("TC send error: {0}")]
|
|
TcSend(#[from] SendError<EcssTcAndToken>),
|
|
#[error("TMTC send error: {0}")]
|
|
TmTcSend(#[from] SendError<StoreAddr>),
|
|
}
|
|
|
|
// Newtype, can not implement necessary traits on MPSC sender directly because of orphan rules.
|
|
#[derive(Clone)]
|
|
pub struct PusTcSourceProviderDynamic(pub Sender<Vec<u8>>);
|
|
|
|
impl ReceivesEcssPusTc for PusTcSourceProviderDynamic {
|
|
type Error = SendError<Vec<u8>>;
|
|
|
|
fn pass_pus_tc(&mut self, _: &SpHeader, pus_tc: &PusTcReader) -> Result<(), Self::Error> {
|
|
self.0.send(pus_tc.raw_data().to_vec())?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
impl ReceivesCcsdsTc for PusTcSourceProviderDynamic {
|
|
type Error = mpsc::SendError<Vec<u8>>;
|
|
|
|
fn pass_ccsds(&mut self, _: &SpHeader, tc_raw: &[u8]) -> Result<(), Self::Error> {
|
|
self.0.send(tc_raw.to_vec())?;
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
// TC source components where the heap is the backing memory of the received telecommands.
|
|
pub struct TcSourceTaskDynamic {
|
|
pub tc_receiver: mpsc::Receiver<Vec<u8>>,
|
|
pus_receiver: PusReceiver<MpscTmAsVecSender>,
|
|
}
|
|
|
|
impl TcSourceTaskDynamic {
|
|
pub fn new(
|
|
tc_receiver: mpsc::Receiver<Vec<u8>>,
|
|
pus_receiver: PusReceiver<MpscTmAsVecSender>,
|
|
) -> Self {
|
|
Self {
|
|
tc_receiver,
|
|
pus_receiver,
|
|
}
|
|
}
|
|
|
|
pub fn periodic_operation(&mut self) {
|
|
self.poll_tc();
|
|
}
|
|
|
|
pub fn poll_tc(&mut self) -> bool {
|
|
match self.tc_receiver.try_recv() {
|
|
Ok(tc) => match PusTcReader::new(&tc) {
|
|
Ok((pus_tc, _)) => {
|
|
self.pus_receiver
|
|
.handle_tc_packet(
|
|
satrs::pus::TcInMemory::Vec(tc.clone()),
|
|
pus_tc.service(),
|
|
&pus_tc,
|
|
)
|
|
.ok();
|
|
true
|
|
}
|
|
Err(e) => {
|
|
log::warn!("error creating PUS TC from raw data: {e}");
|
|
log::warn!("raw data: {:x?}", tc);
|
|
true
|
|
}
|
|
},
|
|
Err(e) => match e {
|
|
TryRecvError::Empty => false,
|
|
TryRecvError::Disconnected => {
|
|
log::warn!("tmtc thread: sender disconnected");
|
|
false
|
|
}
|
|
},
|
|
}
|
|
}
|
|
}
|