# $Id$
import re

class Material():
  '''
     Reads, manipulates and writes material.config files
     See end of file for example of usage
  '''
  __slots__ = ['data']

  def __init__(self):
    self.parts = [
             'homogenization',
             'microstructure',
             'crystallite',
             'phase',
             'texture',
            ]                                       # ordered (!) list of parts
    self.data = {\
              'homogenization': {'__order__': []},
              'microstructure': {'__order__': []},
              'crystallite':    {'__order__': []},
              'phase':          {'__order__': []},
              'texture':        {'__order__': []},
           }
           
  def __repr__(self):
    me = []
    for part in self.parts:
      print('doing',part)
      me += ['','#-----------------------------#','<%s>'%part,'#-----------------------------#',]
      for section in self.data[part]['__order__']:
        section_number=self.data[part]['__order__'].index(section)
        if section_number>0 and section_number<len(self.data[part]['__order__']):
            pass#me+=['\n##-----------------------------------------------------------------##\n']
        me += ['','[%s__No_%i] %s'%(section,section_number+1,'-'*max(0,21-len(section))),'',]
        for key in self.data[part][section]['__order__']:
          if key.startswith('(') and key.endswith(')'):                       # multiple (key)
            me += ['%s\t%s'%(key,' '.join(values)) for values in self.data[part][section][key]]
          else:                                                               # plain key
            me += ['%s\t%s'%(key,' '.join(map(str,self.data[part][section][key])))]
          
    return '\n'.join(me)

  def parse_data(self, part=None, sections=[], content=None):

    re_part = re.compile(r'^<(.+)>$')                     # pattern for part
    re_sec  = re.compile(r'^\[(.+)\]$')                   # pattern for section

    name_section = ''
    idx_section = 0
    active = False

    for line in content:
      line = line.split('#')[0].strip()                   # kill comments and extra whitespace
      line = line.split('#')[0].strip()                   # kill comments and extra whitespace
      if line:                                            # content survives...
        match_part = re_part.match(line)
        if match_part:                                    # found <part> separator
          active = (match_part.group(1) == part)          # only active in <part>
          continue
        if active:
          match_sec  = re_sec.match(line)
          if match_sec:                                   # found [section]
            name_section = match_sec.group(1)             # remember name ...
            if '__order__' not in self.data[part]: self.data[part]['__order__'] = []
            self.data[part]['__order__'].append(name_section)  # ... and position
            self.data[part][name_section] = {'__order__':[]}
            continue
          
          if sections == [] or name_section in sections:  # respect subset
            items = line.split()
            if items[0] not in self.data[part][name_section]:         # first encounter of key?
              self.data[part][name_section][items[0]] = []            # create item
              self.data[part][name_section]['__order__'].append(items[0])
            if items[0].startswith('(') and items[0].endswith(')'):   # multiple "(key)"
              self.data[part][name_section][items[0]].append(items[1:])
            else:                                                     # plain key
              self.data[part][name_section][items[0]] = items[1:]
                  
  def read(self,file=None):
    f=open(file,'r')
    c=f.readlines()
    f.close()
    for p in self.parts:
      self.parse_data(part=p, content=c)
       
  def write(self,file='material.config', overwrite=False):
    import os
    i = 0
    saveFile = file
    while not overwrite and os.path.exists(saveFile):
     i += 1
     saveFile = file+'_%i'%i

    print('Writing material data to file %s'%saveFile)
    f=open(saveFile,'w')
    f.write(str(self)+'\n')                                          #newline at end
    f.close()
    return saveFile

  def add_section(self, part=None, section=None, object=None, merge = False):
    '''adding/updating'''
    
    if part not in self.parts: raise Exception('invalid part %s'%part)

    if type(object) is dict: data = object
    else: data = object.data()
    
    if section not in self.data[part]: self.data[part]['__order__'] += [section]
    if section in self.data[part] and merge:
      for existing in self.data[part][section]['__order__']:                        # replace existing
        if existing in data['__order__']:
          if existing.startswith('(') and existing.endswith(')'):                   # multiple (key)
            self.data[part][section][existing] += data[existing]                    # add new multiple entries to existing ones
          else:                                                                     # regular key
            self.data[part][section][existing] = data[existing]                     # plain replice
      for new in data['__order__']:                                                 # merge new content
        if new not in self.data[part][section]['__order__']:
          self.data[part][section][new] = data[new]
          self.data[part][section]['__order__'] += [new]
    else:
      self.data[part][section] = data


class Section():
  def __init__(self,data = {'__order__':[]},part = ''):
    classes = {
                'homogenization':Homogenization,
                'microstructure':Microstructure,
                'crystallite':Crystallite,
                'phase':Phase,
                'texture':Texture,
              }
    self.parameters = {}
    for key in data:
      if type(data[key]) is not list:
        self.parameters[key] = [data[key]]
      else:
        self.parameters[key] = data[key]

    if '__order__' not in self.parameters:
      self.parameters['__order__'] = self.parameters.keys()
    if part.lower() in classes:
      self.__class__ = classes[part.lower()]
      self.__init__(data)

  def add_multiKey(self,key,data):
    multiKey = '(%s)'%key
    if multiKey not in self.parameters: self.parameters[multiKey] = []
    if multiKey not in self.parameters['__order__']: self.parameters['__order__'] += [multiKey]
    if type(data) == list: self.parameters[multiKey] += [[item] for item in data]
    else:                  self.parameters[multiKey] += [[data]]
      
  def add_key(self,key,data):
    if type(data) == list: self.parameters[key] = [item for item in data]
    else:                  self.parameters[key] = [data]
    if key not in self.parameters['__order__']: self.parameters['__order__'] += [key]
    
  def data(self):
    return self.parameters


class Homogenization(Section):
  def __init__(self,data = {'__order__':[]}):
    Section.__init__(self,data)


class Crystallite(Section):
  def __init__(self,data = {'__order__':[]}):
    Section.__init__(self,data)
  

class Phase(Section):
  def __init__(self,data = {'__order__':[]}):
    Section.__init__(self,data)
  

class Microstructure(Section):
  def __init__(self,data = {'__order__':[]}):
    Section.__init__(self,data)
  
  def add_constituent(self, data,):      # dict of phase,texture, and fraction
    
    theData = ''
    for property in ['phase','texture','fraction']:
      if property not in data.keys():   # suboptimal!!
        print('Croak!')
      else:
        theData += '%s %s\t'%(property,data[property])
    self.add_multiKey('constituent',theData)
  
class Texture(Section):
  def __init__(self,data = {'__order__':[]}):
    Section.__init__(self,data)
    
  def add_component(self,theType,properties):
    
    if 'scatter' not in map(str.lower,properties.keys()):
          scatter = 0.0
    else: scatter = properties['scatter']
    if 'fraction' not in map(str.lower,properties.keys()):
          fraction = 1.0
    else: fraction = properties['fraction']

    multiKey = theType.lower()

    if multiKey == 'gauss':
      self.add_multiKey(multiKey,'phi1 %g\tPhi %g\tphi2 %g\tscatter %g\tfraction %g'%(
                                        properties['eulers'][0],
                                        properties['eulers'][1],
                                        properties['eulers'][2],
                                        scatter,
                                        fraction,
                                        )
                        )

    if multiKey == 'fiber':
      self.add_multiKey(multiKey,'alpha1 %g\talpha2 %g\tbeta1 %g\tbeta2 %g\tscatter %g\tfraction %g'%(
                                        properties['eulers'][0],
                                        properties['eulers'][1],
                                        properties['eulers'][2],
                                        properties['eulers'][3],
                                        scatter,
                                        fraction,
                                        )
                        )
        
    
def example_1(write_file=False):
    '''Useage example of the material.config writer'''
    mat=Material()
    #--- CRYSTALLITE
    c=Crystallite() 
    c.add_multiKey('output',['texture','orientation','grainorientation','f','p'])
    mat.add_section('crystallite','essential_output',c)
    #--- HOMOGENIZATION
    h=Homogenization()
    h.add_key('type','isostrain')
    h.add_key('Ngrains',1)
    mat.add_section('homogenization','homog1',h)
    #--- MICROSTRUCTURE
    m=Microstructure()     
    m.add_key('crystallite','1')    
    m.add_constituent({'phase':1,'texture':2,'fraction':1})
    mat.add_section('microstructure','mustruct1',m)
    #--- TEXTURE    
    t1=Texture()
    t1.add_component('gauss',{'eulers':[1,2,3],'fraction':0.5})
    t1.add_component('gauss',{'eulers':[1,2,3],'scatter':2,'fraction':0.5})
    mat.add_section('texture','tex_1',t1)
    t2=Texture()
    t2.add_component('gauss',{'eulers':[45,0,0]})
    mat.add_section('texture','tex_2',t2)
    #t3=Texture()  # ToDo: fibre texture
    #t3.add_component('fibre',{'eulers':[45,0,0]})
    #mat.add_section('texture','tex_3',t3)
    #---PHASE
    p=Phase({'constitution':'lump'})   
    p.add_key('elasticity','hooke')
    p.add_key('plasticity','phenopowerlaw')
    p.add_key('c11','162.4e9')
    p.add_key('c12',92.4e9)
    p.add_key('Nslip',[12,0,0,0])   
    p.add_key('interaction_slipslip',[1]*27)
    mat.add_section('phase','phase_label',p)
                                           
    print(mat)
    mat.write(file='material.config_example_1')
    mat.write(file='material.config_example_1',overwrite=True)
    return mat