from trisurf.trisurf import Directory, Tape
from trisurf import trisurf as ts
#from hashlib import md5
import os
import django
import sys
os.chdir('/home/samo/trisurf-server')
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "web_trisurf.settings")
django.setup()
from database import models

import numpy as np
import ipyvolume as ipv
import meshio
import matplotlib.pyplot as plt
import uuid

def read_mesh(filename):
    tmpfilename='/tmp/{}.vtu'.format(str(uuid.uuid4()))
    with open (filename) as f:
        contents=f.read()
    valid_vtu=contents[0:contents.find("<trisurfversion>")]+contents[contents.find("<UnstructuredGrid>"):]
    with open (tmpfilename, 'w') as f:
        f.write(valid_vtu)
    mesh=meshio.read(tmpfilename)
    os.unlink(tmpfilename)
    return mesh

def display_vesicle(run, iteration=None, colormap='bwr', point_data='vertices_idx'):
    cmap=plt.get_cmap(colormap)
    if iteration is not None:
        filename=os.path.join(run.Dir.fullpath(),'timestep_{:06d}.vtu'.format(int(iteration)))
    else:
        filename=os.path.join(run.Dir.fullpath(),run.getLastVTU())
    mesh=read_mesh(filename)
    fig=ipv.figure()
    color_data=mesh.point_data[point_data]/mesh.point_data[point_data].max()
    points=mesh.points.transpose()
    trimesh=ipv.plot_trisurf(*points, triangles=mesh.cells['triangle'], color=[c[0:3] for c in cmap(color_data)])
    ipv.xlim(points[0,:].min(),points[0,:].max())
    ipv.ylim(points[1,:].min(),points[1,:].max())
    ipv.zlim(points[2,:].min(),points[2,:].max())
    ipv.squarelim()
    #ipv.style.use('nobox')
    #ipv.pylab.style.use('nobox')
    return fig #ipv.show()
    
    
def database2runs(objs):
    runs=[]
    for obj in objs:
        r=ts.Runner(tape='/home/samo/trisurf-tapes/{}/tape'.format(obj.id))
        r.maindir=obj.simulation.directory
        r.subdir=obj.subdir
        runs.append(r)
    return runs