root/experimental/AWG/trunk/AWG/sources.py @ 4195

#!/usr/bin/env/python

import math
import random

class sources:
    def __init__(self,parent):
        self.parent = parent
        self.available_sources = ['file', 'sine', 'cosine', 'random', 'zeros']
        #IMPORTANT: corresponding_methods must be in the same order as the available_sources
        self.corresponding_methods = ['read_file','gen_sine', 'gen_cosine', 'gen_random_data', 'gen_zeros']


    def get_sources_list(self):
        return self.available_sources


    def gen_signal(self):
        eval_string = "self." + self.corresponding_methods[parent.signal_index] + "()"
        return eval(eval_string)  #return the signal generated by the corresponding method

    def read_file(self):
        
        try:
            my_file = open(self.parent.file_name, 'r')
        except IOError:
            print "no file named " + self.parent.file_name
            return []  #return an empty packet

        data_from_file = my_file.read()   #read the file as a big string
        my_file.close()   #done reading the file.  process the string from now on

        data_from_file.strip('\n')  #\n's will not be tollerated
        data_from_file.strip('[')   #['s will not be tollerated
        data_from_file.strip(']')   #]'s will not be tollerated

        data_from_file = data_from_file.split(self.parent.delimiter)  #break string into a list

        return data_from_file


    def gen_sine(self):
        '''freq is frequency in cycles per packet.  len is the length of the packet.  type is the data type to return (e.g., "float", "short", "int")'''

        count = 0
        sine = []
        while count < self.parent.len:
            sine.append(math.sin(self.parent.freq*2*math.pi*count/self.parent.len))
            count = count + 1

        return sine


    def gen_cosine(self):
        '''freq is frequency in cycles per packet.  len is the length of the packet.  type is the data type to return (e.g., "float", "short", "int")'''

        count = 0
        cosine = []
        while count < self.parent.len:
            cosine.append(math.cos(self.parent.freq*2*math.pi*count/self.parent.len))
            count = count + 1

        return cosine



    def gen_random_data(self):
        '''the rand_type variable defines which function call to use in the random module'''

        data = []
        tmp_rand_type = self.parent.rand_type.lower()  #makes comparison case insensitive
        if tmp_rand_type == 'random':
            count = 0
            while count < self.parent.len:
                data.append(random.random())
                count = count + 1
        elif tmp_rand_type == 'uniform':
            count = 0
            while count < self.parent.len:
                data.append(random.uniform())
                count = count + 1
        else:
            print "random methods other than random and uniform are not supported in sources.random_data"

        return data
                
    def gen_zeros(self):
        '''returns a list of zeros'''
        data = []
        for n in range(self.parent.len):
            data.append(0)
        return data