SparrowOSTS_121127.ISO

SparrowOS/Demo/Graphics/PoleZeros.CPP

@@ -0,0 +1,171 @@
+#define NUM_ZEROS	2
+Complex zeros[NUM_ZEROS]={{10.0,0},{-30,0}};
+#define NUM_POLES	2
+Complex poles[NUM_POLES]={{-20.0,-15.0},{-20.0,15.0}};
+
+F64 scale;
+
+Complex *PoleZeroFind(I64 x,I64 y)
+{
+  I64 i;
+  F64 d,best_d=MAX_F64;
+  Complex *result=NULL;
+  for (i=0;i<NUM_POLES;i++) {
+    d=Sqr(poles[i].x-x)+Sqr(poles[i].y-y);
+    if (d<best_d) {
+      best_d=d;
+      result=&poles[i];
+    }
+  }
+  for (i=0;i<NUM_ZEROS;i++) {
+    d=Sqr(zeros[i].x-x)+Sqr(zeros[i].y-y);
+    if (d<best_d) {
+      best_d=d;
+      result=&zeros[i];
+    }
+  }
+  return result;
+}
+
+F64 F(Complex *x)
+{
+  F64 m,a;
+  Complex num,denom,n1;
+
+  CPoly(&num,  x,zeros,NUM_ZEROS);
+  CPoly(&denom,x,poles,NUM_POLES);
+  CDiv(&num,&denom,&n1);
+ 
+  R2P(&m,&a,n1.x,n1.y);
+  if (ip_rb)
+    return ã+a;
+  else
+    return m;
+}
+
+F64 MPDraw(CTask *task)
+{
+  Complex xx;
+  I64 x,y,w=task->win_pixel_width,h=task->win_pixel_height,cx=w/2,cy=h/2,
+     lo=Gs->num*h/mp_cnt,hi=(Gs->num+1)*h/mp_cnt;
+  F64 yy,y_total=0;
+  CDC *dc=DCAlias(gr_dc2,task);
+  for (y=lo;y<hi;y++) {
+    for (x=0;x<w;x++) {
+      CEqu(&xx,x-cx,cy-y);
+      yy=scale*F(&xx);
+      dc->color=Limit(yy,0,14);
+      y_total+=Limit(yy,-14,14);
+      GrPlot(dc,x,y);
+    }
+    }
+  DCDel(dc);
+  return y_total;
+}
+
+U0 Draw()
+{
+  CMPCmd *tempm[MP_MAX_PROCESSORS];
+  F64 y_total,old_y_total=MAX_F64;
+  I64 i,w=Fs->win_pixel_width,h=Fs->win_pixel_height,cx=w/2,cy=h/2;
+  CDC *dc=DCAlias;
+  while (TRUE) {
+
+    for (i=1;i<mp_cnt;i++)
+      tempm[i]=MPJob(&MPDraw,Fs,i,0);
+    y_total=MPDraw(Fs);
+    for (i=1;i<mp_cnt;i++)
+      y_total+=MPJobResult(tempm[i])(F64);
+
+    if (!y_total) break;
+    scale*=7*GR_WIDTH*GR_HEIGHT/y_total;
+    if (Abs(y_total-old_y_total)<1.0*GR_WIDTH*GR_HEIGHT)
+      break;
+    old_y_total=y_total;
+  }
+  dc->color=15;
+  GrLine(dc,0,cy,w,cy);
+  GrLine(dc,cx,0,cx,h);
+  for (i=0;i<NUM_ZEROS;i++)
+    GrPutS(dc,cx+zeros[i].x-FONT_WIDTH/2,cy-zeros[i].y-FONT_HEIGHT/2,"o");
+  for (i=0;i<NUM_POLES;i++)
+    GrPutS(dc,cx+poles[i].x-FONT_WIDTH/2,cy-poles[i].y-FONT_HEIGHT/2,"x");
+  DCDel(dc);
+}
+
+U0 PoleZeros()
+{
+  I64 msg_code,p1,p2,p11,p22,cx,cy;
+  Complex *tempc=NULL;
+
+  PopUpOk(
+	"Drag the poles and zeros with left mouse.\n"
+	"Hold right mouse for phase plot.\n"
+  );
+  WinMgrSync;
+
+  SettingsPush; //See $LK,"SettingsPush","MN:SettingsPush"$
+  Fs->win_inhibit=WIF_ALL-WIF_BORDER;
+  SetVGAPalette4(vga_palette4_gray);
+  SetVGAPaletteColor(15,0xFF0000); //White is red
+
+  WinMax;
+  WordStat;
+  WinBorder;
+
+  DCFill;
+  DocClear;
+  scale=1.0;
+
+  try {
+    Draw;
+    while (TRUE) {
+      msg_code=GetMsg(&p1,&p2,1<<MSG_KEY_DOWN+1<<MSG_IP_L_DOWN+1<<MSG_IP_L_UP+1<<MSG_IP_R_DOWN+1<<MSG_IP_R_UP+1<<MSG_IP_MOVE);
+pz_msg:
+      cx=Fs->win_pixel_width/2;
+      cy=Fs->win_pixel_height/2;
+      switch (msg_code) {
+	case MSG_IP_L_DOWN:
+	  tempc=PoleZeroFind(p1-cx,cy-p2);
+	  break;
+	case MSG_IP_MOVE:
+	  if (tempc) {
+	    p11=p1;p22=p2;
+	    //get to last mouse move
+	    while (msg_code=ScanMsg(&p1,&p2,1<<MSG_KEY_DOWN+1<<MSG_IP_L_DOWN+1<<MSG_IP_L_UP+1<<MSG_IP_R_DOWN+1<<MSG_IP_R_UP+1<<MSG_IP_MOVE))
+	      if (msg_code==MSG_IP_MOVE) {
+		p11=p1;p22=p2;
+	      } else
+		goto pz_msg;
+
+	    tempc->x=p11-cx;
+	    tempc->y=cy-p22;
+	    Draw;
+	  }
+	  break;
+	case MSG_IP_L_UP:
+	  if (tempc) {
+	    tempc->x=p1-cx;
+	    tempc->y=cy-p2;
+	    tempc=NULL;
+	    Draw;
+	  }
+	  break;
+	case MSG_IP_R_DOWN:
+	case MSG_IP_R_UP:
+	  Draw;
+	  break;
+	case MSG_KEY_DOWN:
+	  if (p1==CH_SHIFT_ESC||p1==CH_ESC)
+	    goto pz_done;
+      }
+      WinMgrSync;
+    }
+pz_done:
+  } catch
+    CatchAll;
+  SettingsPop;
+  DCFill;
+}
+
+PoleZeros;