#!/usr/bin/env python
# coding: utf-8

# # Demo 2: anomalia da gravidade de um prisma poligonal
# 
# Isso é um [IPython notebook](http://ipython.org). Eu posso escrever texto, equações $\int x dx$, inserir imagens
# 
# ![Logo do Python](../img/python-logo.png)

# Algumas células contem código.

# In[ ]:


from fatiando.gravmag import polyprism
from fatiando import mesher, gridder
from fatiando.vis import mpl, myv


# In[ ]:


modelo = [mesher.PolygonalPrism([[0, 0], [0, 100], [100, 100], [100, 0]], 0, 500, {'density': -400})]


# In[ ]:


modelo


# In[ ]:


myv.figure()
myv.polyprisms(modelo)
myv.show()


# In[ ]:


x, y, z = gridder.regular((-500, 500, -500, 500), (50, 50), z=-10)


# In[ ]:


x, y, z


# In[ ]:


gz = polyprism.gz(x, y, z, modelo)


# In[ ]:


gz


# In[ ]:


mpl.contourf(y, x, gz, (50, 50), 30, cmap='Blues_r')
mpl.colorbar().set_label('mGal')
mpl.show()


# ## Exemplo mais interativo

# In[ ]:


vertices = mpl.draw_polygon((-500, 500, -500, 500), mpl.subplot(111))
mpl.show()


# In[ ]:


vertices


# In[ ]:


modelo = [mesher.PolygonalPrism(vertices, 0, 300, {'density': 500})]


# In[ ]:


myv.figure()
myv.polyprisms(modelo)
myv.show()


# In[ ]:


gz = polyprism.gz(x, y, z, modelo)


# In[ ]:


mpl.contourf(y, x, gz, (50, 50), 40)
mpl.show()