コード例 #1
0
 def __init__(self,
              geoFile,
              materialDict,
              bcDict,
              srcDict,
              nGroups=10,
              legOrder=8,
              sNords=2):
     """
     Matierial dict in:
         {'material_str': material_class_instance, ...}
     format
     """
     quadSet = gaussLegQuadSet(sNords)  # quadrature set
     self.sNords = sNords  # number of discrete dirs tracked
     self.sNmu, self.wN = quadSet[0], quadSet[1]  # quadrature weights
     self.maxLegOrder = legOrder  # remember to range(maxLegORder + 1)
     self.nG = nGroups  # number of energy groups
     self.legArray = createLegArray(self.sNmu,
                                    self.maxLegOrder)  # Stores leg polys
     #
     gmshMesh = gmsh1DMesh(geoFile=geoFile)  # Run gmsh
     self.nodes = gmshMesh.nodes
     self.superMesh = superMesh(gmshMesh, materialDict, bcDict, srcDict,
                                nGroups, sNords)  # build the mesh
     self.depth = 0  # scattering source iteration depth
     self.keff = 1
     self.buildTransOp()
     self.buildRHS()
     self.applyBCs()
コード例 #2
0
ファイル: sn1Dcell.py プロジェクト: xiaopingguo165/SpyTran
 def __init__(self,
              xpos,
              deltaX,
              nGroups=10,
              legOrder=8,
              sNords=2,
              **kwargs):
     quadSet = gaussLegQuadSet(sNords)
     self.faceNormals = np.array([-1, 1])
     self.centroid = xpos
     self.deltaX = deltaX
     self.sNords = sNords  # number of discrete dirs tracked
     self.sNmu, self.wN = quadSet[0], quadSet[1]  # quadrature weights
     self.maxLegOrder = legOrder  # remember to range(maxLegORder + 1)
     self.legweights = np.zeros(legOrder + 1)
     self.nG = nGroups  # number of energy groups
     self.legArray = self._createLegArray(
         self.maxLegOrder)  # Stores leg polys
     #
     # initial flux guess
     iguess = kwargs.pop('iFlux', np.ones((nGroups, 3, self.sNords)))
     #
     # Ord flux vector: 0 is cell centered, 1 is left, 2 is right face
     self.ordFlux = iguess
     self.totOrdFlux = iguess
     #
     # Scattering Source term(s)
     self.qin = np.zeros(
         (nGroups, 3, self.sNords))  # init scatter/fission source
     #self.qin[0, 0, :] = 5e10
     self.previousQin = np.ones(
         (nGroups, 3, self.sNords))  # init scatter/fission source
     #
     # optional volumetric source (none by default, fission or user-set possible)
     self.S = kwargs.pop('source', np.zeros((nGroups, 3, self.sNords)))
     #
     # set bc, if any given
     bc = kwargs.pop('bc', None)  # none denotes interior cell
     if bc is not None:
         self.setBC(bc)
     self.multiplying = False
     if type(self.S) is str:
         if self.S == 'fission':
             self.multiplying = True
         else:
             self.S = np.zeros((nGroups, 3, self.sNords))
     elif self.S is None:
         self.S = np.zeros((nGroups, 3, self.sNords))
     elif type(self.S) is np.ndarray:
         if self.S.shape != (nGroups, 3, self.sNords):
             sys.exit(
                 "FATALITY: Invalid shape of source vector. Shape must be (nGrps, 3, sNords)."
             )
コード例 #3
0
 def __init__(self, nodes, fluxStor, source, **kwargs):
     """
     takes ([nodeIDs], [nodePos]) tuple.
     Optionally specify number of groups, leg order and number of ordinates
     """
     #
     # Basic data needed for scattering source calcs
     self.sNords = kwargs.pop("sNords",
                              2)  # number of discrete dirs tracked
     quadSet = kwargs.pop("quadSet",
                          gaussLegQuadSet(self.sNords))  # quadrature set
     self.sNmu, self.wN = quadSet[0], quadSet[1]  # quadrature weights
     self.maxLegOrder = kwargs.pop("legOrder",
                                   8)  # remember to range(maxLegORder + 1)
     self.nG = kwargs.pop("nGroups", 10)  # number of energy groups
     self.legArray = kwargs.pop(
         "legP", createLegArray(self.sNmu,
                                self.maxLegOrder))  # Stores leg polys
     #
     # Store node IDs in element and node positions
     self.nodeIDs, self.nodeVs = nodes
     self._computeDeltaX()  # Compute deltaX
     #
     # Flux and source storage
     self.setEleScFlux(fluxStor[0])
     self.setEleTotFlux(fluxStor[1])
     #
     # Scattering Source term(s)
     self.qin = np.zeros(
         (self.nG, self.sNords))  # init scatter/fission source
     self.previousQin = np.ones(
         self.qin.shape)  # required for over relaxation
     #
     # Volumetric source
     self.S = source
     self.multiplying = False
     if type(self.S) is str:
         if self.S == 'fission':
             self.multiplying = True
         else:
             self.S = np.zeros((self.nG, self.sNords))
     elif self.S is None:
         self.S = np.zeros((self.nG, self.sNords))
     elif type(self.S) is np.ndarray:
         if self.S.shape != (self.nG, self.sNords):
             sys.exit(
                 "FATALITY: Invalid shape of source vector. Shape must be (nGrps, 3, sNords)."
             )
         else:
             pass
コード例 #4
0
ファイル: sn1Dcell.py プロジェクト: wgurecky/spyTran
 def __init__(self, xpos, deltaX, nGroups=10, legOrder=8, sNords=2, **kwargs):
     quadSet = gaussLegQuadSet(sNords)
     self.faceNormals = np.array([-1, 1])
     self.centroid = xpos
     self.deltaX = deltaX
     self.sNords = sNords                                    # number of discrete dirs tracked
     self.sNmu, self.wN = quadSet[0], quadSet[1]             # quadrature weights
     self.maxLegOrder = legOrder                             # remember to range(maxLegORder + 1)
     self.legweights = np.zeros(legOrder + 1)
     self.nG = nGroups                                       # number of energy groups
     self.legArray = self._createLegArray(self.maxLegOrder)  # Stores leg polys
     #
     # initial flux guess
     iguess = kwargs.pop('iFlux', np.ones((nGroups, 3, self.sNords)))
     #
     # Ord flux vector: 0 is cell centered, 1 is left, 2 is right face
     self.ordFlux = iguess
     self.totOrdFlux = iguess
     #
     # Scattering Source term(s)
     self.qin = np.zeros((nGroups, 3, self.sNords))  # init scatter/fission source
     #self.qin[0, 0, :] = 5e10
     self.previousQin = np.ones((nGroups, 3, self.sNords))  # init scatter/fission source
     #
     # optional volumetric source (none by default, fission or user-set possible)
     self.S = kwargs.pop('source', np.zeros((nGroups, 3, self.sNords)))
     #
     # set bc, if any given
     bc = kwargs.pop('bc', None)  # none denotes interior cell
     if bc is not None:
         self.setBC(bc)
     self.multiplying = False
     if type(self.S) is str:
         if self.S == 'fission':
             self.multiplying = True
         else:
             self.S = np.zeros((nGroups, 3, self.sNords))
     elif self.S is None:
         self.S = np.zeros((nGroups, 3, self.sNords))
     elif type(self.S) is np.ndarray:
         if self.S.shape != (nGroups, 3, self.sNords):
             sys.exit("FATALITY: Invalid shape of source vector. Shape must be (nGrps, 3, sNords).")
コード例 #5
0
ファイル: snFE1D.py プロジェクト: wgurecky/spyTran
 def __init__(self, nodes, fluxStor, source, **kwargs):
     """
     takes ([nodeIDs], [nodePos]) tuple.
     Optionally specify number of groups, leg order and number of ordinates
     """
     #
     # Basic data needed for scattering source calcs
     self.sNords = kwargs.pop("sNords", 2)                                    # number of discrete dirs tracked
     quadSet = kwargs.pop("quadSet", gaussLegQuadSet(self.sNords))            # quadrature set
     self.sNmu, self.wN = quadSet[0], quadSet[1]                              # quadrature weights
     self.maxLegOrder = kwargs.pop("legOrder", 8)                             # remember to range(maxLegORder + 1)
     self.nG = kwargs.pop("nGroups", 10)                                      # number of energy groups
     self.legArray = kwargs.pop("legP", createLegArray(self.sNmu, self.maxLegOrder))     # Stores leg polys
     #
     # Store node IDs in element and node positions
     self.nodeIDs, self.nodeVs = nodes
     self._computeDeltaX()  # Compute deltaX
     #
     # Flux and source storage
     self.setEleScFlux(fluxStor[0])
     self.setEleTotFlux(fluxStor[1])
     #
     # Scattering Source term(s)
     self.qin = np.zeros((self.nG, self.sNords))         # init scatter/fission source
     self.previousQin = np.ones(self.qin.shape)  # required for over relaxation
     #
     # Volumetric source
     self.S = source
     self.multiplying = False
     if type(self.S) is str:
         if self.S == 'fission':
             self.multiplying = True
         else:
             self.S = np.zeros((self.nG, self.sNords))
     elif self.S is None:
         self.S = np.zeros((self.nG, self.sNords))
     elif type(self.S) is np.ndarray:
         if self.S.shape != (self.nG, self.sNords):
             sys.exit("FATALITY: Invalid shape of source vector. Shape must be (nGrps, 3, sNords).")
         else:
             pass
コード例 #6
0
ファイル: snFE1D.py プロジェクト: wgurecky/spyTran
 def __init__(self, geoFile, materialDict, bcDict, srcDict, nGroups=10,
              legOrder=8, sNords=2):
     """
     Matierial dict in:
         {'material_str': material_class_instance, ...}
     format
     """
     quadSet = gaussLegQuadSet(sNords)                       # quadrature set
     self.sNords = sNords                                    # number of discrete dirs tracked
     self.sNmu, self.wN = quadSet[0], quadSet[1]             # quadrature weights
     self.maxLegOrder = legOrder                             # remember to range(maxLegORder + 1)
     self.nG = nGroups                                       # number of energy groups
     self.legArray = createLegArray(self.sNmu, self.maxLegOrder)  # Stores leg polys
     #
     gmshMesh = gmsh1DMesh(geoFile=geoFile)  # Run gmsh
     self.nodes = gmshMesh.nodes
     self.superMesh = superMesh(gmshMesh, materialDict, bcDict, srcDict, nGroups, sNords)    # build the mesh
     self.depth = 0  # scattering source iteration depth
     self.keff = 1
     self.buildTransOp()
     self.buildRHS()
     self.applyBCs()