1 files changed, 89 insertions, 0 deletions
diff --git a/miniTamTam/GenRythm.py b/miniTamTam/GenRythm.py
new file mode 100644
index 0000000..0cba969
--- /dev/null
+++ b/miniTamTam/GenRythm.py
@@ -0,0 +1,89 @@
+import random
+import Config
+
+from Generation.GenerationConstants import GenerationConstants
+from Generation.Utils import *
+
+class GenRythm:
+    def __init__( self, instrument, barLength, nbeats ):
+        self.instrument = instrument
+        self.barLength = barLength
+        self.nbeats = nbeats
+
+#############################################################################
+    def drumRythmSequence(self, regularity ):
+        rythmSequence = []
+        binSelection = []
+        downBeats = []
+        upBeats = []
+        beats = []
+        density = 0.8
+        countDown = 0
+        onsetTime = None
+        beatsPerPage = int( self.barLength / Config.TICKS_PER_BEAT )    
+
+        if Config.INSTRUMENTS[ self.instrument ].instrumentRegister == Config.PUNCH:
+            registerDensity = 0.5
+            downBeatRecurence = 4
+            for beat in range( beatsPerPage ):
+                beats.append( beat * Config.TICKS_PER_BEAT )
+            for i in range( len( beats ) ):
+                downBeats.append( ( beats[ GenerationConstants.PUNCH_ACCENTS[ beatsPerPage ][ i ] ], pow( float( len( beats ) - i) / len( beats ), 1.5 ) * 100.) )
+            for downBeat in downBeats:
+                upBeats.append( ( downBeat[ 0 ] +  Config.TICKS_PER_BEAT , downBeat[ 1 ] ) )
+
+        if Config.INSTRUMENTS[ self.instrument ].instrumentRegister == Config.LOW:
+            registerDensity =1.5
+            downBeatRecurence = 4
+            for beat in range( beatsPerPage ):
+                beats.append( beat * Config.TICKS_PER_BEAT )
+            for i in range( len( beats ) ):
+                downBeats.append( ( beats[ GenerationConstants.LOW_ACCENTS[ beatsPerPage ][ i ] ], pow( float( len( beats ) - i) / len( beats ), 1.5 ) * 100.) )
+            for downBeat in downBeats:
+                upBeats.append( ( downBeat[ 0 ] +  Config.TICKS_PER_BEAT / 2 , downBeat[ 1 ] ) )
+
+        if Config.INSTRUMENTS[ self.instrument ].instrumentRegister == Config.MID:
+            registerDensity = .75
+            downBeatRecurence = 1
+            for beat in range( beatsPerPage ):
+                beats.append( beat * Config.TICKS_PER_BEAT )
+                beats.append( beat * Config.TICKS_PER_BEAT + ( Config.TICKS_PER_BEAT / 2 ) )
+            for i in range( len( beats ) ):
+                downBeats.append( ( beats[ GenerationConstants.MID_ACCENTS[ beatsPerPage ][ i ] ], pow( float( len( beats ) - i) / len( beats ), 1.5 ) * 100.) )
+            for downBeat in downBeats:
+                upBeats.append( ( downBeat[ 0 ] +  Config.TICKS_PER_BEAT / 4 , downBeat[ 1 ] ) )
+
+        if Config.INSTRUMENTS[ self.instrument ].instrumentRegister == Config.HIGH:
+            registerDensity = 1.5
+            downBeatRecurence = 1
+            for beat in range( beatsPerPage ):
+                beats.append( beat * Config.TICKS_PER_BEAT )
+                beats.append( beat * Config.TICKS_PER_BEAT + ( Config.TICKS_PER_BEAT / 2 ) )
+            for i in range( len( beats ) ):
+                downBeats.append( ( beats[ GenerationConstants.HIGH_ACCENTS[ beatsPerPage ][ i ] ], pow( float( len( beats ) - i) / len( beats ), 1.5 ) * 100.) )
+            for downBeat in downBeats:
+                upBeats.append( ( downBeat[ 0 ] +  Config.TICKS_PER_BEAT / 4 , downBeat[ 1 ] ) )
+
+        for i in range( int( density * registerDensity * len( downBeats ) ) ):
+            if random.randint( 0, 100 ) < ( regularity * 100 * downBeatRecurence ) and binSelection.count( 1 ) < len( downBeats ): 
+                binSelection.append( 1 )        
+            else:
+                if binSelection.count( 0 ) < len( downBeats ): 
+                    binSelection.append( 0 )
+                else:
+                    binSelection.append( 1 )
+
+        countDown = binSelection.count( 1 )
+
+        for i in range( countDown ):
+            while onsetTime in rythmSequence or onsetTime == None:
+                onsetTime = prob2( downBeats )
+            rythmSequence.append( onsetTime )
+
+        for i in range( len( binSelection ) - countDown ):
+            while onsetTime in rythmSequence or onsetTime == None:
+                onsetTime = prob2( upBeats )
+            rythmSequence.append( onsetTime )
+
+        rythmSequence.sort()
+        return rythmSequence