import numpy as np
import matplotlib.pyplot as plt
from numpy import sin, cos, tan, pi
from scipy.optimize import fsolve

def eq_circuito(s,q):
  A = s[0]
  B = s[1]
  f = np.empty(2)
  f[0] = R*cos(q)+C1*cos(B)+C2*sin(B)-C4+S
  f[1] = R*sin(q)-C1*sin(B)+C2*cos(B)-C3
  return f

# Constantes:
R = 40
C1 = 75
C2 = 8
C3 = 20
C4 = 100

# Chute inicial:
S0 = np.array([pi/4,30])
n = 100
q = np.linspace(0,pi,n)
Si = np.empty([n,2])

#print(Si)
for k in range(n):
  Si[k,:] = fsolve(eq_circuito,S0,q[k])
  S0 = Si[k,:]
plt.plot(q,Si[:,0])
plt.figure()
plt.plot(q,Si[:,1])