def __init__( self, nruns, experiment, ions, desired_order, out, conn=None ):
        #0 = no sym, use order 1 = symmetrize on order 2 = no order specificity
        camera = Luca()
        freq_src = FreqDriver( u'USB0::0x1AB1::0x0641::DG4B142100247::INSTR' )
        ni = NiDriver( self.Chans )
        reorder_time = 0.5

        output = open( out, 'w' )
        ion_order = []
        desired_bright_number = sum(map(lambda x: 1 if x else 0, desired_order))

        try:
            ion_positions = []
            with open( ions, 'r' ) as ionfile:
                for l in ionfile:
                    pos = tuple( int(x) for x in l.split() )
                    ion_positions.append( pos )
    
            bg = []
            for i in range(30):
                data = self.build_data(camera, ion_positions, camera.get_image())
                bg.append( data[-1] )
            threshold = np.mean(bg) + 3*np.std(bg)
            
            experiment.setup( freq_src, ni )
            print( ion_positions )
            while experiment.step( freq_src, ni ):
                for i in range( nruns ):
                    data = self.build_data( 
                        camera, ion_positions, camera.get_image() )
                    ion_order = [ d > threshold for d in data ]

                    bg.append( data[-1] )
                    if len( bg ) > 1000:
                        bg = bg[100:]
                    threshold = np.mean(bg) + 3*np.std(bg)

                    while ion_order != desired_order:
                        curr_bright_number = sum(map(lambda x: 1 if x else 0, ion_order))
                        if curr_bright_number == desired_bright_number:
                            
                            r = NiSimpleDriver( self.RedChan )
                            r.write_single( False )
                            time.sleep( reorder_time )
                            r.write_single( True )
                            r.close()
                        time.sleep( 0.5 )

                        camera.get_image()
                        data = self.build_data( 
                            camera, ion_positions, camera.get_image() )
                        
                        prev_order, ion_order = ion_order, \
                            [ d > threshold for d in data ]
                        print( "{} -> {}".format( prev_order, ion_order ) )

                        if any( prev_order ):
                            if prev_order == ion_order:
                                reorder_time *= 1.1
                            else:
                                reorder_time *= 0.9
                            print( "New reorder time: {}".format( reorder_time ) )
                        if reorder_time < 0.1:  reorder_time = 0.1
                        if reorder_time > 10.0: reorder_time = 10.0
                        
                        if conn is not None:
                            outdata = [str( experiment.control_var() )]
                            outdata.extend( str(d) for d in data )
                            outdata.extend( str(d) for d in data )
                            conn.send( 'reordata ' + '\t'.join(outdata) )
                    
                    ni.run()
                    data.extend( self.build_data( 
                        camera, ion_positions, camera.get_image() ) )

                    outdata = [str( experiment.control_var() )]
                    outdata.extend( str(d) for d in data )
                    print '\t'.join(outdata)
                    if conn is not None:
                        conn.send('\t'.join(outdata) )
                    output.write( '\t'.join(outdata) + '\n' )
                    output.flush()

                    d = NiSimpleDriver( self.OrangeChan )
                    d.write_single( True )
                    d.close()

                    time.sleep( 0.2 )
              
              
        finally:
            camera.shutdown()
 def _control_secular( self, on ):
     driver = NiDriver( "Dev1/port0/line1" )
     driver.write_single( on )
     driver.close()
     self.secular = on
import sys, time

from luca import Luca
from ni_digital_io import NiDriver

if len(sys.argv) < 4:
    print "usage: {0} <runs> <prefix> <wait time (ms)>".format( sys.argv[0] )
    sys.exit( 0 )
    
runs = int( sys.argv[1] )
prefix = sys.argv[2]
wait = int( sys.argv[3] )
# scramble_wait = int( sys.argv[4] )

camera = Luca()
ni = NiDriver( "PXI1Slot9/port0/line7" )

for i in range( runs ):
    print "Run Number {0}".format( i )
    filename = "{0}_{1}ms_{2}_initial.fits".format( prefix, wait, i )
    camera.save_image( filename )

    
    ni.write_single( False )
    time.sleep( 0.001 * wait )
    ni.write_single( True )
    time.sleep(0.1)

    #filename = "{0}_{1}ms_{2}_final.fits".format( prefix, wait, i )
    #camera.save_image( filename )
Beispiel #4
0
    
runs = int( sys.argv[1] )
wait = int( sys.argv[2] )
ions = sys.argv[3]
output = open( sys.argv[4], 'a+' )

tlock = httplib.HTTPConnection( '192.168.1.100', 8321, timeout=10 )
tlock.connect()
tlock.request( 'POST', '/feedback_parameters', 'ch=0&pg={0}&ig={1}&lim={2}'.
    format( 5, 1, 30 ) )
response = tlock.getresponse()
if response.status != 200:
    print( "Could not communicate with temperature lock." )

camera = Luca()
ni = NiDriver( "PXI1Slot9/port0/line6" )

try:
    ion_positions = []
    with open( ions, 'r' ) as ionfile:
        for l in ionfile:
            pos = tuple( int(x) for x in l.split() )
            ion_positions.append( pos )

    for i in range( runs ):
        print "Run Number {0}".format( i )
        #filename = "{0}_{1}ms_{2}_initial.fits".format( prefix, wait, i )
        #camera.save_image( filename )
        image = camera.get_image()
        
        data = []
    def __init__( self, nruns, experiment, ions, desired_order, out, conn=None ):
        #0 = no sym, use order 1 = symmetrize on order 2 = no order specificity
        camera = Luca()
        freq_src = FreqDriver( u'USB0::0x1AB1::0x0641::DG4B142100247::INSTR' )
        ni = NiDriver( self.Chans )
        reorder_time = 0.5

        output = open( out, 'w' )
        ion_order = []
        ion_configs = {} #dictionary to hold different ion configurations (keys) and their reordering time (values)
        ion_configs[str(desired_order)] = 0.0 #don't need a reordering time for the desired ordering
        desired_bright_number = sum(map(lambda x: 1 if x else 0, desired_order))

        try:
            ion_positions = []
            with open( ions, 'r' ) as ionfile:
                for l in ionfile:
                    pos = tuple( int(x) for x in l.split() )
                    ion_positions.append( pos )
    
            bg = []
            for i in range(30):
                data = self.build_data(camera, ion_positions, camera.get_image())
                bg.append( data[-1] )
            threshold = np.mean(bg) + 3*np.std(bg)
            
            experiment.setup( freq_src, ni )
            print( ion_positions )

            while experiment.step( freq_src, ni ):
                num_succ = [0 for i in ion_positions] # number of successes for each ion
                index = 0 #number of time it has run
                while index<nruns: #allows dynamic nruns switching
                    data = self.build_data( 
                        camera, ion_positions, camera.get_image() )
                    ion_order = [ d > threshold for d in data ]

                    bg.append( data[-1] )
                    if len( bg ) > 1000:
                        bg = bg[100:]
                    threshold = np.mean(bg) + 3*np.std(bg)

                    while ion_order != desired_order:
                        #seems this loop is designed to allow ions to reorganize
                        #by turning off cooling lasers

                        ion_key =  ['1' if x else '0' for x in ion_order] #these create the key for our ion_configs dictionary
                        ion_key = ''.join(ion_key) # and turns it into a string so the dictionary can use it
                        if ion_key not in ion_configs.keys() and ion_key!=str(desired_order):
                            ion_configs[ion_key] = reorder_time #this initializes the unseen ion order to reorder time of 0.5

                        curr_bright_number = sum(map(lambda x: 1 if x else 0, ion_order))
                        if curr_bright_number == desired_bright_number:
                            
                            r = NiSimpleDriver( self.RedChan )
                            r.write_single( False )
                            time.sleep(ion_configs[ion_key]) #ion_configs[ion_key] accesses the reorder time stored there
                            r.write_single( True )
                            r.close()
                        time.sleep( 1.0 )

                        camera.get_image()
                        data = self.build_data( 
                            camera, ion_positions, camera.get_image() )
                        
                        prev_order, ion_order = ion_order, \
                            [ d > threshold for d in data ]
                        print( "{} -> {}".format( prev_order, ion_order ) )

                        if any( prev_order ):
                            if prev_order == ion_order:
                                ion_configs[ion_key] *= 1.1
                            else:
                                ion_key =  ['1' if x else '0' for x in ion_order] #these create the key for our ion_configs dictionary
                                ion_key = ''.join(ion_key) # and turns it into a string so the dictionary can use it
                                if ion_key not in ion_configs.keys() and ion_key!=str(desired_order):
                                    ion_configs[ion_key] = reorder_time #this initializes the unseen ion order to reorder time of 0.5
                                ion_configs[ion_key] *= 0.9
                            print( "New reorder time: {}".format( ion_configs[ion_key] ) )
                        if ion_configs[ion_key] < 0.1:  ion_configs[ion_key] = 0.1
                        if ion_configs[ion_key] > 10.0: ion_configs[ion_key] = 10.0
                        
                        if conn is not None:
                            outdata = [str( experiment.control_var() )]
                            outdata.extend( str(d) for d in data )
                            outdata.extend( str(d) for d in data )
                            conn.send( 'reordata ' + '\t'.join(outdata) )
                    
                    ni.run()
                    data.extend( self.build_data( 
                        camera, ion_positions, camera.get_image() ) )
                    #data = [ion before 1762, ion before 1762, ..., fake ion before, ion after 1762, ion after 1762,..., fake ion after 1762]

                    outdata = [str( experiment.control_var() )]
                    outdata.extend( str(d) for d in data )
                    print '\t'.join(outdata)
                    if conn is not None:
                        conn.send('\t'.join(outdata) )
                    output.write( '\t'.join(outdata) + '\n' )
                    output.flush()

                    d = NiSimpleDriver( self.OrangeChan )
                    d.write_single( True )
                    d.close()

                    #ASSUMPTION: at this point the data should reflect the desired ion order,
                    ion_order = [ d > threshold for d in data[len(data)/2:]] #this goes from 1/2 data to the last point("fake ion")
                    #it's the result of using the 1762

                    for i in range(len(num_succ)):
                        if ion_order[i]:
                            num_succ[i]+=1 #counts number of successes (bright) for corresponding ion in that position

                    if index+1==20: #since it starts at 0 need to add 1
                        num_succ = [x/float(index+1) for x in num_succ] #array holding proportion of successes for each ion respectively
                        #subtracts from 0.5 so that we can find which is closest to 50%
                        min_prop = np.abs(0.5-num_succ[0]) #minimum modified proportion (the smallest is the one closest to 50%)
                        prop_succ = num_succ[0] #this is the actual proportion we want

                        #--------finds proportion closest to 50%------------------
                        for i in range(len(num_succ)):
                            mod_prop = np.abs(0.5-num_succ[i])
                            if min_prop > mod_prop: #if still not functioning add in an "and str(desired_order)[i] == 1" to try
                                #to get rid of outliers that could be introduced by dark ions
                                min_prop = mod_prop
                                prop_succ = num_succ[i]

                        binomial_StdErr = self.binStdrderr(index+1,prop_succ) #finds binomial standard error
                        if binomial_StdErr<0.05:
                            nruns = 20 #this will force the program to break out of this while loop
                            print binomial_StdErr, nruns
                        else:
                            nruns = 50
                            print binomial_StdErr, nruns
                    time.sleep( 0.2 )
                    index+=1 #new code

        finally:
            camera.shutdown()