(* ::Package:: *)

Needs["ErrorBarPlots`"]


(* ::Section:: *)
(*Plot successrate over time*)


gsraw=Import[NotebookDirectory[]<>"data/graphdata_ccm_cputime.m"];
(* gsraw=SortBy[gsraw,{#[[1,1]]&,#[[1,2]]&}]; (* Sort by n and then by tau. The {} forces a *stable* sort because otherwise Mathematica sorts also on triangle count and other things. *) *)


gdata=GatherBy[gsraw,{#[[1,2]]&,#[[1,1]]&}];
(* Data format: *)
(* gdata[[ tau index, n index, run index , datatype index ]] *)
(* datatype index:
1: {n,tau}
2: edges
3: HH tri seq
4: {ccm1 #failed-attempts, ccm1 seq}
5: {ccm2 #failed-attempts, ccm2 seq}
*)
nlabels=Map["n = "<>ToString[#]&,gdata[[1,All,1,1,1]]];
taulabels=Map["\[Tau] = "<>ToString[#]&,gdata[[All,1,1,1,2]]];


(* ::Subsection:: *)
(*Canonical dataset (old code for multiple ccm runs per run)*)


(* For export *)
selectedRun=gdata[[1,1,1]];
measureSkip=1;
numEdges=selectedRun[[2]];
maxTris=Max[selectedRun[[3]]];
maxTime=Length[selectedRun[[3]]];
avgTris=Mean[selectedRun[[3,-maxTime/2;;-1]]];
skipPts=Max[1,Round[maxTime/500]]; (* Plotting every point is slow. Plot only once per `skipPts` timesteps *)

getCourseData[array_]:=Map[Mean,Partition[array,skipPts]];
coarseData=getCourseData[selectedRun[[3]]];
fullData=selectedRun[[3]];

shiftedCCMdata[offset_,data_]:=If[Length[data]==0,Nothing,
MapIndexed[{offset+(First[#2]-1)*skipPts,Mean[#1]}&,Partition[data,skipPts]]
];
coarseDatasCCM1=MapIndexed[shiftedCCMdata[numEdges*(First[#2]-1),#1]&,selectedRun[[4]]];
coarseDatasCCM2=MapIndexed[shiftedCCMdata[numEdges*(First[#2]-1),#1]&,selectedRun[[5]]];

getStartPoint[offset_,data_]:=If[Length[data]==0,
{Red,Point[{offset,10}]},
{Black,Point[{offset,data[[1]]}]}
];
startPoints1=MapIndexed[getStartPoint[numEdges*(First[#2]-1),#1]&,selectedRun[[4]]];

plotlabel="n = "<>ToString[selectedRun[[1,1]]]<>", \[Tau] = "<>ToString[selectedRun[[1,2]]];

plotTimeEvol1=ListPlot[coarseData,
Joined->True,PlotRange->{0,3*avgTris},DataRange->{0,measureSkip*maxTime},
Frame->True,FrameLabel->{"timesteps","number of triangles"},
PlotLabel->plotlabel,ImageSize->300,
PlotStyle->{Black},
Epilog->{PointSize[0.015],startPoints1}
];

(*
plotTimeEvol2=ListPlot[fullData,
Joined->True,PlotRange\[Rule]All,DataRange->{0,measureSkip*maxTime},
PlotStyle\[Rule]{Black,Opacity[0.3]}
];*)

plotTimeEvol3=ListPlot[coarseDatasCCM1,
Joined->True,PlotRange->All
];

plotTimeEvol4=Show[plotTimeEvol1,plotTimeEvol3]
(* Map[ListPlot[#,Joined->True,PlotRange\[Rule]{minCount,maxCount},DataRange\[Rule]{0,maxTime}]&,coarseData] *)


(* ::Subsection:: *)
(*New code*)


getCombinedData[run_]:=Module[{numEdges,maxTime,skipPts,getShiftedData,hhData,ccm1Data,ccm2Data},
numEdges=run[[2]];
maxTime=Length[run[[3]]];
skipPts=Max[1,Round[maxTime/500]];

getShiftedData[offset_,data_]:=If[Length[data]==0,{},
MapIndexed[{offset+(First[#2]-1)*skipPts,Mean[#1]}&,Partition[data,skipPts]]
];

hhData=getShiftedData[0,run[[3]]];
ccm1Data=getShiftedData[0*numEdges*run[[4,1]],run[[4,2]]];
ccm2Data=getShiftedData[0*numEdges*run[[5,1]],run[[5,2]]];

{Legended[hhData,"\[Tau] = "<>ToString[run[[1,2]]]],ccm1Data,ccm2Data}
]

dataSets=Map[getCombinedData,gdata,{3}];


dataSetsFlattened=Flatten[dataSets,3];
colorList=Table[ColorData[97,"ColorList"][[1+Floor[i/3]]],{i,0,Length[dataSetsFlattened]-1}];


plot1=ListPlot[dataSetsFlattened,Joined->True,PlotRange->{{0,40000},{0,6000}},PlotStyle->colorList,ImageSize->300,PlotLabel->nlabels[[1]],Frame->True,FrameLabel->{"timesteps","number of triangles"}]
plot1log=ListLogPlot[dataSetsFlattened,Joined->True,PlotRange->{{0,40000},{1,15000}},PlotStyle->colorList,ImageSize->300,PlotLabel->nlabels[[1]],Frame->True,FrameLabel->{"timesteps","number of triangles"}]


Export[NotebookDirectory[]<>"plots/timeevol_ccm.pdf",plot1]
Export[NotebookDirectory[]<>"plots/timeevol_ccm_log.pdf",plot1log]


(* ::Subsection:: *)
(*New code with CPU time*)


getCombinedTimeData[run_]:=Module[{maxTime,skipPts,avg,hhData,ccm1Data,ccm2Data,hhSlowData,line1,line2,line3,line4,lineAvg},
maxTime=Length[run[[3]]];
skipPts=Max[1,Round[maxTime/500]];
avg=Mean[run[[3,Floor[(1/2)*Length[run[[3]]]];;-1,2]]];

hhData=  run[[3,1;;-1;;skipPts]];
(*ccm1Data=run[[4,2,1;;-1;;skipPts]];*)
ccm2Data=run[[5,2,1;;-1;;skipPts]];
hhSlowData=run[[6,1;;-1;;skipPts]];
line1={{0,run[[3,1,2]]},run[[3,1]]};
line2={{0,run[[4,2,1,2]]},run[[4,2,1]]};
line3={{0,run[[5,2,1,2]]},run[[5,2,1]]};
line4={{0,run[[6,1,2]]},run[[6,1]]};
lineAvg={{0,avg},{3*run[[3,-1,1]],avg}};

{Legended[hhData,"\[Tau] = "<>ToString[run[[1,2]]]],
(*ccm1Data,*)
ccm2Data,
hhSlowData,
line1,
(*line2,*)
line3,
line4,
lineAvg}
]

getStartPoints[run_]:={run[[3,1]],run[[5,2,1]],run[[6,1]]};
getLogScaleStartPoints[run_]:={{run[[3,1,1]],Log[run[[3,1,2]]]},{run[[5,2,1,1]],Log[run[[5,2,1,2]]]},{run[[6,1,1]],Log[run[[6,1,2]]]}};

dataSets=Map[getCombinedTimeData,gdata,{3}];
startPoints=Map[getStartPoints,gdata,{3}];
logScaleStartPoints=Map[getLogScaleStartPoints,gdata,{3}];


dataSetsFlattened=Flatten[dataSets,2];
styleList=Table[{
Directive[Dashing[None],ColorData[97,"ColorList"][[i]]],
Directive[Dashing[None],ColorData[97,"ColorList"][[i]]],
Directive[Dashing[None],ColorData[97,"ColorList"][[i]]],
Directive[Dashed,ColorData[97,"ColorList"][[i]]],
Directive[Dashed,ColorData[97,"ColorList"][[i]]],
Directive[Dashed,ColorData[97,"ColorList"][[i]]],
Directive[Dashing[None],Black,Thin]
},{i,1,Length[dataSetsFlattened]}];
dataSetsFlattened=Flatten[dataSetsFlattened,1];
styleList=Flatten[styleList,1];
pointList={Red,PointSize[Medium],Point[Flatten[startPoints,3]]};
logPointList={Red,PointSize[Medium],Point[Flatten[logScaleStartPoints,3]]};


plot1=ListPlot[dataSetsFlattened,Joined->True,PlotRange->{{0,0.55},All},PlotStyle->styleList,Epilog->pointList,ImageSize->300,PlotLabel->nlabels[[1]],Frame->True,FrameLabel->{"seconds","number of triangles"}]
plot1log=ListLogPlot[dataSetsFlattened,Joined->True,PlotRange->{{0,0.55},All},PlotStyle->styleList,Epilog->logPointList,ImageSize->300,PlotLabel->nlabels[[1]],Frame->True,FrameLabel->{"seconds","number of triangles"}]


Export[NotebookDirectory[]<>"plots/timeevol_ccm.pdf",plot1]
Export[NotebookDirectory[]<>"plots/timeevol_ccm_log.pdf",plot1log]