Plugins/Euclid/Euclid.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 EuclidInfo : IInfo
  10         {
  11                 public override string Name
  12                 {
  13                         get { return Catalog.GetString("Euclid metric"); }
  14                 }
  15
  16                 public override string ShortName
  17                 {
  18                         get { return "L2"; }
  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 Euclid metric between images."); }
  29                 }
  30         }
  31
  32         public class EuclidFactory : IFactory
  33         {
  34                 IInfo _info = new EuclidInfo();
  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 IPlugin Create()
  50                 {
  51                         return new EuclidPlugin();
  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 progress = 0;
  63
  64                 public TaskInfo(IResult[] a_out, IImage[] a_in1, IImage[] a_in2, int start, int end)
  65                 {
  66                         this.a_out = a_out;
  67                         this.a_in1 = a_in1;
  68                         this.a_in2 = a_in2;
  69                         this.start = start;
  70                         this.end = end;
  71                 }
  72
  73                 public void TaskWork()
  74                 {
  75                         for(int i=start; i<end; i++)
  76                         {
  77                                 double[] data = new double[a_in1.Length];
  78
  79                                 for(int j=0; j<a_in1.Length; j++)
  80                                 {
  81                                         data[j] = Compare(a_in1[j], a_in2[i]);
  82                                         progress++;
  83                                 }
  84
  85                                 a_out[i] = new IResult(data);
  86                         }
  87                 }
  88
  89                 private double Compare(IImage img1, IImage img2)
  90                 {
  91                         if(img1.BPP != img2.BPP)
  92                                 throw new PluginException(Catalog.GetString("Images BPP do not match."));
  93                         if(img1.H != img2.H || img1.W != img2.W)
  94                                 throw new PluginException(Catalog.GetString("Images dimensions do not match."));
  95
  96                         double sum = 0;
  97
  98                         if(img1.BPP == BPP.Grayscale)
  99                                 for(int i=0; i<img1.Data.Length; i++)
 100                                 {
 101                                         int diff = img1.Data[i] - img2.Data[i];
 102                                         sum += diff * diff;
 103                                 }
 104                         else
 105                                 for(int y=0; y<img1.H; y++)
 106                                         for(int x=0; x<img1.W; x++)
 107                                         {
 108                                                 float diff = (float)img1[x, y] - (float)img2[x, y];
 109                                                 sum += diff * diff;
 110                                         }
 111
 112
 113                         return Math.Sqrt(sum);
 114                 }
 115
 116                 public int Progress
 117                 {
 118                         get { return progress; }
 119                 }
 120         }
 121
 122         public class EuclidPlugin : IComparatorPlugin
 123         {
 124                 private ArrayList tasks = new ArrayList();
 125                 private int totalImages;
 126
 127                 public EuclidPlugin()
 128                 {
 129                         _info = new EuclidInfo();
 130                 }
 131
 132                 public override void Setup()
 133                 {
 134                 }
 135
 136                 public override bool HasSetup
 137                 {
 138                         get { return false; }
 139                 }
 140
 141                 public override void Work()
 142                 {
 143                         tasks.Clear();
 144
 145                         bool MultiThreading = Eithne.Config.Get("engine/blockthreads", false);
 146
 147                         ICommImage socket1 = _in[0] as ICommImage;
 148                         ICommImage socket2 = _in[1] as ICommImage;
 149
 150                         IImage[] img1 = socket1.Images;
 151                         IImage[] img2 = socket2.Images;
 152
 153                         _out = new CommSocket(1);
 154
 155                         IResult[] res = new IResult[img2.Length];
 156
 157                         totalImages = img1.Length * img2.Length;
 158
 159                         if(MultiThreading)
 160                         {
 161                                 TaskInfo ti1 = new TaskInfo(res, img1, img2, 0, img2.Length/2);
 162                                 TaskInfo ti2 = new TaskInfo(res, img1, img2, img2.Length/2, img2.Length);
 163
 164                                 tasks.Add(ti1);
 165                                 tasks.Add(ti2);
 166
 167                                 Thread t1 = new Thread(ti1.TaskWork);
 168                                 Thread t2 = new Thread(ti2.TaskWork);
 169
 170                                 t1.Start();
 171                                 t2.Start();
 172
 173                                 t1.Join();
 174                                 t2.Join();
 175                         }
 176                         else
 177                         {
 178                                 TaskInfo t = new TaskInfo(res, img1, img2, 0, img2.Length);
 179                                 tasks.Add(t);
 180                                 t.TaskWork();
 181                         }
 182
 183                         _out[0] = new ICommResult(res, 0, socket1.OriginalImages, socket2.OriginalImages, socket1.Categories, socket2.Categories);
 184
 185                         tasks.Clear();
 186
 187                         _workdone = true;
 188                 }
 189
 190                 public override int NumIn               { get { return 2; } }
 191                 public override int NumOut              { get { return 1; } }
 192
 193                 public override string DescIn(int n)
 194                 {
 195                         if(n == 0)
 196                                 return Catalog.GetString("Base images.");
 197                         else
 198                                 return Catalog.GetString("Test images.");
 199                 }
 200
 201                 public override string DescOut(int n)
 202                 {
 203                         return Catalog.GetString("Calculated Euclid metric.");
 204                 }
 205
 206                 private static string[] matchin   = new string[] { "image/grayscale", "image/float" };
 207                 private static string[] matchout  = new string[] { "result" };
 208
 209                 public override string[] MatchIn        { get { return matchin; } }
 210                 public override string[] MatchOut       { get { return matchout; } }
 211
 212                 public override float Progress
 213                 {
 214                         get
 215                         {
 216                                 int done = 0;
 217
 218                                 if(tasks.Count == 0)
 219                                         return 1;
 220
 221                                 for(int i=0; i<tasks.Count; i++)
 222                                 {
 223                                         done += ((TaskInfo)tasks[i]).Progress;
 224                                 }
 225
 226                                 return (float)done/totalImages;
 227                         }
 228                 }
 229         }
 230 }