def sphere(R, ro):
    S = 4*np.pi*R**2
    V = float(4/3)*np.pi*R**3
    M = ro*V
    print("Pole kuli o promieniu "+str(R)+"j wynosi " +str(S)+"j^2")
    print("Objętość kuli o promieniu "+str(R)+"j wynosi " +str(V)+"j^3")
    print("Masa kuli o promieniu "+str(R)+"j i gęstości "+str(ro)+" jednostek masy / j^3 wynosi "+str(M)+" jednostek masy")

def tetrahedron(a, ro):
    S = np.sqrt(3)*a**2
    V = float(2/np.sqrt(12))*a**3
    M = ro*V
    print("Pole czworościanu foremnego o krawędzi "+str(a)+"j wynosi " +str(S)+"j^2")
    print("Objętość czworościanu foremnego o krawędzi "+str(a)+"j wynosi " +str(V)+"j^3")
    print("Masa czworościanu foremnego o krawędzi "+str(a)+"j i gęstości "+str(ro)+" jednostek masy / j^3 wynosi "+str(M)+" jednostek masy")

def rec_pyramid(a, b, h, ro):
    h_a = np.sqrt(h**2+a**2) #wysokość ściany bocznej
    h_b = np.sqrt(h**2+b**2)
    S = a*b+b*h_b+a*h_a
    V = float(1/3)*a*b*h
    M = ro*V
    print("Pole ostrosłupa prostego prostokątnego o wysokości "+str(h)+"j oraz podstawie o bokach "+str(a)+" i " +str(b)+" wynosi " +str(S)+"j^2")
    print("Objętość ostrosłupa prostego prostokątnego o wysokości "+str(h)+"j oraz podstawie o bokach "+str(a)+" i " +str(b)+" wynosi " +str(V)+"j^3")
    print("Masa ostrosłupa prostego prostokątnego o wysokości "+str(h)+"j oraz podstawie o bokach "+str(a)+" i " +str(b)+" oraz gęstości "+str(ro)+
          " jednostek masy / j^3 wynosi "+str(M)+" jednostek masy")

def cone(R, h, ro):
    l = np.sqrt(h**2+R**2) #długość tworzącej
    S = np.pi*R*(R+l)
    V = float(1/3)*h*np.pi*R**2
    M = ro*V
    print("Pole stożka prostego o promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j wynosi "+str(S)+"j^2")
    print("Objętość stożka prostego o promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j wynosi "+str(V)+"j^3")
    print("Masa stożka prostego o promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j oraz gęstości "+str(ro)+" jednostek masy / j^3 wynosi "+str(M)+" jednostek masy")

def cylinder(R, h, ro):
    S = 2*np.pi*R*h+4*np.pi*R**2
    V = h*np.pi*R**2
    M = ro*V
    print("Pole walca promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j wynosi "+str(S)+"j^2")
    print("Objętość walca o promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j wynosi "+str(V)+"j^3")
    print("Masa walca o promieniu podstawy "+str(R)+"j i wysokości "+str(h)+"j oraz gęstości "+str(ro)+" jednostek masy / j^3 wynosi "+str(M)+" jednostek masy")

def parametry(i):
    if i==1:
        R = float(input("Podaj promień kuli: "))
        ro = float(input("Podaj gęstość kuli: "))
        sphere(R, ro)
    elif i==2:
        a = float(input("Podaj długość krawędzi czworościanu: "))
        ro = float(input("Podaj gęstość czworościanu: "))
        tetrahedron(a, ro)
    elif i==3:
        a = float(input("Podaj długość pierwszego boku podstawy: "))
        b = float(input("Podaj długość drugiego boku podstawy: "))
        h = float(input("Podaj wysokość ostrosłupa: "))
        ro = float(input("Podaj gęstość ostrosłupa: "))
        rec_pyramid(a, b, h, ro)
    elif i==4:
        R = float(input("Podaj promień podstawy stożka: "))
        h = float(input("Podaj wysokość stożka: "))
        ro = float(input("Podaj gęstość stożka: "))
        cone(R, h, ro)
    elif i==5:
        R = float(input("Podaj promień podstawy walca: "))
        h = float(input("Podaj wysokość walca: "))
        ro = float(input("Podaj gęstość walca: "))
        cylinder(R, h, ro)

i = int(input("Wpisz numer bryły, której parametry chcesz policzyć\n1. Kula\n2. Czworościan foremny\n3. Ostorsłup prosty o podstawie prostokąta\n4. Stożek prosty\n5. Walec\n"))
while(i not in [1,2,3,4,5]):
    i = int(input("Niepoprawny numer. Wprowadź poprawny numer: "))
parametry(i)