from pymol import cmd

prefix="116_pc3_00"
frames=34
a=10 ##colored frames
b=26 


cmd.hide('all')
cmd.show('cartoon')
cmd.set('cartoon_trace_atoms', '1')
cmd.spectrum("count",selection="("+prefix+str(a)+")&e. c")
#cmd.set('cartoon_color', 'grey50')
cmd.set('cartoon_color', 'default', prefix+str(a))
cmd.set('cartoon_transparency', 0,  prefix+str(a))
cmd.set_view([-0.782458127,   -0.126783773,   -0.609655738,
     0.603195488,    0.088785999,   -0.792632282,
     0.154623434,   -0.987947404,    0.007004964,
     0.000000000,    0.000000000, -241.713378906,
    35.640064240,  -49.162437439,  -30.586669922,
   197.522979736,  285.903869629,  -20.000000000] )

for i in range(a+1,b):
    cmd.disable(prefix + str(i))
for i in range(b,frames,2):
    cmd.disable(prefix + str(i))
for i in range(2,a,2):
    if len(str(i))<2:
        i='0'+str(i)
    cmd.disable(prefix + str(i))

j=0
col=90
t=0.7
for i in range(b,frames+1):
    print(t)
    cmd.set('cartoon_transparency', t, prefix + str(i))

    j+=1
    if j%3==0:
        col-=10
    
    cmd.set('cartoon_color', 'grey'+str(col), prefix + str(i))
    t-=0.025


for i in range(1,a):
    if len(str(i))<2:
        i='0'+str(i)
    cmd.set('cartoon_transparency', t, prefix + str(i))

    j+=1
    if j%3==0:
        col-=10
    
    cmd.set('cartoon_color', 'grey'+str(col), prefix + str(i))
    t-=0.025