Plugins/L0Mod/L0Mod.cs

   1 using System;
   2 using System.Collections;
   3 using System.Threading;
   4 using System.Xml;
   5 using Mono.Unix;
   6
   7 namespace Eithne
   8 {
   9         public class L0ModInfo : IInfo
  10         {
  11                 public override string Name
  12                 {
  13                         get { return Catalog.GetString("Modified L<sub>0</sub> metric"); }
  14                 }
  15
  16                 public override string ShortName
  17                 {
  18                         get { return "L0mod"; }
  19                 }
  20
  21                 public override string Author
  22                 {
  23                         get { return "Bartosz Taudul"; }
  24                 }
  25
  26                 public override string Description
  27                 {
  28                         get { return Catalog.GetString("This plugin calculates modified L<sub>0</sub> metric between images."); }
  29                 }
  30         }
  31
  32         public class L0ModFactory : IFactory
  33         {
  34                 IInfo _info = new L0ModInfo();
  35                 public IInfo Info
  36                 {
  37                         get { return _info; }
  38                 }
  39
  40                 public IType Type
  41                 {
  42                         get { return IType.Comparator; }
  43                 }
  44
  45                 public void Initialize()
  46                 {
  47                 }
  48
  49                 public Plugin.Base Create()
  50                 {
  51                         return new L0ModPlugin();
  52                 }
  53         }
  54
  55         public class TaskInfo
  56         {
  57                 private IResult[] a_out;
  58                 private IImage[] a_in1;
  59                 private IImage[] a_in2;
  60                 private int start;
  61                 private int end;
  62                 private int delta;
  63                 private int progress = 0;
  64
  65                 public TaskInfo(IResult[] a_out, IImage[] a_in1, IImage[] a_in2, int delta, int start, int end)
  66                 {
  67                         this.a_out = a_out;
  68                         this.a_in1 = a_in1;
  69                         this.a_in2 = a_in2;
  70                         this.start = start;
  71                         this.end = end;
  72                         this.delta = delta;
  73                 }
  74
  75                 public void TaskWork()
  76                 {
  77                         for(int i=start; i<end; i++)
  78                         {
  79                                 double[] data = new double[a_in1.Length];
  80
  81                                 for(int j=0; j<a_in1.Length; j++)
  82                                 {
  83                                         data[j] = Compare(a_in1[j], a_in2[i]);
  84                                         progress++;
  85                                 }
  86
  87                                 a_out[i] = new IResult(data);
  88                         }
  89                 }
  90
  91                 private double Compare(IImage img1, IImage img2)
  92                 {
  93                         if(img1.BPP != img2.BPP)
  94                                 throw new PluginException(Catalog.GetString("Images BPP do not match."));
  95                         if(img1.H != img2.H || img1.W != img2.W)
  96                                 throw new PluginException(Catalog.GetString("Images dimensions do not match."));
  97
  98                         double sum = 0;
  99
 100                         if(img1.BPP == BPP.Grayscale)
 101                                 for(int i=0; i<img1.Data.Length; i++)
 102                                 {
 103                                         int diff = Math.Abs(img1.Data[i] - img2.Data[i]);
 104                                         if(diff > delta)
 105                                                 sum++;
 106                                 }
 107                         else
 108                                 for(int y=0; y<img1.H; y++)
 109                                         for(int x=0; x<img1.W; x++)
 110                                         {
 111                                                 float diff = Math.Abs((float)img1[x, y] - (float)img2[x, y]);
 112                                                 if(diff > delta)
 113                                                         sum++;
 114                                         }
 115
 116                         return sum / (img1.W * img1.H);
 117                 }
 118
 119                 public int Progress
 120                 {
 121                         get { return progress; }
 122                 }
 123         }
 124
 125         public class L0ModPlugin : Plugin.Comparator
 126         {
 127                 private int delta = 20;
 128                 private ArrayList tasks = new ArrayList();
 129                 private int totalImages;
 130
 131                 public L0ModPlugin()
 132                 {
 133                         _info = new L0ModInfo();
 134                 }
 135
 136                 public override XmlNode Config
 137                 {
 138                         get { return GetConfig(); }
 139                         set { LoadConfig(value); }
 140                 }
 141
 142                 private void UpdateValue(int n)
 143                 {
 144                         delta = n;
 145                         _block.Invalidate();
 146                 }
 147
 148                 public override void Setup()
 149                 {
 150                         new L0ModSetup(delta, UpdateValue);
 151                 }
 152
 153                 public override void Work()
 154                 {
 155                         tasks.Clear();
 156
 157                         bool MultiThreading = Eithne.Config.Get("engine/blockthreads", false);
 158
 159                         ICommImage socket1 = _in[0] as ICommImage;
 160                         ICommImage socket2 = _in[1] as ICommImage;
 161
 162                         IImage[] img1 = socket1.Images;
 163                         IImage[] img2 = socket2.Images;
 164
 165                         _out = new CommSocket(1);
 166
 167                         IResult[] res = new IResult[img2.Length];
 168
 169                         totalImages = img1.Length * img2.Length;
 170
 171                         if(MultiThreading)
 172                         {
 173                                 TaskInfo ti1 = new TaskInfo(res, img1, img2, delta, 0, img2.Length/2);
 174                                 TaskInfo ti2 = new TaskInfo(res, img1, img2, delta, img2.Length/2, img2.Length);
 175
 176                                 tasks.Add(ti1);
 177                                 tasks.Add(ti2);
 178
 179                                 Thread t1 = new Thread(ti1.TaskWork);
 180                                 Thread t2 = new Thread(ti2.TaskWork);
 181
 182                                 t1.Start();
 183                                 t2.Start();
 184
 185                                 t1.Join();
 186                                 t2.Join();
 187                         }
 188                         else
 189                         {
 190                                 TaskInfo t = new TaskInfo(res, img1, img2, delta, 0, img2.Length);
 191                                 tasks.Add(t);
 192                                 t.TaskWork();
 193                         }
 194
 195                         _out[0] = new ICommResult(res, 0, socket1.OriginalImages, socket2.OriginalImages,
 196                                         socket1.Categories, socket2.Categories);
 197
 198                         tasks.Clear();
 199
 200                         _workdone = true;
 201                 }
 202
 203                 private XmlNode GetConfig()
 204                 {
 205                         XmlNode root = _xmldoc.CreateNode(XmlNodeType.Element, "config", "");
 206
 207                         root.InnerText = delta.ToString();
 208
 209                         return root;
 210                 }
 211
 212                 private void LoadConfig(XmlNode root)
 213                 {
 214                         UpdateValue(Int32.Parse(root.InnerText));
 215                 }
 216
 217                 public override int NumIn               { get { return 2; } }
 218                 public override int NumOut              { get { return 1; } }
 219
 220                 public override string DescIn(int n)
 221                 {
 222                         if(n == 0)
 223                                 return Catalog.GetString("Base images.");
 224                         else
 225                                 return Catalog.GetString("Test images.");
 226                 }
 227
 228                 public override string DescOut(int n)
 229                 {
 230                         return Catalog.GetString("Calculated modified L0 metric.");
 231                 }
 232
 233                 private static string[] matchin   = new string[] { "image/grayscale", "image/float" };
 234                 private static string[] matchout  = new string[] { "result" };
 235
 236                 public override string[] MatchIn        { get { return matchin; } }
 237                 public override string[] MatchOut       { get { return matchout; } }
 238
 239                 public override float Progress
 240                 {
 241                         get
 242                         {
 243                                 int done = 0;
 244
 245                                 if(tasks.Count == 0)
 246                                         return 1;
 247
 248                                 for(int i=0; i<tasks.Count; i++)
 249                                 {
 250                                         done += ((TaskInfo)tasks[i]).Progress;
 251                                 }
 252
 253                                 return (float)done/totalImages;
 254                         }
 255                 }
 256         }
 257 }