#region ================== Copyright (c) 2007 Pascal vd Heiden /* * Copyright (c) 2007 Pascal vd Heiden, www.codeimp.com * This program is released under GNU General Public License * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * */ #endregion #region ================== Namespaces using System; using System.Collections; using System.Collections.Generic; using System.Globalization; using System.Text; using System.Windows.Forms; using System.IO; using System.Reflection; using CodeImp.DoomBuilder.Windows; using CodeImp.DoomBuilder.IO; using CodeImp.DoomBuilder.Map; using CodeImp.DoomBuilder.Rendering; using CodeImp.DoomBuilder.Geometry; using CodeImp.DoomBuilder.Editing; using System.Drawing; using CodeImp.DoomBuilder.Actions; using CodeImp.DoomBuilder.Data; #endregion namespace CodeImp.DoomBuilder.BuilderModes { [EditMode(DisplayName = "Brightness Mode", SwitchAction = "brightnessmode", ButtonImage = "BrightnessMode.png", ButtonOrder = int.MinValue + 201, ButtonGroup = "000_editing", AllowCopyPaste = false, UseByDefault = true)] public sealed class BrightnessMode : BaseClassicMode { #region ================== Enums private enum ModifyMode : int { None, Adjusting } #endregion #region ================== Constants #endregion #region ================== Variables // Highlighted item private Sector highlighted; // Interface private bool editpressed; // The methods GetSelected* and MarkSelected* on the MapSet do not // retain the order in which items were selected. // This list keeps in order while sectors are selected/deselected. protected List<Sector> orderedselection; // Labels private Dictionary<Sector, TextLabel[]> labels; // Modifying private ModifyMode mode; private Point editstartpos; private List<int> sectorbrightness; private int undoticket; #endregion #region ================== Properties public ICollection<Sector> OrderedSelection { get { return orderedselection; } } #endregion #region ================== Constructor / Disposer // Constructor public BrightnessMode() { // Make ordered selection list orderedselection = new List<Sector>(); } // Disposer public override void Dispose() { // Not already disposed? if(!isdisposed) { // Dispose old labels foreach(KeyValuePair<Sector, TextLabel[]> lbl in labels) foreach(TextLabel l in lbl.Value) l.Dispose(); // Dispose base base.Dispose(); } } #endregion #region ================== Methods // This sets the ordered selection public void SetOrderedSelection(ICollection<Sector> list) { orderedselection = new List<Sector>(list); } // This sets up new labels private void SetupLabels() { if(labels != null) { // Dispose old labels foreach(KeyValuePair<Sector, TextLabel[]> lbl in labels) foreach(TextLabel l in lbl.Value) l.Dispose(); } // Make text labels for sectors labels = new Dictionary<Sector, TextLabel[]>(General.Map.Map.Sectors.Count); foreach(Sector s in General.Map.Map.Sectors) { // Setup labels TextLabel[] labelarray = new TextLabel[s.Triangles.IslandVertices.Count]; for(int i = 0; i < s.Triangles.IslandVertices.Count; i++) { Vector2D v = s.Labels[i].position; labelarray[i] = new TextLabel(20); labelarray[i].TransformCoords = true; labelarray[i].Rectangle = new RectangleF(v.x, v.y, 0.0f, 0.0f); labelarray[i].AlignX = TextAlignmentX.Center; labelarray[i].AlignY = TextAlignmentY.Middle; labelarray[i].Scale = 14f; labelarray[i].Color = General.Colors.Highlight.WithAlpha(255); labelarray[i].Backcolor = General.Colors.Background.WithAlpha(255); } labels.Add(s, labelarray); } } // This updates the overlay private void UpdateOverlay() { if(renderer.StartOverlay(true)) { // Editing a selection? if(mode == ModifyMode.Adjusting) { // Go for all sectors that are being edited foreach(Sector s in orderedselection) { // We use the overlay to dim the brightness of the sectors PixelColor brightnesscolor = new PixelColor((byte)(255 - s.Brightness), 0, 0, 0); int brightnessint = brightnesscolor.ToInt(); // Render the geometry FlatVertex[] verts = new FlatVertex[s.FlatVertices.Length]; s.FlatVertices.CopyTo(verts, 0); for(int i = 0; i < verts.Length; i++) verts[i].c = brightnessint; renderer.RenderGeometry(verts, null, true); } } if(BuilderPlug.Me.ViewSelectionNumbers) { // Go for all sectors foreach(Sector s in orderedselection) { // Render labels TextLabel[] labelarray = labels[s]; for(int i = 0; i < s.Labels.Count; i++) { TextLabel l = labelarray[i]; // Render only when enough space for the label to see float requiredsize = (l.TextSize.Height / 2) / renderer.Scale; if(requiredsize < s.Labels[i].radius) renderer.RenderText(l); } } } renderer.Finish(); } } // This highlights a new item protected void Highlight(Sector s) { // Highlight actually changes? if(s != highlighted) { // Update display if(renderer.StartPlotter(false)) { if((highlighted != null) && !highlighted.IsDisposed) { // Undraw previous highlight renderer.PlotSector(highlighted); // Change label color TextLabel[] labelarray = labels[highlighted]; foreach(TextLabel l in labelarray) l.Color = General.Colors.Selection; } // Set new highlight highlighted = s; if((highlighted != null) && !highlighted.IsDisposed) { // Render highlighted item renderer.PlotSector(highlighted, General.Colors.Highlight); // Change label color TextLabel[] labelarray = labels[highlighted]; foreach(TextLabel l in labelarray) l.Color = General.Colors.Highlight; } renderer.Finish(); } UpdateOverlay(); renderer.Present(); } // Show highlight info if((highlighted != null) && !highlighted.IsDisposed) General.Interface.ShowSectorInfo(highlighted); else General.Interface.HideInfo(); } // This selectes or deselects a sector protected void SelectSector(Sector s, bool selectstate, bool update) { bool selectionchanged = false; if(!s.IsDisposed) { // Select the sector? if(selectstate && !s.Selected) { orderedselection.Add(s); s.Selected = true; selectionchanged = true; // Setup labels TextLabel[] labelarray = labels[s]; foreach(TextLabel l in labelarray) { l.Text = orderedselection.Count.ToString(); l.Color = General.Colors.Selection; } } // Deselect the sector? else if(!selectstate && s.Selected) { orderedselection.Remove(s); s.Selected = false; selectionchanged = true; // Clear labels TextLabel[] labelarray = labels[s]; foreach(TextLabel l in labelarray) l.Text = ""; // Update all other labels UpdateSelectedLabels(); } // Selection changed? if(selectionchanged) { // Make update lines selection foreach(Sidedef sd in s.Sidedefs) { bool front, back; if(sd.Line.Front != null) front = sd.Line.Front.Sector.Selected; else front = false; if(sd.Line.Back != null) back = sd.Line.Back.Sector.Selected; else back = false; sd.Line.Selected = front | back; } } if(update) { UpdateOverlay(); renderer.Present(); } } else { // Remove from list orderedselection.Remove(s); } } // This updates labels from the selected sectors private void UpdateSelectedLabels() { // Update labels for editing mode? if(mode == ModifyMode.Adjusting) { // Go for all labels in all selected sectors for(int i = 0; i < orderedselection.Count; i++) { Sector s = orderedselection[i]; TextLabel[] labelarray = labels[s]; foreach(TextLabel l in labelarray) { // Make sure the text and color are right int labelnum = s.Brightness; l.Text = labelnum.ToString(); l.Color = General.Colors.Indication; } } } // Updating for normal mode else { // Go for all labels in all selected sectors for(int i = 0; i < orderedselection.Count; i++) { Sector s = orderedselection[i]; TextLabel[] labelarray = labels[s]; foreach(TextLabel l in labelarray) { // Make sure the text and color are right int labelnum = i + 1; l.Text = labelnum.ToString(); l.Color = General.Colors.Selection; } } } } #endregion #region ================== Events // Mode engages public override void OnEngage() { base.OnEngage(); // Add toolbar buttons General.Interface.AddButton(BuilderPlug.Me.MenusForm.ViewSelectionNumbers); General.Interface.AddButton(BuilderPlug.Me.MenusForm.SeparatorSectors1); General.Interface.AddButton(BuilderPlug.Me.MenusForm.MakeGradientBrightness); // Make custom presentation CustomPresentation p = new CustomPresentation(); p.AddLayer(new PresentLayer(RendererLayer.Background, BlendingMode.Mask, General.Settings.BackgroundAlpha)); p.AddLayer(new PresentLayer(RendererLayer.Grid, BlendingMode.Mask)); p.AddLayer(new PresentLayer(RendererLayer.Surface, BlendingMode.Mask)); p.AddLayer(new PresentLayer(RendererLayer.Overlay, BlendingMode.Alpha, 1f, true)); //p.AddLayer(new PresentLayer(RendererLayer.Things, BlendingMode.Alpha, Presentation.THINGS_BACK_ALPHA, false)); p.AddLayer(new PresentLayer(RendererLayer.Geometry, BlendingMode.Alpha, 1f, true)); renderer.SetPresentation(p); // Make text labels for sectors SetupLabels(); // Convert geometry selection to sectors only General.Map.Map.ClearAllMarks(false); General.Map.Map.MarkSelectedVertices(true, true); ICollection<Linedef> lines = General.Map.Map.LinedefsFromMarkedVertices(false, true, false); foreach(Linedef l in lines) l.Selected = true; General.Map.Map.ClearMarkedSectors(true); foreach(Linedef l in General.Map.Map.Linedefs) { if(!l.Selected) { if(l.Front != null) l.Front.Sector.Marked = false; if(l.Back != null) l.Back.Sector.Marked = false; } } General.Map.Map.ClearAllSelected(); foreach(Sector s in General.Map.Map.Sectors) { if(s.Marked) { s.Selected = true; foreach(Sidedef sd in s.Sidedefs) sd.Line.Selected = true; } } // Fill the list with selected sectors (these are not in order, but we have no other choice) ICollection<Sector> selectedsectors = General.Map.Map.GetSelectedSectors(true); if(orderedselection.Count < selectedsectors.Count) { General.Map.Map.ClearSelectedSectors(); foreach(Sector s in selectedsectors) SelectSector(s, true, false); } // Update UpdateSelectedLabels(); UpdateOverlay(); } // When disengaged public override void OnDisengage() { base.OnDisengage(); // Remove toolbar buttons General.Interface.RemoveButton(BuilderPlug.Me.MenusForm.ViewSelectionNumbers); General.Interface.RemoveButton(BuilderPlug.Me.MenusForm.SeparatorSectors1); General.Interface.RemoveButton(BuilderPlug.Me.MenusForm.MakeGradientBrightness); // Going to EditSelectionMode? if(General.Editing.NewMode is EditSelectionMode) { // No selection made? But we have a highlight! if((General.Map.Map.GetSelectedSectors(true).Count == 0) && (highlighted != null)) { // Make the highlight the selection SelectSector(highlighted, true, false); } } // Going to SectorsMode? else if(General.Editing.NewMode is SectorsMode) { // Pass on the ordered selection (General.Editing.NewMode as SectorsMode).SetOrderedSelection(orderedselection); } // Hide highlight info General.Interface.HideInfo(); } // This redraws the display public override void OnRedrawDisplay() { renderer.RedrawSurface(); // Render lines and vertices if(renderer.StartPlotter(true)) { renderer.PlotLinedefSet(General.Map.Map.Linedefs); renderer.PlotVerticesSet(General.Map.Map.Vertices); if((highlighted != null) && !highlighted.IsDisposed) renderer.PlotSector(highlighted, General.Colors.Highlight); renderer.Finish(); } // Render things if(renderer.StartThings(true)) { renderer.RenderThingSet(General.Map.ThingsFilter.HiddenThings, Presentation.THINGS_HIDDEN_ALPHA); renderer.RenderThingSet(General.Map.ThingsFilter.VisibleThings, 1.0f); renderer.Finish(); } // Render overlay UpdateOverlay(); renderer.Present(); } // Mouse moves public override void OnMouseMove(MouseEventArgs e) { base.OnMouseMove(e); // Not in any editing mode? if((mode == ModifyMode.None) && (e.Button == MouseButtons.None)) { // Find the nearest linedef within highlight range Linedef l = General.Map.Map.NearestLinedef(mousemappos); if(l != null) { // Check on which side of the linedef the mouse is float side = l.SideOfLine(mousemappos); if(side > 0) { // Is there a sidedef here? if(l.Back != null) { // Highlight if not the same if(l.Back.Sector != highlighted) Highlight(l.Back.Sector); } else { // Highlight nothing if(highlighted != null) Highlight(null); } } else { // Is there a sidedef here? if(l.Front != null) { // Highlight if not the same if(l.Front.Sector != highlighted) Highlight(l.Front.Sector); } else { // Highlight nothing if(highlighted != null) Highlight(null); } } } else { // Highlight nothing if(highlighted != null) Highlight(null); } } // Adjusting mode? else if(mode == ModifyMode.Adjusting) { // Calculate change in position Point delta = Cursor.Position - new Size(editstartpos); if(General.Interface.ShiftState) { // Adjust selected sectors for(int i = 0; i < orderedselection.Count; i++) { Sector s = orderedselection[i]; int basebrightness = sectorbrightness[i]; // Adjust brightness s.Brightness = basebrightness - delta.Y; if(s.Brightness > 255) s.Brightness = 255; if(s.Brightness < 0) s.Brightness = 0; } } else { // Adjust selected sectors for(int i = 0; i < orderedselection.Count; i++) { Sector s = orderedselection[i]; int basebrightness = sectorbrightness[i]; // Adjust brightness s.Brightness = General.Map.Config.BrightnessLevels.GetNearest(basebrightness - delta.Y); } } // Update General.Interface.RefreshInfo(); UpdateSelectedLabels(); UpdateOverlay(); renderer.Present(); } } // Selecting with mouse protected override void OnSelectBegin() { // Not modifying? if(mode == ModifyMode.None) { // Item highlighted? if((highlighted != null) && !highlighted.IsDisposed) { // Flip selection SelectSector(highlighted, !highlighted.Selected, true); // Update display if(renderer.StartPlotter(false)) { // Redraw highlight to show selection renderer.PlotSector(highlighted); renderer.Finish(); renderer.Present(); } } else { // Start making a selection StartMultiSelection(); } } base.OnSelectBegin(); } // End selection protected override void OnSelectEnd() { // Not stopping from multiselection or modifying if(!selecting && (mode == ModifyMode.None)) { // Item highlighted? if((highlighted != null) && !highlighted.IsDisposed) { // Update display if(renderer.StartPlotter(false)) { // Render highlighted item renderer.PlotSector(highlighted, General.Colors.Highlight); renderer.Finish(); } // Update overlay TextLabel[] labelarray = labels[highlighted]; foreach(TextLabel l in labelarray) l.Color = General.Colors.Highlight; UpdateOverlay(); renderer.Present(); } } base.OnSelectEnd(); } // This is called wheh selection ends protected override void OnEndMultiSelection() { if(General.Interface.ShiftState ^ BuilderPlug.Me.AdditiveSelect) { // Go for all lines foreach(Linedef l in General.Map.Map.Linedefs) { l.Selected |= ((l.Start.Position.x >= selectionrect.Left) && (l.Start.Position.y >= selectionrect.Top) && (l.Start.Position.x <= selectionrect.Right) && (l.Start.Position.y <= selectionrect.Bottom) && (l.End.Position.x >= selectionrect.Left) && (l.End.Position.y >= selectionrect.Top) && (l.End.Position.x <= selectionrect.Right) && (l.End.Position.y <= selectionrect.Bottom)); } } else { // Go for all lines foreach(Linedef l in General.Map.Map.Linedefs) { l.Selected = ((l.Start.Position.x >= selectionrect.Left) && (l.Start.Position.y >= selectionrect.Top) && (l.Start.Position.x <= selectionrect.Right) && (l.Start.Position.y <= selectionrect.Bottom) && (l.End.Position.x >= selectionrect.Left) && (l.End.Position.y >= selectionrect.Top) && (l.End.Position.x <= selectionrect.Right) && (l.End.Position.y <= selectionrect.Bottom)); } } // Go for all sectors foreach(Sector s in General.Map.Map.Sectors) { // Go for all sidedefs bool allselected = true; foreach(Sidedef sd in s.Sidedefs) { if(!sd.Line.Selected) { allselected = false; break; } } // Sector completely selected? SelectSector(s, allselected, false); } // Make sure all linedefs reflect selected sectors foreach(Sidedef sd in General.Map.Map.Sidedefs) if(!sd.Sector.Selected && ((sd.Other == null) || !sd.Other.Sector.Selected)) sd.Line.Selected = false; base.OnEndMultiSelection(); UpdateOverlay(); General.Interface.RedrawDisplay(); } // This is called when the selection is updated protected override void OnUpdateMultiSelection() { base.OnUpdateMultiSelection(); // Render selection UpdateOverlay(); if(renderer.StartOverlay(false)) { RenderMultiSelection(); renderer.Finish(); renderer.Present(); } } // Editing protected override void OnEditBegin() { base.OnEditBegin(); // No selection? if(orderedselection.Count == 0) { // Make the highlight a selection if we have a highlight if((highlighted != null) && !highlighted.IsDisposed) SelectSector(highlighted, true, false); } // Anything selected? if(orderedselection.Count > 0) { // Create undo undoticket = General.Map.UndoRedo.CreateUndo("Adjust brightness"); // Start editing mode = ModifyMode.Adjusting; editstartpos = Cursor.Position; // Keep sector brightness offsets and make the sector full brightness so we can use // the overlay to adjust the brightness. The surface is only updated here and again // with correct brightness when editing is done. sectorbrightness = new List<int>(orderedselection.Count); foreach(Sector s in orderedselection) { int realbrightness = s.Brightness; sectorbrightness.Add(realbrightness); s.Brightness = 255; s.UpdateCache(); s.Brightness = realbrightness; } // Update surface to render full bright sectors renderer.RedrawSurface(); // Update UpdateSelectedLabels(); UpdateOverlay(); renderer.Present(); } } // Done editing protected override void OnEditEnd() { base.OnEditEnd(); // Stop editing mode = ModifyMode.None; sectorbrightness = null; // Nothing changed? Then writhdraw the undo if(editstartpos.Y == Cursor.Position.Y) General.Map.UndoRedo.WithdrawUndo(undoticket); // Update General.Map.Map.Update(); UpdateSelectedLabels(); General.Interface.RefreshInfo(); General.Interface.RedrawDisplay(); renderer.Present(); // If only one sector was selected, deselect it if(orderedselection.Count == 1) SelectSector(orderedselection[0], false, true); } // When undo is used public override bool OnUndoBegin() { // Clear selection General.Map.Map.ClearAllSelected(); orderedselection.Clear(); return base.OnUndoBegin(); } // When undo is performed public override void OnUndoEnd() { // Clear labels SetupLabels(); } // When redo is used public override bool OnRedoBegin() { // Clear selection General.Map.Map.ClearAllSelected(); orderedselection.Clear(); return base.OnRedoBegin(); } // When redo is performed public override void OnRedoEnd() { // Clear labels SetupLabels(); } #endregion #region ================== Actions [BeginAction("gradientbrightness")] public void MakeGradientBrightness() { General.Interface.DisplayStatus(StatusType.Action, "Created gradient brightness over selected sectors."); General.Map.UndoRedo.CreateUndo("Gradient brightness"); // Need at least 3 selected sectors // The first and last are not modified if(orderedselection.Count > 2) { float startbrightness = (float)orderedselection[0].Brightness; float endbrightness = (float)orderedselection[orderedselection.Count - 1].Brightness; float delta = endbrightness - startbrightness; // Go for all sectors in between first and last for(int i = 1; i < (orderedselection.Count - 1); i++) { float u = (float)i / (float)(orderedselection.Count - 1); float b = startbrightness + delta * u; orderedselection[i].Brightness = (int)b; } } // Update General.Map.Map.Update(); UpdateOverlay(); renderer.Present(); General.Interface.RedrawDisplay(); General.Map.IsChanged = true; } // This clears the selection [BeginAction("clearselection", BaseAction = true)] public void ClearSelection() { // Clear selection General.Map.Map.ClearAllSelected(); orderedselection.Clear(); // Clear labels foreach(TextLabel[] labelarray in labels.Values) foreach(TextLabel l in labelarray) l.Text = ""; // Redraw General.Interface.RedrawDisplay(); } #endregion } }