import ROOT

class calibCluster:
    def __init__(self):
        self.digipos=[]
        self.digiamp=[]
        self.plane=ROOT.GFDetPlane()
        self.params=[9,0,2]
        
    def addDigi(self,pos,amp):
        self.digipos.append(pos)
        self.digiamp.append(amp)
        
    def setPlane(self,o,u,v):
        self.plane.set(o,u,v)
        
    def produceFakePulls(self):
        retarr=[]
        #print 'making fake data with',self.params[0],self.params[1],self.params[2]
        for j in range(100):
            retarr.append([])
        for i in range(500):
            x=ROOT.gRandom.Uniform(0,100)
            bin=int(x)
            retarr[bin].append(self.params[0]+x**2*self.params[1]+ROOT.gRandom.Gaus(0,self.params[2]))
        return retarr
    
    def setParams(self,params):
        self.params=params