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 __init__( self, nruns, experiment, ions, desired_order, out, conn=None ):

        camera = Luca()
        freq_src = FreqDriver( u'USB0::0x1AB1::0x0641::DG4B142100247::INSTR' )
        output = open( out, 'w' )
        ion_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 = 0
            for i in range(20):
                data = self.build_data(camera, ion_positions, camera.get_image())
                bg += data[-1]/20
            
            experiment.setup()
            while experiment.step():
                for i in range( nruns ):
                    while ion_order_before != desired_order and wrongbrights.all():
                        data = self.build_data( 
				camera, ion_positions, camera.get_image() )
                        ion_order_before = [True if x > bg else False for x in data]
                        ni = NiDriver( self.Chans, wf )
                        data2 = self.build_data( 
                            camera, ion_positions, camera.get_image() )
                        ion_order_after = [True if x > bg else False for x in data]
                        wrongbrights = [False for i,j in zip(desired_order,ion_order_after) if j != i and j == 1]
                        data.extend(data2) 
                    	d = NiSimpleDriver( self.OrangeChan )
                    	d.write_single( True )
                    	d.close()
			if ion_order_before != desired_order and wrongbrights.all():
			    r = NiSimpleDriver( self.RedChan )
			    r.write_single( False )
			    time.sleep( 0.2 )
			    r.write(single( True ) )
			    r.close()
                    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()
Example #3
0
    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 )

    while current_time < stop_time:
        wf = build_waveform( dark_time, current_time )
        for i in range( nruns ):
            data = build_data( ion_positions, camera.get_image() )
            ni = NiDriver( Chans, wf )
            data.extend( build_data( ion_positions, camera.get_image() ) )
            image = camera.get_image()

            outdata = [str(current_time)]
            outdata.extend( str(d) for d in data )
            print "Data: {0}".format( outdata )
            output.write( '\t'.join(outdata) + '\n' )
            output.flush()

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

            time.sleep( 0.2 )
        current_time += time_step

      
finally:
    camera.shutdown()
    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()