#!/usr/bin/python3
import numpy as np
import random
import matplotlib.pyplot as plt
import pdb
import time
np.random.seed(int(time.time()))

def randSum(num, sum):
valArr = np.random.rand(num)
valArr = valArr/np.sum(valArr)
valArr = np.floor(valArr*sum)
if np.amin(valArr) == 0:
   valArr[valArr == np.amin(valArr)] = 1
if num == 1:
   rndIdx = 0
else:
   err = sum - np.sum(valArr)
   if err > 0:
         rndIdx = np.random.randint(0,high=num-1,size=int(err))
         for i in range(int(err)):
            valArr[rndIdx[i]] = valArr[rndIdx[i]] + 1.0
#print('Sum of array is %d and required sum is %d' %(np.sum(valArr),sum))
return valArr.astype(int)

def polymerMixInitialize():
nPolyTypes = np.random.randint(2, high = nPolyTypesMax)
print('Types of polymers used in this simulation are %i.' %(nPolyTypes))
nCountEachPolyType = randSum(nPolyTypes, nPolyChains)
print('Number chains of each polymer type in the mix are as follows.')
print(*nCountEachPolyType, sep = ",")
nMonoEachPolyType = []
for i in range(nPolyTypes):
   tmp = randSum(nMonomers, nPolyBeads)
   if i > 0:
         tmp1 = [row[0] for row in nMonoEachPolyType]
         while (tmp[0] in tmp1):
            tmp = randSum(nMonomers, nPolyBeads)
         nMonoEachPolyType.append(tmp)
   else:
         nMonoEachPolyType.append(tmp)
nMonoEachPolyType = np.asarray(nMonoEachPolyType)
MonoSequenceEachPolyType = []
for i in range(nPolyTypes):
   tmp = ['A' for _ in range(nPolyBeads)]
   LbIdx = random.sample(range(nPolyBeads), nMonoEachPolyType[i,1])
   for index in LbIdx:
         tmp[index] = 'B'
   MonoSequenceEachPolyType.append(tmp)
mySystem = {}
mrkrs = ["o","P","s","*","p","v"]
colrs = colors(nPolyTypes)
for i in range(nPolyTypes):

Write code to simulate and study the self-assembly of co-polymers c...