owlib.py

   1 #!/usr/bin/python
   2 # Copyright (c) 2017, Tobias Mueller tm(at)tm3d.de
   3 # All rights reserved.
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions are
   7 # met:
   8 #
   9 #  * Redistributions of source code must retain the above copyright
  10 #    notice, this list of conditions and the following disclaimer.
  11 #  * 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
  14 #    distribution.
  15 #  * All advertising materials mentioning features or use of this
  16 #    software must display the following acknowledgement: This product
  17 #    includes software developed by tm3d.de and its contributors.
  18 #  * Neither the name of tm3d.de nor the names of its contributors may
  19 #    be used to endorse or promote products derived from this software
  20 #    without specific prior written permission.
  21 #
  22 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  23 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  24 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  25 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  26 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  27 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  28 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  29 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  30 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  31 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  32 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  33
  34
  35
  36
  37 import time
  38 import subprocess
  39 import sys
  40 devInfo= ["","temperature","pressure","illuminance","humidity","constant","voltage","current","VOC","counter","CO2","resistance"]
  41 devUnit= ["","°C","hPa","lux","%","","V","mA","ppm","","ppm","kOhm"]
  42
  43 devController=["","Old code","Attiny84A","Attiny44","Atmega328"]
  44
  45 devChip=["","DS18B20","DS2438","DS2438","DS2438","DS2450","Thermoelement","SHT21","SHT25","DHT22","HIH9021","HDC1080","HIH4030","HIH5030","BMP280","MAX44009","CDM7160","MAX1164/TGS8100","TGS8100","DS2423","SHT35","SHT31"]
  46
  47
  48 def calcValue(code,vn,V):
  49         if code==0:
  50                 return V[vn]
  51         if code==1:
  52                 return V[vn] / 16.0;
  53         if code==2:
  54                 return V[vn]/1.6;
  55         if code==3:
  56                 return V[vn]*0.2 + 700;
  57         if code==4:
  58                 return exp(V[vn] / 160.0);
  59         if code==5:
  60                 return V[vn]*62.5 + 55000;
  61         if code==6:
  62                 return V[vn] / 256.0;
  63         if code==7:
  64                 return ((float(V[2]) / float(V[1]) - 0.16) / 0.0062) / (1.0546 - 0.00216*V[0]/256.0);
  65         if code==8:
  66                 return V[vn] / 100.0;
  67         if code==9:
  68                 return V[vn] / 65535.0*5.1;
  69         if code==10:
  70                 return V[vn] / 65535.0*2.55;
  71         if code==11:
  72                 return V[vn] / 65535.0*1.1;
  73         if code==12:
  74                 return V[vn] / 10.0;
  75         if code==13:
  76                 return V[vn];
  77         if code==14:
  78                 return (V[vn] - 32767.0) / 100.0;
  79         if code==15:
  80                 return exp((V[vn]-32767.0)/1000.0);
  81         if code==16:
  82                 return V[vn]/32.0;
  83         return 0;
  84
  85
  86
  87 def owcom(dev,send,rc):
  88         res=[]
  89         try:
  90                 f=open("/sys/bus/w1/devices/%s/rw" %(dev),"r+b",0)
  91         except IOError:
  92                 l=subprocess.check_output("rmmod w1_therm",shell=True)
  93                 f=open("/sys/bus/w1/devices/%s/rw" %(dev),"r+b",0)
  94         f.write("".join(map(chr, send)))
  95         if (rc!=0):
  96                 res=map(ord,f.read(rc))
  97         f.close()
  98         return res
  99
 100
 101 def crc8(arr):
 102         lscrc=0x0;
 103         for v in arr:
 104                 bit=1;
 105                 while bit<256:
 106                         if (v&bit)==bit:
 107                                 lb=1
 108                         else:
 109                                 lb=0
 110                         if (lscrc&1)!=lb:
 111                                 lscrc=(lscrc>>1)^0x8c
 112                         else:
 113                                 lscrc=(lscrc>>1)
 114                         bit=bit*2
 115         return (lscrc==0)
 116
 117
 118 oddparity = [ 0, 1, 1, 0, 1, 0, 0, 1, 1, 0, 0, 1, 0, 1, 1, 0 ]
 119
 120
 121 def crc16(arr):
 122         crc16=0
 123         for cdata in arr:
 124                 cdata = (cdata ^ (crc16 & 0xff)) & 0xff;
 125                 crc16 = crc16>>8;
 126                 if (oddparity[cdata & 0xf] ^ oddparity[cdata >> 4]):
 127                         crc16 = crc16 ^ 0xc001;
 128                 cdata = (cdata << 6)&0xffff;
 129                 crc16  = crc16 ^ cdata;
 130                 cdata = (cdata << 1)&0xffff;
 131                 crc16  = crc16^cdata;
 132         #r=crc16==0xB001
 133         return (crc16==0xB001)
 134
 135
 136 def ow_fconvert(b1, b2):
 137         tsht = b1 | (b2 << 8)
 138         if (b2 & 0x080):
 139                 tsht=-(tsht&0x7FFF)
 140         return tsht
 141
 142
 143 def testnr(s):
 144         for c in s.lower()[:]:
 145                 if not((c) in ['0','1','2','3','4','5','6','7','8','9','a','b','c','d','e','f','-']):
 146                         return False
 147         return True
 148
 149 def addid(id,val):
 150         for i in range(7):
 151                 id[i+1]=id[i+1]+val
 152                 if id[i+1]>254:
 153                         id[i+1]=id[i+1]-254
 154                         val=1
 155                 else:
 156                         return id
 157         return id
 158 #
 159
 160 def owfind(famcodes):
 161         l=subprocess.check_output("ls /sys/bus/w1/devices/", shell=True)
 162         dl=[]
 163         for g in  (l.split("\n")):
 164                 if len(g)>2:
 165                         if testnr(g[0:2]):
 166                                 if (g[0:2] in famcodes) or len(famcodes)==0:
 167                                         dl.append(g)
 168                         print g
 169         return dl;
 170
 171
 172 class owDevice:
 173         raw=[0,0,0,0]
 174         values=[0.0,0.0,0.0,0.0]
 175         owid=""
 176         config=[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
 177         def __init__(self, owid):
 178                 self.owid = owid
 179
 180         def setdefaultConfig(self):
 181                 raise NotImplementedError()
 182
 183         def readConfig(self):
 184                 self.config=owcom(self.owid,[0x85],26)
 185                 print self.config
 186                 if self.config[0]==0xFF:
 187                         print("No Deviceconfig. Not a Device form tm3d.de. Set Default");
 188                         self.setdefaultConfig()
 189                 else:
 190                         if self.config[25]==0xFF:
 191                                 if crc8(self.config[0:24]):
 192                                         print("CRC Error reading Deviceconfig. Set Default")
 193                                         self.setdefaultConfig()
 194                         else:
 195                                 if not(crc16([0x85]+self.config)):
 196                                         print("CRC Error reading Deviceconfig. Set Default")
 197                                         self.setdefaultConfig()
 198
 199         def convertAll(self):
 200                 raise NotImplementedError()
 201
 202         def getChips(self):
 203                 l=[];
 204                 for i in range(0,4):
 205                         l.append(devChip[self.config[i+9]])
 206                 return l
 207
 208         def getUnits(self):
 209                 l=[];
 210                 for i in range(0,4):
 211                         l.append(devUnit[self.config[i*2]])
 212                 return l
 213
 214         def getProperties(self):
 215                 l=[];
 216                 for i in range(0,4):
 217                         l.append(devInfo[self.config[i*2]])
 218                 return l
 219
 220
 221 class owDS2450(owDevice):
 222         def setdefaultConfig(self):
 223                 self.config=[6,9, 6,9, 6,9, 6,9, 0,5,5,5,5,0,0,0,0,0,0,0,0,0,0,0]
 224
 225         def convertAll(self):
 226                 ret=owcom(self.owid,[0x3C,0x0F,0x00],2)
 227                 if not(crc16([0x3C,0x0F,0x00]+ret)):
 228                         print("CRC ERROR")
 229                         return
 230                 time.sleep(1)
 231                 ret=owcom(self.owid,[0xAA,0x00,0x00],10)
 232                 if not(crc16([0xAA,0x00,0x00]+ret)):
 233                         print("CRC ERROR")
 234                         return
 235                 for i in range(0,4):
 236                         self.raw[i]=ret[2*i]|(ret[2*i+1]<<8);
 237                 for i in range(0,4):
 238                         self.values[i]=calcValue(self.config[i*2+1],i,self.raw)
 239
 240
 241 class owDS18B20(owDevice):
 242         def setdefaultConfig(self):
 243                 self.config=[1,1, 0,0, 0,0, 0,0, 0,1,0,0,0,0,0,0,0,0,0,0,0,0,0,0]
 244
 245         def convertAll(self):
 246                 owcom(self.owid,[0x44],0)
 247                 time.sleep(1)
 248                 ret=owcom(self.owid,[0xBE],9)
 249                 if not(crc8(ret)):
 250                         print("CRC ERROR")
 251                         return
 252                 self.raw[0]=ow_fconvert(ret[0],ret[1]);
 253                 self.values[0]=calcValue(self.config[0*2+1],0,self.raw)
 254                 self.values[1]=0;
 255                 self.values[2]=0;
 256                 self.values[3]=0;
 257
 258 class owDS2438(owDevice):
 259         def setdefaultConfig(self):
 260                 self.config=[1,6, 6,8, 4,7, 7,17, 0,2,3,12,4,0,0,0,0,0,0,0,0,0,0,0]
 261
 262         def readScratchpad(self,page,recall):
 263                 if (recall):
 264                         owcom(self.owid,[0xB8,page],0)
 265                 ret=owcom(self.owid,[0xBE,page],9)
 266                 if not(crc8(ret)):
 267                         print("CRC ERROR")
 268                         return []
 269                 return ret
 270
 271         def setConfigByte(self, cb):
 272                 for i in range(3):
 273                         owcom(self.owid,[0x4E,0x00,cb],0)
 274                         sp=self.readScratchpad(0,False)
 275                         if (sp!=[]):
 276                                 if sp[0]==cb:
 277                                         return True
 278                 return False
 279
 280
 281
 282         def convertAll(self):
 283                 owcom(self.owid,[0x44],0)
 284                 time.sleep(0.01)
 285                 self.setConfigByte(0x08)
 286                 owcom(self.owid,[0xB4],0)
 287                 time.sleep(0.01)
 288                 sp=self.readScratchpad(0,True)
 289                 if sp==[]:
 290                         return
 291                 temp=ow_fconvert(sp[1],sp[2]);
 292                 VDD=ow_fconvert(sp[3],sp[4]);
 293                 self.setConfigByte(0x00)
 294                 owcom(self.owid,[0xB4],0)
 295                 time.sleep(0.01)
 296                 sp=self.readScratchpad(0,True);
 297                 if sp==[]:
 298                         return
 299                 I=ow_fconvert(sp[5],sp[6]);
 300                 VAD=ow_fconvert(sp[3],sp[4]);
 301                 self.raw[0]=temp;
 302                 self.raw[1]=VDD;
 303                 self.raw[2]=VAD;
 304                 self.raw[3]=I;
 305                 for i in range(0,4):
 306                         self.values[i]=calcValue(self.config[i*2+1],i,self.raw)
 307
 308 class owDS2423(owDevice):
 309         def setdefaultConfig(self):
 310                 self.config=[9,13, 9,13, 9,13, 9,13, 0,19,19,19,19,0,0,0,0,0,0,0,0,0,0,0]
 311
 312         def readCounter(self,page):
 313                 addr=(page<<5)+31
 314                 ret=owcom(self.owid,[0xA5,addr&0xFF,addr>>8],11)
 315                 if not(crc16([0xA5,addr&0xFF,addr>>8]+ret)):
 316                         print("CRC ERROR")
 317                         return 0
 318                 c=0
 319                 for i in range (4):
 320                         c<<=8;
 321                         c|=ret[4-i];
 322                 return c;
 323
 324
 325         def convertAll(self):
 326                 for i in range(4):
 327                         self.raw[i]=self.readCounter(i+12);
 328                 for i in range(0,4):
 329                         self.values[i]=calcValue(self.config[i*2+1],i,self.raw)
 330