#!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 = linspace(0,N,0.002)

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

for t in t_range(0,2000,100):
  print t
# Opening file in file_t format
  #fname = 'file_' + str(t)
  #fo = open('fname','r+')
  for i in range(0,N):
    p_za[i] = p_initial[i] + t*K*cos(K*p_initial[i]);

#Imposing Periodic Bondary condition
    if p_za[i] < 0.0:
      p_za[i] = (4.0-abs(p_za[i]) % 4.0)
    if p_za[i] > 4.0:
      p_za[i] = p_za[i] % 4.0;

    for i in range(0,N):
      for j in range(0,N):
        if p_za[i] < p_za[j]:
          temp = p_za[i]
          p_za[i] = p_za[j]
          p_za[j]=temp;

#Calculation of Density

    for i in range(0,N):
      temp_za[i] = p_za/0.02

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

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