1234567891011121314151617181920212223242526272829303132 |
- """Implementation of :class:`CompositeDomain` class. """
- from sympy.polys.domains.domain import Domain
- from sympy.polys.polyerrors import GeneratorsError
- from sympy.utilities import public
- @public
- class CompositeDomain(Domain):
- """Base class for composite domains, e.g. ZZ[x], ZZ(X). """
- is_Composite = True
- gens, ngens, symbols, domain = [None]*4
- def inject(self, *symbols):
- """Inject generators into this domain. """
- if not (set(self.symbols) & set(symbols)):
- return self.__class__(self.domain, self.symbols + symbols, self.order)
- else:
- raise GeneratorsError("common generators in %s and %s" % (self.symbols, symbols))
- def drop(self, *symbols):
- """Drop generators from this domain. """
- symset = set(symbols)
- newsyms = tuple(s for s in self.symbols if s not in symset)
- domain = self.domain.drop(*symbols)
- if not newsyms:
- return domain
- else:
- return self.__class__(domain, newsyms, self.order)
|