#!/usr/bin/env python
# encoding: utf-8
"""
_SICDeltaAOperator.py

Created by Graham Dennis on 2008-08-07.

Copyright (c) 2008-2012, Graham Dennis

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""

from xpdeint.Operators.DeltaAOperator import DeltaAOperator

from xpdeint.ParserException import ParserException

class _SICDeltaAOperator (DeltaAOperator):
  def __init__(self, *args, **KWs):
    DeltaAOperator.__init__(self, *args, **KWs)
    
    # Set default variables
    self.crossPropagationDimension           = None
    self.crossPropagationDirection           = None
    self.crossIntegrationVectors             = set()
    self.crossIntegrationVectorsEntity       = None
    
  
  @property
  def integrator(self):
    # Our parent is the operator container, and its parent is the integrator
    return self.parent.parent
  
  def bindNamedVectors(self):
    super(_SICDeltaAOperator, self).bindNamedVectors()
    
    self.primaryCodeBlock.dependencies.update(self.codeBlocks['boundaryCondition'].dependencies)
    self.primaryCodeBlock.dependencies.update(self.codeBlocks['crossPropagation'].dependencies)
    
    if self.crossIntegrationVectorsEntity:
      self.crossIntegrationVectors.update(self.vectorsFromEntity(self.crossIntegrationVectorsEntity))
      if self.integrationVectors.intersection(self.crossIntegrationVectors):
        badVectors = self.integrationVectors.intersection(self.crossIntegrationVectors)
        raise ParserException(self.crossIntegrationVectorsEntity.xmlElement,
                              "Can't have a vector being integrated by both a cross-propagator and an integration block.\n"
                              "The vectors causing the problems are: %s" % ', '.join([v.name for v in badVectors]))
      self.dependencies.update(self.crossIntegrationVectors)
      
  
  
  def preflight(self):
    super(_SICDeltaAOperator, self).preflight()
    # Construct the operator components dictionary for the cross-propagation vectors
    for crossIntegrationVector in self.crossIntegrationVectors:
      for componentName in crossIntegrationVector.components:
        derivativeString = "d%s_d%s" % (componentName, self.crossPropagationDimension)
        
        # Map of operator names to vector -> component list dictionary
        self.operatorComponents[derivativeString] = {crossIntegrationVector: [componentName]}
    
    crossDimRep = self.loopingField.dimensionWithName(self.crossPropagationDimension).inBasis(self.operatorBasis)
    crossPropDimOverrides = [(v.field, crossDimRep.loopIndex) for v in self.codeBlocks['crossPropagation'].dependencies 
                                if v.field.hasDimensionName(self.crossPropagationDimension)]
    
    self.codeBlocks['crossPropagation'].loopArguments['indexOverrides'] = \
      {self.crossPropagationDimension: crossPropDimOverrides}