import numpy as np

class buffer(object):
    def __init__(self):
        self.states = []
        self.actions = []
        self.rewards = []
        print("√√√√√Buffer Initialized Success√√√√√")
    def clearBuffer(self):
        self.states = []
        self.actions = []
        self.rewards = []
    def getStates(self):
        return np.asarray(self.states)
    def getActions(self):
        return np.asarray(self.actions)
    def getRewards(self):
        return np.asarray(self.rewards)
    
    def saveState(self,state):
        self.states.append(state)
    def saveAction(self,action):
        self.actions.append(action)
    def saveReward(self,reward):
        self.rewards.append(reward)
    def saveBuffers(self,state,action,reward):
        self.states.append(state)
        self.actions.append(action)
        self.rewards.append(reward)