rg_power_space.py 1 KB
Newer Older
theos's avatar
theos committed
1
2
# -*- coding: utf-8 -*-

theos's avatar
theos committed
3
4
import numpy as np

theos's avatar
theos committed
5
from nifty.power import PowerSpace
theos's avatar
theos committed
6
from nifty.nifty_paradict import rg_power_space_paradict
csongor's avatar
csongor committed
7
# from nifty.power.power_index_factory import RGPowerIndexFactory
theos's avatar
theos committed
8
9
10


class RGPowerSpace(PowerSpace):
theos's avatar
theos committed
11
12
13
14
15
16
17
18
    def __init__(self, shape, dgrid, distribution_strategy, zerocentered=False,
                 dtype=np.dtype('float'), log=False, nbin=None,
                 binbounds=None):
        self.dtype = np.dtype(dtype)
        self.paradict = rg_power_space_paradict(
                                    shape=shape,
                                    dgrid=dgrid,
                                    zerocentered=zerocentered,
csongor's avatar
csongor committed
19
                                    distribution_strategy=distribution_strategy,
theos's avatar
theos committed
20
21
22
23
                                    log=log,
                                    nbin=nbin,
                                    binbounds=binbounds)

csongor's avatar
csongor committed
24
25
        # self.power_indices = RGPowerIndexFactory.get_power_indices(
        #                         **self.paradict.parameters)