Measure/SubTenderGather/stgSelectFileFrm.pas

unit stgSelectFileFrm;

interface

uses
  stgGatherControl,
  Windows, Messages, SysUtils, Variants, Classes, Graphics, Controls, Forms,
  Dialogs, StdCtrls, ExtCtrls, sdGridDBA, sdGridTreeDBA, ZJGrid, sdIDTree;

type
  TstgSelectFileForm = class(TForm)
    pnlTop: TPanel;
    leSumBaseFile: TLabeledEdit;
    btnSelectSbf: TButton;
    pnlGatherFiles: TPanel;
    pnlSelect: TPanel;
    zgTenderSelect: TZJGrid;
    pnlSelectTitle: TPanel;
    lblTenderList: TLabel;
    stdTenderSelect: TsdGridTreeDBA;
    pnlDivision: TPanel;
    pnlResult: TPanel;
    zgResult: TZJGrid;
    pnlResultTitle: TPanel;
    lblResultList: TLabel;
    pnlBottom: TPanel;
    btnOk: TButton;
    btnCancel: TButton;
    procedure zgTenderSelectDrawCellText(ACanvas: TCanvas;
      const ARect: TRect; const ACoord: TPoint; AGrid: TZJGrid;
      const Text: String; var ADefaultDraw: Boolean);
    procedure zgTenderSelectShowHint(var HintStr: String;
      var CanShow: Boolean; var HintInfo: THintInfo; const ACoord: TPoint);
    procedure zgTenderSelectSetCellText(Sender: TObject;
      const ACoord: TPoint; var Value: String; DisplayText: Boolean);
    procedure zgTenderSelectGetCellText(Sender: TObject;
      const ACoord: TPoint; var Value: String; DisplayText: Boolean);
    procedure zgTenderSelectCellTextChanged(Sender: TObject; Col,
      Row: Integer);
    procedure btnOkClick(Sender: TObject);
    procedure btnSelectSbfClick(Sender: TObject);
  private
    FSelects: TList;

    procedure AddRows(ANode: TsdIDTreeNode);
    procedure RemoveRows(ANode: TsdIDTreeNode);

    procedure InitResultGrid;
    procedure AssignResultGrid;
    procedure AssignResult;
    function GetSelectCount: Integer;
  public
    constructor Create(AGatherControl: TstgGatherControl);
    destructor Destroy; override;

    procedure AssignSelect(AGatherControl: TstgGatherControl);
    property SelectCount: Integer read GetSelectCount;
  end;

function SelectFileForSubTenderGather(AGatherControl: TstgGatherControl): Boolean;

implementation

uses MainFrm, Globals, UtilMethods;

{$R *.dfm}

function SelectFileForSubTenderGather(AGatherControl: TstgGatherControl): Boolean;
var
  vSelectFrm: TstgSelectFileForm;
begin
  vSelectFrm := TstgSelectFileForm.Create(AGatherControl);
  try
    Result := vSelectFrm.ShowModal = mrOk;
    if Result then
      vSelectFrm.AssignSelect(AGatherControl);
  finally
    vSelectFrm.Free;
  end;
end;

procedure TstgSelectFileForm.zgTenderSelectDrawCellText(ACanvas: TCanvas;
  const ARect: TRect; const ACoord: TPoint; AGrid: TZJGrid;
  const Text: String; var ADefaultDraw: Boolean);

  procedure GetBitmap(AImage: TBitmap);
  begin
    with stdTenderSelect.IDTree.Items[ACoord.Y - 1] do
      if Rec.ValueByName('Type').AsInteger = 0 then
        if Expanded and HasChildren then
          MainForm.Images.GetBitmap(34, AImage)
        else
          MainForm.Images.GetBitmap(34, AImage)
      else
        MainForm.Images.GetBitmap(11, AImage);
  end;

const
  rIconWidth = 16;
  rIconHeight = 16;
var
  Img: TBitmap;
  Cell: TZjCell;
  rImg: TRect;
begin
 if (ACoord.X = 2) and (ACoord.Y > zgTenderSelect.FixedRowCount - 1) then
  begin
    Cell := zgTenderSelect.Cells[ACoord.X, ACoord.Y];
    Img := TBitmap.Create;
    try
      GetBitmap(Img);
      case Cell.Align of
        gaTopLeft, gaTopCenter, gaTopRight:
          rImg := Rect(ARect.Left + 2, ARect.Top, ARect.Left + rIconWidth, ARect.Top + rIconHeight);
        gaCenterLeft, gaCenterCenter, gaCenterRight:
          rImg := Rect(ARect.Left + 2, ARect.Top + (ARect.Bottom - ARect.Top - rIconHeight) div 2, ARect.Left + rIconWidth, ARect.Bottom - (ARect.Bottom - ARect.Top - rIconHeight) div 2);
        gaBottomLeft, gaBottomCenter, gaBottomRight:
          rImg := Rect(ARect.Left + 2, ARect.Bottom - rIconHeight, ARect.Left + rIconWidth, ARect.Bottom);
      end;
      ACanvas.StretchDraw(rImg, Img);
      WriteText(ACanvas, Rect(ARect.Left + rIconWidth, ARect.Top, ARect.Right, ARect.Bottom)
        , 2, 2, Text, Cell.Align, False);
      ADefaultDraw := False;
    finally
      Img.Free;
    end;
  end;
end;

procedure TstgSelectFileForm.zgTenderSelectShowHint(var HintStr: String;
  var CanShow: Boolean; var HintInfo: THintInfo; const ACoord: TPoint);
var
  vCell: TZjCell;
  vNode: TsdIDTreeNode;
  iLevelWidth: Integer;
  rText: TRect;

  procedure CalcTextRect(var R: TRect);
  var
    DC: HDC;
    iTextHeight: Integer;
  begin
    DC := CreateCompatibleDC(0);
    try
      SelectObject(DC, vCell.Font.Handle);
      iTextHeight := DrawText(DC, PChar(vCell.Text), Length(vCell.Text), R, DT_SINGLELINE or DT_VCenter
        or DT_NOCLIP or DT_CALCRECT);
    finally
      DeleteDC(DC);
    end;
  end;

begin
  if ACoord.Y < 1 then Exit;

  vCell := zgTenderSelect.Cells[ACoord.X, ACoord.Y];
  with HintInfo do
  begin
    vNode := stdTenderSelect.IDTree.Items[ACoord.Y - 1];
    if not Assigned(vNode) then Exit;
    iLevelWidth := (vNode.Level + 1) * 20 + 16;

    rText := CursorRect;
    CalcTextRect(rText);
    if (rText.Right - rText.Left + iLevelWidth > CursorRect.Right - CursorRect.Left) or
      (rText.Right > ClientWidth) then
    begin
      CanShow := True;
      HintStr := vCell.Text;
      GetCursorPos(HintPos);
    end;
  end;
end;

procedure TstgSelectFileForm.zgTenderSelectSetCellText(Sender: TObject;
  const ACoord: TPoint; var Value: String; DisplayText: Boolean);
var
  stnNode: TsdIDTreeNode;
begin
  if ACoord.X = 1 then
  begin
    stnNode := stdTenderSelect.IDTree.Items[ACoord.Y - 1];
    if Value = 'True' then
      AddRows(stnNode)
    else
      RemoveRows(stnNode);
    AssignResult;
  end;
end;

procedure TstgSelectFileForm.zgTenderSelectGetCellText(Sender: TObject;
  const ACoord: TPoint; var Value: String; DisplayText: Boolean);
var
  stnNode: TsdIDTreeNode;
begin
  if ACoord.X = 1 then
  begin
    stnNode := stdTenderSelect.IDTree.Items[ACoord.Y - 1];
    if Assigned(stnNode) and (FSelects.IndexOf(Pointer(stnNode.ID)) > -1) then
      Value := 'True';
  end;
end;

procedure TstgSelectFileForm.zgTenderSelectCellTextChanged(Sender: TObject;
  Col, Row: Integer);
begin
  if (Col = 1) then
    zgTenderSelect.InvalidateCol(1);
end;

procedure TstgSelectFileForm.AddRows(ANode: TsdIDTreeNode);
var
  iChild: Integer;
begin
  if not Assigned(ANode) then Exit;

  if FSelects.IndexOf(Pointer(ANode.ID)) = -1 then
    FSelects.Add(Pointer(ANode.ID));

  if ANode.HasChildren then
    for iChild := 0 to ANode.ChildCount - 1 do
      AddRows(ANode.ChildNodes[iChild]);
end;

procedure TstgSelectFileForm.RemoveRows(ANode: TsdIDTreeNode);
var
  iChild: Integer;
begin
  if not Assigned(ANode) then Exit;

  if FSelects.IndexOf(Pointer(ANode.ID)) > -1 then
    FSelects.Remove(Pointer(ANode.ID));

  if ANode.HasChildren then
    for iChild := 0 to ANode.ChildCount - 1 do
      RemoveRows(ANode.ChildNodes[iChild]);
end;

procedure TstgSelectFileForm.InitResultGrid;
begin
  zgResult.ColCount := 2;
  zgResult.RowCount := 1;
  zgResult.Cells[1, 0].Text := '所选项目';
  zgResult.ColWidths[1] := 365;
end;

procedure TstgSelectFileForm.AssignResultGrid;
var
  i, iID: Integer;
  vNode: TsdIDTreeNode;
begin
  for i := 0 to FSelects.Count - 1 do
  begin
    iID := Integer(FSelects.Items[i]);
    vNode := stdTenderSelect.IDTree.FindNode(iID);
    if vNode.Rec.ValueByName('Type').AsInteger = 1 then
    begin
      zgResult.RowCount := zgResult.RowCount + 1;
      zgResult.Cells[1, zgResult.RowCount - 1].Text := vNode.Rec.ValueByName('Name').AsString;
      zgResult.Cells[1, zgResult.RowCount - 1].Align := gaCenterLeft;
      zgResult.Rows[zgResult.RowCount - 1].Data := vNode;
    end;
  end;
end;

procedure TstgSelectFileForm.AssignSelect(AGatherControl: TstgGatherControl);
begin
  AGatherControl.SumBaseFile := leSumBaseFile.Text;
  AGatherControl.Projects.Assign(FSelects);
end;

constructor TstgSelectFileForm.Create(AGatherControl: TstgGatherControl);
begin
  inherited Create(nil);
  stdTenderSelect.DataView := ProjectManager.sdvProjectsSpare;
  FSelects := TList.Create;
  FSelects.Assign(AGatherControl.Projects);
  InitResultGrid;
  leSumBaseFile.Text := AGatherControl.SumBaseFile;
end;

destructor TstgSelectFileForm.Destroy;
begin
  FSelects.Free;
  inherited;
end;

procedure TstgSelectFileForm.AssignResult;
begin
  InitResultGrid;
  AssignResultGrid;
end;

procedure TstgSelectFileForm.btnOkClick(Sender: TObject);
begin
  if (leSumBaseFile.Text = '') then
    WarningMessage('请选择总包基准文件。', Handle)
  else if not FileExists(leSumBaseFile.Text) then
    WarningMessage('当前选择的总包基准文件不存在，请重新选择。', Handle)
  else if SelectCount = 0 then
    WarningMessage('请选择需要汇总的分包标段。', Handle)
  else
    ModalResult := mrOk;
end;

function TstgSelectFileForm.GetSelectCount: Integer;
begin
  Result := zgResult.RowCount - zgResult.FixedRowCount;
end;

procedure TstgSelectFileForm.btnSelectSbfClick(Sender: TObject);
var
  sFileName: string;
begin
  sFileName := leSumBaseFile.Text;
  if SelectFile(sFileName, '.sbf') then
    leSumBaseFile.Text := sFileName;
end;

end.