Source code for pychangcooper.scenarios.generic_cooling_component

import numpy as np


class GenericCoolingComponent(object):
    def __init__(self, C0, cooling_index):
        """
        A generic cooling component that must be co-inherited with
        ChangCooper to produce a cooling only solution

        :param C0: the cooling constant
        :param cooling_index: the cooling index
        """

        self._cooling_index = cooling_index
        self._C0 = C0

    def _define_terms(self):

        self._dispersion_term = np.zeros(self._n_grid_points)
        self._heating_term = self._C0 * np.power(self._half_grid, self._cooling_index)