Plugins/SimpleDB/SimpleDB.cs

   1 using System;
   2 using System.Collections;
   3 using System.Xml;
   4 using Mono.Unix;
   5
   6 namespace Eithne
   7 {
   8         public class SimpleDBInfo : IInfo
   9         {
  10                 public override string Name
  11                 {
  12                         get { return Catalog.GetString("Simple Image Database"); }
  13                 }
  14
  15                 public override string ShortName
  16                 {
  17                         get { return "SimpleDB"; }
  18                 }
  19
  20                 public override string Author
  21                 {
  22                         get { return "Bartosz Taudul"; }
  23                 }
  24
  25                 public override string Description
  26                 {
  27                         get { return Catalog.GetString("This plugin allows to pass images to image recognition system."); }
  28                 }
  29         }
  30
  31         public class SimpleDBFactory : IFactory
  32         {
  33                 IInfo _info = new SimpleDBInfo();
  34                 public IInfo Info
  35                 {
  36                         get { return _info; }
  37                 }
  38
  39                 public IType Type
  40                 {
  41                         get { return IType.In; }
  42                 }
  43
  44                 public void Initialize()
  45                 {
  46                 }
  47
  48                 public Plugin.Base Create()
  49                 {
  50                         return new SimpleDBPlugin();
  51                 }
  52         }
  53
  54         public class SimpleDBPlugin : Plugin.In
  55         {
  56                 private ArrayList _fl = new ArrayList();
  57                 private float progress;
  58
  59                 public SimpleDBPlugin()
  60                 {
  61                         _info = new SimpleDBInfo();
  62                 }
  63
  64                 public override XmlNode Config
  65                 {
  66                         get { return GetConfig(); }
  67                         set { LoadConfig(value); }
  68                 }
  69
  70                 public override void Setup()
  71                 {
  72                         new SimpleDBSetup(_fl, _block);
  73                 }
  74
  75                 public override void Work()
  76                 {
  77                         progress = 0;
  78
  79                         if(_fl.Count == 0)
  80                                 throw new PluginException(Catalog.GetString("No images in list"));
  81
  82                         int i = 0;
  83                         IImage[] imgarray = new IImage[_fl.Count];
  84                         int[] categories = new int[_fl.Count];
  85
  86                         foreach(string fn in _fl)
  87                         {
  88                                 Gdk.Pixbuf buf = new Gdk.Pixbuf(fn);
  89
  90                                 imgarray[i] = IImage.Create(buf, Utility.IsBW(buf) ? BPP.Grayscale : BPP.RGB);
  91                                 categories[i] = i;
  92
  93                                 i++;
  94
  95                                 progress = (float)i/_fl.Count;
  96                         }
  97
  98                         _out = new CommSocket(1);
  99                         _out[0] = new ICommImage(imgarray, imgarray, categories);
 100
 101                         _workdone = true;
 102                 }
 103
 104                 private XmlNode GetConfig()
 105                 {
 106                         XmlNode root = _xmldoc.CreateNode(XmlNodeType.Element, "config", "");
 107
 108                         foreach(string s in _fl)
 109                         {
 110                                 XmlNode n = _xmldoc.CreateNode(XmlNodeType.Element, "image", "");
 111                                 n.InnerText = s;
 112                                 root.AppendChild(n);
 113                         }
 114
 115                         return root;
 116                 }
 117
 118                 private void LoadConfig(XmlNode root)
 119                 {
 120                         XmlNodeList nl = root.SelectNodes("image");
 121                         ArrayList errors = new ArrayList();
 122
 123                         foreach(XmlNode n in nl)
 124                         {
 125                                 try
 126                                 {
 127                                         new Gdk.Pixbuf(n.InnerText);
 128
 129                                         _fl.Add(n.InnerText);
 130                                 }
 131                                 catch(GLib.GException)
 132                                 {
 133                                         errors.Add(n.InnerText);
 134                                 }
 135                         }
 136
 137                         if(errors.Count != 0)
 138                                 new LoadError(errors);
 139                 }
 140
 141                 public override int NumOut              { get { return 1; } }
 142
 143                 public override string DescOut(int n)
 144                 {
 145                         return Catalog.GetString("Images.");
 146                 }
 147
 148                 public override float Progress          { get { return progress; } }
 149
 150                 private static string[] matchout  = new string[] { "image/rgb", "image/grayscale" };
 151                 public override string[] MatchOut       { get { return matchout; } }
 152         }
 153 }