#NEAT Composer text version. This version uses OOP to make it cleaner

#and hopefully make transfering to a GUI easier. Also puts fitness

#functions and other messy stuff in the fitness_func.py file.

import sys

import MultiNEAT as NEAT

from mingus.midi import fluidsynth # Commented out until they work on lab machines

from mingus.midi import MidiFileOut

from mingus.containers.Track import Track

from mingus.containers.Bar import Bar

from mingus.containers.Note import Note

from mingus.core import *

import random

import fitness_func #contains fitness functions and other useful stuff

from Tkinter import *

from Tkinter import Tk

import tkSimpleDialog

import tkFileDialog

import novelty

fluidsynth.init("FluidR3_GM.sf2", "alsa")

LENGTH = 12.0 #Global var for song length, must be a multiple of 4 for now

GEN_NUM = 0

#Song class that keeps all the information about the song in one place, including

#the track itself, it's fitness, what generation it belongs to, it's length, etc

params = NEAT.Parameters()

params.PopulationSize = 12

params.DynamicCompatibility = True

params.CompatTreshold = 2.0

params.YoungAgeTreshold = 15

params.SpeciesMaxStagnation = 100

params.OldAgeTreshold = 35

params.MinSpecies = 5

params.MaxSpecies = 25

params.RouletteWheelSelection = False

params.RecurrentProb = 0

params.OverallMutationRate = 0.33

params.MutateWeightsProb = 0.90

params.WeightMutationMaxPower = 5.0

params.WeightReplacementMaxPower = 5.0

params.MutateWeightsSevereProb = 0.5

params.WeightMutationRate = 0.75

params.MaxWeight = 20

params.MutateAddNeuronProb = 0.01

params.MutateAddLinkProb = 0.05

params.MutateRemLinkProb = 0.05

rng = NEAT.RNG()

#rng.TimeSeed()

rng.Seed(0)

genome = NEAT.Genome(0,2,0,7,False,NEAT.ActivationFunction.UNSIGNED_SIGMOID,NEAT.ActivationFunction.UNSIGNED_SIGMOID, 0, params)

class Song:

#pass

#End of Song Class

def evaluate(a_song, fitness_func, genome, *args): #Takes a song and fitness function, then evaluates

return a_song.fitness #song_fitness attribute

def advance_gen(genome_list): #Advances population to next generation

return song_list2

def redraw_listbox():

listbox.delete(0, END);

for song in song_list:

listbox.insert(END, "Song: %d.%d" %(GEN_NUM, song.song_id) + " Fitness:" + str( song.fitness) + " Novelty:" + str(song.novelty_score))

def song_clicked(i):

index = int(listbox.curselection()[0])

print(song_list[index].print_song())

song_list[index].play_song()

def print_all():

for song in song_list:

print song.print_song() + "\n"

print "\n"

def print_song_prop():

pass

def in_key_choice():

global choice

choice = tkSimpleDialog.askstring("Key Choice", "Enter key you with to restrict to")

print "In key choice is", choice

def evaluate_pop():

redraw_listbox()

def evolve_pop():

redraw_listbox()

def export():

song_list[song].export_song(filename)

def novelty_func():

novelty.set_novelty_params(song_list)

novelty.novelty_search(song_list)

redraw_listbox()

choice = "C"

population = NEAT.Population(genome, params, True, 1.0)

genome_list = NEAT.GetGenomeList(population)

song_list = advance_gen(genome_list)

root = Tk()

root.wm_title("NEAT Composer")

frame = Frame(root)

frame.grid(row=0, column=0)

framenovelty = Frame(root)

framenovelty.grid(row=0, column=1)

framelistbox = Frame(root)

framelistbox.grid(row = 0, column = 2)

nov1 = IntVar()

nov1.set(0)

nov2 = IntVar()

nov2.set(0)

nov3 = IntVar()

nov3.set(0)

evaluateButton = Button(frame, text = "Evaluate population", command = evaluate_pop)

evaluateButton.grid(row = 4, column = 0)

noveltyButton = Button(framenovelty, text = "Use Novelty", command = novelty_func)

noveltyButton.grid(row=4, column = 0)

evolveButton = Button(frame, text = "Evolve population", command = evolve_pop)

evolveButton.grid(row = 5, column = 0)

menu = Menu(root)

filemenu = Menu(menu)

menu.add_cascade(label="File", menu = filemenu)

filemenu.add_command(label = "Export Song", command = export)

viewmenu = Menu(menu)

menu.add_cascade(label = "View", menu = viewmenu)

viewmenu.add_command(label = "Print all", command = print_all)

root.config(menu = menu)

yscroll = Scrollbar(framelistbox, relief = 'raised')

yscroll.grid(row=0, column=1)

listbox = Listbox(framelistbox, yscrollcommand = yscroll.set, width = 40, height = 8)

for song in song_list:

listbox.insert(END, "Song: %d.%d" %(GEN_NUM, song.song_id) + " Fitness:" + str( song.fitness) + " Novelty:" + str(song.novelty_score))

listbox.grid(row = 0, column = 0)

yscroll.config(command = listbox.yview)

listbox.bind('<<ListboxSelect>>', song_clicked)

fitnesslabel = Label(frame, text = "Fitness Functions")

fitnesslabel.grid(row = 0, column = 0)

noveltylabel = Label(framenovelty, text= "Novelty Parameters")

noveltylabel.grid(row=0, column=0)

song_slope_button = Checkbutton(framenovelty, text = "Song Slope", variable = nov1, offvalue=0, onvalue=1)

song_slope_button.grid(row = 1, column = 0)

song_var_button = Checkbutton(framenovelty, text = "Song Variance", variable = nov2, offvalue = 0, onvalue=1)

song_var_button.grid(row = 2, column = 0)

song_range_button = Checkbutton(framenovelty, text = "Song Range", variable = nov3, offvalue = 0, onvalue = 1)

song_range_button.grid(row=3, column = 0)

in_key_button = Button(frame, text = "Set Key", command = in_key_choice)

in_key_button.grid(row=1, column=0)

in_key_slider = Scale(frame, from_=-100, to=100, orient=HORIZONTAL, label="InKey")

in_key_slider.grid(row=2, column=0)

diatonic_slider = Scale(frame, from_=-100, to=100, orient=HORIZONTAL, label="Diatonic")

diatonic_slider.grid(row=3, column=0)

root.mainloop()