context1.m

%function context1(iprocess)
% Playing around with ways to ramp up context
clear classes;
close all;
% choose representative environments
% load('datasets/ch4Dat.mat');
%
% m1 = LL{1,1};
% Ehf = m1.EhfEnv;
% [a,i] = sort(Ehf);
% plot(a,'b.');
% ikeep = i(5:10:90);
% ikeep = sort(ikeep);
% hold on;
% plot(ikeep,Ehf(ikeep),'ro');
% save('ch4keep.mat','ikeep');
%
iprocess = 2;
topDir = 'C:/matdl/yaron/8-15-12/context/';
%topDir = '/brashear/yaron/matdl/8-12-12/context-psc/';
ftype = 3;
runParallel = 1;
showPlots = 1;

ics = 1;

if (iprocess == 1)
   trainC{1} = {'h2',2:3,'envs',1:5};
   testC{1} = []; %{'h2',4,'envs',1:100};
   filePrefix{1} = 'h2-2-3';
elseif (iprocess == 2)
   %   load('ch4keep.mat');
   ikeep = [ 12    26    37    41    47    48    62    92    94];
   trainC{1} = {'ch4',1:3,'envs',ikeep};
   testC{1} = {'ch4',4:19,'envs',ikeep};
   filePrefix{1} = 'ch4-23';
elseif (iprocess == 3)
   %   load('ch4keep.mat');
   ikeep = [ 12    26    37    41    47    48    62    92    94];
   trainC{1} = {'ethane',[1 3 5 7],'envs',ikeep};
   testC{1} = []; %{'h2',4,'envs',1:100};
   filePrefix{1} = 'ethane-4geoms';
end
filePre = filePrefix{1};
dataDir = [topDir,filePre];
if (exist(dataDir,'dir') ~= 7)
   status = mkdir(dataDir);
end
summaryName = [topDir,filePre,'/summary.txt'];
% if (exist(summaryName,'file'))
%    delete(summaryName);
% end
summaryFile = fopen(summaryName,'a');
diaryName = [topDir,filePre,'/cfit.diary'];
% if (exist(diaryName,'file'))
%    delete(diaryName);
% end
diary(diaryName);
diary on;
commonIn = {'silent',0};
%
iC = 1;
trainIn = trainC{iC};
testIn = testC{iC};
filePre = filePrefix{iC};
ke = [];
en = [];
e2 = [];
f1 = [];

% Assumes atom and bond context have 3 elements
iP2 = [1 0 0 0];
iP3 = [1 0 0 0 0];
diagNames = {'val','rho','avg r','avg bo','shift'};
bondNames = {'val','r','bo','drho'};

ke.H = Mixer(iP3,11,'ke.H',3);
ke.Cs = Mixer(iP3,11,'ke.C',3);
ke.Cp = ke.Cs;
ke.HH = Mixer(iP2,12,'ke.HH',2);
ke.CH = Mixer(iP2,12,'ke.CH',2);
ke.CH.hybrid = 1;
ke.CCs = Mixer(iP2,12,'ke.CCs',2);
ke.CCs.hybrid = 1;
ke.CCp = Mixer(iP2,12,'ke.CCp',2);
ke.CCp.hybrid = 2;

en.H = Mixer(iP3,11,'en.H',3);
en.Cs = Mixer(iP3,11,'en.C',3);
en.Cp = en.Cs;
en.HH = Mixer(iP2,12,'en.HH',2);
en.CH = Mixer(iP2,12,'en.CH',2);
en.CH.hybrid = 1;
en.HC = en.CH;
en.CCs = Mixer(iP2,12,'en.CCs',2);
en.CCs.hybrid = 1;
en.CCp = Mixer(iP2,12,'en.CCp',2);
en.CCp.hybrid = 2;

e2.H = Mixer(iP2,11,'e2.H',2);
e2.C = Mixer(iP2,11,'e2.C',2);
if (iprocess == 1) % h2
   e2.HH = Mixer(iP2,12,'e2.HH',2);
else % want only bond length dependence in H-H for methane
   e2.HH = Mixer([1 0],4,'e2.HH',2);
end
e2.CC = Mixer(iP2,12,'e2.CC',2);
e2.CH = Mixer(iP2,12,'e2.CH',2);


% Create fitme object
f1 = makeFitme(trainIn{:},commonIn{:},'enstructh',en, ...
   'kestructh',ke,'e2struct',e2);
%f1 = makeFitme(trainIn{:},commonIn{:},'enmods',0, ...
%   'kestructh',ke);

% Fix all context sensitive parameters
for imix = 1:length(f1.mixers)
   mix = f1.mixers{imix};
   if ((mix.mixType == 11) && mix.funcType == 3)
      mix.fixed = [0 1 1 1 0];
   elseif ((mix.mixType == 11) && mix.funcType == 2)
      mix.fixed = [0 1 1 1];
   elseif (mix.mixType == 12)
      mix.fixed = [0 1 1 1];
   elseif (mix.mixType == 4)
      mix.fixed = [0 1];
   end
end

startName = [topDir,filePre,'/start.mat'];
if (exist(startName,'file'))
   fprintf(1,'LOADING START \n');
   fprintf(summaryFile,'LOADING START \n');
   load(startName,'f1','currentError','currentPar');
else
   [currentError,currentPar] = contextFit(f1,0,0);
   save(startName);
end

disp(['starting error ', num2str(currentError)]);
fprintf(summaryFile,'initial error %12.5f \n',currentError);
ticID = tic;
for iter = 1:30
   allName = [topDir,filePre,'/all-',num2str(iter),'.mat'];
   if (exist(allName,'file'))
      fprintf(1,'LOADING ITERATION %i \n',iter);
      fprintf(summaryFile,'LOADING ITERATION %i \n',iter);
      load(allName,'f1','currentError','currentPar');
   else
      fprintf(1,'STARTING ITERATION %i \n',iter);
      fprintf(summaryFile,'STARTING ITERATION %i \n',iter);
      if (runParallel)
         save('f1temp.mat','f1');
      end
      % set up loop over imix and ipar, so we can do one big parfor loop
      ic = 0;
      mixes = {};
      for imix = 1:length(f1.mixers)
         mix = f1.mixers{imix};
         for ipar = 1:length(mix.par)
            if (mix.fixed(ipar) == 1)
               temp2.imix = imix;
               temp2.ipar = ipar;
               if (mix.mixType == 11)
                  names = diagNames;
               elseif (mix.mixType == 12)
                  names = bondNames;
               elseif (mix.mixType == 4)
                  names = {'val','r'};
               end
               temp2.name = [mix.desc,' ',names{ipar}];
               ic = ic+1;
               mixes{ic} = temp2;
            end
         end
      end
      nSave = length(mixes);
      errors = zeros(nSave,1);
      pars = cell(nSave,1);
      parfor ic = 1:nSave
         imix = mixes{ic}.imix;
         ipar = mixes{ic}.ipar;
         %name = mixes{ic}.name;
         %      if (runParallel)
         [etemp,ptemp] = contextFit([],imix,ipar);
         %      else
         %         [etemp, ptemp] = contextFit(f1,imix,ipar);
         %      end
         errors(ic) = etemp;
         pars{ic} = ptemp;
         %       temp.imix = imix;
         %       temp.ipar = ipar;
         %       temp.name = [desc,' ',names{ipar}];
         %       mixes{ic} = temp;
         %      disp([desc,' context ',names{ipar},' err ', ...
         %         num2str(etemp - currentError)]);
         %       if (~runParallel)
         %          f1.printMixers;
         %          %            input('junk');
         %          f1.mixers{imix}.fixed(ipar) = 1;
         %          f1.mixers{imix}.par(ipar) = 0;
         %          f1.setPars(currentPar);
         %       end
         %      fprintf(summaryFile, ...
         %         '%s %12.5f \n',name,etemp);
         %       fprintf(summaryFile, ' %i %12.5f \n',ic,etemp);
         
      end
      disp('Done with loop over all mixers');
      for ic=1:nSave
         fprintf(summaryFile, ...
            '%s %12.5f \n',mixes{ic}.name,errors(ic));
      end
      [lowErr,lowi] = min(errors);
      currentError = lowErr;
      currentPar = pars{lowi};
      imix = mixes{lowi}.imix;
      ipar = mixes{lowi}.ipar;
      name = mixes{lowi}.name;
      fprintf(1,'Lowest error of %12.5f is from %s \n', ...
         currentError,name);
      fprintf(summaryFile,'Lowest error of %12.5f is from %s \n', ...
         currentError,name);
      f1.mixers{imix}.fixed(ipar) = 0;
      f1.setPars(pars{lowi});
      f1.printMixers;
      save(allName);
   end
end
diary off;
fclose(summaryFile);
%    %%
%    for i=1:length(errors)
%       mix = f1.mixers{mixes{i}.imix};
%       ipar = mixes{i}.ipar;
%       etemp = errors(i);
%       disp([mix.desc,' context ',num2str(ipar),' err ', ...
%          num2str(etemp)]);
%       disp(['pars ',num2str(pars{i})]);
%    end
runTime = toc(ticID)