main: close tx half of sockets with closed rx half

fixes Gitea issue #9

pytec/plot.py

@@ -6,15 +6,19 @@ from pytec.client import Client
 
 TIME_WINDOW = 300.0
 
+tec = Client()
+target_temperature = tec.get_pid()[0]['target']
+print("Channel 0 target temperature: {:.3f}".format(target_temperature))
+
 class Series:
-    def __init__(self, scale=1.0):
-        self.scale = scale
+    def __init__(self, conv=lambda x: x):
+        self.conv = conv
         self.x_data = []
         self.y_data = []
 
     def append(self, x, y):
         self.x_data.append(x)
-        self.y_data.append(self.scale * y)
+        self.y_data.append(self.conv(y))
 
     def clip(self, min_x):
         drop = 0
@@ -25,8 +29,8 @@ class Series:
         
 series = {
     'adc': Series(),
-    'sens': Series(0.0001),
-    'temperature': Series(),
+    'sens': Series(lambda x: x * 0.0001),
+    'temperature': Series(lambda t: t - target_temperature),
     'i_set': Series(),
     'pid_output': Series(),
     'vref': Series(),
@@ -55,7 +59,6 @@ def recv_data(tec):
         if quit:
             break
 
-tec = Client()
 thread = Thread(target=recv_data, args=(tec,))
 thread.start()
 

pytec/pytec/client.py

@@ -1,6 +1,8 @@
 import socket
 import json
 
+CHANNELS = 2
+
 class Client:
     def __init__(self, host="192.168.1.26", port=23, timeout=None):
         self._socket = socket.create_connection((host, port), timeout)
@@ -22,6 +24,27 @@ class Client:
         self._lines = self._lines[1:]
         return line
 
+    def _get_conf(self, topic):
+        self._command(topic)
+        result = []
+        for channel in range(0, CHANNELS):
+            line = self._read_line()
+            conf = json.loads(line)
+            result.append(conf)
+        return result
+
+    def get_pwm(self):
+        return self._get_conf("pwm")
+
+    def get_pid(self):
+        return self._get_conf("pid")
+
+    def get_steinhart_hart(self):
+        return self._get_conf("s-h")
+
+    def get_postfilter(self):
+        return self._get_conf("postfilter")
+
     def report_mode(self):
         """Start reporting measurement values
 

src/main.rs

@@ -54,7 +54,7 @@ mod net;
 mod server;
 use server::Server;
 mod session;
-use session::{Session, SessionOutput};
+use session::{Session, SessionInput};
 mod command_parser;
 use command_parser::{Command, ShowCommand, PwmPin};
 mod timer;
@@ -194,10 +194,12 @@ fn main() -> ! {
                     if ! socket.is_active() {
                         let _ = socket.listen(TCP_PORT);
                         session.reset();
-                    } else if socket.can_send() && socket.can_recv() && socket.send_capacity() - socket.send_queue() > 1024 {
+                    } else if socket.may_send() && !socket.may_recv() {
+                        socket.close()
+                    } else if socket.can_send() && socket.can_recv() {
                         match socket.recv(|buf| session.feed(buf)) {
-                            Ok(SessionOutput::Nothing) => {}
-                            Ok(SessionOutput::Command(command)) => match command {
+                            Ok(SessionInput::Nothing) => {}
+                            Ok(SessionInput::Command(command)) => match command {
                                 Command::Quit =>
                                     socket.close(),
                                 Command::Reporting(reporting) => {
@@ -413,7 +415,7 @@ fn main() -> ! {
                                     SCB::sys_reset();
                                 }
                             }
-                            Ok(SessionOutput::Error(e)) => {
+                            Ok(SessionInput::Error(e)) => {
                                 let _ = writeln!(socket, "Command error: {:?}", e);
                             }
                             Err(_) =>

src/session.rs

@@ -38,16 +38,16 @@ impl LineReader {
     }
 }
 
-pub enum SessionOutput {
+pub enum SessionInput {
     Nothing,
     Command(Command),
     Error(ParserError),
 }
 
-impl From<Result<Command, ParserError>> for SessionOutput {
+impl From<Result<Command, ParserError>> for SessionInput {
     fn from(input: Result<Command, ParserError>) -> Self {
-        input.map(SessionOutput::Command)
-            .unwrap_or_else(SessionOutput::Error)
+        input.map(SessionInput::Command)
+            .unwrap_or_else(SessionInput::Error)
     }
 }
 
@@ -106,7 +106,7 @@ impl Session {
         self.report_pending[channel] = false;
     }
 
-    pub fn feed(&mut self, buf: &[u8]) -> (usize, SessionOutput) {
+    pub fn feed(&mut self, buf: &[u8]) -> (usize, SessionInput) {
         let mut buf_bytes = 0;
         for (i, b) in buf.iter().enumerate() {
             buf_bytes = i + 1;
@@ -125,6 +125,6 @@ impl Session {
                 None => {}
             }
         }
-        (buf_bytes, SessionOutput::Nothing)
+        (buf_bytes, SessionInput::Nothing)
     }
 }