# -*- coding: UTF-8 no BOM -*-

# $Id$

import numpy as np
import sys

try:
  import h5py
except:
  sys.stderr.write('Could not import h5py.\n') 

class Result():
  '''
     General class for result parsing.
     Needs h5py to be installed
  '''
  
  def __init__(self,resultsFile):
    self.data=h5py.File(resultsFile,"r")
    self.Npoints=self.data.attrs['Number of Materialpoints']
    print("Opened "+resultsFile+" with %i points"%self.Npoints)

  def getCrystallite(self,labels,inc,constituent=1,points=None):
    if points is None: points = np.array(np.array(xrange(self.Npoints)))
    results = {}
    mapping=self.data['mapping/crystallite']
    for instance in self.data['increments/%s/crystallite/'%inc]:
      dsets = self.data['increments/%s/crystallite/%s'%(inc,instance)].keys()
      for label in labels:
        if label in dsets and label not in results:
          shape = np.shape(self.data['increments/%s/crystallite/%s/%s'%(inc,instance,label)])[1:]
          results[label] = np.nan*np.ones(np.array((self.Npoints,)+shape))
    for myPoint in xrange(len(points)):
      matPoint = points[myPoint]
      pos = mapping[matPoint,constituent-1]
      if pos[0] != 0:
        try:
          for label in labels:
            results[label][matPoint,...] = self.data['increments/%s/crystallite/%s/%s'%(inc,pos[0],label)][pos[1],...]
        except:
          pass
    return results