Plugins/FFT/DCT.cs

   1 using System;
   2 using System.Xml;
   3 using Mono.Unix;
   4
   5 namespace Eithne
   6 {
   7         public class DCTInfo : IInfo
   8         {
   9                 public override string Name
  10                 {
  11                         get { return Catalog.GetString("Discrete Cosine Transform"); }
  12                 }
  13
  14                 public override string ShortName
  15                 {
  16                         get { return "DCT"; }
  17                 }
  18
  19                 public override string Author
  20                 {
  21                         get { return "Bartosz Taudul"; }
  22                 }
  23
  24                 public override string Description
  25                 {
  26                         get { return Catalog.GetString("This plugin performs discrete cosine transform."); }
  27                 }
  28         }
  29
  30         public class DCTFactory : IFactory
  31         {
  32                 IInfo _info = new DCTInfo();
  33                 public IInfo Info
  34                 {
  35                         get { return _info; }
  36                 }
  37
  38                 public IType Type
  39                 {
  40                         get { return IType.ImgProc; }
  41                 }
  42
  43                 public void Initialize()
  44                 {
  45                 }
  46
  47                 public Plugin.Base Create()
  48                 {
  49                         return new DCTPlugin();
  50                 }
  51         }
  52
  53         public class DCTPlugin : Plugin.ImgProc
  54         {
  55                 private bool zero = true;
  56                 private float progress;
  57
  58                 public DCTPlugin()
  59                 {
  60                         _info = new DCTInfo();
  61                 }
  62
  63                 public override XmlNode Config
  64                 {
  65                         get { return GetConfig(); }
  66                         set { LoadConfig(value); }
  67                 }
  68
  69                 private void UpdateValue(bool z)
  70                 {
  71                         zero = z;
  72                         _block.SlotsChanged();
  73                 }
  74
  75                 public override void Setup()
  76                 {
  77                         new FFTSetup(zero, UpdateValue, true);
  78                 }
  79
  80                 public override void Work()
  81                 {
  82                         progress = 0;
  83
  84                         ICommImage socket = _in[0] as ICommImage;
  85                         IImage[] img = socket.Images;
  86                         IImage[] res = new IImage[img.Length];
  87                         _out = new CommSocket(1);
  88
  89                         for(int i=0; i<img.Length; i++)
  90                         {
  91                                 res[i] = DCT(img[i]);
  92                                 progress = (float)i/img.Length;
  93                         }
  94
  95                         _out[0] = new ICommImage(res, socket.OriginalImages, socket.Categories);
  96
  97                         FFTW.Cleanup();
  98
  99                         _workdone = true;
 100                 }
 101
 102                 private IImage DCT(IImage img)
 103                 {
 104                         double[] datain = new double[img.W * img.H];
 105                         double[] dataout = new double[img.W * img.H];
 106
 107                         for(int y=0; y<img.H; y++)
 108                                 for(int x=0; x<img.W; x++)
 109                                         datain[x + y*img.W] = (byte)img[x, y];
 110
 111                         IntPtr plan = FFTW.PlanR2R2D(img.H, img.W, datain, dataout, FFTW.Kind.FFTW_REDFT10,
 112                                         FFTW.Kind.FFTW_REDFT10, 0);
 113
 114                         FFTW.Execute(plan);
 115
 116                         FFTW.DestroyPlan(plan);
 117
 118                         IImage ret = new IImage(BPP.Float, img.W, img.H);
 119
 120                         for(int y=0; y<img.H; y++)
 121                                 for(int x=0; x<img.W; x++)
 122                                         ret[x, y] = (float)dataout[x + y*img.W] / 255f;
 123
 124                         if(zero)
 125                                 ret[0, 0] = 0f;
 126
 127                         return ret;
 128                 }
 129
 130                 public override void Lock()
 131                 {
 132                         FFTW.mutex.WaitOne();
 133                 }
 134
 135                 public override void Unlock()
 136                 {
 137                         FFTW.mutex.ReleaseMutex();
 138                 }
 139
 140                 private XmlNode GetConfig()
 141                 {
 142                         XmlNode root = _xmldoc.CreateNode(XmlNodeType.Element, "config", "");
 143
 144                         if(zero)
 145                                 root.InnerText = "1";
 146                         else
 147                                 root.InnerText = "0";
 148
 149                         return root;
 150                 }
 151
 152                 private void LoadConfig(XmlNode root)
 153                 {
 154                         if(root.InnerText == "1")
 155                                 UpdateValue(true);
 156                         else
 157                                 UpdateValue(false);
 158                 }
 159
 160                 public override int NumIn               { get { return 1; } }
 161                 public override int NumOut              { get { return 1; } }
 162
 163                 public override string DescIn(int n)
 164                 {
 165                         return Catalog.GetString("Input image.");
 166                 }
 167
 168                 public override string DescOut(int n)
 169                 {
 170                         return Catalog.GetString("DCT of image.");
 171                 }
 172
 173                 private static string[] matchin   = new string[] { "image/grayscale" };
 174                 private static string[] matchout  = new string[] { "image/float" };
 175
 176                 public override string[] MatchIn        { get { return matchin; } }
 177                 public override string[] MatchOut       { get { return matchout; } }
 178
 179                 public override float Progress { get { return progress; } }
 180         }
 181 }