77from .memory import MemoryInterface
88
99
10- class UnsignedInteger ( object ) :
10+ class UnsignedInteger :
1111 def __init__ (self , size ):
1212 self .size = size
1313
@@ -18,7 +18,7 @@ def to_binary(self, val):
1818 return intbv (val , max = 2 ** self .size , min = 0 )._val
1919
2020
21- class SignedInteger ( object ) :
21+ class SignedInteger :
2222 def __init__ (self , size ):
2323 self .size = size
2424
@@ -36,7 +36,6 @@ def to_binary(self, val):
3636class EnumTypeWrapper (UnsignedInteger ):
3737 def __init__ (self , enum_type ):
3838 self ._enum_type = enum_type
39- maximum = max (self ._enum_type ._reverse_dct .keys ())
4039 size = int (ceil (log (16 , 2 )))
4140 super (EnumTypeWrapper , self ).__init__ (size = size )
4241
@@ -90,7 +89,7 @@ def to_binary(self, val):
9089 return out [32 :]._val
9190
9291
93- class RegisterProperty ( object ) :
92+ class RegisterProperty :
9493 def __init__ (self , addr , register_type , size = None ):
9594 if isinstance (register_type , type ) and issubclass (register_type , Enum ):
9695 register_type = EnumTypeWrapper (register_type )
@@ -121,7 +120,7 @@ class GetSetRegister(SetRegister, GetRegister):
121120 pass
122121
123122
124- class GetSetBit ( object ) :
123+ class GetSetBit :
125124 def __init__ (self , addr , pos , bit_type = None ):
126125 self ._bit_type = bit_type
127126 self .addr = addr
@@ -141,13 +140,6 @@ def __set__(self, instance, value):
141140 return instance .write (self .addr , int (new_value ))
142141
143142
144- # class EnumRegister(GetSetRegister):
145- # def __init__(self, addr, enum_type):
146- # assert issubclass(enum_type, Enum)
147- # register_type = EnumType(enum_type)
148- # super(EnumRegister, self).__init__(addr, register_type)
149-
150-
151143class HK (MemoryInterface ):
152144 def __init__ (self , addr_base = 0x40000000 , ** kwd ):
153145 kwd ["addr_base" ] = addr_base
@@ -200,10 +192,10 @@ def __init__(self, addr_base=0x40400000, **kwd):
200192class TriggerSource (Enum ):
201193 none = 0
202194 immediately = 1
203- chA_posedge = 2
204- chA_negedge = 3
205- chB_posedge = 4
206- chB_negedge = 5
195+ cha_posedge = 2
196+ cha_negedge = 3
197+ chb_posedge = 4
198+ chb_negedge = 5
207199 ext_posedge = 6
208200 ext_negedge = 7
209201 awg_posedge = 8
@@ -248,7 +240,7 @@ def arm_trigger(self, v=True):
248240 dac2_on_ch1 = GetSetBit (0x50 , pos = 0 )
249241 dac1_on_ch2 = GetSetBit (0x50 , pos = 1 )
250242
251- ## Function specific to read the array of data
243+ # Function specific to read the array of data
252244 def get_rawdata (self , addr ):
253245 x = self .reads (addr , self .data_length )
254246 y = x .copy ()
@@ -295,7 +287,7 @@ def setup(self, frequency=1, trigger_source=TriggerSource.immediately):
295287 self .arm_trigger ()
296288
297289 def rearm (self , frequency = None , trigger_source = 8 ):
298- if not frequency is None :
290+ if frequency is not None :
299291 self .frequency = frequency
300292 self .trigger_delay = self .data_length
301293 self .trigger_source = trigger_source
@@ -515,7 +507,7 @@ def setup(self, frequency=1):
515507 self .sm_reset = False
516508
517509 def trig (self , frequency = None ):
518- if not frequency is None :
510+ if frequency is not None :
519511 self .frequency = frequency
520512 self .start_offset = 0
521513 self .trig_selector = 1
@@ -582,15 +574,15 @@ def derivative(self, val):
582574
583575 def initialize (self , setpoint = None , integral = 0 , proportional = 0 , derivative = 0 ):
584576 self .reset = True
585- if not setpoint is None :
577+ if setpoint is not None :
586578 self .setpoint = setpoint
587579 self .integral = integral
588580 self .proportional = proportional
589581 self .derivative = derivative
590582 self .reset = False
591583
592584
593- class InterfaceDescriptor ( object ) :
585+ class InterfaceDescriptor :
594586 def __init__ (self , cls , ** kwd ):
595587 self ._cls = cls
596588 self .kwd = kwd
@@ -601,7 +593,7 @@ def __get__(self, instance, owner):
601593 return self ._cls (parent_memory = instance , ** self .kwd )
602594
603595
604- class RedPitaya ( object ) :
596+ class RedPitaya :
605597 hk = InterfaceDescriptor (HK )
606598 ams = InterfaceDescriptor (AMS )
607599 scope = InterfaceDescriptor (Scope )
@@ -613,21 +605,10 @@ class RedPitaya(object):
613605 asgb = InterfaceDescriptor (ASG , channel = "B" )
614606
615607
616- # class BoardRedPitaya(RedPitaya, BoardRawMemory):
617- # pass
618-
619-
620- # class PCRedPitaya(RedPitaya, ClientMemory):
621- # pass
622-
623608if __name__ == "__main__" :
624609 from time import sleep , time
625610
626- import rpyc
627-
628- # conn = rpyc.connect('134.157.6.206', 18861)
629- # red_pitaya = PCRedPitaya(remote_connection = conn)
630- red_pitaya = BoardRedPitaya ()
611+ red_pitaya = RedPitaya ()
631612 red_pitaya .scope .arm_trigger ()
632613 red_pitaya .scope .trigger_source = 1
633614 sleep (1 )
0 commit comments