EiTpart.m

classdef EiTpart < Greedy.Interface
  properties (SetAccess=private, Dependent)
    generated_basis_type;
  end

  properties (Access = private)
    data_injection_path = [];
  end

  properties (Access = private)
    child;
    tpart_map;
  end

  methods
    function eitpbg = EiTpart(child_basis_generation)
      eitpbg = eitpbg@Greedy.Interface(child_basis_generation.model);
      eitpbg.child = child_basis_generation;
      eitpbg.tpart_map = [0 child_basis_generation.model.T];
    end

    function set_trainingsset(this, paramset)
      set_trainingsset(this.child, paramset);
    end

    function update_trainingsset(this, paramset)
      update_trainingsset(this.child, paramset);
    end

    function set_validation_generator(this, val_gen)
      set_validation_generator(this.child, val_gen);
    end

    function prepare(this, model_data)
      prepare(this.child, model_data);
    end

    function detailed_data = gen_detailed_data(this, detailed_data)
    end

    function detailed_data = init_basis(this, model_data)
      detailed_data = init_basis(this.child, model_data);
      detailed_data = inject_tpart_into_detailed_data(this, detailed_data);
      injected_ei_dd = get_injected_detailed_data(this, detailed_data);
    end

  end
  methods (Access = private)
    function detailed_data = inject_tpart_into_detailed_data(this, detailed_data)
      ei_child = detailed_data;
      if isempty(this.injection_path)
        while true
          if length(ei_child) > 1
            error('%s\n',...
                  'EiTpart is not allows to be used on top of another',...
                  'basis generation algorithm paritioning the EI bases.');
          elseif isequal(this.child.generated_basis_type, 'ei')
            break;
          elseif isequal(this.child.generated_basis_type, 'eirb')
            if isa(ei_child, 'Greedy.DataTree.Detailed.PODEILeafNode')
              ei_child = ei_child.ei;
            else
              ei_child = get(ei_child, 1);
              this.injection_path = [this.injection_path, 1];
            end
          elseif isequal(this.child.generated_basis_type, 'rb')
            error('%s\n',...
                  'You tried to use EiTpart on top of a basis generation', ...
                  'algorithm producing no empirical interpolation basis');
          end
        end

        injected_dd_ei = Greedy.DataTree.Detailed.TpartNode(ei_child, tpart_map);


      end
    end
  end
end