Mercurial > ~darius > hgwebdir.cgi > pyinst
comparison scpisock.py @ 22:ef713fe03db8
Add SCPI socket class.
author | Daniel O'Connor <darius@dons.net.au> |
---|---|
date | Thu, 11 Aug 2011 21:18:09 +0930 |
parents | |
children | c6be52360c2f |
comparison
equal
deleted
inserted
replaced
21:6e7619f2dffd | 22:ef713fe03db8 |
---|---|
1 #!/usr/bin/env python | |
2 | |
3 # Copyright (c) 2011 | |
4 # Daniel O'Connor <darius@dons.net.au>. All rights reserved. | |
5 # | |
6 # Redistribution and use in source and binary forms, with or without | |
7 # modification, are permitted provided that the following conditions | |
8 # are met: | |
9 # 1. Redistributions of source code must retain the above copyright | |
10 # notice, this list of conditions and the following disclaimer. | |
11 # 2. Redistributions in binary form must reproduce the above copyright | |
12 # notice, this list of conditions and the following disclaimer in the | |
13 # documentation and/or other materials provided with the distribution. | |
14 # | |
15 # THIS SOFTWARE IS PROVIDED BY AUTHOR AND CONTRIBUTORS ``AS IS'' AND | |
16 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |
17 # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
18 # ARE DISCLAIMED. IN NO EVENT SHALL AUTHOR OR CONTRIBUTORS BE LIABLE | |
19 # FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |
20 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |
21 # OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |
22 # HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |
23 # LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |
24 # OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |
25 # SUCH DAMAGE. | |
26 # | |
27 | |
28 import select | |
29 import socket | |
30 | |
31 # See this for more info | |
32 # http://www.staff.uni-bayreuth.de/~btp918/cmt2007/geraete/agilent_34410A_dmm/manual/SCPI_Sockets.pdf | |
33 # | |
34 # ToDo: Implement the SRQ & DCL channel (I don't have any supported hardware) | |
35 | |
36 SCPI_PORT = 5025 | |
37 | |
38 class SCPISockDevice(object): | |
39 def __init__(self, host, port = SCPI_PORT): | |
40 self.sock = socket.create_connection((host, port)) | |
41 | |
42 def flush(self): | |
43 while True: | |
44 r, w, x = select.select([self.sock], [], [], 0) | |
45 if len(r) == 0: | |
46 break | |
47 self.sock.recv(1024) | |
48 | |
49 def write(self, data): | |
50 trail = '' | |
51 if data[-1] != '\n': | |
52 trail = '\n' | |
53 | |
54 self.sock.send(data + trail) | |
55 | |
56 def read(self, timeout = None): | |
57 res = '' | |
58 if timeout == None: | |
59 timeout = 0.1 | |
60 | |
61 while True: | |
62 r, w, x = select.select([self.sock], [], [], timeout) | |
63 if len(r) == 0: | |
64 break | |
65 res = res + self.sock.recv(1024) | |
66 if res[-1] == '\n': | |
67 break | |
68 | |
69 return res.rstrip('\n') | |
70 |