#!usr/bin/python

import os
import numpy as np
import matplotlib.pyplot as plt

from numpy import *

# Declaration of constants

K = 3.14159265359
N= 2000
p_za=[]
#-----------------------
#Allotment of particles
#-----------------------

p_initial = np.linspace(0,4,num=2000)

print p_initial


#Displacement of particles using Zeldovich Approximation
#def t_range(start, end, step):
 # while start <= end:
  #  yield start
   # start += step

for t in range(2000,20002):
  print t
# Opening file in file_t format
  fname = 'file_' + str(t)
  fo = open(fname,'w')
  
  p_za.append(p_initial + t*K*np.cos(K*p_initial))
  print 'p_za=',p_za
    
  for i in range(0,N):
#Imposing Periodic Bondary condition
    if p_za < 0.0:
      p_za = (4.0-abs(p_za) % 4.0)
      
    if p_za > 4:
      p_za = np.array(p_za) % 4
      
    
    print p_za   
    
  
#Calculation of Density


    temp_za = p_za/0.02
    print temp_za

    for j in range(0,200):
      for i in range(0,N):

        if temp_za == j:
          counter+=1
      print("%d      %d", j, counter)
      counter+=1